/*
 * Copyright (c) 2004, Yannick Verschueren
 * Copyright (c) 2005, Hervˇ Drolon, FreeImage Team
 * Copyright (c) 2002-2005, Communications and remote sensing Laboratory, Universite catholique de Louvain, Belgium
 * 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"


/* ----------------------------------------------------------------------- */

static bool jp2_read_boxhdr(opj_common_ptr cinfo, opj_cio_t *cio, opj_jp2_box_t *box) {
  box->init_pos = cio_tell(cio);
  box->length = cio_read(cio, 4);
  box->type = cio_read(cio, 4);
  if (box->length == 1) {
    if (cio_read(cio, 4) != 0) {
      opg_event_msg(cinfo, EVT_ERROR, "Cannot handle box sizes higher than 2^32\n");
      return false;
    }
    box->length = cio_read(cio, 4);
    if (box->length == 0) 
      box->length = cio_numbytesleft(cio) + 12;
  }
  else if (box->length == 0) {
    box->length = cio_numbytesleft(cio) + 8;
  }
  
  return true;
}

static void jp2_write_url(opj_cio_t *cio, char *Idx_file) {
  unsigned int i;
  opj_jp2_box_t box;

  box.init_pos = cio_tell(cio);
  cio_skip(cio, 4);
  cio_write(cio, JP2_URL, 4); /* DBTL */
  cio_write(cio, 0, 1);   /* VERS */
  cio_write(cio, 0, 3);   /* FLAG */

  if(Idx_file) {
    for (i = 0; i < strlen(Idx_file); i++) {
      cio_write(cio, Idx_file[i], 1);
    }
  }

  box.length = cio_tell(cio) - box.init_pos;
  cio_seek(cio, box.init_pos);
  cio_write(cio, box.length, 4);  /* L */
  cio_seek(cio, box.init_pos + box.length);
}

static bool jp2_read_ihdr(opj_jp2_t *jp2, opj_cio_t *cio) {
  opj_jp2_box_t box;

  opj_common_ptr cinfo = jp2->cinfo;

  jp2_read_boxhdr(cinfo, cio, &box);
  if (JP2_IHDR != box.type) {
    opg_event_msg(cinfo, EVT_ERROR, "Expected IHDR Marker\n");
    return false;
  }

  jp2->h = cio_read(cio, 4);      /* HEIGHT */
  jp2->w = cio_read(cio, 4);      /* WIDTH */
  jp2->numcomps = cio_read(cio, 2); /* NC */

  jp2->bpc = cio_read(cio, 1);    /* BPC */

  jp2->C = cio_read(cio, 1);      /* C */
  jp2->UnkC = cio_read(cio, 1);   /* UnkC */
  jp2->IPR = cio_read(cio, 1);    /* IPR */

  if (cio_tell(cio) - box.init_pos != box.length) {
    opg_event_msg(cinfo, EVT_ERROR, "Error with IHDR Box\n");
    return false;
  }

  return true;
}

static void jp2_write_ihdr(opj_jp2_t *jp2, opj_cio_t *cio) {
  opj_jp2_box_t box;

  box.init_pos = cio_tell(cio);
  cio_skip(cio, 4);
  cio_write(cio, JP2_IHDR, 4);    /* IHDR */

  cio_write(cio, jp2->h, 4);      /* HEIGHT */
  cio_write(cio, jp2->w, 4);      /* WIDTH */
  cio_write(cio, jp2->numcomps, 2); /* NC */

  cio_write(cio, jp2->bpc, 1);    /* BPC */

  cio_write(cio, jp2->C, 1);      /* C : Always 7 */
  cio_write(cio, jp2->UnkC, 1);   /* UnkC, colorspace unknown */
  cio_write(cio, jp2->IPR, 1);    /* IPR, no intellectual property */

  box.length = cio_tell(cio) - box.init_pos;
  cio_seek(cio, box.init_pos);
  cio_write(cio, box.length, 4);  /* L */
  cio_seek(cio, box.init_pos + box.length);
}

