/*
* Copyright (c) 2002-2014, Universite catholique de Louvain (UCL), Belgium
* Copyright (c) 2002-2014, Professor Benoit Macq
* Copyright (c) 2003-2007, Francois-Olivier Devaux and Antonin Descampe
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in the
*    documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/

#include "opj_includes.h"
#include "mj2_convert.h"

/*  -----------------------	      */
/*				      */
/*				      */
/*  Count the number of frames	      */
/*  in a YUV file		      */
/*				      */
/*  -----------------------	      */

unsigned int OPJ_CALLCONV yuv_num_frames(mj2_tk_t * tk, char *infile)
{
  unsigned int prec_size;
  long end_of_f, frame_size;
	FILE *f;

  f = fopen(infile,"rb");
  if (!f) {  
    fprintf(stderr, "failed to open %s for reading\n",infile);
    return 0;
  }
	prec_size = (tk->depth + 7)/8;/* bytes of precision */

  frame_size = (long) (tk->w * tk->h * (1.0 + (double) 2 / (double) (tk->CbCr_subsampling_dx * tk->CbCr_subsampling_dy)));	/* Calculate frame size */
	frame_size *= prec_size; 
	
  fseek(f, 0, SEEK_END);
  end_of_f = ftell(f);		/* Calculate file size */
	
  if (end_of_f < frame_size) {
    fprintf(stderr,
			"YUV does not contains any frame of %d x %d size\n", tk->w,
			tk->h);
    return 0;
  }
  fclose(f);
	
  return (unsigned int)(end_of_f / frame_size);
}

/*  ----------------------- */
/* */
/* */
/*  YUV to IMAGE */
/* */
/*  ----------------------- */

opj_image_t * OPJ_CALLCONV mj2_image_create(mj2_tk_t * tk, opj_cparameters_t *parameters)
{
	opj_image_cmptparm_t cmptparm[3];
	opj_image_t * img;
	int i;
	int numcomps = 3;
	int subsampling_dx = parameters->subsampling_dx;
	int subsampling_dy = parameters->subsampling_dy;

	/* initialize image components */
	memset(&cmptparm[0], 0, 3 * sizeof(opj_image_cmptparm_t));
	for(i = 0; i < numcomps; i++) {
		cmptparm[i].prec = tk->depth;
		cmptparm[i].bpp = tk->depth;
		cmptparm[i].sgnd = 0;		
		cmptparm[i].dx = i ? subsampling_dx * tk->CbCr_subsampling_dx : subsampling_dx;
		cmptparm[i].dy = i ? subsampling_dy * tk->CbCr_subsampling_dy : subsampling_dy;
		cmptparm[i].w = tk->w;
		cmptparm[i].h = tk->h;
	}
	/* create the image */
	img = opj_image_create(numcomps, cmptparm, CLRSPC_SRGB);
	return img;
}

char OPJ_CALLCONV yuvtoimage(mj2_tk_t * tk, opj_image_t * img, int frame_num, opj_cparameters_t *parameters, char* infile)
{
  int i, compno;
  int offset, size, max, prec_bytes, is_16, v;
  long end_of_f, position;
	int numcomps = 3;
	int subsampling_dx = parameters->subsampling_dx;
	int subsampling_dy = parameters->subsampling_dy;
	FILE *yuvfile;
	int *data;
	unsigned char uc;

  yuvfile = fopen(infile,"rb");
  if (!yuvfile) {  
    fprintf(stderr, "failed to open %s for readings\n",parameters->infile);
    return 1;
  }
	is_16 = (tk->depth > 8);
	prec_bytes = (is_16?2:1);

  offset = (int) ((double) (frame_num * tk->w * tk->h) * (1.0 +
		1.0 * (double) 2 / (double) (tk->CbCr_subsampling_dx * tk->CbCr_subsampling_dy)));
  offset *= prec_bytes;

  fseek(yuvfile, 0, SEEK_END);
  end_of_f = ftell(yuvfile);
  fseek(yuvfile, sizeof(unsigned char) * offset, SEEK_SET);
  position = ftell(yuvfile);
  if (position >= end_of_f) {
    fprintf(stderr, "Cannot reach frame number %d in yuv file !!\n",
			frame_num);
		fclose(yuvfile);
    return 1;
  }
	
  img->x0 = tk->Dim[0];
  img->y0 = tk->Dim[1];
  img->x1 = !tk->Dim[0] ? (tk->w - 1) * subsampling_dx + 1 : tk->Dim[0] +
    (tk->w - 1) * subsampling_dx + 1;
  img->y1 = !tk->Dim[1] ? (tk->h - 1) * subsampling_dy + 1 : tk->Dim[1] +
    (tk->h - 1) * subsampling_dy + 1;

	size = tk->w * tk->h * prec_bytes;
	
	for(compno = 0; compno < numcomps; compno++) 
   {
	max = size/(img->comps[compno].dx * img->comps[compno].dy);
	data = img->comps[compno].data;

	for (i = 0; i < max && !feof(yuvfile); i++)
  {
	v = 0;
	fread(&uc, 1, 1, yuvfile);
	v = uc;

	if(is_16)
 {
	fread(&uc, 1, 1, yuvfile);
	v |= (uc<<8);
 }
	*data++ = v;
  }
   }
	fclose(yuvfile);
	
  return 0;
}



