#include <stdlib.h>
#include <stdio.h>
#include <jpeglib.h>
#include <UDPTCPNetwork.h>

#include "videoframe.h"
#include "inmemoryfile.h"

VideoFrame::VideoFrame() {
    mem = NULL;
    width = 0;
    height = 0;
    mem_allocated = 0;
};

VideoFrame::~VideoFrame() {
    FreeFrame();
};


void VideoFrame::FreeFrame() {
    if (mem != NULL) {
        free (mem);
        mem = NULL;
        width = 0;
        height = 0;
        mem_allocated = 0;
    }
};


void VideoFrame::AllocateFrame() {
    int memnewsize = width * height * 3;
    if (memnewsize < mem_allocated) return;
    else if (memnewsize == 0) FreeFrame();

    mem = (unsigned char *) realloc (mem, memnewsize);
    mem_allocated = memnewsize;

    if (mem == NULL) {
        debug ("Error on allocation new frame\n");
        exit (1);
    }
};


int VideoFrame::SetSize(int w, int h) {
    if (w < 0 && h < 0) return 0;

    width = w;
    height = h;

    AllocateFrame();

    return 1;
};


int VideoFrame::ConvertToJpeg(InMemoryFile *imf, int quality) {
    unsigned char *outbuffer = NULL;
	size_t outbuffersize;
    struct jpeg_compress_struct cinfo;
    struct jpeg_error_mgr jerr;
    JSAMPROW row_pointer[1];	/* pointer to JSAMPLE row[s] */
    int row_stride;		        /* physical row width in image buffer */

	if (imf == NULL) return 0;
    if (height == 0 || width == 0) return 0;

    cinfo.err = jpeg_std_error(&jerr);
    jpeg_create_compress(&cinfo);

    jpeg_mem_dest(&cinfo, &outbuffer, &outbuffersize);

    cinfo.image_width = width; 	/* image width and height, in pixels */
    cinfo.image_height = height;
    cinfo.input_components = 3;		/* # of color components per pixel */
    cinfo.in_color_space = JCS_RGB; 	/* colorspace of input image */
    jpeg_set_defaults(&cinfo);
    jpeg_set_quality(&cinfo, quality, TRUE /* limit to baseline-JPEG values */);
    jpeg_start_compress(&cinfo, TRUE);
    row_stride = width * 3;	/* JSAMPLEs per row in image_buffer */

    while (cinfo.next_scanline < cinfo.image_height) {
        row_pointer[0] = & mem[cinfo.next_scanline * row_stride];
        (void) jpeg_write_scanlines(&cinfo, row_pointer, 1);
    }

    jpeg_finish_compress(&cinfo);
    jpeg_destroy_compress(&cinfo);

	imf->CopyFrom(outbuffer, outbuffersize);
	free (outbuffer);
	
    return 1;
};


int VideoFrame::TestScreen(int w, int h) {
    int x, y;
    unsigned char r, g, b;

    SetSize (w, h);

    r = g = b = 0;
    for (x = 0; x < w; x++) {
        for (g = 0, y = 0; y < h; y++) {
            b = r + g;

            mem[3*(x + y * width) + 0] = r;
            mem[3*(x + y * width) + 1] = g;
            mem[3*(x + y * width) + 2] = b;

            g++;
        }
        r ++;
    }

    return 1;
}


/*********************************************************************/



void VideoFrameFloat::AllocateFrame() {
    printf ("VideoFrameFloat::AllocateFrame()\n");
    int memnewsize = width * height * 3 * sizeof(float);
    if (memnewsize >= mem_allocated) return;
    else if (memnewsize == 0) FreeFrame();

    mem = (unsigned char *) realloc (mem, memnewsize);
    mem_allocated = memnewsize;

    if (mem == NULL) {
        debug ("Error on allocation new frame\n");
        exit (1);
    }
};