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

#include "video.h"

cMarkAdLogo::cMarkAdLogo(MarkAdContext *maContext)
{
    macontext=maContext;

    // 3x3 GX Sobel mask

    GX[0][0] = -1;
    GX[0][1] =  0;
    GX[0][2] =  1;
    GX[1][0] = -2;
    GX[1][1] =  0;
    GX[1][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;
    GY[0][2] =  1;
    GY[1][0] =  0;
    GY[1][1] =  0;
    GY[1][2] =  0;
    GY[2][0] = -1;
    GY[2][1] = -2;
    GY[2][2] = -1;

    memset(&area,0,sizeof(area));

    LOGOHEIGHT=LOGO_DEFHEIGHT;
    if (maContext->Info.VPid.Type==MARKAD_PIDTYPE_VIDEO_H264)
    {
        LOGOWIDTH=LOGO_DEFHDWIDTH;
    }
    else
    {
        LOGOWIDTH=LOGO_DEFWIDTH;
    }

    area.status=UNINITIALIZED;
}

cMarkAdLogo::~cMarkAdLogo()
{
}

int cMarkAdLogo::Load(char *directory, char *file)
{
    char *path;
    if (asprintf(&path,"%s/%s.pgm",directory,file)==-1) return -3;

    // Load mask
    FILE *pFile;
    area.valid=false;
    area.corner=-1;
    pFile=fopen(path, "rb");
    free(path);
    if (!pFile) return -1;

    fscanf(pFile, "P5\n#C%i %i\n%d %d\n255\n#", &area.corner,&area.mpixel,&LOGOWIDTH,&LOGOHEIGHT);

    if (LOGOHEIGHT==255)
    {
        LOGOHEIGHT=LOGOWIDTH;
        LOGOWIDTH=area.mpixel;
        area.mpixel=0;
    }

    if ((LOGOWIDTH<=0) || (LOGOHEIGHT<=0) || (LOGOWIDTH>LOGO_MAXWIDTH) || (LOGOHEIGHT>LOGO_MAXHEIGHT) ||
            (area.corner<TOP_LEFT) || (area.corner>BOTTOM_RIGHT))
    {
        fclose(pFile);
        return -2;
    }

    fread(&area.mask,1,LOGOWIDTH*LOGOHEIGHT,pFile);

    if (!area.mpixel)
    {
        for (int i=0; i<LOGOWIDTH*LOGOHEIGHT; i++)
        {
            if (!area.mask[i]) area.mpixel++;
        }
    }

    fclose(pFile);
    area.valid=true;
    return 0;
}



void cMarkAdLogo::Save(int lastiframe, uchar *picture)
{
    if (!macontext) return;


    char *buf=NULL;
    if (asprintf(&buf,"%s/%06d-%s-A%i_%i.pgm","/tmp/",lastiframe,
                 macontext->Info.ChannelID,
                 area.aspectratio.Num,area.aspectratio.Den)!=-1)
    {
        // Open file
        FILE *pFile=fopen(buf, "wb");
        if (pFile==NULL)
        {
            free(buf);
            return;
        }

        // Write header
        fprintf(pFile, "P5\n#C%i\n%d %d\n255\n", area.corner, LOGOWIDTH,LOGOHEIGHT);

        // Write pixel data
        fwrite(picture,1,LOGOWIDTH*LOGOHEIGHT,pFile);
        // Close file
        fclose(pFile);
        free(buf);
    }
}

int cMarkAdLogo::Detect(int lastiframe, int *logoiframe)
{
    // Detection is made with Sobel-Operator

    if (!macontext) return 0;
    if (!macontext->Video.Info.Width) return 0;
    if (!macontext->Video.Info.Height) return 0;
    if (!macontext->Video.Data.Valid) return 0;

    if (area.corner>BOTTOM_RIGHT) return 0;
    if (area.corner<TOP_LEFT) return 0;

    int xstart,xend,ystart,yend;

    switch (area.corner)
    {
    case TOP_LEFT:
        xstart=0;
        xend=LOGOWIDTH;
        ystart=0;
        yend=LOGOHEIGHT;
        break;
    case TOP_RIGHT:
        xstart=macontext->Video.Info.Width-LOGOWIDTH;
        xend=macontext->Video.Info.Width;
        ystart=0;
        yend=LOGOHEIGHT;
        break;
    case BOTTOM_LEFT:
        xstart=0;
        xend=LOGOWIDTH;
        ystart=macontext->Video.Info.Height-LOGOHEIGHT;
        yend=macontext->Video.Info.Height;
        break;
    case BOTTOM_RIGHT:
        xstart=macontext->Video.Info.Width-LOGOWIDTH;
        xend=macontext->Video.Info.Width;
        ystart=macontext->Video.Info.Height-LOGOHEIGHT;
        yend=macontext->Video.Info.Height;
        break;
    default:
        return 0;
    }

    int SUMA=0;
    for (int Y=ystart; Y<=yend-1; Y++)
    {
        for (int X=xstart; X<=xend-1; X++)
        {
            int val=macontext->Video.Data.Plane[0][X+(Y*macontext->Video.Data.PlaneLinesize[0])];
            area.source[(X-xstart)+(Y-ystart)*LOGOWIDTH]=val;
            SUMA+=val;
        }
    }

    SUMA/=(LOGOWIDTH*LOGOHEIGHT);
#if 0
    if (SUMA>=100)
    {
        int maxval=(int) SUMA;
        SUMA=0;
        for (int Y=ystart; Y<=yend-1; Y++)
        {
            for (int X=xstart; X<=xend-1; X++)
            {
                int val=macontext->Video.Data.Plane[0][X+(Y*macontext->Video.Data.PlaneLinesize[0])];
                val=(int) (((double) val- (double) maxval/1.4)*1.4);
                if (val>maxval) val=maxval;
                if (val<0) val=0;
                area.source[(X-xstart)+(Y-ystart)*LOGOWIDTH]=val;
                SUMA+=val;
            }
        }
        SUMA/=(LOGOWIDTH*LOGOHEIGHT);
    }
#endif
    int ret=NOCHANGE;

    if ((SUMA<100) || ((area.status==LOGO) && (SUMA<180)))
    {

        int SUM;
        int sumX,sumY;
        area.rpixel=0;
        for (int Y=ystart; Y<=yend-1; Y++)
        {
            for (int X=xstart; X<=xend-1; X++)
            {
                sumX=0;
                sumY=0;

                // image boundaries
                if (Y<(ystart+15) || Y>(yend-15))
                    SUM=0;
                else if (X<(xstart+15) || X>(xend-15))
                    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.Data.PlaneLinesize[0]))
                                              *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.Data.PlaneLinesize[0]))*
                                              GY[I+1][J+1]);
                        }
                    }

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

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

                int val = 255-(uchar) SUM;

                area.sobel[(X-xstart)+(Y-ystart)*LOGOWIDTH]=val;

                area.result[(X-xstart)+(Y-ystart)*LOGOWIDTH]=
                    (area.mask[(X-xstart)+(Y-ystart)*LOGOWIDTH] + val) & 255;

                if (!area.result[(X-xstart)+(Y-ystart)*LOGOWIDTH]) area.rpixel++;

            }
        }
        if (macontext->Options.LogoExtraction==-1)
        {
            if (area.status==UNINITIALIZED)
            {
                // Initialize
                if (area.rpixel>(area.mpixel*LOGO_VMARK))
                {
                    area.status=LOGO;
                }
                else
                {
                    area.status=NOLOGO;
                }
            }

            if (area.rpixel>=(area.mpixel*LOGO_VMARK))
            {
                if (area.status==NOLOGO)
                {
                    if (area.counter>=LOGO_VMAXCOUNT)
                    {
                        area.status=ret=LOGO;
                        *logoiframe=area.lastiframe;
                        area.counter=0;
                    }
                    else
                    {
                        if (!area.counter) area.lastiframe=lastiframe;
                        area.counter++;
                    }
                }
                else
                {
                    area.lastiframe=lastiframe;
                    area.counter=0;
                }
            }

            if (area.rpixel<(area.mpixel*LOGO_IMARK))
            {
                if (area.status==LOGO)
                {
                    if (area.counter>=LOGO_IMAXCOUNT)
                    {
                        area.status=ret=NOLOGO;
                        *logoiframe=area.lastiframe;
                        area.counter=0;
                    }
                    else
                    {
                        area.counter++;
                    }
                }
                else
                {
                    area.counter=0;
                }
            }

            if ((area.rpixel<(area.mpixel*LOGO_VMARK)) && (area.rpixel>(area.mpixel*LOGO_IMARK)))
            {
                area.counter=0;
            }

#if 0
            printf("%5i %3i %4i %4i %i %i %i\n",lastiframe,SUMA,area.rpixel,area.mpixel,
                   (area.rpixel>=(area.mpixel*LOGO_VMARK)),(area.rpixel<(area.mpixel*LOGO_IMARK)),
                   area.counter  );
            Save(lastiframe,area.sobel); // TODO: JUST FOR DEBUGGING!
#endif
        }
        else
        {
            Save(lastiframe,area.sobel);
        }
    }
    return ret;
}

