/*
 * video.cpp: A plugin for the Video Disk Recorder
 *
 * See the README file for copyright information and how to reach the author.
 *
 * $Id$
 */

#include "video.h"

cMarkAdLogo::cMarkAdLogo(int RecvNumber, MarkAdContext *maContext)
{
    macontext=maContext;

    // 3x3 GX Sobel mask
    GX[0][0] = -1;
    GX[0][1] =  0;
    GX[0][2] =  1;
    GX[1][0] = -2; // 2
    GX[1][1] =  0;
    GX[1][2] =  2; // 2
    GX[2][0] = -1;
    GX[2][1] =  0;
    GX[2][2] =  1;

    // 3x3 GY Sobel mask
    GY[0][0] =  1;
    GY[0][1] =  2; // 2
    GY[0][2] =  1;
    GY[1][0] =  0;
    GY[1][1] =  0;
    GY[1][2] =  0;
    GY[2][0] = -1;
    GY[2][1] = -2; //2
    GY[2][2] = -1;

    plane=NULL;
    first=true;
}

cMarkAdLogo::~cMarkAdLogo()
{
    if (plane) free(plane);
}

void cMarkAdLogo::SaveFrame(int LastIFrame)
{
    if (!macontext) return;
    if (!macontext->Video.Info.Width) return;
    if (!plane) return;

    FILE *pFile;
    char szFilename[32];

    // Open file
    sprintf(szFilename, "frame%06d.pgm", LastIFrame);
    pFile=fopen(szFilename, "wb");
    if (pFile==NULL)
        return;

    // Write header
    fprintf(pFile, "P5\n%d %d\n255\n", macontext->Video.Info.Width,LOGOHEIGHT);

    // Write pixel data
    fwrite(plane,1,macontext->Video.Info.Width*LOGOHEIGHT,pFile);
    // Close file
    fclose(pFile);

    dsyslog("markad saved frame %i",LastIFrame);
    if (LastIFrame>750) abort();
}

int cMarkAdLogo::Process(int LastIFrame)
{
    if (!macontext) return 0;
    if (!macontext->Video.Info.Width) return 0;
    if (!macontext->Video.Data.Valid) return 0;

    if (!plane)
    {
        plane=(uchar *) malloc(LOGOHEIGHT*macontext->Video.Info.Width);
        if (!plane) return 0;
    }

    int SUM;
    int sumX,sumY;

    for (int Y=0; Y<=LOGOHEIGHT-1; Y++)
    {
        for (int X=0; X<=macontext->Video.Info.Width-1; X++)
        {
            sumX=0;
            sumY=0;

            // image boundaries
            if (Y==0 || Y==LOGOHEIGHT-1)
                SUM=0;
            else if (X==0 || X==macontext->Video.Info.Width-1)
                SUM=0;
            // convolution starts here
            else
            {
                // X Gradient approximation
                for (int I=-1; I<=1; I++)
                {
                    for (int J=-1; J<=1; J++)
                    {
                        sumX=sumX+ (int) ((*(macontext->Video.Data.Plane[0]+X+I+
                                             (Y+J)*macontext->Video.Info.Width))*GX[I+1][J+1]);
                    }
                }

                // Y Gradient approximation
                for (int I=-1; I<=1; I++)
                {
                    for (int J=-1; J<=1; J++)
                    {
                        sumY=sumY+ (int) ((*(macontext->Video.Data.Plane[0]+X+I+
                                             (Y+J)*macontext->Video.Info.Width))*GY[I+1][J+1]);
                    }
                }

                // Gradient Magnitude approximation
                SUM = abs(sumX) + abs(sumY);
            }

            if (SUM>=127) SUM=255;
            if (SUM<127) SUM=0;

            plane[X+Y*macontext->Video.Info.Width]=255-(uchar) (SUM);
        }
    }
    //SaveFrame(LastIFrame);

    return 0;
}

cMarkAdBlackBordersHoriz::cMarkAdBlackBordersHoriz(int RecvNumber, MarkAdContext *maContext)
{
    macontext=maContext;

    borderstatus=false;
    borderiframe=-1;
    borderstarttime=0;
}

void cMarkAdBlackBordersHoriz::SaveFrame(int LastIFrame)
{
    if (!macontext) return;
    if (!macontext->Video.Data.Valid) return;

    if (macontext->Video.Data.PlaneLinesize[0]!=macontext->Video.Info.Width) return;

    FILE *pFile;
    char szFilename[32];

    // Open file
    sprintf(szFilename, "frame%06d.pgm", LastIFrame);
    pFile=fopen(szFilename, "wb");
    if (pFile==NULL)
        return;

    // Write header
    fprintf(pFile, "P5\n%d %d\n255\n", macontext->Video.Info.Width,
            macontext->Video.Info.Width);

    // Write pixel data
    fwrite(macontext->Video.Data.Plane[0],1,macontext->Video.Info.Width*
           macontext->Video.Info.Height,pFile);
    // Close file
    fclose(pFile);
}

