rtsp.cpp

#include <dirent.h>
#include <queue>
#include "RecButton.h"
#include "ReconnectButton.h"
#include "env.h"

// Defined inside env.h
//#define WIDTH 1024
//#define HEIGHT 600
//#define REC_WIDTH 640
//#define REC_HEIGHT 480
//#define MAX_LENGHT 10
//std::string rtsp = "rtsp://user:pass@ip:port"
// End env.h

// Buscar video en directorio y guardarlo en grabaciones(sobreescribir archivo en grabaciones).
void mvVideo();
void displayAndRec();
void connect();
void readFrame(cv::Mat* frame);
void displayFrame();
void recFrame(cv::Mat &frame);
void receive(cv::Mat* frame);
void mouseCallBack(int event, int x, int y, int flags, void* userdata);
char* readCredentials();
int readData(FILE* file, char* data, int size);

enum recording {OFF, ON};
enum fileData {PROTOCOL, USER, PASS, IP, PORT, ROUTE};
//--- INITIALIZE VIDEOCAPTURE
RecButton* rec_Button;
Button* reconnect_Button;
cv::Mat bar, tmp;
cv::VideoCapture cap;
cv::VideoWriter video;
std::queue<cv::Mat> frames;
char* url; // rtsp://user:pass@ip:port/route

int main(int, char**){
    //printf("%s\n", cv::getBuildInformation().c_str());
    mvVideo();
    //Create window
    cv::namedWindow("Live", cv::WINDOW_NORMAL);
    cv::resizeWindow("Live", WIDTH, HEIGHT);
    //cv::setWindowProperty("Live", cv::WND_PROP_FULLSCREEN, cv::WINDOW_AUTOSIZE);
    //cv::createTrackbar( "Grabar", "Live", &value, 1, pressRecord, &keepRecording);
    cap.set(cv::CAP_PROP_BUFFERSIZE, 1);
    cap.set(cv::CAP_PROP_FOURCC, cv::VideoWriter::fourcc('M', 'J', 'P', 'G'));
    cap.set(cv::CAP_PROP_CODEC_PIXEL_FORMAT, cv::VideoWriter::fourcc('M', 'J', 'P', 'G'));
    //cap.set(cv::CAP_PROP_FRAME_WIDTH, REC_WIDTH);
    //cap.set(cv::CAP_PROP_FRAME_HEIGHT, REC_HEIGHT);
    //set the callback function for any mouse event
    cv::setMouseCallback("Live", mouseCallBack, NULL);
    //createButton("b",NULL); //Need QT
    //cap.setExceptionMode(true);
    //cap.set(CAP_PROP_FPS, 15);
    //createButton("Grabar",record,NULL,QT_CHECKBOX,0);// Need QT
    // Set OPENCV environment RTSP UDP
    _putenv_s("OPENCV_FFMPEG_CAPTURE_OPTIONS", "rtsp_transport;udp");

    if(frames.empty()){
        printf("\nqueue empty\n");
    }
    // Define the codec and create VideoWriter object.The output is stored in 'video.avi' file.
	// Define the fps to be equal to 10. Also frame size is passed.
	//VideoWriter video;//("video.avi", VideoWriter::fourcc('M','J','P','G'),10, Size(WIDTH, HEIGHT));

    // open the default camera using default API
    // cap.open(0);
    // OR advance usage: select any API backend
    
    //int deviceID = 0;             // 0 = open default camera
    //int apiID = cv::CAP_ANY;      // 0 = autodetect default API

    // open selected camera using selected API
    //"rtsp://user:pass@ip:port"
    cv::Mat frame;

    connect();

    cv::Scalar color(100, 50, 0); //BGR --> Azul
    bar = cv::Mat::zeros(cv::Size(REC_WIDTH, 50), CV_8UC3);
    bar = cv::Mat(bar.rows, bar.cols, CV_8UC3, color);
    rec_Button = new RecButton(bar, cv::Point(bar.cols/6, bar.rows/2), 10, cv::Scalar(0, 0, 255),
    cv::Scalar(0, 255, 0), &video);
    reconnect_Button = new ReconnectButton(bar, cv::Point(bar.cols*7/8, bar.rows/2), 10,
    cv::Scalar(0, 0, 255), cv::Scalar(0, 255, 0), cap, url);

    while(true){

        readFrame(&frame);
        displayFrame();

        //Cierra la ventana al apretar la X de la esquina superior derecha
        if(!cv::getWindowProperty("Live",cv::WND_PROP_VISIBLE)){
            break;
        }

        // show live and wait for a key with timeout long enough to show images
        if(cv::waitKey(1) == 27){
            break;
        }
        //printf("\nmain\n");
    }
    // the camera will be deinitialized automatically in VideoCapture destructor
    // When everything done, release the video capture and write object
	cap.release();
	video.release();
    cv::destroyAllWindows();
    delete reconnect_Button;
    delete rec_Button;
    delete url;
    
    return 0;
}

void connect(){
    printf("try to open camera...\n");
    url = readCredentials();
    while(true){
        try{
            //printf("%s\n", url);
            if(url == NULL){
                printf("camera_credentials.txt file does not exist!, please create the file and fill it\n");
                continue;
            }
            //rtsp.c_str()
            cap.open(url, cv::CAP_FFMPEG);
            if (!cap.isOpened()) {
                std::cerr << "ERROR! Unable to open camera\n";
                continue;
            }
            break;
        }
        catch(std::string err){
            printf("Trying to connect...\n");
        }
    }
    printf("Camera open successfully");
}