int cMarkAdLogo::Process(int LastIFrame, int *LogoIFrame)
{
    if (!macontext) return ERROR;
    if (!macontext->Video.Data.Valid) return ERROR;
    if (!macontext->Video.Info.Width) return ERROR;
    if (!macontext->LogoDir) return ERROR;
    if (!macontext->Info.ChannelID) return ERROR;

    if (macontext->Options.LogoExtraction==-1)
    {

        if ((area.aspectratio.Num!=macontext->Video.Info.AspectRatio.Num) ||
                (area.aspectratio.Den!=macontext->Video.Info.AspectRatio.Den))
        {
            area.valid=false;  // just to be sure!
            char *buf=NULL;
            if (asprintf(&buf,"%s-A%i_%i",macontext->Info.ChannelID,
                         macontext->Video.Info.AspectRatio.Num,macontext->Video.Info.AspectRatio.Den)!=-1)
            {
                int ret=Load(macontext->LogoDir,buf);
                switch (ret)
                {
                case -1:
                    isyslog("no logo for %s",buf);
                    break;
                case -2:
                    esyslog("format error in %s",buf);
                    break;
                case -3:
                    esyslog("cannot load %s",buf);
                    break;
                }
                free(buf);
            }
            area.aspectratio.Num=macontext->Video.Info.AspectRatio.Num;
            area.aspectratio.Den=macontext->Video.Info.AspectRatio.Den;
        }
    }
    else
    {
        area.aspectratio.Num=macontext->Video.Info.AspectRatio.Num;
        area.aspectratio.Den=macontext->Video.Info.AspectRatio.Den;
        area.corner=macontext->Options.LogoExtraction;
        if (macontext->Options.LogoWidth!=-1)
        {
            LOGOWIDTH=macontext->Options.LogoWidth;
        }
        if (macontext->Options.LogoHeight!=-1)
        {
            LOGOHEIGHT=macontext->Options.LogoHeight;
        }
        area.valid=true;
    }

    if (!area.valid) return ERROR;

    return Detect(LastIFrame,LogoIFrame);
}

