1 files changed, 295 insertions, 0 deletions
diff --git a/conversion/test/test_jpeg.cc b/conversion/test/test_jpeg.cc
new file mode 100644
index 0000000..e56a3d4
--- /dev/null
+++ b/conversion/test/test_jpeg.cc
@@ -0,0 +1,295 @@
+#include <stdlib.h>
+#include <stdio.h>
+#include <iostream>
+#include <string.h>
+#include <math.h>
+
+
+#include "conversion/convertor.hh"
+#include "conversion/jpeg.hh"
+
+#ifdef EARLY_TEMPLATE
+#define __EXECUTABLE__
+#include "raslib/template_inst.hh"
+#endif
+
+static int init_identity(int offset)
+{
+  return offset;
+}
+
+static int init_square(int offset)
+{
+  return offset * offset;
+}
+
+static int SinTabOK = 0;
+static int SinusTable[512];
+
+static int init_sine(int offset)
+{
+  if (SinTabOK == 0)
+  {
+    int i;
+
+    for (i=0; i<512; i++)
+    {
+      SinusTable[i] = (int)(255*sin(M_PI * offset / 512.0));
+    }
+    SinTabOK = 1;
+  }
+  return SinusTable[offset & 511];
+}
+
+static int JpegErrorFileNumber;
+
+int test_picture(int width, int height, int bpp, const char *params, int (*init)(int))
+{
+  char *data;
+  r_Minterval interv(2);
+  int datasize;
+  int baseType;
+  int bytespp;
+  int i;
+
+  switch (bpp)
+  {
+    case 1: baseType = r_Convertor::ctype_bool; bytespp = 1; break;
+    case 8: baseType = r_Convertor::ctype_char; bytespp = 1; break;
+    case 24: baseType = r_Convertor::ctype_rgb; bytespp = 3; break;
+    default:
+      cerr << "Unknown bpp value " << bpp << endl;
+      return -1;
+  }
+
+  interv << r_Sinterval((r_Range)0, (r_Range)width-1)
+	 << r_Sinterval((r_Range)0, (r_Range)height-1);
+
+  datasize = (width * bytespp * height);
+  data = new char[datasize];
+
+  cout << "Test: bpp = " << bpp << ", domain = " << interv << ", params = " << ((params == NULL) ? "" : params) << ", size = " << datasize << endl;
+
+  if (bpp > 1)
+  {
+    for (i=0; i<datasize; i++) data[i] = (char)init(i);
+  }
+  else
+  {
+    int j, count;
+
+    for (i=0; i<datasize; i+=8)
+    {
+      count = datasize - i; if (count > 8) count = 8;
+      for (j=0; j<count; j++) data[i+j] = ((init(i>>3) & (1<<j)) == 0) ? 0 : 1;
+    }
+  }
+
+  r_Conv_JPEG *jpeg = NULL;
+  char *dest = NULL;
+
+  try
+  {
+    r_convDesc desc;
+    int destsize;
+    int status = 0;
+
+    jpeg = new r_Conv_JPEG(data, interv, baseType);
+    desc = jpeg->convertTo(params);
+    destsize = desc.destInterv[0].high() - desc.destInterv[0].low() + 1;
+    cout << "\tConverted to JPEG, domain " << desc.destInterv << endl;
+    dest = desc.dest;
+    r_Minterval destInterv(desc.destInterv);
+    delete desc.destType;
+    delete jpeg;
+    jpeg = new r_Conv_JPEG(dest, destInterv, r_Convertor::ctype_char);
+    desc = jpeg->convertFrom();
+    cout << "\tConverted from JPEG, domain " << desc.destInterv << endl;
+    
+    if (interv == desc.destInterv)
+    {
+      unsigned char *b, *d;
+      double totalError;
+      unsigned char maxError;
+      int i;
+
+      // Bitmaps get expanded to greyscale...
+      if (baseType == r_Convertor::ctype_bool)
+      {
+	for (i=0; i<datasize; i++)
+	{
+	  if (data[i] != 0) data[i] = 0xff;
+	}
+      }
+
+      b = (unsigned char*)data; d = (unsigned char*)desc.dest;
+      totalError = 0.0; maxError = abs(*b - *d);
+      for (i=0; i<datasize; i++)
+      {
+	b[i] = abs(b[i] - d[i]);
+	if (b[i] > maxError) maxError = b[i];
+	totalError += (double)(b[i]);
+      }
+
+      r_Conv_JPEG *errpeg;
+      r_convDesc cdsc;
+      char errfile[256];
+      FILE *fp;
+      int errType;
+
+      errType = (baseType == r_Convertor::ctype_bool) ? r_Convertor::ctype_char : baseType;
+      sprintf(errfile, "jerror%d.jpg", JpegErrorFileNumber++);
+      cout << "\tCreating error file <" << errfile << "> ..." << endl;
+      cout << "\t(max error = " << (int)maxError << ", avg error = "
+	   << totalError / (double)datasize << ")" << endl;
+
+      errpeg = new r_Conv_JPEG(data, interv, errType);
+      cdsc = errpeg->convertTo("quality=90");
+      i = (int)(cdsc.destInterv[0].high() - cdsc.destInterv[0].low() + 1);
+      if ((fp = fopen(errfile, "wb")) == NULL)
+      {
+	cerr << "\tUnable to open output file!" << endl;
+      }
+      else
+      {
+	fwrite(cdsc.dest, 1, i, fp);
+	fclose(fp);
+      }
+      delete cdsc.destType;
+      free(cdsc.dest);
+      delete errpeg;
+    }
+    else
+    {
+      cout << "\t Domain differs!!!" << endl;
+      status = -1;
+    }
+
+    delete jpeg;
+    delete [] data;
+    free(dest);
+    free(desc.dest);
+    delete desc.destType;
+
+    return status;
+  }
+  catch (r_Error &err)
+  {
+    cerr << "Error: " << err.what() << endl;
+
+    delete [] data;
+    if (dest != NULL) delete [] dest;
+    if (jpeg != NULL) delete jpeg;
+
+    return -1;
+  }
+}
+
+int main(int argc, char *argv[])
+{
+  int i;
+  char jpegfile[256] = "";
+  char params[256];
+  const char *paramptr = NULL;
+
+  i = 1;
+  while (i < argc)
+  {
+    if (strcmp(argv[i], "-file") == 0)
+    {
+      strcpy(jpegfile, argv[++i]);
+    }
+    else if (strcmp(argv[i], "-quality") == 0)
+    {
+      sprintf(params, "quality=%s", argv[++i]);
+      paramptr = params;
+    }
+    i++;
+  }
+
+  if (jpegfile[0] == '\0')
+  {
+    JpegErrorFileNumber = 0;
+    test_picture(400, 300, 1, paramptr, init_identity);
+    test_picture(400, 300, 8, paramptr, init_identity);
+    test_picture(400, 300, 24, paramptr, init_identity);
+    test_picture(701, 333, 1, paramptr, init_square);
+    test_picture(701, 333, 8, paramptr, init_square);
+    test_picture(701, 333, 24, paramptr, init_square);
+    test_picture(3007, 3999, 1, paramptr, init_sine);
+    test_picture(3007, 3999, 8, paramptr, init_sine);
+    test_picture(3007, 3999, 24, paramptr, init_sine);
+  }
+  else
+  {
+    FILE *fp;
+    long fileSize;
+    char *data = NULL;
+    char *cdata = NULL;
+    r_convDesc desc;
+    r_Type *baseType;
+    r_Conv_JPEG *jpeg;
+    
+    if ((fp = fopen(jpegfile, "rb")) == NULL)
+    {
+      cerr << "Unable to open input file" << endl;
+      return -1;
+    }
+    fseek(fp, 0, SEEK_END); fileSize = ftell(fp); fseek(fp, 0, SEEK_SET);
+    data = new char[fileSize];
+    fread(data, 1, fileSize, fp);
+    fclose(fp);
+
+    jpeg = NULL; baseType = NULL;
+    try
+    {
+      r_Minterval interv(1);
+      interv << r_Sinterval((r_Range)0, (r_Range)fileSize-1);
+
+      jpeg = new r_Conv_JPEG(data, interv, 0);
+      cout << "Converting file <" << jpegfile << "> to MDD..." << endl;
+      desc = jpeg->convertFrom();
+
+      r_Minterval imgInterv(desc.destInterv);
+      delete [] data; data = NULL;
+      cdata = desc.dest;
+      baseType = desc.destType;
+      delete jpeg;
+
+      cout << "Resulting domain " << imgInterv << ", ";
+      baseType->print_status();
+      cout << endl;
+
+      jpeg = new r_Conv_JPEG(cdata, imgInterv, baseType);
+      cout << "Converting MDD back to JPEG..." << endl;
+      desc = jpeg->convertTo(paramptr);
+      free(cdata);
+      cdata = desc.dest;
+  
+      cout << "Resulting domain " << desc.destInterv << ", ";
+      desc.destType->print_status();
+      cout << endl;
+
+      fp = fopen("result.jpg", "wb");
+      fileSize = desc.destInterv[0].high() - desc.destInterv[0].low() + 1;
+      fwrite(cdata, 1, fileSize, fp);
+      fclose(fp);
+
+      delete baseType;
+      free(cdata);
+
+      delete desc.destType;
+      delete jpeg;
+    }
+    catch (r_Error &err)
+    {
+      cerr << "Conversion failed: " << err.what() << endl;
+      if (baseType != NULL) delete baseType;
+      if (cdata != NULL) delete [] cdata;
+      if (jpeg != NULL) delete jpeg;
+    }
+    if (data != NULL)
+      delete [] data;
+  }
+  return 0;
+}