void readFrame(cv::Mat* frame){
    try{
        //printf("%s\n", url);
        if (cap.read(*frame)) {

            //printf("Read");
            //printf("\nMat size: %d\n", frame->size);
            //printf("1");
            resize(*frame, *frame, cv::Size(REC_WIDTH, REC_HEIGHT));
            //printf("2");
            if (frame->empty()) {
                std::cerr << "ERROR! blank frame grabbed\n";
            }
            if(frames.size() < MAX_LENGHT){
                frames.push(frame->clone());
                //printf("push\n");
            }
            //printf("\nReceive\n");
        }
        Sleep(1000/24);
    }

    catch(std::string str){
        printf("Ha ocurrido un error (ERROR: %s) al leer el frame\n", str.c_str());
    }
}

void displayFrame(){
    try{
        if(!frames.empty()){
            //printf("display");
            tmp = frames.front();

            if(rec_Button->getState()){
                //printf("State: %d", rec_Button->getState());
                //resize(tmp, tmp, cv::Size(REC_WIDTH, REC_HEIGHT));
                //Sleep(1000/24);
                recFrame(tmp);
            }

            //hconcat(a, b, dst) // horizontal
            vconcat(bar, tmp, tmp); // vertical
            // Write the frame into the file 'video.avi'
            //printf("queue size: %d\n", frames.size());
            imshow("Live", tmp);
            //printf("show");
            //waitKey(500);
            frames.pop();
        }
    }
    catch(std::string str){
        printf("Hubo un error al grabar o mostrar la imagen");
    }
}

void recFrame(cv::Mat &frame){
    //resize(tmp, tmp, Size(WIDTH, HEIGHT));
    try{
        video.write(frame);
        cv::putText(frame, "Grabando", cv::Point(10, frame.rows / 10), //top-left position
        cv::FONT_HERSHEY_TRIPLEX, 1.0, CV_RGB(255, 0, 0), 2);
    }
    catch(std::string str){
        printf("Hubo un error al grabar. Error: %s", str.c_str());
    }
}

void mvVideo(){
    DIR *d;
    struct dirent *dir;
    d = opendir(".");
    if (d){
        while ((dir = readdir(d)) != NULL){
            if(dir->d_name[6] == 'a' && dir->d_name[7] == 'v' && dir->d_name[8] == 'i'){
                system("move video.avi ./grabaciones/");
            }
        }
        closedir(d);
    }
}

char* readCredentials(){
    char* data = new char[MAX_ARRAY_LENGHT];
    char tmp;
    char* all_Data[CREDENTIALS];
    FILE* file = fopen("camera_credentials.txt", "r");
    int size;
    memset(data, 0, MAX_ARRAY_LENGHT);

    while(tmp != EOF){
        tmp = fgetc(file);
        //printf("%c", tmp);
        if(tmp == '#'){
            size = 0;
            while(tmp != '='){
                tmp = fgetc(file);
                if(tmp != '='){
                    data[size] = tmp;
                    size++;
                }
            }
            //printf("%s ", data);
            if(strcmp(data, "PROTOCOL") == 0){
                size = readData(file, data, size);
                all_Data[PROTOCOL] = new char[size];
                strcpy(all_Data[PROTOCOL], data);
                //printf("PROTOCOL: %s\n", all_Data[PROTOCOL]);
            }
            else if(strcmp(data, "USER") == 0){
                size = readData(file, data, size);
                all_Data[USER] = new char[size];
                strcpy(all_Data[USER], data);
                //printf("user: %s", all_Data[USER]);
            }
            else if(strcmp(data, "PASS") == 0){
                size = readData(file, data, size);
                all_Data[PASS] = new char[size];
                strcpy(all_Data[PASS], data);
                //printf("pass: %s", all_Data[PASS]);
            }
            else if(strcmp(data, "IP") == 0){
                size = readData(file, data, size);
                all_Data[IP] = new char[size];
                strcpy(all_Data[IP], data);
                //printf("ip: %s", all_Data[IP]);
            }
            else if(strcmp(data, "PORT") == 0){
                size = readData(file, data, size);
                all_Data[PORT] = new char[size];
                strcpy(all_Data[PORT], data);
                //printf("port: %s", all_Data[PORT]);
            }
            else if(strcmp(data, "ROUTE") == 0){
                size = readData(file, data, size);
                all_Data[ROUTE] = new char[size];
                strcpy(all_Data[ROUTE], data);
                //printf("route: %s", all_Data[ROUTE]);
            }

            //printf("\naqui\n");
            memset(data, 0, size);
        }
    }

    fclose(file);

    strcpy(data, all_Data[PROTOCOL]);
    strcat(data, "://");
    strcat(data, all_Data[USER]);
    strcat(data, ":");
    strcat(data, all_Data[PASS]);
    strcat(data, "@");
    strcat(data, all_Data[IP]);
    strcat(data, ":");
    strcat(data, all_Data[PORT]);
    strcat(data, all_Data[ROUTE]);
    //printf("data: %s", data);

    for(int i = PROTOCOL; i <= ROUTE; i++){
        delete all_Data[i];
    }

    return data;
}

int readData(FILE* file, char* data, int size){
    char tmp = ' ';
    memset(data, 0, size);
    size = 0;
    while(tmp != '\n' && tmp != EOF){
        tmp = fgetc(file);
        if(tmp != '\n' && tmp != EOF){
            data[size] = tmp;
            size++;
        }
    }
    //printf("%s\n", data);
    return size;
}

void mouseCallBack(int event, int x, int y, int flags, void* userdata){
    if  ( event == cv::EVENT_LBUTTONDOWN ){
        rec_Button->pressed(x, y);
        reconnect_Button->pressed(x, y);
    }
    else if  ( event == cv::EVENT_RBUTTONDOWN ){
        
    }
    else if  ( event == cv::EVENT_MBUTTONDOWN ){

    }
    else if ( event == cv::EVENT_MOUSEMOVE ){

    }
}