cMarkAdBlackBordersHoriz::cMarkAdBlackBordersHoriz(MarkAdContext *maContext)
{
    macontext=maContext;

    borderstatus=UNINITIALIZED;
    borderiframe=-1;
}

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

    int height=macontext->Video.Info.Height-OFFSET;

    int start=(height-CHECKHEIGHT)*macontext->Video.Data.PlaneLinesize[0];
    int end=height*macontext->Video.Data.PlaneLinesize[0];
    bool ftop=true,fbottom=true;
    int val=0,cnt=0,xz=0;

    for (int x=start; x<end; x++)
    {
        if (xz<macontext->Video.Info.Width)
        {
            val+=macontext->Video.Data.Plane[0][x];
            cnt++;
        }
        xz++;
        if (xz>=macontext->Video.Data.PlaneLinesize[0]) xz=0;
    }
    val/=cnt;
    if (val>BRIGHTNESS) fbottom=false;

    if (fbottom)
    {
        start=OFFSET*macontext->Video.Data.PlaneLinesize[0];
        end=macontext->Video.Data.PlaneLinesize[0]*(CHECKHEIGHT+OFFSET);
        val=0;
        cnt=0;
        xz=0;
        for (int x=start; x<end; x++)
        {
            if (xz<macontext->Video.Info.Width)
            {
                val+=macontext->Video.Data.Plane[0][x];
                cnt++;
            }
            xz++;
            if (xz>=macontext->Video.Data.PlaneLinesize[0]) xz=0;
        }
        val/=cnt;
        if (val>BRIGHTNESS) ftop=false;
    }

    if ((fbottom) && (ftop))
    {
        if (borderiframe==-1)
        {
            borderiframe=LastIFrame;
        }
        else
        {
#define MINSECS 420
            switch (borderstatus)
            {
            case UNINITIALIZED:
                if (LastIFrame>(borderiframe+macontext->Video.Info.FramesPerSecond*MINSECS))
                {
                    borderstatus=BORDER;
                }
                break;

            case NOBORDER:
                if (LastIFrame>(borderiframe+macontext->Video.Info.FramesPerSecond*MINSECS))
                {
                    *BorderIFrame=borderiframe;
                    borderstatus=BORDER;
                    return 1; // detected start of black border
                }
                break;

            case BORDER:
                borderiframe=LastIFrame;
                break;
            }
        }
    }
    else
    {
        if (borderiframe!=-1)
        {
            if (borderstatus==BORDER)
            {
                *BorderIFrame=borderiframe;
                borderstatus=NOBORDER;
                borderiframe=-1;
                return -1; // detected stop of black border
            }
            else
            {
                borderiframe=-1;
            }
        }
        else
        {
            borderiframe=-1;
        }
    }
    return 0;
}


