apps/png/pnghist_nocgi.c

Go to the documentation of this file.

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <math.h>

#include <asm/elphel/autoexp.h>
#include <asm/elphel/hist.h>

#include <sys/mman.h>
#include <asm/elphel/c313a.h>

 
//#include "zlib.h"
#include "libpng/png.h"
//#define PNGHISTDEBUG 1
#define HIST_SIZE 256 * 4 * 4

#define HPNG_PLOT_BARS    1
#define HPNG_CONNECT_DOTS 2
#define HPNG_FILL_ZEROS   4
#define HPNG_LIN_INTERP   8
#define HPNG_RUN_SQRT     16
struct autoexp_t autoexp;
int write_png(char *file_name, int height, int data[1024], int mode, int colors);
int main(argc, argv)
    int argc;
    char *argv[];
{
  int hists[1536];
  int rava[256];
//  int fd;
  int autoexp_fd;
  int i,j,k,a,l;
//  short int hsz[2];
  double dscale,dcoeff;
  int coeff;
  int png_height;
  int ww,wh;
  int devfd;
  int mode;
  int colors=57; // colors 1 - r, 2 - g1, 4 - g2, 8 - b, 16 - w, 32 - g1+g2 
  int lastmiddle_i,newmiddle_i;
  int rav=0; // running average
  int z;
  void *_map;  
  const char colors_calc[]={  0,  1,  2,  3,  4,  5,  6,  7,  8,  9,0xa,0xb,0xc,0xd,0xe,0xf,
                      0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,
                        6,  7,  6,  7,  6,  7,  6,  7,0xe,0xf,0xe,0xf,0xe,0xf,0xe,0xf,
                      0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf,0xf};
  if (argc<5) {printf("pnghist height scale mode colors average"); return 0;}
  png_height=atoi(argv[1]);
  if (png_height>256) png_height=256;
  dscale=atof(argv[2]);
  mode=atoi(argv[3]);
  colors=atoi(argv[4]) & 0x3f;
  
 
  rav=atoi(argv[5]);
  
 
  if ((devfd = open("/dev/ccam_img", O_RDONLY)) < 0) {printf("Device busy (%s)\r\n", "/dev/ccam_img"); return -1;}

  ww=ioctl(devfd, _CCCMD( CCAM_RPARS , P_ACTUAL_WIDTH), 0);
  wh=ioctl(devfd, _CCCMD( CCAM_RPARS , P_ACTUAL_HEIGHT), 0);
  close (devfd);
//  printf ("w=%d, h=%d\n",ww,wh);
  
//  autoexp_fd = open(AUTOEXP_DEV_NAME, O_RDWR);
  autoexp_fd = open(AUTOEXP_DEV_NAME, O_RDONLY);
  if(autoexp_fd < 0) {
     printf("open %s failed - may be busy, or not started\r\n", AUTOEXP_DEV_NAME);
     return -1;
  }
  z = ioctl(autoexp_fd, _IO(IOC_AUTOEXP_GET, 0x00), &autoexp);
  
  
//  if((_map = mmap(NULL, HIST_SIZE, PROT_READ | PROT_WRITE, MAP_FIXED | MAP_SHARED, autoexp_fd, 0)) == MAP_FAILED) {
  if((_map = mmap(NULL, HIST_SIZE, PROT_READ, MAP_FIXED | MAP_SHARED, autoexp_fd, 0)) == MAP_FAILED) {
     printf("mmap %s failed - may be busy, or not started\r\n", AUTOEXP_DEV_NAME);
     return -1;
  }
  memcpy (hists,_map,HIST_SIZE);
  close (autoexp_fd);
#if 0  
  fd = open("/tmp/histogram", O_RDONLY);
  if(fd <= 0) {
      printf("No histogram file (%s)\r\n", "/tmp/histogram");
      return -1;
  }
  read(fd, hsz, 4);
  ww=(ww*hsz[0])/100;
  wh=(wh*hsz[1])/100;
  read(fd, hists, HIST_SIZE);
  close (fd);
#endif  
  ww=(ww*autoexp.width)/100;
  wh=(wh*autoexp.height)/100;
  dcoeff=(png_height*1024)/(dscale*ww*wh);
  coeff=dcoeff*65536;
#ifdef PNGHISTDEBUG
  printf ("w=%d, h=%d\n",ww,wh);
  for (i=0;i<1024;i++) {
    printf ("%5d ",hists[i]);
    if ((i & 0x0f)== 0x0f) printf ("\n");
  if ((i & 0xff) == 0xff) printf ("-----------------------------------------------\n");
    
  }
  printf ("===============================================\n");
#endif  

  

  if ((mode & (HPNG_FILL_ZEROS | HPNG_LIN_INTERP)) | (rav>1)) { // spread between zeros - needed at high digital gains (near black with low gammas)
    for (j=0;j<4;j++)  if (colors_calc[colors] & (1<<j)) {
      lastmiddle_i=255;
      i=255;
      while ((hists[i+(j<<8)]==0) && (i>0)) i--;// skip high zeros
      while (i>0) {
       k=i;
       i--;
       while ((i>=0) && (hists[i+(j<<8)]==0)) i--;
       if ((k-i)>1) {
         a=hists[k+(j<<8)]/(k-i);
         for (l=k;l>i;l--) hists[l+(j<<8)] = a;
       }
       if ((mode & HPNG_LIN_INTERP) | (rav>1)) { //linear interpolation - also for high digital gains (near black with low gammas)
         newmiddle_i= (k+i+1)/2;
         for (l=lastmiddle_i-1;l>newmiddle_i;l--) { 
            hists[l+(j<<8)]= hists[newmiddle_i+(j<<8)] + ((l-newmiddle_i)* (hists[lastmiddle_i+(j<<8)]-hists[newmiddle_i+(j<<8)]))/(lastmiddle_i-newmiddle_i);
         }   
         lastmiddle_i=newmiddle_i;
       
       }
      }
// apply running average (if any)
      if (rav>0) {
        k=0;
        l=0;
        for (i=0;i<256;i++) {
         rava[i]=hists[i+(j<<8)];
         k+=rava[i];
         if ((i-rav)>=0) k-=rava[i-rav];
         else l++;
         hists[i+(j<<8)]=k/l;
        }
      } 
      
      
    }
  }
// calculate white if needed
  if (colors & 0x10) for (i=0;i<256;i++) hists[i+1024]=(hists[i]+hists[i+256]+hists[i+512]+hists[i+768])>>2;
// calculate green if needed
  if (colors & 0x20) for (i=0;i<256;i++) hists[i+1280]=(hists[i+256]+hists[i+512])>>1;
  
  if (mode & HPNG_RUN_SQRT) {
    dcoeff=sqrt((32768.0*png_height)/coeff);
    for (j=0;j<6;j++) if (colors & (1 << j)) for (i=0;i<256;i++) hists[i+(j<<8)]=sqrt(hists[i+(j<<8)])*dcoeff;
  }
// now scale data
  for (j=0;j<6;j++) if (colors & (1 << j)) for (i=0;i<256;i++) {
   hists[i+(j<<8)]=(hists[i+(j<<8)]*coeff) >> 16;
   if (hists[i+(j<<8)] > (png_height-1)) hists[i+(j<<8)] = png_height-1;
  }   

#ifdef PNGHISTDEBUG
  printf ("w=%d, h=%d\n",ww,wh);
  for (i=0;i<1536;i++) {
    printf ("%5d ",hists[i]);
    if ((i & 0x0f)== 0x0f) printf ("\n");
  }
#endif  
//      sq = state.width * state.height;

  write_png("/var/html/hist.png", png_height, hists, mode, colors);
  
  return 0;
}
#define ERROR -1
#define OK 0
//#define HPNG_PLOT_BARS    1
//#define HPNG_CONNECT_DOTS 2