static void jp2_write_bpcc(opj_jp2_t *jp2, opj_cio_t *cio) {
  unsigned int i;
  opj_jp2_box_t box;

  box.init_pos = cio_tell(cio);
  cio_skip(cio, 4);
  cio_write(cio, JP2_BPCC, 4);  /* BPCC */

  for (i = 0; i < jp2->numcomps; i++) {
    cio_write(cio, jp2->comps[i].bpcc, 1);
  }

  box.length = cio_tell(cio) - box.init_pos;
  cio_seek(cio, box.init_pos);
  cio_write(cio, box.length, 4);  /* L */
  cio_seek(cio, box.init_pos + box.length);
}


static bool jp2_read_bpcc(opj_jp2_t *jp2, opj_cio_t *cio) {
  unsigned int i;
  opj_jp2_box_t box;

  opj_common_ptr cinfo = jp2->cinfo;

  jp2_read_boxhdr(cinfo, cio, &box);
  if (JP2_BPCC != box.type) {
    opg_event_msg(cinfo, EVT_ERROR, "Expected BPCC Marker\n");
    return false;
  }

  for (i = 0; i < jp2->numcomps; i++) {
    jp2->comps[i].bpcc = cio_read(cio, 1);
  }

  if (cio_tell(cio) - box.init_pos != box.length) {
    opg_event_msg(cinfo, EVT_ERROR, "Error with BPCC Box\n");
    return false;
  }

  return true;
}

static void jp2_write_colr(opj_jp2_t *jp2, opj_cio_t *cio) {
  opj_jp2_box_t box;

  box.init_pos = cio_tell(cio);
  cio_skip(cio, 4);
  cio_write(cio, JP2_COLR, 4);    /* COLR */

  cio_write(cio, jp2->meth, 1);   /* METH */
  cio_write(cio, jp2->precedence, 1); /* PRECEDENCE */
  cio_write(cio, jp2->approx, 1);   /* APPROX */

  if (jp2->meth == 1) {
    cio_write(cio, jp2->enumcs, 4); /* EnumCS */
  } else {
    cio_write(cio, 0, 1);     /* PROFILE (??) */
  }

  box.length = cio_tell(cio) - box.init_pos;
  cio_seek(cio, box.init_pos);
  cio_write(cio, box.length, 4);  /* L */
  cio_seek(cio, box.init_pos + box.length);
}

static bool jp2_read_colr(opj_jp2_t *jp2, opj_cio_t *cio) {
  opj_jp2_box_t box;
  int skip_len;

  opj_common_ptr cinfo = jp2->cinfo;

  jp2_read_boxhdr(cinfo, cio, &box);
  do {
    if (JP2_COLR != box.type) {
      cio_skip(cio, box.length - 8);
      jp2_read_boxhdr(cinfo, cio, &box);
    }
  } while(JP2_COLR != box.type);

  jp2->meth = cio_read(cio, 1);   /* METH */
  jp2->precedence = cio_read(cio, 1); /* PRECEDENCE */
  jp2->approx = cio_read(cio, 1);   /* APPROX */

  if (jp2->meth == 1) {
    jp2->enumcs = cio_read(cio, 4); /* EnumCS */
  } else {
    /* skip PROFILE */
    skip_len = box.init_pos + box.length - cio_tell(cio);
    if (skip_len < 0) {
      opg_event_msg(cinfo, EVT_ERROR, "Error with JP2H box size\n");
      return false;
    }
    cio_skip(cio, box.init_pos + box.length - cio_tell(cio));
  }

  if (cio_tell(cio) - box.init_pos != box.length) {
    opg_event_msg(cinfo, EVT_ERROR, "Error with BPCC Box\n");
    return false;
  }
  return true;
}

static void jp2_write_jp2h(opj_jp2_t *jp2, opj_cio_t *cio) {
  opj_jp2_box_t box;

  box.init_pos = cio_tell(cio);
  cio_skip(cio, 4);
  cio_write(cio, JP2_JP2H, 4);  /* JP2H */

  jp2_write_ihdr(jp2, cio);

  if (jp2->bpc == 255) {
    jp2_write_bpcc(jp2, cio);
  }
  jp2_write_colr(jp2, cio);

  box.length = cio_tell(cio) - box.init_pos;
  cio_seek(cio, box.init_pos);
  cio_write(cio, box.length, 4);  /* L */
  cio_seek(cio, box.init_pos + box.length);
}