cMarkAdVideo::cMarkAdVideo(MarkAdContext *maContext)
{
    macontext=maContext;

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

    hborder=new cMarkAdBlackBordersHoriz(maContext);
    logo = new cMarkAdLogo(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;
    mark.Type=0;
}

bool cMarkAdVideo::AddMark(int Type, 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;
    mark.Type=Type;
    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;

    if (!macontext->Video.Options.IgnoreLogoDetection)
    {
        int logoiframe;
        int lret=logo->Process(LastIFrame,&logoiframe);
        if ((lret>=-1) && (lret!=0))
        {
            char *buf=NULL;
            if (lret>0)
            {
                if (asprintf(&buf,"detected logo start (%i)",logoiframe)!=-1)
                {
                    AddMark(MT_LOGOSTART,logoiframe,buf);
                    free(buf);
                }
            }
            else
            {
                if (asprintf(&buf,"detected logo stop (%i)",logoiframe)!=-1)
                {
                    AddMark(MT_LOGOSTOP,logoiframe,buf);
                    free(buf);
                }
            }
        }
    }

    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)
        {
            AddMark(MT_BORDERSTART,borderiframe,buf);
            free(buf);
        }
    }

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

    if (!macontext->Video.Options.IgnoreAspectRatio)
    {
        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)
            {
                if ((macontext->Info.AspectRatio.Num) && (macontext->Info.AspectRatio.Den))
                {
                    if ((macontext->Video.Info.AspectRatio.Num==macontext->Info.AspectRatio.Num) &&
                            (macontext->Video.Info.AspectRatio.Num==macontext->Info.AspectRatio.Num))
                    {
                        AddMark(MT_ASPECTSTART,LastIFrame,buf);
                    }
                    else
                    {
                        AddMark(MT_ASPECTSTOP,LastIFrame,buf);
                    }
                }
                else
                {
                    AddMark(MT_ASPECTCHANGE,LastIFrame,buf);
                }
                free(buf);
            }
        }

        aspectratio.Num=macontext->Video.Info.AspectRatio.Num;
        aspectratio.Den=macontext->Video.Info.AspectRatio.Den;
    }
    return &mark;
}