#include "raw_merge.h"
#include "stdlib.h"
#include "platform.h"
#include "gui_mpopup.h"
#include "gui_mbox.h"
#include "raw.h"
#include "gui_lang.h"
#include "lang.h"
#include "conf.h"
#define TEMP_FILE_NAME   "A/raw16.tmp"
#define TEMP_FILE_NAME_1 "A/raw16_1.tmp"

static int raw_action;
static int raw_count;
static unsigned short *row;
static unsigned char *rawrow;
static char namebuf[100];

// note: if processing with dcraw etc, zeros may get replaced with interpolated values
// this may or may not be what you want
static int raw_subtract_values(int from, int sub) {
/*    if ( sub > conf.sub_in_dark_value ) {
        int result = from - sub;
        if ( result < conf.sub_out_dark_value ) {
            return conf.sub_out_dark_value;
        }
        else {
            return result;
        }
    }
    else {
        return from;
    }*/

 int result;
 if ((from==0) || (sub==0)) return 0; // bad pixel
 result = from - sub + CAM_BLACK_LEVEL;
 if (result<CAM_BLACK_LEVEL) result=CAM_BLACK_LEVEL;
 if (result>CAM_WHITE_LEVEL) result=CAM_WHITE_LEVEL;
 return result;

}
/* subtract "sub" from "from" and store the result in "dest"*/
/* TODO allow replacing if dest == from or sub*/
int raw_subtract(const char *from, const char *sub, const char *dest) {
    unsigned req=(hook_raw_size()>>10) + 1;
    unsigned avail=GetFreeCardSpaceKb();
    FILE *ffrom = NULL, *fsub = NULL, *fdest = NULL;
    char *baccum = 0,*bsub = 0;
    int status = 0;
    unsigned short s,d;
    int i,j;

    static struct utimbuf t;

    struct stat st;

    if (stat((char *)from,&st) != 0 || st.st_size!=hook_raw_size()) 
        return 0;

    if (stat((char *)sub,&st) != 0 || st.st_size!=hook_raw_size()) 
        return 0;

     if( (baccum=malloc(RAW_ROWLEN)) &&
        (bsub=malloc(RAW_ROWLEN)) &&
        (ffrom=fopen(from, "rb")) &&
        (fsub=fopen(sub, "rb")) &&
        (fdest=fopen(dest, "wb")) &&
        avail > req)
    {
        started();
        for (j = 0; j < CAM_RAW_ROWS; j++) {
            fread(baccum,1, RAW_ROWLEN,ffrom);
            fread(bsub,1, RAW_ROWLEN,fsub);

#if CAM_SENSOR_BITS_PER_PIXEL==10

            for(i = 0;i<RAW_ROWLEN; i+=10) {
                s =((0x3fc&(((unsigned short)bsub[i+1])<<2)) | (bsub[i+0] >> 6));
                d =((0x3fc&(((unsigned short)baccum[i+1])<<2)) | (baccum[i+0] >> 6));
                d = raw_subtract_values(d,s);
                baccum[i]=(baccum[i]&0x3F)|(d<<6); 
                baccum[i+1]=d>>2;

                s =((0x3f0&(((unsigned short)bsub[i+0])<<4)) | (bsub[i+3] >> 4));
                d =((0x3f0&(((unsigned short)baccum[i+0])<<4)) | (baccum[i+3] >> 4));
                d = raw_subtract_values(d,s);
                baccum[i]=(baccum[i]&0xC0)|(d>>4);
                baccum[i+3]=(baccum[i+3]&0x0F)|(d<<4);

                s =((0x3c0&(((unsigned short)bsub[i+3])<<6)) | (bsub[i+2] >> 2));
                d =((0x3c0&(((unsigned short)baccum[i+3])<<6)) | (baccum[i+2] >> 2));
                d = raw_subtract_values(d,s);
                baccum[i+2]=(baccum[i+2]&0x03)|(d<<2);
                baccum[i+3]=(baccum[i+3]&0xF0)|(d>>6);

                s =((0x300&(((unsigned short)bsub[i+2])<<8)) | (bsub[i+5])); 
                d =((0x300&(((unsigned short)baccum[i+2])<<8)) | (baccum[i+5])); 
                d = raw_subtract_values(d,s);
                baccum[i+2]=(baccum[i+2]&0xFC)|(d>>8); 
                baccum[i+5]=d;


                s =((0x3fc&(((unsigned short)bsub[i+4])<<2)) | (bsub[i+7] >> 6)); 
                d =((0x3fc&(((unsigned short)baccum[i+4])<<2)) | (baccum[i+7] >> 6)); 
                d = raw_subtract_values(d,s);
                baccum[i+4]=d>>2;
                baccum[i+7]=(baccum[i+7]&0x3F)|(d<<6);


                s =((0x3f0&(((unsigned short)bsub[i+7])<<4)) | (bsub[i+6] >> 4)); 
                d =((0x3f0&(((unsigned short)baccum[i+7])<<4)) | (baccum[i+6] >> 4)); 
                d = raw_subtract_values(d,s);
                baccum[i+6]=(baccum[i+6]&0x0F)|(d<<4);
                baccum[i+7]=(baccum[i+7]&0xC0)|(d>>4);

                s =((0x3c0&(((unsigned short)bsub[i+6])<<6)) | (bsub[i+9] >> 2)); 
                d =((0x3c0&(((unsigned short)baccum[i+6])<<6)) | (baccum[i+9] >> 2)); 
                d = raw_subtract_values(d,s);
                baccum[i+6]=(baccum[i+6]&0xF0)|(d>>6);
                baccum[i+9]=(baccum[i+9]&0x03)|(d<<2);


                s =((0x300&(((unsigned short)bsub[i+9])<<8)) | (bsub[i+8])); 
                d =((0x300&(((unsigned short)baccum[i+9])<<8)) | (baccum[i+8])); 
                d = raw_subtract_values(d,s);
                baccum[i+8]=d;
                baccum[i+9]=(baccum[i+9]&0xFC)|(d>>8);
            }

#elif CAM_SENSOR_BITS_PER_PIXEL==12

            for(i = 0;i<RAW_ROWLEN; i+=6) {

                s=((0xFF0&(((unsigned short)bsub[i+1])<<4))   | (bsub[i+0] >> 4));
                d=((0xFF0&(((unsigned short)baccum[i+1])<<4)) | (baccum[i+0] >> 4));
                d = raw_subtract_values(d,s);
                baccum[i+0]=(baccum[i+0]&0x0F)|(d<<4);
                baccum[i+1]=d>>4;

                s=((0xF00&(((unsigned short)bsub[i+0])<<8))   | (bsub[i+3]     ));
                d=((0xF00&(((unsigned short)baccum[i+0])<<8)) | (baccum[i+3]   ));
                d = raw_subtract_values(d,s);
                baccum[i+0]=(baccum[i+0]&0xF0)|(d>>8);
                baccum[i+3]=d;

                s=((0xFF0&(((unsigned short)bsub[i+2])<<4))   | (bsub[i+5] >> 4));
                d=((0xFF0&(((unsigned short)baccum[i+2])<<4)) | (baccum[i+5] >> 4));
                d = raw_subtract_values(d,s);
                baccum[i+2]=d>>4;
                baccum[i+5]=(baccum[i+5]&0x0F)|(d<<4);

                s=((0xF00&(((unsigned short)bsub[i+5])<<8))   | (bsub[i+4]     ));
                d=((0xF00&(((unsigned short)baccum[i+5])<<8)) | (baccum[i+4]     ));
                d = raw_subtract_values(d,s);
                baccum[i+4]=d;
                baccum[i+5]=(baccum[i+5]&0xF0)|(d>>8);
            }

#else 
 #error define set_raw_pixel for sensor bit depth
#endif

            fwrite(baccum,1,RAW_ROWLEN,fdest);
            if ( (j & 0x1F) == 0 ) {
                gui_browser_progress_show((char *)dest, j*100/CAM_RAW_ROWS);
            }
        }
        gui_browser_progress_show((char *)dest, 100);
        finished();
        status = 1;
    }
    free(baccum);
    free(bsub);
    if(ffrom)
        fclose(ffrom);
    if(fsub)
        fclose(fsub);
    if(fdest) {
        fclose(fdest);
        t.actime = t.modtime = time(NULL);
        utime((char *)dest, &t);
    }
    return status;
}