static bool jp2_read_jp2h(opj_jp2_t *jp2, opj_cio_t *cio) {
  opj_jp2_box_t box;
  int skip_len;

  opj_common_ptr cinfo = jp2->cinfo;

  jp2_read_boxhdr(cinfo, cio, &box);
  do {
    if (JP2_JP2H != box.type) {
      if (box.type == JP2_JP2C) {
        opg_event_msg(cinfo, EVT_ERROR, "Expected JP2H Marker\n");
        return false;
      }
      cio_skip(cio, box.length - 8);
      jp2_read_boxhdr(cinfo, cio, &box);
    }
  } while(JP2_JP2H != box.type);

  if (!jp2_read_ihdr(jp2, cio))
    return false;

  if (jp2->bpc == 255) {
    if (!jp2_read_bpcc(jp2, cio))
      return false;
  }
  if (!jp2_read_colr(jp2, cio))
    return false;

  skip_len = box.init_pos + box.length - cio_tell(cio);
  if (skip_len < 0) {
    opg_event_msg(cinfo, EVT_ERROR, "Error with JP2H Box\n");
    return false;
  }
  cio_skip(cio, box.init_pos + box.length - cio_tell(cio));

  return true;
}

static void jp2_write_ftyp(opj_jp2_t *jp2, opj_cio_t *cio) {
  unsigned int i;
  opj_jp2_box_t box;

  box.init_pos = cio_tell(cio);
  cio_skip(cio, 4);
  cio_write(cio, JP2_FTYP, 4);    /* FTYP */

  cio_write(cio, jp2->brand, 4);    /* BR */
  cio_write(cio, jp2->minversion, 4); /* MinV */

  for (i = 0; i < jp2->numcl; i++) {
    cio_write(cio, jp2->cl[i], 4);  /* CL */
  }

  box.length = cio_tell(cio) - box.init_pos;
  cio_seek(cio, box.init_pos);
  cio_write(cio, box.length, 4);  /* L */
  cio_seek(cio, box.init_pos + box.length);
}

static bool jp2_read_ftyp(opj_jp2_t *jp2, opj_cio_t *cio) {
  int i;
  opj_jp2_box_t box;

  opj_common_ptr cinfo = jp2->cinfo;

  jp2_read_boxhdr(cinfo, cio, &box);

  if (JP2_FTYP != box.type) {
    opg_event_msg(cinfo, EVT_ERROR, "Expected FTYP Marker\n");
    return false;
  }

  jp2->brand = cio_read(cio, 4);    /* BR */
  jp2->minversion = cio_read(cio, 4); /* MinV */
  jp2->numcl = (box.length - 16) / 4;
  jp2->cl = (unsigned int *) opj_malloc(jp2->numcl * sizeof(unsigned int));

  for (i = 0; i < (int)jp2->numcl; i++) {
    jp2->cl[i] = cio_read(cio, 4);  /* CLi */
  }

  if (cio_tell(cio) - box.init_pos != box.length) {
    opg_event_msg(cinfo, EVT_ERROR, "Error with FTYP Box\n");
    return false;
  }

  return true;
}

static int jp2_write_jp2c(opj_jp2_t *jp2, opj_cio_t *cio, char *index) {
  unsigned int j2k_codestream_offset, j2k_codestream_length;
  opj_jp2_box_t box;

  opj_j2k_t *j2k = jp2->j2k;
  opj_image_t *image = jp2->image;

  box.init_pos = cio_tell(cio);
  cio_skip(cio, 4);
  cio_write(cio, JP2_JP2C, 4);  /* JP2C */

  /* J2K encoding */
  j2k_codestream_offset = cio_tell(cio);
  if(!j2k_encode(j2k, cio, image, index)) {
    opg_event_msg(j2k->cinfo, EVT_ERROR, "Failed to encode image\n");
    return 0;
  }
  j2k_codestream_length = cio_tell(cio) - j2k_codestream_offset;

  jp2->j2k_codestream_offset = j2k_codestream_offset;
  jp2->j2k_codestream_length = j2k_codestream_length;

  box.length = 8 + jp2->j2k_codestream_length;
  cio_seek(cio, box.init_pos);
  cio_write(cio, box.length, 4);  /* L */
  cio_seek(cio, box.init_pos + box.length);

  return box.length;
}