int write_png(char *file_name, int png_height, int data[1536], int mode, int colors)
{
   FILE *fp;
   png_structp png_ptr;
   png_infop info_ptr;
//   const char clrs[]={1,2,2,4,15,2}; // dim white
   const char clrs[]={1,2,2,4,7,2}; // bright white
//   png_colorp palette;
   unsigned char cpalette[]= {0,   0,   0,
                        0xff,0,   0,
                        0,   0xff,0,
                        0xff,0xff,0,
                        0,   0,   0xff,
                        0xff,0,   0xff,
                        0,   0xff,0xff,
                        0xff,0xff,0xff,
                        0,   0,   0,
                        0x80,0,   0,
                        0,   0x80,0,
                        0x80,0x80,0,
                        0,   0,   0x80,
                        0x80,0,   0x80,
                        0,   0x80,0x80,
                        0x80,0x80,0x80};
  unsigned char ctransparency[]={0,  255,255,255,255,255,255,255,
                                 255,255,255,255,255,255,255,255};
  png_bytep   trans =   (png_bytep) ctransparency;
  
  png_colorp  palette = (png_colorp ) cpalette;
//   palette = (png_colorp)png_malloc(png_ptr, PNG_MAX_PALETTE_LENGTH
//             * png_sizeof (png_color));
   png_uint_32 k;
   png_uint_32 height=png_height;
   png_uint_32 width=256;
   png_uint_32 bytes_per_pixel=1;
   png_byte image[256][256];
   png_bytep row_pointers[256];
   int i,j,c0,c,d,l,last;
   if (height>256) height=256;
   
   memset(image, 0, sizeof(image));
   for (j=0;j<6;j++) if (colors & (1 << j)){
     c0= clrs[j];
     last=png_height-1;
     for (i=0;i<256;i++) {
       c=c0 << ((i&1)?0:4);
       d=png_height-1-data[i+(j<<8)];
       if (d<=0) d=0;
       if (d>=png_height) d=png_height-1;
       image[d][i>>1] |= c;
       if (mode & 1){
//         l=d-mode;
//         if ((i&0x10)==0) l-=mode;
//         l-= (i & 0xf);
//         if (l<0) l=0;
         for (l=d+1; l<png_height; l++)image[l][i>>1] |= c;
//         for (l=png_height; l>d; l--)image[l][i] |= c;
       } else if (mode & 2) {
         for (l=d+1; l<last; l++) image[l][i>>1] |= c; // only one of "for " can fire
         for (l=d-1; l>last; l--) image[l][i>>1] |= c; 
         last=d;
       
       }
       
     }
   }
   
   /* open the file */
   fp = fopen(file_name, "wb");
   if (fp == NULL)
      return (ERROR);

   /* Create and initialize the png_struct with the desired error handler
    * functions.  If you want to use the default stderr and longjump method,
    * you can supply NULL for the last three parameters.  We also check that
    * the library version is compatible with the one used at compile time,
    * in case we are using dynamically linked libraries.  REQUIRED.
    */
   png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING,
      NULL,NULL,NULL);

   if (png_ptr == NULL)
   {
      fclose(fp);
      return (ERROR);
   }

   /* Allocate/initialize the image information data.  REQUIRED */
   info_ptr = png_create_info_struct(png_ptr);
   if (info_ptr == NULL)
   {
      fclose(fp);
      png_destroy_write_struct(&png_ptr,  png_infopp_NULL);
      return (ERROR);
   }

   /* Set error handling.  REQUIRED if you aren't supplying your own
    * error handling functions in the png_create_write_struct() call.
    */
   if (setjmp(png_jmpbuf(png_ptr)))
   {
      /* If we get here, we had a problem reading the file */
      fclose(fp);
      png_destroy_write_struct(&png_ptr, &info_ptr);
      return (ERROR);
   }

   /* One of the following I/O initialization functions is REQUIRED */
