|
|
|
|
@ -196,7 +196,8 @@ public: |
|
|
|
|
}; |
|
|
|
|
// ----------------------------------------
|
|
|
|
|
|
|
|
|
|
void send_mjpeg(IplImage* ipl, int port, int timeout, int quality) { |
|
|
|
|
void send_mjpeg(IplImage* ipl, int port, int timeout, int quality) |
|
|
|
|
{ |
|
|
|
|
static MJPGWriter wri(port, timeout, quality); |
|
|
|
|
cv::Mat mat = cv::cvarrToMat(ipl); |
|
|
|
|
wri.write(mat); |
|
|
|
|
@ -204,6 +205,164 @@ void send_mjpeg(IplImage* ipl, int port, int timeout, int quality) { |
|
|
|
|
} |
|
|
|
|
// ----------------------------------------
|
|
|
|
|
|
|
|
|
|
class JSON_sender |
|
|
|
|
{ |
|
|
|
|
SOCKET sock; |
|
|
|
|
SOCKET maxfd; |
|
|
|
|
fd_set master; |
|
|
|
|
int timeout; // master sock timeout, shutdown after timeout millis.
|
|
|
|
|
|
|
|
|
|
int _write(int sock, char const*const s, int len) |
|
|
|
|
{ |
|
|
|
|
if (len < 1) { len = strlen(s); } |
|
|
|
|
return ::send(sock, s, len, 0); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public: |
|
|
|
|
|
|
|
|
|
JSON_sender(int port = 0, int _timeout = 200000) |
|
|
|
|
: sock(INVALID_SOCKET) |
|
|
|
|
, timeout(_timeout) |
|
|
|
|
{ |
|
|
|
|
FD_ZERO(&master); |
|
|
|
|
if (port) |
|
|
|
|
open(port); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
~JSON_sender() |
|
|
|
|
{ |
|
|
|
|
release(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool release() |
|
|
|
|
{ |
|
|
|
|
if (sock != INVALID_SOCKET) |
|
|
|
|
::shutdown(sock, 2); |
|
|
|
|
sock = (INVALID_SOCKET); |
|
|
|
|
return false; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool open(int port) |
|
|
|
|
{ |
|
|
|
|
sock = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); |
|
|
|
|
|
|
|
|
|
SOCKADDR_IN address; |
|
|
|
|
address.sin_addr.s_addr = INADDR_ANY; |
|
|
|
|
address.sin_family = AF_INET; |
|
|
|
|
address.sin_port = htons(port); // ::htons(port);
|
|
|
|
|
if (::bind(sock, (SOCKADDR*)&address, sizeof(SOCKADDR_IN)) == SOCKET_ERROR) |
|
|
|
|
{ |
|
|
|
|
cerr << "error : couldn't bind sock " << sock << " to port " << port << "!" << endl; |
|
|
|
|
return release(); |
|
|
|
|
} |
|
|
|
|
if (::listen(sock, 10) == SOCKET_ERROR) |
|
|
|
|
{ |
|
|
|
|
cerr << "error : couldn't listen on sock " << sock << " on port " << port << " !" << endl; |
|
|
|
|
return release(); |
|
|
|
|
} |
|
|
|
|
FD_ZERO(&master); |
|
|
|
|
FD_SET(sock, &master); |
|
|
|
|
maxfd = sock; |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool isOpened() |
|
|
|
|
{ |
|
|
|
|
return sock != INVALID_SOCKET; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool write(char *outputbuf) |
|
|
|
|
{ |
|
|
|
|
fd_set rread = master; |
|
|
|
|
struct timeval to = { 0,timeout }; |
|
|
|
|
if (::select(maxfd + 1, &rread, NULL, NULL, &to) <= 0) |
|
|
|
|
return true; // nothing broken, there's just noone listening
|
|
|
|
|
|
|
|
|
|
size_t outlen = strlen(outputbuf); |
|
|
|
|
|
|
|
|
|
#ifdef _WIN32 |
|
|
|
|
for (unsigned i = 0; i<rread.fd_count; i++) |
|
|
|
|
{ |
|
|
|
|
int addrlen = sizeof(SOCKADDR); |
|
|
|
|
SOCKET s = rread.fd_array[i]; // fd_set on win is an array, while ...
|
|
|
|
|
#else |
|
|
|
|
for (int s = 0; s <= maxfd; s++) |
|
|
|
|
{ |
|
|
|
|
socklen_t addrlen = sizeof(SOCKADDR); |
|
|
|
|
if (!FD_ISSET(s, &rread)) // ... on linux it's a bitmask ;)
|
|
|
|
|
continue; |
|
|
|
|
#endif |
|
|
|
|
if (s == sock) // request on master socket, accept and send main header.
|
|
|
|
|
{ |
|
|
|
|
SOCKADDR_IN address = { 0 }; |
|
|
|
|
SOCKET client = ::accept(sock, (SOCKADDR*)&address, &addrlen); |
|
|
|
|
if (client == SOCKET_ERROR) |
|
|
|
|
{ |
|
|
|
|
cerr << "error : couldn't accept connection on sock " << sock << " !" << endl; |
|
|
|
|
return false; |
|
|
|
|
} |
|
|
|
|
maxfd = (maxfd>client ? maxfd : client); |
|
|
|
|
FD_SET(client, &master); |
|
|
|
|
_write(client, "HTTP/1.0 200 OK\r\n", 0); |
|
|
|
|
_write(client, |
|
|
|
|
"Server: Mozarella/2.2\r\n" |
|
|
|
|
"Accept-Range: bytes\r\n" |
|
|
|
|
"Connection: close\r\n" |
|
|
|
|
"Max-Age: 0\r\n" |
|
|
|
|
"Expires: 0\r\n" |
|
|
|
|
"Cache-Control: no-cache, private\r\n" |
|
|
|
|
"Pragma: no-cache\r\n" |
|
|
|
|
"Content-Type: application/json\r\n" |
|
|
|
|
//"Content-Type: multipart/x-mixed-replace; boundary=boundary\r\n"
|
|
|
|
|
"\r\n", 0); |
|
|
|
|
cerr << "new client " << client << endl; |
|
|
|
|
} |
|
|
|
|
else // existing client, just stream pix
|
|
|
|
|
{ |
|
|
|
|
char head[400]; |
|
|
|
|
// application/json
|
|
|
|
|
// application/x-resource+json or application/x-collection+json - when you are representing REST resources and collections
|
|
|
|
|
// text/json or text/javascript or text/plain.
|
|
|
|
|
// https://stackoverflow.com/questions/477816/what-is-the-correct-json-content-type
|
|
|
|
|
//sprintf(head, "\r\nContent-Length: %zu\r\n\r\n", outlen);
|
|
|
|
|
//sprintf(head, "--boundary\r\nContent-Type: application/json\r\nContent-Length: %zu\r\n\r\n", outlen);
|
|
|
|
|
_write(s, head, 0); |
|
|
|
|
int n = _write(s, outputbuf, outlen); |
|
|
|
|
//cerr << "known client " << s << " " << n << endl;
|
|
|
|
|
if (n < outlen) |
|
|
|
|
{ |
|
|
|
|
cerr << "kill client " << s << endl; |
|
|
|
|
::shutdown(s, 2); |
|
|
|
|
FD_CLR(s, &master); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
}; |
|
|
|
|
// ----------------------------------------
|
|
|
|
|
|
|
|
|
|
// JSON format:
|
|
|
|
|
//{
|
|
|
|
|
// "frame_id":8990,
|
|
|
|
|
// "objects":[
|
|
|
|
|
// {"class_id":4, "name":"aeroplane", "relative coordinates":{"center_x":0.398831, "center_y":0.630203, "width":0.057455, "height":0.020396}, "confidence":0.793070},
|
|
|
|
|
// {"class_id":14, "name":"bird", "relative coordinates":{"center_x":0.398831, "center_y":0.630203, "width":0.057455, "height":0.020396}, "confidence":0.265497}
|
|
|
|
|
// ]
|
|
|
|
|
//},
|
|
|
|
|
|
|
|
|
|
void send_json(detection *dets, int nboxes, int classes, char **names, long long int frame_id, int port, int timeout) |
|
|
|
|
{ |
|
|
|
|
static JSON_sender js(port, timeout); |
|
|
|
|
char *send_buf = detection_to_json(dets, nboxes, classes, names, frame_id); |
|
|
|
|
|
|
|
|
|
js.write(send_buf); |
|
|
|
|
std::cout << " JSON-stream sent. \n"; |
|
|
|
|
free(send_buf); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// ----------------------------------------
|
|
|
|
|
|
|
|
|
|
CvCapture* get_capture_video_stream(char *path) { |
|
|
|
|
CvCapture* cap = NULL; |
|
|
|
|
try { |
|
|
|
|
|
AlexeyAB
6 years ago