static bool jp2_read_jp2c(opj_jp2_t *jp2, opj_cio_t *cio, unsigned int *j2k_codestream_length, unsigned int *j2k_codestream_offset) {
  opj_jp2_box_t box;

  opj_common_ptr cinfo = jp2->cinfo;

  jp2_read_boxhdr(cinfo, cio, &box);
  do {
    if(JP2_JP2C != box.type) {
      cio_skip(cio, box.length - 8);
      jp2_read_boxhdr(cinfo, cio, &box);
    }
  } while(JP2_JP2C != box.type);

  *j2k_codestream_offset = cio_tell(cio);
  *j2k_codestream_length = box.length - 8;

  return true;
}

static void jp2_write_jp(opj_cio_t *cio) {
  opj_jp2_box_t box;

  box.init_pos = cio_tell(cio);
  cio_skip(cio, 4);
  cio_write(cio, JP2_JP, 4);    /* JP2 signature */
  cio_write(cio, 0x0d0a870a, 4);

  box.length = cio_tell(cio) - box.init_pos;
  cio_seek(cio, box.init_pos);
  cio_write(cio, box.length, 4);  /* L */
  cio_seek(cio, box.init_pos + box.length);
}

static bool jp2_read_jp(opj_jp2_t *jp2, opj_cio_t *cio) {
  opj_jp2_box_t box;

  opj_common_ptr cinfo = jp2->cinfo;

  jp2_read_boxhdr(cinfo, cio, &box);
  if (JP2_JP != box.type) {
    opg_event_msg(cinfo, EVT_ERROR, "Expected JP Marker\n");
    return false;
  }
  if (0x0d0a870a != cio_read(cio, 4)) {
    opg_event_msg(cinfo, EVT_ERROR, "Error with JP Marker\n");
    return false;
  }
  if (cio_tell(cio) - box.init_pos != box.length) {
    opg_event_msg(cinfo, EVT_ERROR, "Error with JP Box size\n");
    return false;
  }

  return true;
}


static bool jp2_read_struct(opj_jp2_t *jp2, opj_cio_t *cio) {
  if (!jp2_read_jp(jp2, cio))
    return false;
  if (!jp2_read_ftyp(jp2, cio))
    return false;
  if (!jp2_read_jp2h(jp2, cio))
    return false;
  if (!jp2_read_jp2c(jp2, cio, &jp2->j2k_codestream_length, &jp2->j2k_codestream_offset))
    return false;
  
  return true;
}

/* ----------------------------------------------------------------------- */
/* JP2 decoder interface                                             */
/* ----------------------------------------------------------------------- */

opj_jp2_t* jp2_create_decompress(opj_common_ptr cinfo) {
  opj_jp2_t *jp2 = (opj_jp2_t*)opj_malloc(sizeof(opj_jp2_t));
  if(jp2) {
    jp2->cinfo = cinfo;
    /* create the J2K codec */
    jp2->j2k = j2k_create_decompress(cinfo);
    if(jp2->j2k == NULL) {
      jp2_destroy_decompress(jp2);
      return NULL;
    }
  }
  return jp2;
}

void jp2_destroy_decompress(opj_jp2_t *jp2) {
  if(jp2) {
    /* destroy the J2K codec */
    j2k_destroy_decompress(jp2->j2k);

    if(jp2->comps) {
      opj_free(jp2->comps);
    }
    if(jp2->cl) {
      opj_free(jp2->cl);
    }
    opj_free(jp2);
  }
}

void jp2_setup_decoder(opj_jp2_t *jp2, opj_dparameters_t *parameters) {
  /* setup the J2K codec */
  j2k_setup_decoder(jp2->j2k, parameters);
  /* further JP2 initializations go here */
}

opj_image_t* jp2_decode(opj_jp2_t *jp2, opj_cio_t *cio) {
  opj_common_ptr cinfo;
  opj_image_t *image = NULL;

  if(!jp2 || !cio) {
    return NULL;
  }

  cinfo = jp2->cinfo;

  /* JP2 decoding */
  if(!jp2_read_struct(jp2, cio)) {
    opg_event_msg(cinfo, EVT_ERROR, "Failed to decode jp2 structure\n");
    return NULL;
  }

  /* J2K decoding */
  image = j2k_decode(jp2->j2k, cio);
  if(!image) {
    opg_event_msg(cinfo, EVT_ERROR, "Failed to decode J2K image\n");
  }

  return image;
}

/* ----------------------------------------------------------------------- */
/* JP2 encoder interface                                             */
/* ----------------------------------------------------------------------- */