//#ifdef streams /* I/O initialization method 1 */
   /* set up the output control if you are using standard C streams */
   png_init_io(png_ptr, fp);
//#else  no_streams /* I/O initialization method 2 */
   /* If you are using replacement read functions, instead of calling
    * png_init_io() here you would call */
//   png_set_write_fn(png_ptr, (void *)user_io_ptr, user_write_fn,
//      user_IO_flush_function);
   /* where user_io_ptr is a structure you want available to the callbacks */
//#endif// no_streams /* only use one initialization method */

   /* This is the hard way */

   /* Set the image information here.  Width and height are up to 2^31,
    * bit_depth is one of 1, 2, 4, 8, or 16, but valid values also depend on
    * the color_type selected. color_type is one of PNG_COLOR_TYPE_GRAY,
    * PNG_COLOR_TYPE_GRAY_ALPHA, PNG_COLOR_TYPE_PALETTE, PNG_COLOR_TYPE_RGB,
    * or PNG_COLOR_TYPE_RGB_ALPHA.  interlace is either PNG_INTERLACE_NONE or
    * PNG_INTERLACE_ADAM7, and the compression_type and filter_type MUST
    * currently be PNG_COMPRESSION_TYPE_BASE and PNG_FILTER_TYPE_BASE. REQUIRED
    */
   png_set_IHDR(png_ptr,
                info_ptr,
                256, // width
                png_height, // height
                4,          // bit_depth,
                PNG_COLOR_TYPE_PALETTE, //PNG_COLOR_TYPE_???,
                PNG_INTERLACE_NONE, //PNG_INTERLACE_????,
                PNG_COMPRESSION_TYPE_BASE,
                PNG_FILTER_TYPE_BASE);

   /* set the palette if there is one.  REQUIRED for indexed-color images */