int raw_merge_start(int action){
  unsigned int req, avail;
  req=((CAM_RAW_ROWPIX*CAM_RAW_ROWS)>>18)+1;
  avail=GetFreeCardSpaceKb()>>10;
  if (avail<req) {
    sprintf(namebuf,lang_str(LANG_AVERAGE_NO_CARD_SPACE),req,avail);
    gui_mbox_init((int)"", (int)namebuf, MBOX_BTN_OK|MBOX_TEXT_CENTER, NULL);
    return 0;
  }
  raw_action=action;
  raw_count=0;
  row=malloc(CAM_RAW_ROWPIX*sizeof(unsigned short));
  if (!row)
    return 0;
  rawrow=malloc(RAW_ROWLEN);
  if (!rawrow) {
    free(row);
    return 0;
  }
  return 1;
}

void raw_merge_add_file(const char * filename) {
  int  t,src,i,j,nrow;
  FILE *fbrawin=NULL,*fbrawout,*fcraw;
  struct stat st;

  if (!filename)
    return;

  stat(filename,&st);
  if (st.st_size!=hook_raw_size())
    return;

  started();

  fcraw=fopen(filename,"rb");
  if (fcraw) {
    if (raw_count)
      fbrawin=fopen(TEMP_FILE_NAME,"rb");
    if (!raw_count || fbrawin){
      fbrawout=fopen(TEMP_FILE_NAME_1,"w+b");
      if (fbrawout){
        fread(rawrow, 1, RAW_ROWLEN, fcraw);
        if (raw_count) 
          fread(row, 1, CAM_RAW_ROWPIX*sizeof(unsigned short), fbrawin); 
        else
          for (i=0;i<CAM_RAW_ROWPIX;i++)
            row[i]=0;

        for (nrow=0,j=0;nrow<CAM_RAW_ROWS;nrow++,j++){

#if CAM_SENSOR_BITS_PER_PIXEL==10

          for (i=0,src=0; i<CAM_RAW_ROWPIX; i+=8, src+=10){
            row[i+0]+=((0x3fc&(((unsigned short)rawrow[src+1])<<2)) | (rawrow[src+0] >> 6));
            row[i+1]+=((0x3f0&(((unsigned short)rawrow[src+0])<<4)) | (rawrow[src+3] >> 4));
            row[i+2]+=((0x3c0&(((unsigned short)rawrow[src+3])<<6)) | (rawrow[src+2] >> 2));
            row[i+3]+=((0x300&(((unsigned short)rawrow[src+2])<<8)) | (rawrow[src+5])); 
            row[i+4]+=((0x3fc&(((unsigned short)rawrow[src+4])<<2)) | (rawrow[src+7] >> 6)); 
            row[i+5]+=((0x3f0&(((unsigned short)rawrow[src+7])<<4)) | (rawrow[src+6] >> 4)); 
            row[i+6]+=((0x3c0&(((unsigned short)rawrow[src+6])<<6)) | (rawrow[src+9] >> 2)); 
            row[i+7]+=((0x300&(((unsigned short)rawrow[src+9])<<8)) | (rawrow[src+8])); 
          }

#elif CAM_SENSOR_BITS_PER_PIXEL==12

          for (i=0,src=0; i<CAM_RAW_ROWPIX; i+=4, src+=6){
            row[i+0]+=((0xFF0&(((unsigned short)rawrow[src+1])<<4)) | (rawrow[src+0] >> 4));
            row[i+1]+=((0xF00&(((unsigned short)rawrow[src+0])<<8)) | (rawrow[src+3]     ));
            row[i+2]+=((0xFF0&(((unsigned short)rawrow[src+2])<<4)) | (rawrow[src+5] >> 4));
            row[i+3]+=((0xF00&(((unsigned short)rawrow[src+5])<<8)) | (rawrow[src+4]     ));
          }

#else
 #error define set_raw_pixel for sensor bit depth
#endif

          fwrite(row, 1, CAM_RAW_ROWPIX*sizeof(unsigned short), fbrawout);
          if (raw_count)
            fread(row, 1, CAM_RAW_ROWPIX*sizeof(unsigned short), fbrawin);
          else
            for (i=0;i<CAM_RAW_ROWPIX;i++)
              row[i]=0;
          fread(rawrow, 1, RAW_ROWLEN, fcraw);
          if (j>=CAM_RAW_ROWS/10) {
            j-=CAM_RAW_ROWS/10;
            gui_browser_progress_show(filename, nrow*100/CAM_RAW_ROWS);
          }
        }
        raw_count++;
        strcpy(namebuf,filename);
        fclose(fbrawout);
      }
      if (fbrawin)
        fclose(fbrawin);
    }
    fclose(fcraw);
  } 
  remove(TEMP_FILE_NAME);
  rename(TEMP_FILE_NAME_1,TEMP_FILE_NAME);
  finished();
}