opj_jp2_t* jp2_create_compress(opj_common_ptr cinfo) {
  opj_jp2_t *jp2 = (opj_jp2_t*)opj_malloc(sizeof(opj_jp2_t));
  if(jp2) {
    jp2->cinfo = cinfo;
    /* create the J2K codec */
    jp2->j2k = j2k_create_compress(cinfo);
    if(jp2->j2k == NULL) {
      jp2_destroy_compress(jp2);
      return NULL;
    }
  }
  return jp2;
}

void jp2_destroy_compress(opj_jp2_t *jp2) {
  if(jp2) {
    /* destroy the J2K codec */
    j2k_destroy_compress(jp2->j2k);

    if(jp2->comps) {
      opj_free(jp2->comps);
    }
    if(jp2->cl) {
      opj_free(jp2->cl);
    }
    opj_free(jp2);
  }
}

void jp2_setup_encoder(opj_jp2_t *jp2, opj_cparameters_t *parameters, opj_image_t *image) {
  int i;
  int depth_0, sign;

  if(!jp2 || !parameters || !image)
    return;

  /* setup the J2K codec */
  /* ------------------- */

  j2k_setup_encoder(jp2->j2k, parameters, image);

  /* setup the JP2 codec */
  /* ------------------- */
  
  /* Profile box */

  jp2->brand = JP2_JP2; /* BR */
  jp2->minversion = 0;  /* MinV */
  jp2->numcl = 1;
  jp2->cl = (unsigned int*) opj_malloc(jp2->numcl * sizeof(unsigned int));
  jp2->cl[0] = JP2_JP2; /* CL0 : JP2 */

  /* Image Header box */

  jp2->image = image;

  jp2->numcomps = image->numcomps;  /* NC */
  jp2->comps = (opj_jp2_comps_t*) opj_malloc(jp2->numcomps * sizeof(opj_jp2_comps_t));
  jp2->h = image->y1 - image->y0;   /* HEIGHT */
  jp2->w = image->x1 - image->x0;   /* WIDTH */
  /* BPC */
  depth_0 = image->comps[0].prec - 1;
  sign = image->comps[0].sgnd;
  jp2->bpc = depth_0 + (sign << 7);
  for (i = 1; i < image->numcomps; i++) {
    int depth = image->comps[i].prec - 1;
    sign = image->comps[i].sgnd;
    if (depth_0 != depth)
      jp2->bpc = 255;
  }
  jp2->C = 7;     /* C : Always 7 */
  jp2->UnkC = 0;    /* UnkC, colorspace specified in colr box */
  jp2->IPR = 0;   /* IPR, no intellectual property */
  
  /* BitsPerComponent box */

  for (i = 0; i < image->numcomps; i++) {
    jp2->comps[i].bpcc = image->comps[i].prec - 1 + (image->comps[i].sgnd << 7);
  }

  /* Colour Specification box */

  if ((image->numcomps == 1 || image->numcomps == 3) && (jp2->bpc != 255)) {
    jp2->meth = 1;  /* METH: Enumerated colourspace */
  } else {
    jp2->meth = 2;  /* METH: Restricted ICC profile */
  }
  if (jp2->meth == 1) {
    if (image->color_space == 1)
      jp2->enumcs = 16; /* sRGB as defined by IEC 61966–2–1 */
    else if (image->color_space == 2)
      jp2->enumcs = 17; /* greyscale */
    else if (image->color_space == 3)
      jp2->enumcs = 18; /* YUV */
  } else {
    jp2->enumcs = 0;    /* PROFILE (??) */
  }
  jp2->precedence = 0;  /* PRECEDENCE */
  jp2->approx = 0;    /* APPROX */

}

bool jp2_encode(opj_jp2_t *jp2, opj_cio_t *cio, opj_image_t *image, char *index) {

  /* JP2 encoding */

  /* JPEG 2000 Signature box */
  jp2_write_jp(cio);
  /* File Type box */
  jp2_write_ftyp(jp2, cio);
  /* JP2 Header box */
  jp2_write_jp2h(jp2, cio);

  /* J2K encoding */

  if(!jp2_write_jp2c(jp2, cio, index)) {
    opg_event_msg(jp2->cinfo, EVT_ERROR, "Failed to encode image\n");
    return false;
  }

  return true;
}