/*  ----------------------- */
/* */
/* */
/*  IMAGE to YUV */
/* */
/*  ----------------------- */


opj_bool OPJ_CALLCONV imagetoyuv(opj_image_t * img, char *outfile)
{
  FILE *f;
  int *data;
  int i, v, is_16, prec_bytes;
  unsigned char buf[2];

  if (img->numcomps == 3) {
    if (img->comps[0].dx != img->comps[1].dx / 2
      || img->comps[1].dx != img->comps[2].dx) {
      fprintf(stderr,
				"Error with the input image components size: cannot create yuv file)\n");
      return OPJ_FALSE;
    }
  } else if (!(img->numcomps == 1)) {
    fprintf(stderr,
      "Error with the number of image components(must be one or three)\n");
    return OPJ_FALSE;
  }
  
  f = fopen(outfile, "a+b");
  if (!f) {
    fprintf(stderr, "failed to open %s for writing\n", outfile);
    return OPJ_FALSE;
  }
  is_16 = (img->comps[0].prec > 8);
  prec_bytes = (is_16?2:1);
  data = img->comps[0].data;
  
  for (i = 0; i < (img->comps[0].w * img->comps[0].h); i++) {
    v = *data++;
    buf[0] = (unsigned char)v;

	if(is_16) buf[1] = (unsigned char)(v>>8);

    fwrite(buf, 1, prec_bytes, f);
  }
  
  
  if (img->numcomps == 3) {
	data = img->comps[1].data;

    for (i = 0; i < (img->comps[1].w * img->comps[1].h); i++) {
      v = *data++;
      buf[0] = (unsigned char)v;

      if(is_16) buf[1] = (unsigned char)(v>>8);

      fwrite(buf, 1, prec_bytes, f);
    }
    data = img->comps[2].data;
    
    for (i = 0; i < (img->comps[2].w * img->comps[2].h); i++) {
      v = *data++;
      buf[0] = (unsigned char)v;

      if(is_16) buf[1] = (unsigned char)(v>>8);

      fwrite(buf, 1, prec_bytes, f);
    }
  } else if (img->numcomps == 1) {
/* fake CbCr values */
	if(is_16) 
  { 
	buf[0] = 255;
	if(img->comps[0].prec == 10) buf[1] = 1;
	else
	if(img->comps[0].prec == 12) buf[1] = 3;
	else
	 buf[1] = 125;
  } 
	else buf[0] = 125;

    for (i = 0; i < (img->comps[0].w * img->comps[0].h * 0.25); i++) {
      fwrite(buf, 1, prec_bytes, f);
    }
    
    
    for (i = 0; i < (img->comps[0].w * img->comps[0].h * 0.25); i++) {
      fwrite(buf, 1, prec_bytes, f);
    }
  }  
  fclose(f);
  return OPJ_TRUE;
}

/*  ----------------------- */
/* */
/* */
/*  IMAGE to BMP */
/* */
/*  ----------------------- */

