libs/libkdcraw/libraw/samples/dcraw_emu.cpp

   1 /* -*- C++ -*-
   2  * File: dcraw_emu.cpp
   3  * Copyright 2008-2009 Alex Tutubalin <lexa@lexa.ru>
   4  * Created: Sun Mar 23,   2008
   5  *
   6  * LibRaw simple C++ API  (almost complete dcraw emulator)
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2, or (at your option)
  11  * any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  21  * 02111-1307, USA.
  22  */
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <stdlib.h>
  26 #include <math.h>
  27 #include <ctype.h>
  28 #ifndef WIN32
  29 #include <fcntl.h>
  30 #include <unistd.h>
  31 #include <sys/stat.h>
  32 #include <sys/mman.h>
  33 #endif
  34
  35 #include "libraw/libraw.h"
  36 #ifdef WIN32
  37 #define snprintf _snprintf
  38 #endif
  39
  40
  41 void usage(const char *prog)
  42 {
  43     printf("dcraw_emu: almost complete dcraw emulator\n");
  44     printf("Usage:  %s [OPTION]... [FILE]...\n", prog);
  45     printf(
  46 "-v        Verbose: print progress messages (repeated -v will add verbosity)\n"
  47 "-w        Use camera white balance, if possible\n"
  48 "-a        Average the whole image for white balance\n"
  49 "-A <x y w h> Average a grey box for white balance\n"
  50 "-r <r g b g> Set custom white balance\n"
  51 "+M/-M     Use/don't use an embedded color matrix\n"
  52 "-C <r b>  Correct chromatic aberration\n"
  53 "-P <file> Fix the dead pixels listed in this file\n"
  54 "-K <file> Subtract dark frame (16-bit raw PGM)\n"
  55 "-k <num>  Set the darkness level\n"
  56 "-S <num>  Set the saturation level\n"
  57 "-n <num>  Set threshold for wavelet denoising\n"
  58 "-H [0-9]  Highlight mode (0=clip, 1=unclip, 2=blend, 3+=rebuild)\n"
  59 "-t [0-7]  Flip image (0=none, 3=180, 5=90CCW, 6=90CW)\n"
  60 "-o [0-5]  Output colorspace (raw,sRGB,Adobe,Wide,ProPhoto,XYZ)\n"
  61 #ifndef NO_LCMS
  62 "-o file   Output ICC profile\n"
  63 "-p file   Camera input profile (use \'embed\' for embedded profile)\n"
  64 #endif
  65 "-j        Don't stretch or rotate raw pixels\n"
  66 "-W        Don't automatically brighten the image\n"
  67 "-b <num>  Adjust brightness (default = 1.0)\n"
  68 "-q [0-3]  Set the interpolation quality\n"
  69 "-h        Half-size color image (twice as fast as \"-q 0\")\n"
  70 "-f        Interpolate RGGB as four colors\n"
  71 "-m <num>  Apply a 3x3 median filter to R-G and B-G\n"
  72 "-s [0..N-1] Select one raw image from input file\n"
  73 "-4        Write 16-bit linear instead of 8-bit with gamma\n"
  74 "-g pow ts Set gamma curve to gamma pow and toe slope ts (default = 2.222 4.5)\n"
  75 "-T        Write TIFF instead of PPM\n"
  76 #ifndef WIN32
  77 "-B        Use mmap()-ed buffer instead of plain FILE I/O\n"
  78 #endif
  79         );
  80     exit(1);
  81 }
  82
  83 static int verbosity=0;
  84
  85 int cnt=0;
  86 int my_progress_callback(void *d,enum LibRaw_progress p,int iteration, int expected)
  87 {
  88     char *passed  = (char*)(d?d:"default string"); // data passed to callback at set_callback stage
  89
  90     if(verbosity>2) // verbosity set by repeat -v switches
  91         {
  92             printf("CB: %s  pass %d of %d (data passed=%s)\n",libraw_strprogress(p),iteration,expected,passed);
  93         }
  94     else if (iteration == 0) // 1st iteration of each step
  95         printf("Starting %s (expecting %d iterations)\n", libraw_strprogress(p),expected);
  96     else if (iteration == expected-1)
  97         printf("%s finished\n",libraw_strprogress(p));
  98
  99 ///    if(++cnt>10) return 1; // emulate user termination on 10-th callback call
 100
 101     return 0; // always return 0 to continue processing
 102 }
 103
 104
 105 int main(int argc, char *argv[])
 106 {
 107     if(argc==1) usage(argv[0]);
 108
 109     LibRaw RawProcessor;
 110     int i,arg,c,ret;
 111     char opm,opt,*cp,*sp;
 112     int use_mmap=0, msize;
 113     void *iobuffer;
 114
 115 #define OUT RawProcessor.imgdata.params
 116
 117   argv[argc] = "";
 118   for (arg=1; (((opm = argv[arg][0]) - 2) | 2) == '+'; )
 119       {
 120           opt = argv[arg++][1];
 121           if ((cp = strchr (sp="nbrkStqmHACg", opt)))
 122               for (i=0; i < "11411111142"[cp-sp]-'0'; i++)
 123                   if (!isdigit(argv[arg+i][0]))
 124                       {
 125                           fprintf (stderr,"Non-numeric argument to \"-%c\"\n", opt);
 126                           return 1;
 127                       }
 128           switch (opt)
 129               {
 130               case 'v':  verbosity++;  break;
 131
 132               case 'n':  OUT.threshold   = atof(argv[arg++]);  break;
 133               case 'b':  OUT.bright      = atof(argv[arg++]);  break;
 134               case 'P':  OUT.bad_pixels  = argv[arg++];        break;
 135               case 'K':  OUT.dark_frame  = argv[arg++];        break;
 136               case 'r':
 137                   for(c=0;c<4;c++)
 138                       OUT.user_mul[c] = atof(argv[arg++]);
 139                   break;
 140               case 'C':
 141                   OUT.aber[0] = 1 / atof(argv[arg++]);
 142                   OUT.aber[2] = 1 / atof(argv[arg++]);
 143                   break;
 144               case 'g':
 145                   OUT.gamm[0] = 1 / atof(argv[arg++]);
 146                   OUT.gamm[1] =     atof(argv[arg++]);
 147                   break;
 148               case 'k':  OUT.user_black  = atoi(argv[arg++]);  break;
 149               case 'S':  OUT.user_sat    = atoi(argv[arg++]);  break;
 150               case 't':  OUT.user_flip   = atoi(argv[arg++]);  break;
 151               case 'q':  OUT.user_qual   = atoi(argv[arg++]);  break;
 152               case 'm':  OUT.med_passes  = atoi(argv[arg++]);  break;
 153               case 'H':  OUT.highlight   = atoi(argv[arg++]);  break;
 154               case 's':  OUT.shot_select = abs(atoi(argv[arg++])); break;
 155               case 'o':
 156                   if(isdigit(argv[arg+1][0]) && !isdigit(argv[arg+1][1]))
 157                           OUT.output_color = atoi(argv[arg++]);
 158 #ifndef NO_LCMS
 159                    else
 160                          OUT.output_profile = argv[arg++];
 161                   break;
 162               case 'p':  OUT.camera_profile = argv[arg++];
 163 #endif
 164                   break;
 165               case 'h':  OUT.half_size         = 1;
 166                   // no break:  "-h" implies "-f"
 167               case 'f':
 168                   OUT.four_color_rgb    = 1;
 169                   break;
 170               case 'A':  for(c=0; c<4;c++) OUT.greybox[c]  = atoi(argv[arg++]);
 171               case 'a':  OUT.use_auto_wb       = 1;  break;
 172               case 'w':  OUT.use_camera_wb     = 1;  break;
 173               case 'M':  OUT.use_camera_matrix = (opm == '+');  break;
 174               case 'j':  OUT.use_fuji_rotate   = 0;  break;
 175               case 'W':  OUT.no_auto_bright    = 1;  break;
 176               case 'T':  OUT.output_tiff       = 1;  break;
 177               case '4':  OUT.output_bps       = 16;  break;
 178               case '1':  OUT.gamma_16bit       = 1;  break;
 179 #ifndef WIN32
 180               case 'B':  use_mmap              = 1;  break;
 181 #endif
 182               default:
 183                   fprintf (stderr,"Unknown option \"-%c\".\n", opt);
 184                   return 1;
 185               }
 186       }
 187   putenv ((char*)"TZ=UTC"); // dcraw compatibility, affects TIFF datestamp field
 188   OUT.filtering_mode = LIBRAW_FILTERING_AUTOMATIC;
 189 #define P1 RawProcessor.imgdata.idata
 190 #define S RawProcessor.imgdata.sizes
 191 #define C RawProcessor.imgdata.color
 192 #define T RawProcessor.imgdata.thumbnail
 193 #define P2 RawProcessor.imgdata.other
 194
 195   if(verbosity>1)
 196           RawProcessor.set_progress_handler(my_progress_callback,(void*)"Sample data passed");
 197 #ifdef _OPENMP
 198   if(verbosity)
 199           printf ("Using %d threads\n", omp_get_max_threads());
 200 #endif
 201
 202   for ( ; arg < argc; arg++)
 203         {
 204             char outfn[1024];
 205
 206             if(verbosity) printf("Processing file %s\n",argv[arg]);
 207 #ifndef WIN32
 208             if(use_mmap)
 209                 {
 210                     int file = open(argv[arg],O_RDONLY);
 211                     struct stat st;
 212                     if(file<0)
 213                         {
 214                             fprintf(stderr,"Cannot open %s: %s\n",argv[arg],strerror(errno));
 215                             continue;
 216                         }
 217                     if(fstat(file,&st))
 218                         {
 219                             fprintf(stderr,"Cannot stat %s: %s\n",argv[arg],strerror(errno));
 220                             close(file);
 221                             continue;
 222                         }
 223                     int pgsz = getpagesize();
 224                     msize = ((st.st_size+pgsz-1)/pgsz)*pgsz;
 225                     iobuffer = mmap(NULL,msize,PROT_READ,MAP_PRIVATE,file,0);
 226                     if(!iobuffer)
 227                         {
 228                             fprintf(stderr,"Cannot mmap %s: %s\n",argv[arg],strerror(errno));
 229                             close(file);
 230                             continue;
 231                         }
 232                     close(file);
 233                     if( (ret = RawProcessor.open_buffer(iobuffer,st.st_size) != LIBRAW_SUCCESS))
 234                         {
 235                             fprintf(stderr,"Cannot open_buffer %s: %s\n",argv[arg],libraw_strerror(ret));
 236                             continue; // no recycle b/c open file will recycle itself
 237                         }
 238
 239                 }
 240             else
 241 #endif
 242                 {
 243                     if( (ret = RawProcessor.open_file(argv[arg])) != LIBRAW_SUCCESS)
 244                         {
 245                             fprintf(stderr,"Cannot open %s: %s\n",argv[arg],libraw_strerror(ret));
 246                             continue; // no recycle b/c open_file will recycle itself
 247                         }
 248                 }
 249             if( (ret = RawProcessor.unpack() ) != LIBRAW_SUCCESS)
 250                 {
 251                     fprintf(stderr,"Cannot unpack %s: %s\n",argv[arg],libraw_strerror(ret));
 252                     continue;
 253                 }
 254             if (LIBRAW_SUCCESS != (ret = RawProcessor.dcraw_process()))
 255                 {
 256                     fprintf(stderr,"Cannot do postpocessing on %s: %s\n",argv[arg],libraw_strerror(ret));
 257                     if(LIBRAW_FATAL_ERROR(ret))
 258                         continue;
 259                 }
 260             snprintf(outfn,sizeof(outfn),
 261                      "%s.%s",
 262                      argv[arg], OUT.output_tiff ? "tiff" : (P1.colors>1?"ppm":"pgm"));
 263
 264             if(verbosity) printf("Writing file %s\n",outfn);
 265
 266             if( LIBRAW_SUCCESS != (ret = RawProcessor.dcraw_ppm_tiff_writer(outfn)))
 267                 fprintf(stderr,"Cannot write %s: %s\n",outfn,libraw_strerror(ret));
 268
 269 #ifndef WIN32
 270             if(use_mmap && iobuffer)
 271                 {
 272                     munmap(iobuffer,msize);
 273                     iobuffer=0;
 274                 }
 275 #endif
 276
 277             RawProcessor.recycle(); // just for show this call
 278         }
 279     return 0;
 280 }