grabber.h

Go to the documentation of this file.

#ifndef __GRABBER_H_
#define __GRABBER_H_

#include <libavformat/avformat.h>  // for AVStream
#include <libavutil/frame.h>       // for AVFrame
#include <libswscale/swscale.h>    // for SWS_BICUBIC
#include <pthread.h>               // for pthread_mutex_t, pthread_cond_t
#include <stdint.h>                // for uint8_t
#include <X11/Xlib.h>
#include "buffer_sizes.h"          // for BUF_SIZE
#include "socket.h"                // for socket_t

#define PORT_GRAB 32500

#define STREAM_DURATION 60.0

#define STREAM_FRAME_RATE 60

#define STREAM_PIX_FMT AV_PIX_FMT_YUV420P

#define SCALE_FLAGS SWS_BICUBIC

#define CLIP(X) ((X) > 255 ? 255 : (X) < 0 ? 0 : X)

// RGB -> YUV
#define RGB2Y(R, G, B) CLIP(((66 * (R) + 129 * (G) + 25 * (B) + 128) >> 8) + 16)
#define RGB2U(R, G, B) CLIP(((-38 * (R) -74 * (G) + 112 * (B) + 128) >> 8) + 128)
#define RGB2V(R, G, B) CLIP(((112 * (R) -94 * (G) -18 * (B) + 128) >> 8) + 128)


typedef struct read_args
{
    pthread_mutex_t mtx;
    pthread_cond_t cond;
    uint8_t* buffer;
    socket_t sock;
    int len;
    int flagSnapRec;
    int flagRecording;
} s_read_args;

typedef struct thread_args
{
    pthread_mutex_t mtx;
    char record_filename[BUF_SIZE];
} s_thread_args;

void grab_snapshot(char* snap_filename);

typedef struct OutputStream
{
    AVStream* st;

    /* pts of the next frame that will be generated */
    int64_t next_pts;
    int samples_count;

    AVFrame* frame;
    AVFrame* tmp_frame;

    float t, tincr, tincr2;

    struct SwsContext* sws_ctx;
    struct SwrContext* swr_ctx;
} OutputStream;

int ffmpeg_grabber(void* arg);

unsigned char* xgrabber();

void* grab_handler_sock();

void* grab_handler_amqp(void* args);

void grabber_set_path_results(char* results);

void grabber_set_display(Display* display);
#endif