(no commit message)
[openjpeg.git] / mj2 / mj2_to_frames.c
index f5b222af2d02f7294ae2dad4e8d88ec58ba33590..69357d3189d998fa1a8656273b7b3a9a8d4de5cd 100644 (file)
@@ -1,8 +1,8 @@
-/* Copyright (c) 2003-2004, Fran�ois-Olivier Devaux
-* Copyright (c) 2003-2004,  Communications and remote sensing Laboratory, Universite catholique de Louvain, Belgium
+/*
+* Copyright (c) 2003-2004, Fran�ois-Olivier Devaux
+* Copyright (c) 2002-2004,  Communications and remote sensing Laboratory, Universite catholique de Louvain, Belgium
 * All rights reserved.
 *
-* All rights reserved. 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * POSSIBILITY OF SUCH DAMAGE.
 */
 
-#include <openjpeg.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
+#include "opj_includes.h"
 #include "mj2.h"
 #include "mj2_convert.h"
 
-int ceildiv(int a, int b)
-{
-  return (a + b - 1) / b;
-}
-
-int main(int argc, char **argv)
-{
-  FILE *f;
-  char *src, *src_name;
-  char *dest, S1, S2, S3;
-  int len;
-  j2k_cp_t cp;
-  mj2_movie_t mj2_movie;
+/* -------------------------------------------------------------------------- */
 
+/**
+sample error callback expecting a FILE* client object
+*/
+void error_callback(const char *msg, void *client_data) {
+       FILE *stream = (FILE*)client_data;
+       fprintf(stream, "[ERROR] %s", msg);
+}
+/**
+sample warning callback expecting a FILE* client object
+*/
+void warning_callback(const char *msg, void *client_data) {
+       FILE *stream = (FILE*)client_data;
+       fprintf(stream, "[WARNING] %s", msg);
+}
+/**
+sample debug callback expecting a FILE* client object
+*/
+void info_callback(const char *msg, void *client_data) {
+       FILE *stream = (FILE*)client_data;
+       fprintf(stream, "[INFO] %s", msg);
+}
 
-  if (argc < 3) {
-    fprintf(stderr,
-           "usage: %s j2k-file image-file -reduce n (<- optional)\n",
-           argv[0]);
+/* -------------------------------------------------------------------------- */
+
+
+int main(int argc, char *argv[]) {
+       mj2_dparameters_t mj2_parameters;                       /* decompression parameters */
+       opj_dinfo_t* dinfo; 
+       opj_event_mgr_t event_mgr;              /* event manager */     
+       opj_cio_t *cio = NULL;
+  unsigned int tnum, snum;
+  opj_mj2_t *movie;
+  mj2_tk_t *track;
+  mj2_sample_t *sample;
+  unsigned char* frame_codestream;
+  FILE *file, *outfile;
+  char outfilename[50];
+  opj_image_t *img = NULL;
+       unsigned int max_codstrm_size = 0;
+       double total_time = 0;
+       unsigned int numframes = 0;
+                       
+  if (argc != 3) {
+    printf("Bad syntax: Usage: mj2_to_frames inputfile.mj2 outputfile.yuv\n"); 
+    printf("Example: MJ2_decoder foreman.mj2 foreman.yuv\n");
     return 1;
   }
-
-  f = fopen(argv[1], "rb");
-  if (!f) {
+  
+  file = fopen(argv[1], "rb");
+  
+  if (!file) {
     fprintf(stderr, "failed to open %s for reading\n", argv[1]);
     return 1;
   }
-
-  dest = argv[2];
-
-  cp.reduce_on = 0;
-  cp.reduce_value = 0;
-
-  /* OPTION REDUCE IS ACTIVE */
-  if (argc == 5) {
-    if (strcmp(argv[3], "-reduce")) {
-      fprintf(stderr,
-             "usage: options " "-reduce n"
-             " where n is the factor of reduction [%s]\n", argv[3]);
-      return 1;
-    }
-    cp.reduce_on = 1;
-    sscanf(argv[4], "%d", &cp.reduce_value);
-  }
-
-  while (*dest) {
-    dest++;
-  }
-  dest--;
-  S3 = *dest;
-  dest--;
-  S2 = *dest;
-  dest--;
-  S1 = *dest;
-
-  if (!((S1 == 'y' && S2 == 'u' && S3 == 'v')
-       || (S1 == 'Y' && S2 == 'U' && S3 == 'V'))) {
-    fprintf(stderr,
-           "!! Unrecognized format for infile : %c%c%c [accept only *.yuv] !!\n\n",
-           S1, S2, S3);
+       
+  // Checking output file
+  outfile = fopen(argv[2], "w");
+  if (!file) {
+    fprintf(stderr, "failed to open %s for writing\n", argv[2]);
     return 1;
   }
-
-  fseek(f, 0, SEEK_END);
-  len = ftell(f);
-  fseek(f, 0, SEEK_SET);
-  src = (char *) malloc(len);
-  fread(src, 1, len, f);
-  fclose(f);
-
-  src_name = argv[1];
-  while (*src_name) {
-    src_name++;
-  }
-  src_name--;
-  S3 = *src_name;
-  src_name--;
-  S2 = *src_name;
-  src_name--;
-  S1 = *src_name;
-
-  /* MJ2 format */
-  if ((S1 == 'm' && S2 == 'j' && S3 == '2')
-      || (S1 == 'M' && S2 == 'J' && S3 == '2')) {
-    mj2_movie.num_stk = 0;
-    mj2_movie.num_htk = 0;
-    mj2_movie.num_vtk = 0;
-    mj2_movie.mj2file = argv[1];
-    if (mj2_decode(src, len, &mj2_movie, &cp, argv[2])) {
-      fprintf(stderr, "mj2_to_frames: failed to decode image!\n");
-      return 1;
+  fclose(outfile);
+       
+       /*
+       configure the event callbacks (not required)
+       setting of each callback is optionnal
+       */
+       memset(&event_mgr, 0, sizeof(opj_event_mgr_t));
+       event_mgr.error_handler = error_callback;
+       event_mgr.warning_handler = warning_callback;
+       event_mgr.info_handler = NULL;
+       
+       /* get a MJ2 decompressor handle */
+       dinfo = mj2_create_decompress();
+       movie = dinfo->mj2_handle;
+       
+       /* catch events using our callbacks and give a local context */
+       opj_set_event_mgr((opj_common_ptr)dinfo, &event_mgr, stderr);           
+
+       /* set J2K decoding parameters to default values */
+       opj_set_default_decoder_parameters(&mj2_parameters.j2k_parameters);
+       
+       /* setup the decoder decoding parameters using user parameters */
+       mj2_setup_decoder(dinfo->mj2_handle, &mj2_parameters);
+                       
+  if (mj2_read_struct(file, movie)) // Creating the movie structure
+    return 1;  
+       
+  // Decode first video track 
+       for (tnum=0; tnum < (unsigned int)(movie->num_htk + movie->num_stk + movie->num_vtk); tnum++) {
+               if (movie->tk[tnum].track_type == 0) 
+                       break;
+       }
+       
+       if (movie->tk[tnum].track_type != 0) {
+               printf("Error. Movie does not contain any video track\n");
+               return 1;
+       }
+       
+  track = &movie->tk[tnum];
+       
+  // Output info on first video tracl
+  fprintf(stdout,"The first video track contains %d frames.\nWidth: %d, Height: %d \n\n",
+    track->num_samples, track->w, track->h);
+       
+       max_codstrm_size = track->sample[0].sample_size-8;
+       frame_codestream = (unsigned char*) malloc(max_codstrm_size * sizeof(unsigned char)); 
+
+       numframes = track->num_samples;
+       
+  for (snum=0; snum < numframes; snum++)
+  {
+               double init_time = opj_clock();
+               double elapsed_time;
+    sample = &track->sample[snum];
+               if (sample->sample_size-8 > max_codstrm_size) {
+                       max_codstrm_size =  sample->sample_size-8;
+                       if ((frame_codestream = realloc(frame_codestream, max_codstrm_size)) == NULL) {
+                               printf("Error reallocation memory\n");
+                               return 1;
+                       };              
+               }
+    fseek(file,sample->offset+8,SEEK_SET);
+    fread(frame_codestream, sample->sample_size-8, 1, file);  // Assuming that jp and ftyp markers size do
+               
+               /* open a byte stream */
+               cio = opj_cio_open((opj_common_ptr)dinfo, frame_codestream, sample->sample_size-8);
+               
+               img = opj_decode(dinfo, cio); // Decode J2K to image
+                               
+    if (((img->numcomps == 3) && (img->comps[0].dx == img->comps[1].dx / 2) 
+      && (img->comps[0].dx == img->comps[2].dx / 2 ) && (img->comps[0].dx == 1)) 
+      || (img->numcomps == 1)) {
+      
+      if (!imagetoyuv(img, argv[2]))   // Convert image to YUV
+                               return 1;
     }
-    mj2_memory_free(&mj2_movie);
-  } else {
-    fprintf(stderr,
-           "mj2_to_frames : Unknown format image *.%c%c%c [only *.mj2]!! \n",
-           S1, S2, S3);
-    return 1;
-  }
-
-  free(src);
+    else if ((img->numcomps == 3) && 
+      (img->comps[0].dx == 1) && (img->comps[1].dx == 1)&&
+      (img->comps[2].dx == 1))// If YUV 4:4:4 input --> to bmp
+    {
+      fprintf(stdout,"The frames will be output in a bmp format (output_1.bmp, ...)\n");
+      sprintf(outfilename,"output_%d.bmp",snum);
+      if (imagetobmp(img, outfilename))        // Convert image to YUV
+                               return 1;
+      
+    }
+    else {
+      fprintf(stdout,"Image component dimensions are unknown. Unable to output image\n");
+      fprintf(stdout,"The frames will be output in a j2k file (output_1.j2k, ...)\n");
+                       
+      sprintf(outfilename,"output_%d.j2k",snum);
+      outfile = fopen(outfilename, "wb");
+      if (!outfile) {
+                               fprintf(stderr, "failed to open %s for writing\n",outfilename);
+                               return 1;
+      }
+      fwrite(frame_codestream,sample->sample_size-8,1,outfile);
+      fclose(outfile);
+    }
+               /* close the byte stream */
+               opj_cio_close(cio);     
+               /* free image data structure */
+               opj_image_destroy(img);
+               elapsed_time = opj_clock()-init_time;
+               fprintf(stderr, "Frame number %d/%d decoded in %.2f mseconds\n", snum + 1, numframes, elapsed_time*1000);
+               total_time += elapsed_time;
 
+  }
+       
+       free(frame_codestream); 
+  fclose(file);        
+       /* free remaining structures */
+       if(dinfo) {
+               mj2_destroy_decompress((opj_mj2_t*)dinfo->mj2_handle);
+       }
+       free(dinfo);
+       
+       fprintf(stdout, "%d frame(s) correctly decompressed\n", snum);
+       fprintf(stdout,"Total decoding time: %.2f seconds (%.1f fps)\n", total_time, (float)numframes/total_time);
+               
   return 0;
 }