int cMarkAdBlackBordersHoriz::Process(int LastIFrame, int *BorderIFrame)
{
#define CHECKHEIGHT 20
#define BRIGHTNESS 20
    if (!macontext) return 0;
    if (!macontext->Video.Data.Valid) return 0;
    return 0;

    *BorderIFrame=borderiframe;

    int x,y;
    bool ftop=true,fbottom=true;

    if (macontext->Video.Data.PlaneLinesize[0]!=macontext->Video.Info.Width)
    {
        // slow (?) method
        for (y=(macontext->Video.Info.Height-CHECKHEIGHT); y<macontext->Video.Info.Height; y++)
        {
            for (x=0; x<macontext->Video.Info.Width; x++)
            {
                if (macontext->Video.Data.Plane[0][y*macontext->Video.Data.PlaneLinesize[0]+x]>
                        BRIGHTNESS)
                {
                    fbottom=false;
                    y=macontext->Video.Info.Height;
                    break;
                }
            }
        }

        if (fbottom)
        {
            for (y=0; y<CHECKHEIGHT; y++)
            {
                for (x=0; x<macontext->Video.Info.Width; x++)
                {
                    if (macontext->Video.Data.Plane[0][y*macontext->Video.Data.PlaneLinesize[0]+x]
                            >BRIGHTNESS)
                    {
                        ftop=false;
                        y=CHECKHEIGHT;
                        break;
                    }
                }
            }
        }
    }
    else
    {
        // "fast" method
        for (x=(macontext->Video.Info.Height-CHECKHEIGHT)*macontext->Video.Info.Width;
                x<macontext->Video.Info.Height*macontext->Video.Info.Width; x++)
        {
            if (macontext->Video.Data.Plane[0][x]>BRIGHTNESS) fbottom=false;
        }

        if (fbottom)
        {
            for (x=0; x<(macontext->Video.Info.Width*CHECKHEIGHT); x++)
            {
                if (macontext->Video.Data.Plane[0][x]>BRIGHTNESS) ftop=false;
            }
        }
    }

    if ((fbottom) && (ftop))
    {
        if (!borderstatus)
        {
            if (!borderstarttime)
            {
                borderiframe=LastIFrame;
                borderstarttime=time(NULL);
                borderstatus=false;
            }
            else
            {
                if ((time(NULL)>(borderstarttime+20)))
                {
                    borderstatus=true;
                    return 1; // detected black border
                }
            }
        }
    }
    else
    {
        if (borderstatus)
        {
            borderiframe=LastIFrame;
            borderstarttime=0;
            borderstatus=false;
            return -1;
        }
        else
        {
            borderiframe=-1;
            borderstarttime=0;
            return 0;
        }
    }
    return 0;
}


cMarkAdVideo::cMarkAdVideo(int RecvNumber,MarkAdContext *maContext)
{
    macontext=maContext;
    recvnumber=RecvNumber;

    aspectratio.Num=0;
    aspectratio.Den=0;
    mark.Comment=NULL;
    mark.Position=0;

    hborder=new cMarkAdBlackBordersHoriz(RecvNumber,maContext);
    logo = new cMarkAdLogo(RecvNumber,maContext);
}

cMarkAdVideo::~cMarkAdVideo()
{
    ResetMark();
    if (hborder) delete hborder;
    if (logo) delete logo;
}

void cMarkAdVideo::ResetMark()
{
    if (mark.Comment) free(mark.Comment);
    mark.Comment=NULL;
    mark.Position=0;
}

bool cMarkAdVideo::AddMark(int Position, const char *Comment)
{
    if (!Comment) return false;
    if (mark.Comment)
    {
        int oldlen=strlen(mark.Comment);
        mark.Comment=(char *) realloc(mark.Comment,oldlen+10+strlen(Comment));
        if (!mark.Comment)
        {
            mark.Position=0;
            return false;
        }
        strcat(mark.Comment," [");
        strcat(mark.Comment,Comment);
        strcat(mark.Comment,"]");
    }
    else
    {
        mark.Comment=strdup(Comment);
    }
    mark.Position=Position;
    return true;
}

bool cMarkAdVideo::AspectRatioChange(MarkAdAspectRatio *a, MarkAdAspectRatio *b)
{
    if ((!a) || (!b)) return false;

    if (a->Num==0 || a->Den==0 || b->Num==0 || b->Den==0) return false;
    if ((a->Num!=b->Num) && (a->Den!=b->Den)) return true;
    return false;

}


MarkAdMark *cMarkAdVideo::Process(int LastIFrame)
{
    ResetMark();
    if (!LastIFrame) return NULL;

    logo->Process(LastIFrame);

    if (macontext->State.ContentStarted)
    {
        int borderiframe;
        int hret=hborder->Process(LastIFrame,&borderiframe);

        if ((hret>0) && (borderiframe))
        {
            char *buf=NULL;
            if (asprintf(&buf,"detected start of horiz. borders (%i)",borderiframe)!=-1)
            {
                isyslog("markad [%i]: %s",recvnumber,buf);
                AddMark(borderiframe,buf);
                free(buf);
            }
        }

        if ((hret<0) && (borderiframe))
        {
            char *buf=NULL;
            if (asprintf(&buf,"detected stop of horiz. borders (%i)",borderiframe)!=-1)
            {
                isyslog("markad [%i]: %s",recvnumber,buf);
                AddMark(borderiframe,buf);
                free(buf);
            }
        }
    }

    if (AspectRatioChange(&macontext->Video.Info.AspectRatio,&aspectratio))
    {
        char *buf=NULL;
        if (asprintf(&buf,"aspect ratio change from %i:%i to %i:%i (%i)",
                     aspectratio.Num,aspectratio.Den,
                     macontext->Video.Info.AspectRatio.Num,
                     macontext->Video.Info.AspectRatio.Den,LastIFrame)!=-1)
        {
            isyslog("markad [%i]: %s",recvnumber, buf);
            AddMark(LastIFrame,buf);
            free(buf);
        }
    }

    aspectratio.Num=macontext->Video.Info.AspectRatio.Num;
    aspectratio.Den=macontext->Video.Info.AspectRatio.Den;

    return &mark;
}