myvtm/Module/mod_livenessdetection/RvcWsServer.h

#pragma once

#include "websocketpp/config/asio_no_tls.hpp"
#include "websocketpp/config/asio.hpp"
#include "websocketpp/server.hpp"
#include <boost/asio.hpp>
#include <boost/thread/mutex.hpp>
#include <functional>
#include <memory>
#include <vector>
#if defined(RVC_OS_LINUX)
#include <semaphore.h>
#endif //RVC_OS_LINUX


#ifndef RVC_LIVENESS_WS_PORT
#define RVC_LIVENESS_WS_PORT 9100
#endif


#ifndef RVC_WS_INIT_STR
#define RVC_WS_INIT_STR "rvc wbsocket trans init"
#endif


#ifndef RVC_WS_START_TRANS_STR
#define RVC_WS_START_TRANS_STR "rvc wbsocket start trans"
#endif


#ifndef RVC_WS_STOP_TRANS_STR
#define RVC_WS_STOP_TRANS_STR "rvc wbsocket stop trans"
#endif


#ifndef RVC_WS_CHANGE_CAMERA_STR
#define RVC_WS_CHANGE_CAMERA_STR "rvc wbsocket change camera"
#endif

#ifndef RVC_WS_START_CAPTURE_STR
#define RVC_WS_START_CAPTURE_STR "rvc wbsocket start capture"
#endif

#ifndef RVC_WS_CONNECT_IDENTIFIER
#define RVC_WS_CONNECT_IDENTIFIER "+"
#endif


#ifndef RVC_DEFAULT_FPS
#define RVC_DEFAULT_FPS 5
#endif


#ifndef RVC_MIN_FPS
#define RVC_MIN_FPS	1
#endif


#ifndef RVC_MAX_FPS
#define RVC_MAX_FPS	30
#endif


namespace LivenessDetection{

enum eVideoType{
	ePreview_Type,
	eCapture_Type
};

enum eCameraType{
	eCamera_Env,
	eCamera_Opt
};

typedef websocketpp::server<websocketpp::config::asio> server;

typedef struct websocket_callback_s  {
	int (*on_get_videodata)(eVideoType eType, eCameraType ecameraid, int* width, int* height, unsigned char* bmpdata, int isize, void* user_data);
	void *user_data;
}websocket_callback_t;


typedef struct rvc_video_param_s{
	int iwidth;
	int iheight;
	int icapwidth;
	int icapheight;
}rvc_video_param_t;



class RvcWsServer
{
public:
	RvcWsServer(void);
	~RvcWsServer(void);
	int Init_WsServer(websocket_callback_t* pcallback, rvc_video_param_t* pparam, int iport = RVC_LIVENESS_WS_PORT);
	int StartVideoTransmit();
	int StopVideoTransmit();
	int StartVideoCapTransmit();

private:
	bool validate(websocketpp::connection_hdl hdl);
	void on_http(websocketpp::connection_hdl hdl);
	void on_fail(websocketpp::connection_hdl hdl);
	void on_open(websocketpp::connection_hdl hdl);
	void on_close(websocketpp::connection_hdl hdl);
	void on_message(websocketpp::connection_hdl hdl, server::message_ptr msg);

	int handle_initial_instructions(websocketpp::connection_hdl hdl, std::string strinstrut);
	int handle_change_camera_instructions(std::string strinstrut);

public:
	volatile bool m_bconnected;			// connect flag
	volatile bool m_bstarttrans;		// start transmit
	volatile eCameraType m_ecameraid;			// camera id
#if defined(RVC_OS_WIN)
	HANDLE m_work_thread;
	HANDLE m_evt;
#else
	pthread_t m_work_threadid;
	sem_t m_semt;
#endif //RVC_OS_WIN
	websocket_callback_t m_callback;
	server m_wsserver;					// websocket server,用于和业务层传输视频流,默认端口9100
	websocketpp::connection_hdl m_hdl;
	int m_cameraid;
	int m_fps;
	unsigned char* m_buffer;
	size_t m_ubuffer_size;
	unsigned char* m_capbuffer;
	size_t m_ucapbuffer_size;
	unsigned int m_utranstime;

private:
	int m_listenport;				  // websocket server监听端口
	std::string m_struuid;		      // websocket connet flag
};
}