//   palette = (png_colorp)png_malloc(png_ptr, PNG_MAX_PALETTE_LENGTH
//             * png_sizeof (png_color));
   /* ... set palette colors ... */
   png_set_PLTE(png_ptr,
                info_ptr,
                palette,
                PNG_MAX_PALETTE_LENGTH);
   png_set_tRNS(png_ptr,
                info_ptr,
                trans,
                sizeof(trans), //16,
                NULL); //, png_color_16p trans_values)                
   /* You must not free palette here, because png_set_PLTE only makes a link to
      the palette that you malloced.  Wait until you are about to destroy
      the png structure. */

   /* optional significant bit chunk */
   /* if we are dealing with a grayscale image then */
#if 0   
   sig_bit.gray = 1; //true_bit_depth;
   /* otherwise, if we are dealing with a color image then */
   sig_bit.red = 1; //true_red_bit_depth;
   sig_bit.green = 1; //true_green_bit_depth;
   sig_bit.blue = 1; //true_blue_bit_depth;
   /* if the image has an alpha channel then */
   sig_bit.alpha = 1; //true_alpha_bit_depth;
   png_set_sBIT(png_ptr, info_ptr, sig_bit);
#endif

   /* Optional gamma chunk is strongly suggested if you have any guess
    * as to the correct gamma of the image.
    */
//   png_set_gAMA(png_ptr, info_ptr, gamma);

   /* Optionally write comments into the image */
#if 0  
   text_ptr[0].key = "Title";
   text_ptr[0].text = "Histogram";// "Mona Lisa";
   text_ptr[0].compression = PNG_TEXT_COMPRESSION_NONE;
   text_ptr[1].key = "Author";
   text_ptr[1].text = "Elphel"; //"Leonardo DaVinci";
   text_ptr[1].compression = PNG_TEXT_COMPRESSION_NONE;
   text_ptr[2].key = "Description";
   text_ptr[2].text = "color histograms for the camera image/video"; //"<long text>";
   text_ptr[2].compression = PNG_TEXT_COMPRESSION_zTXt;
#ifdef PNG_iTXt_SUPPORTED
   text_ptr[0].lang = NULL;
   text_ptr[1].lang = NULL;
   text_ptr[2].lang = NULL;
#endif
   png_set_text(png_ptr, info_ptr, text_ptr, 3);