int OPJ_CALLCONV imagetobmp(opj_image_t * img, char *outfile) {
  int w,wr,h,hr,i,pad;
  FILE *f;
  
  if (img->numcomps == 3 && img->comps[0].dx == img->comps[1].dx
    && img->comps[1].dx == img->comps[2].dx
    && img->comps[0].dy == img->comps[1].dy
    && img->comps[1].dy == img->comps[2].dy
    && img->comps[0].prec == img->comps[1].prec
    && img->comps[1].prec == img->comps[2].prec) {
    /* -->> -->> -->> -->>
    
      24 bits color
      
    <<-- <<-- <<-- <<-- */
    
    f = fopen(outfile, "wb");
    if (!f) {
      fprintf(stderr, "failed to open %s for writing\n", outfile);
      return 1;
    }   
    
    w = img->comps[0].w;
    wr = int_ceildivpow2(img->comps[0].w, img->comps[0].factor);
    
    h = img->comps[0].h;
    hr = int_ceildivpow2(img->comps[0].h, img->comps[0].factor);
    
    fprintf(f, "BM");
    
    /* FILE HEADER */
    /* ------------- */
    fprintf(f, "%c%c%c%c",
      (unsigned char) (hr * wr * 3 + 3 * hr * (wr % 2) +
      54) & 0xff,
      (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2) + 54)
      >> 8) & 0xff,
      (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2) + 54)
      >> 16) & 0xff,
      (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2) + 54)
      >> 24) & 0xff);
    fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
      ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
    fprintf(f, "%c%c%c%c", (54) & 0xff, ((54) >> 8) & 0xff,
      ((54) >> 16) & 0xff, ((54) >> 24) & 0xff);
    
    /* INFO HEADER   */
    /* ------------- */
    fprintf(f, "%c%c%c%c", (40) & 0xff, ((40) >> 8) & 0xff,
      ((40) >> 16) & 0xff, ((40) >> 24) & 0xff);
    fprintf(f, "%c%c%c%c", (unsigned char) ((wr) & 0xff),
      (unsigned char) ((wr) >> 8) & 0xff,
      (unsigned char) ((wr) >> 16) & 0xff,
      (unsigned char) ((wr) >> 24) & 0xff);
    fprintf(f, "%c%c%c%c", (unsigned char) ((hr) & 0xff),
      (unsigned char) ((hr) >> 8) & 0xff,
      (unsigned char) ((hr) >> 16) & 0xff,
      (unsigned char) ((hr) >> 24) & 0xff);
    fprintf(f, "%c%c", (1) & 0xff, ((1) >> 8) & 0xff);
    fprintf(f, "%c%c", (24) & 0xff, ((24) >> 8) & 0xff);
    fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
      ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
    fprintf(f, "%c%c%c%c",
      (unsigned char) (3 * hr * wr +
      3 * hr * (wr % 2)) & 0xff,
      (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2)) >>
      8) & 0xff,
      (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2)) >>
      16) & 0xff,
      (unsigned char) ((hr * wr * 3 + 3 * hr * (wr % 2)) >>
      24) & 0xff);
    fprintf(f, "%c%c%c%c", (7834) & 0xff, ((7834) >> 8) & 0xff,
      ((7834) >> 16) & 0xff, ((7834) >> 24) & 0xff);
    fprintf(f, "%c%c%c%c", (7834) & 0xff, ((7834) >> 8) & 0xff,
      ((7834) >> 16) & 0xff, ((7834) >> 24) & 0xff);
    fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
      ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
    fprintf(f, "%c%c%c%c", (0) & 0xff, ((0) >> 8) & 0xff,
      ((0) >> 16) & 0xff, ((0) >> 24) & 0xff);
    
    for (i = 0; i < wr * hr; i++) {
      unsigned char R, G, B;
      /* a modifier */
      /* R = img->comps[0].data[w * h - ((i) / (w) + 1) * w + (i) % (w)];*/
      R = img->comps[0].data[w * hr - ((i) / (wr) + 1) * w + (i) % (wr)];
      /* G = img->comps[1].data[w * h - ((i) / (w) + 1) * w + (i) % (w)];*/
      G = img->comps[1].data[w * hr - ((i) / (wr) + 1) * w + (i) % (wr)];
      /* B = img->comps[2].data[w * h - ((i) / (w) + 1) * w + (i) % (w)];*/
      B = img->comps[2].data[w * hr - ((i) / (wr) + 1) * w + (i) % (wr)];
      fprintf(f, "%c%c%c", B, G, R);
      
      if ((i + 1) % wr == 0) {
				for (pad = (3 * wr) % 4 ? 4 - (3 * wr) % 4 : 0; pad > 0; pad--)	/* ADD */
					fprintf(f, "%c", 0);
      }
    }
    fclose(f);
  }
  return 0;
}