void raw_merge_end(void) {
  int src,i,j,nrow;
  FILE *fbraw, *fcraw;
  static struct utimbuf t;
  if (!raw_count)
    return;

  i=strlen(namebuf)-3;
  if (strncmp(namebuf+i,"CR",2)==0)
    strcpy(namebuf+i,"WAV");
  else
    strcpy(namebuf+i,"CRW");

  started();
  fbraw=fopen(TEMP_FILE_NAME,"r+b");
  if (fbraw) {
    fcraw=fopen(namebuf,"w+b");
    if (fcraw) {
      fread(row, 1, CAM_RAW_ROWPIX*sizeof(unsigned short), fbraw);
      for (nrow=0,j=0;nrow<CAM_RAW_ROWS;nrow++,j++) {
        for (i=0;i<CAM_RAW_ROWPIX;i++) {
          if (raw_action==RAW_OPERATION_AVERAGE)
            row[i]/=raw_count;
          else {
            if (row[i]>CAM_BLACK_LEVEL*(raw_count-1))
              row[i]-=CAM_BLACK_LEVEL*(raw_count-1);
            else
              row[i]=0;
            if (row[i]>CAM_WHITE_LEVEL)
              row[i]=CAM_WHITE_LEVEL;
          }
        }
#if CAM_SENSOR_BITS_PER_PIXEL==10
        for (i=0,src=0;i<CAM_RAW_ROWPIX;i+=8,src+=10) {
          rawrow[src+0]=(row[i+0]<<6)|(row[i+1]>>4);
          rawrow[src+1]=(row[i+0]>>2);
          rawrow[src+2]=(row[i+2]<<2)|(row[i+3]>>8);
          rawrow[src+3]=(row[i+1]<<4)|(row[i+2]>>6);
          rawrow[src+4]=(row[i+4]>>2);
          rawrow[src+5]=(row[i+3]);
          rawrow[src+6]=(row[i+5]<<4)|(row[i+6]>>6);
          rawrow[src+7]=(row[i+4]<<6)|(row[i+5]>>4);
          rawrow[src+8]=(row[i+7]);
          rawrow[src+9]=(row[i+6]<<2)|(row[i+7]>>8);
        }
#elif CAM_SENSOR_BITS_PER_PIXEL==12
        for (i=0,src=0; i<CAM_RAW_ROWPIX; i+=4, src+=6){
          rawrow[src+0]=(row[i+0]<<4)|(row[i+1]>>8);
          rawrow[src+1]=(row[i+0]>>4);
          rawrow[src+2]=(row[i+2]>>4);
          rawrow[src+3]= row[i+1];
          rawrow[src+4]= row[i+3];
          rawrow[src+5]=(row[i+2]<<4)|(row[i+3]>>8);
        }
#else 
 #error define set_raw_pixel for sensor bit depth
#endif

        fwrite(rawrow, 1, RAW_ROWLEN, fcraw);
        fread(row, 1, CAM_RAW_ROWPIX*sizeof(unsigned short), fbraw);
        if (j>=CAM_RAW_ROWS/5) {
          j-=CAM_RAW_ROWS/5;
          gui_browser_progress_show(namebuf, nrow*100/CAM_RAW_ROWS);
        }
      }
      fclose(fcraw);
    }
    fclose(fbraw);
  }

  t.actime = t.modtime = time(NULL);
  utime(namebuf, &t);
  remove(TEMP_FILE_NAME);

  finished();
  free(rawrow);
  free(row);
}