#endif

   /* other optional chunks like cHRM, bKGD, tRNS, tIME, oFFs, pHYs, */
   /* note that if sRGB is present the gAMA and cHRM chunks must be ignored
    * on read and must be written in accordance with the sRGB profile */

   /* Write the file header information.  REQUIRED */
   png_write_info(png_ptr, info_ptr);

   /* If you want, you can write the info in two steps, in case you need to
    * write your private chunk ahead of PLTE:
    *
    *   png_write_info_before_PLTE(write_ptr, write_info_ptr);
    *   write_my_chunk();
    *   png_write_info(png_ptr, info_ptr);
    *
    * However, given the level of known- and unknown-chunk support in 1.1.0
    * and up, this should no longer be necessary.
    */

   /* Once we write out the header, the compression type on the text
    * chunks gets changed to PNG_TEXT_COMPRESSION_NONE_WR or
    * PNG_TEXT_COMPRESSION_zTXt_WR, so it doesn't get written out again
    * at the end.
    */

   /* set up the transformations you want.  Note that these are
    * all optional.  Only call them if you want them.
    */

   /* invert monochrome pixels */
//   png_set_invert_mono(png_ptr);

   /* Shift the pixels up to a legal bit depth and fill in
    * as appropriate to correctly scale the image.
    */
//   png_set_shift(png_ptr, &sig_bit);

   /* pack pixels into bytes */
//   png_set_packing(png_ptr);

   /* swap location of alpha bytes from ARGB to RGBA */
//   png_set_swap_alpha(png_ptr);

   /* Get rid of filler (OR ALPHA) bytes, pack XRGB/RGBX/ARGB/RGBA into
    * RGB (4 channels -> 3 channels). The second parameter is not used.
    */
//   png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE);

   /* flip BGR pixels to RGB */
//   png_set_bgr(png_ptr);

   /* swap bytes of 16-bit files to most significant byte first */
//   png_set_swap(png_ptr);

   /* swap bits of 1, 2, 4 bit packed pixel formats */
//   png_set_packswap(png_ptr);

   /* turn on interlace handling if you are not using png_write_image() */
#if 0   
   if (interlacing)
      number_passes = png_set_interlace_handling(png_ptr);
   else
      number_passes = 1;
#endif   

   /* The easiest way to write the image (you may have a different memory
    * layout, however, so choose what fits your needs best).  You need to
    * use the first method if you aren't handling interlacing yourself.
    */
//   png_uint_32 k, height, width;
//   png_byte image[height][width*bytes_per_pixel];
//   png_bytep row_pointers[height];

   if (height > PNG_UINT_32_MAX/png_sizeof(png_bytep))
     png_error (png_ptr, "Image is too tall to process in memory");

   for (k = 0; k < height; k++)
     row_pointers[k] = (png_bytep) image + k*width*bytes_per_pixel;

   /* One of the following output methods is REQUIRED */
   png_write_image(png_ptr, row_pointers);


   /* You can write optional chunks like tEXt, zTXt, and tIME at the end
    * as well.  Shouldn't be necessary in 1.1.0 and up as all the public
    * chunks are supported and you can use png_set_unknown_chunks() to
    * register unknown chunks into the info structure to be written out.
    */

   /* It is REQUIRED to call this to finish writing the rest of the file */
   png_write_end(png_ptr, info_ptr);

   /* If you png_malloced a palette, free it here (don't free info_ptr->palette,
      as recommended in versions 1.0.5m and earlier of this example; if
      libpng mallocs info_ptr->palette, libpng will free it).  If you
      allocated it with malloc() instead of png_malloc(), use free() instead
      of png_free(). */
//   png_free(png_ptr, palette);
//   palette=NULL;

   /* Similarly, if you png_malloced any data that you passed in with
      png_set_something(), such as a hist or trans array, free it here,
      when you can be sure that libpng is through with it. */
//   png_free(png_ptr, trans);
//   trans=NULL;

   /* clean up after the write, and free any memory allocated */
   png_destroy_write_struct(&png_ptr, &info_ptr);

   /* close the file */
   fclose(fp);

   /* that's it */
   return (OK);
}

---