Actual source code: image.c

      1: #include <petsc/private/petscimpl.h>

      3: PETSC_EXTERN PetscErrorCode PetscDrawImageSave(const char[],const char[],unsigned char[][3],unsigned int,unsigned int,const unsigned char[]);
      4: PETSC_EXTERN PetscErrorCode PetscDrawMovieSave(const char[],PetscInt,const char[],PetscInt,const char[]);
      5: PETSC_EXTERN PetscErrorCode PetscDrawImageCheckFormat(const char *[]);
      6: PETSC_EXTERN PetscErrorCode PetscDrawMovieCheckFormat(const char *[]);

      8: /*
      9:    Code to write images in PPM format
     10: */
     11: PETSC_EXTERN PetscErrorCode PetscDrawImageSavePPM(const char filename[],unsigned char palette[][3],unsigned int w,unsigned int h,const unsigned char pixels[])
     12: {
     13:   int            fd;
     14:   char           header[32];
     15:   size_t         hdrlen;
     16:   unsigned char  *rgb;

     23:   /* map pixels to RGB colors */
     24:   if (palette) {
     25:     int k,p,n = (int)(w*h);
     26:     const unsigned char *colordef;
     27:     PetscMalloc1(3*w*h,&rgb);
     28:     for (k=p=0; k<n; k++) {
     29:       colordef = palette[pixels[k]];
     30:       rgb[p++] = colordef[0];
     31:       rgb[p++] = colordef[1];
     32:       rgb[p++] = colordef[2];
     33:     }
     34:   } else { /* assume pixels are RGB colors */
     35:     rgb = (unsigned char*)pixels;
     36:   }
     37:   /* open file and write PPM header */
     38:   PetscBinaryOpen(filename,FILE_MODE_WRITE,&fd);
     39:   PetscSNPrintf(header,sizeof(header),"P6\n%d %d\n255\n\0",(int)w,(int)h);
     40:   PetscStrlen(header,&hdrlen);
     41:   PetscBinaryWrite(fd,header,hdrlen,PETSC_CHAR);
     42:   /* write image data and close file */
     43:   PetscBinaryWrite(fd,rgb,3*w*h,PETSC_CHAR);
     44:   PetscBinaryClose(fd);
     45:   if (palette) {PetscFree(rgb);}
     46:   return(0);
     47: }

     49: static PetscErrorCode PetscDrawImageSave_PPM(const char filename[],unsigned char palette[][3],unsigned int w,unsigned int h,const unsigned char pixels[])
     50: { return PetscDrawImageSavePPM(filename,palette,w,h,pixels); }

     53: /*
     54:    Code to write images in PNG format
     55: */
     56: #if defined(PETSC_HAVE_LIBPNG)

     58: #include <png.h>

     60: #if defined(PNG_SETJMP_SUPPORTED)
     61: # ifndef png_jmpbuf
     62: #   define png_jmpbuf(png_ptr) ((png_ptr)->jmpbuf)
     63: # endif
     64: #endif

     66: PETSC_EXTERN PetscErrorCode PetscDrawImageSavePNG(const char filename[],unsigned char palette[][3],unsigned int w,unsigned int h,const unsigned char pixels[])
     67: {
     68:   FILE           *fp;
     69:   png_struct     *png_ptr;
     70:   png_info       *info_ptr;
     71:   unsigned int   row, stride = palette ? w : 3*w;

     79:   /* open file and create libpng structures */
     80:   PetscFOpen(PETSC_COMM_SELF,filename,"wb",&fp);
     81:   png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING,NULL,NULL,NULL);
     82:   if (!png_ptr) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot create PNG context");
     83:   info_ptr = png_create_info_struct(png_ptr);
     84:   if (!info_ptr) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot create PNG context");

     86:   /* setup libpng error handling */
     87: #if defined(PNG_SETJMP_SUPPORTED)
     88:   if (setjmp(png_jmpbuf(png_ptr))) {
     89:     png_destroy_write_struct(&png_ptr,&info_ptr);
     90:     (void)PetscFClose(PETSC_COMM_SELF,fp);
     91:     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error writing PNG file %s",filename);
     92:   }
     93: #endif

     95:   /* setup PNG image metadata */
     96:   png_init_io(png_ptr, fp);
     97:   png_set_IHDR(png_ptr, info_ptr, w, h, /*depth*/8,
     98:                palette ? PNG_COLOR_TYPE_PALETTE : PNG_COLOR_TYPE_RGB,
     99:                PNG_INTERLACE_NONE,
    100:                PNG_COMPRESSION_TYPE_DEFAULT,
    101:                PNG_FILTER_TYPE_DEFAULT);
    102:   if (palette)
    103:     png_set_PLTE(png_ptr, info_ptr, (png_color*)palette, 256);

    105:   /* write PNG image header and data */
    106:   png_write_info(png_ptr, info_ptr);
    107:   for (row = 0; row < h; row++)
    108:     png_write_row(png_ptr, pixels + row*stride);
    109:   png_write_end(png_ptr, NULL);

    111:   /* destroy libpng structures and close file */
    112:   png_destroy_write_struct(&png_ptr, &info_ptr);
    113:   PetscFClose(PETSC_COMM_SELF,fp);
    114:   return(0);
    115: }

    117: static PetscErrorCode PetscDrawImageSave_PNG(const char filename[],unsigned char palette[][3],unsigned int w,unsigned int h,const unsigned char pixels[])
    118: { return PetscDrawImageSavePNG(filename,palette,w,h,pixels); }

    120: #endif/*!PETSC_HAVE_LIBPNG*/

    123: /*
    124:    Code to write images in GIF format
    125: */
    126: #if defined(PETSC_HAVE_GIFLIB)

    128: #include <gif_lib.h>

    130: #if !defined(GIFLIB_MAJOR) || GIFLIB_MAJOR < 5
    131: #define GifMakeMapObject          MakeMapObject
    132: #define GifFreeMapObject          FreeMapObject
    133: #define EGifOpenFileName(n,b,err) EGifOpenFileName(n,b)
    134: #define EGifOpenFileHandle(h,err) EGifOpenFileName(h)
    135: #define EGifCloseFile(f,err)      EGifCloseFile(f)
    136: #define DGifOpenFileName(n,err)   DGifOpenFileName(n)
    137: #define DGifOpenFileHandle(h,err) DGifOpenFileName(h)
    138: #define DGifCloseFile(f,err)      DGifCloseFile(f)
    139: #endif

    141: PETSC_EXTERN PetscErrorCode PetscDrawImageSaveGIF(const char filename[],unsigned char palette[][3],unsigned int w,unsigned int h,const unsigned char pixels[])
    142: {
    143:   int            Row, Error;
    144:   int            Width  = (int)w;
    145:   int            Height = (int)h;
    146:   int            ColorRes   = 8;
    147:   int            ColorCount = 256;
    148:   ColorMapObject *GifCMap = NULL;
    149:   GifFileType    *GifFile = NULL;
    150: # define         SETERRGIF(msg) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,msg", GIF file: %s",filename)
    151: # define         CHKERRGIF(msg) do {if (Error != GIF_OK) SETERRGIF(msg);} while (0)

    158:   GifCMap = GifMakeMapObject(ColorCount, (GifColorType*)palette); if (!GifCMap) SETERRGIF("Allocating colormap");
    159:   GifFile = EGifOpenFileName(filename, 0, NULL); if (!GifFile) SETERRGIF("Opening");
    160:   Error = EGifPutScreenDesc(GifFile, Width, Height, ColorRes, 0, GifCMap); CHKERRGIF("Writing screen descriptor");
    161:   Error = EGifPutImageDesc(GifFile, 0, 0, Width, Height, 0, NULL); CHKERRGIF("Writing image descriptor");
    162:   for (Row = 0; Row < Height; Row++) {
    163:     Error = EGifPutLine(GifFile, (GifPixelType*)pixels + Row*Width, Width); CHKERRGIF("Writing image pixels");
    164:   }
    165:   Error = EGifCloseFile(GifFile, NULL); CHKERRGIF("Closing");
    166:   GifFreeMapObject(GifCMap); GifCMap = NULL;

    168: # undef SETERRGIF
    169: # undef CHKERRGIF
    170:   return(0);
    171: }

    173: static PetscErrorCode PetscDrawImageSave_GIF(const char filename[],unsigned char palette[][3],unsigned int w,unsigned int h,const unsigned char pixels[])
    174: { return PetscDrawImageSaveGIF(filename,palette,w,h,pixels); }

    176: PETSC_EXTERN PetscErrorCode PetscDrawMovieSaveGIF(const char pattern[],PetscInt count,const char movie[])
    177: {
    178:   int            i,j,Row;
    179:   char           image[PETSC_MAX_PATH_LEN];
    180:   GifFileType    *GifMovie = NULL;
    181:   GifFileType    *GifImage = NULL;
    183: # define         SETERRGIF(msg,fn) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,msg" GIF file %s",fn)

    188:   if (count < 1) return(0);

    190:   for (i = 0; i < count; i++) {
    191:     PetscSNPrintf(image,sizeof(image),pattern,(int)i);
    192:     /* open and read image file */
    193:     if ((GifImage = DGifOpenFileName(image, NULL)) == NULL) SETERRGIF("Opening input",image);
    194:     if (DGifSlurp(GifImage) != GIF_OK) SETERRGIF("Reading input",image);
    195:     /* open movie file and write header */
    196:     if (i == 0) {
    197:       if ((GifMovie = EGifOpenFileName(movie, 0, NULL)) == NULL) SETERRGIF("Opening output",movie);
    198:       if (EGifPutScreenDesc(GifMovie,
    199:                             GifImage->SWidth,
    200:                             GifImage->SHeight,
    201:                             GifImage->SColorResolution,
    202:                             GifImage->SBackGroundColor,
    203:                             GifImage->SColorMap) != GIF_OK) SETERRGIF("Writing screen descriptor,",movie);
    204:     }
    205:     /* loop over all frames in image */
    206:     for (j = 0; j < GifImage->ImageCount; j++) {
    207:       SavedImage *sp = &GifImage->SavedImages[j];
    208:       GifImageDesc *GifFrame = &sp->ImageDesc;
    209:       ColorMapObject *FrameColorMap = GifFrame->ColorMap ? GifFrame->ColorMap : GifImage->SColorMap;
    210:       if (GifMovie->SColorMap && GifMovie->SColorMap->ColorCount == FrameColorMap->ColorCount &&
    211:           !memcmp(GifMovie->SColorMap->Colors,FrameColorMap->Colors,
    212:                   (size_t)FrameColorMap->ColorCount*sizeof(GifColorType)))
    213:         FrameColorMap = NULL;
    214:       /* add frame to movie */
    215:       if (EGifPutImageDesc(GifMovie,
    216:                            GifFrame->Left,
    217:                            GifFrame->Top,
    218:                            GifFrame->Width,
    219:                            GifFrame->Height,
    220:                            GifFrame->Interlace,
    221:                            FrameColorMap) != GIF_OK) SETERRGIF("Writing image descriptor,",movie);
    222:       for (Row = 0; Row < GifFrame->Height; Row++) {
    223:         if (EGifPutLine(GifMovie,
    224:                         sp->RasterBits + Row * GifFrame->Width,
    225:                         GifFrame->Width) != GIF_OK) SETERRGIF("Writing image pixels,",movie);
    226:       }
    227:     }
    228:     if (DGifCloseFile(GifImage, NULL) != GIF_OK) SETERRGIF("Closing input",image);
    229:   }
    230:   if (EGifCloseFile(GifMovie, NULL) != GIF_OK) SETERRGIF("Closing output",movie);

    232: # undef SETERRGIF
    233:   return(0);
    234: }

    236: #endif/*!PETSC_HAVE_GIFLIB*/

    238: /*
    239:    Code to write images in JPEG format
    240: */
    241: #if defined(PETSC_HAVE_LIBJPEG)

    243: #include <jpeglib.h>

    245: #if defined(PETSC_HAVE_SETJMP_H)
    246: #include <setjmp.h>
    247: static jmp_buf petsc_jpeg_jumpbuf;
    248: static void petsc_jpeg_error_longjmp (j_common_ptr cinfo) { (void)cinfo; longjmp(petsc_jpeg_jumpbuf,1); }
    249: #endif

    251: PETSC_EXTERN PetscErrorCode PetscDrawImageSaveJPG(const char filename[],unsigned char palette[][3],unsigned int w,unsigned int h,const unsigned char pixels[])
    252: {
    253:   unsigned char               *rgbpixels;
    254:   FILE                        *fp;
    255:   struct jpeg_compress_struct cinfo;
    256:   struct jpeg_error_mgr       jerr;
    257:   PetscErrorCode              ierr;

    263:   /* map pixels to RGB colors */
    264:   if (palette) {
    265:     int k,p,n = (int)(w*h);
    266:     const unsigned char *colordef;
    267:     PetscMalloc1(3*w*h,&rgbpixels);
    268:     for (k=p=0; k<n; k++) {
    269:       colordef = palette[pixels[k]];
    270:       rgbpixels[p++] = colordef[0];
    271:       rgbpixels[p++] = colordef[1];
    272:       rgbpixels[p++] = colordef[2];
    273:     }
    274:   } else { /* assume pixels are RGB colors */
    275:     rgbpixels = (unsigned char*)pixels;
    276:   }
    277:   PetscFOpen(PETSC_COMM_SELF,filename,"wb",&fp);

    279:   cinfo.err = jpeg_std_error(&jerr);
    280: #if defined(PETSC_HAVE_SETJMP_H)
    281:   jerr.error_exit = petsc_jpeg_error_longjmp;
    282:   if (setjmp(petsc_jpeg_jumpbuf)) {
    283:     char message[JMSG_LENGTH_MAX];
    284:     jerr.format_message((j_common_ptr)&cinfo,message);
    285:     jpeg_destroy_compress(&cinfo);
    286:     (void)PetscFClose(PETSC_COMM_SELF,fp);
    287:     SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error writing JPEG file %s\n%s",filename,message);
    288:   }
    289: #endif
    290:   jpeg_create_compress(&cinfo);
    291:   jpeg_stdio_dest(&cinfo,fp);
    292:   cinfo.image_width      = w;
    293:   cinfo.image_height     = h;
    294:   cinfo.input_components = 3;
    295:   cinfo.in_color_space   = JCS_RGB;
    296:   jpeg_set_defaults(&cinfo);
    297:   jpeg_start_compress(&cinfo,TRUE);
    298:   while (cinfo.next_scanline < cinfo.image_height) {
    299:     unsigned char *rowptr = rgbpixels + cinfo.next_scanline * 3*w;
    300:     (void)jpeg_write_scanlines(&cinfo,&rowptr,1);
    301:   }
    302:   jpeg_finish_compress(&cinfo);
    303:   jpeg_destroy_compress(&cinfo);

    305:   PetscFClose(PETSC_COMM_SELF,fp);
    306:   if (palette) {PetscFree(rgbpixels);}
    307:   return(0);
    308: }

    310: static PetscErrorCode PetscDrawImageSave_JPG(const char filename[],unsigned char palette[][3],unsigned int w,unsigned int h,const unsigned char pixels[])
    311: { return PetscDrawImageSaveJPG(filename,palette,w,h,pixels); }

    313: #endif/*!PETSC_HAVE_LIBJPEG*/

    315: static struct {
    316:   const char      *extension;
    317:   PetscErrorCode (*SaveImage)(const char[],unsigned char[][3],unsigned int,unsigned int,const unsigned char[]);
    318: } PetscDrawImageSaveTable[] = {
    319: #if defined(PETSC_HAVE_LIBPNG)
    320:   {".png", PetscDrawImageSave_PNG},
    321: #endif
    322: #if defined(PETSC_HAVE_GIFLIB)
    323:   {".gif", PetscDrawImageSave_GIF},
    324: #endif
    325: #if defined(PETSC_HAVE_LIBJPEG)
    326:   {".jpg", PetscDrawImageSave_JPG},
    327: #endif
    328:   {".ppm", PetscDrawImageSave_PPM}
    329: };

    331: PetscErrorCode PetscDrawImageCheckFormat(const char *ext[])
    332: {
    333:   size_t         k;
    334:   PetscBool      match = PETSC_FALSE;

    338:   /* if extension is empty, return default format to caller */
    340:   if (!*ext || !**ext) {
    341:     *ext = PetscDrawImageSaveTable[0].extension;
    342:     return(0);
    343:   }
    344:   /* check the extension matches a supported format */
    346:   for (k=0; k<sizeof(PetscDrawImageSaveTable)/sizeof(PetscDrawImageSaveTable[0]); k++) {
    347:     PetscStrcasecmp(*ext,PetscDrawImageSaveTable[k].extension,&match);
    348:     if (match && PetscDrawImageSaveTable[k].SaveImage) return(0);
    349:   }
    350:   SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Image extension %s not supported, use .ppm or see PetscDrawSetSave() for what ./configure option you may need",*ext);
    351: }

    353: PetscErrorCode PetscDrawImageSave(const char basename[],const char ext[],unsigned char palette[][3],unsigned int w,unsigned int h,const unsigned char pixels[])
    354: {
    355:   size_t         k;
    356:   PetscBool      match = PETSC_FALSE;
    357:   char           filename[PETSC_MAX_PATH_LEN];

    366:   PetscDrawImageCheckFormat(&ext);
    367:   PetscSNPrintf(filename,sizeof(filename),"%s%s",basename,ext);
    368:   for (k=0; k<sizeof(PetscDrawImageSaveTable)/sizeof(PetscDrawImageSaveTable[0]); k++) {
    369:     PetscStrcasecmp(ext,PetscDrawImageSaveTable[k].extension,&match);
    370:     if (match && PetscDrawImageSaveTable[k].SaveImage) {
    371:       PetscDrawImageSaveTable[k].SaveImage(filename,palette,w,h,pixels);
    372:       return(0);
    373:     }
    374:   }
    375:   SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Image extension %s not supported, use .ppm",ext);
    376: }

    378: PetscErrorCode PetscDrawMovieCheckFormat(const char *ext[])
    379: {
    382:   if (!*ext || !**ext) *ext = ".m4v";
    383:   return(0);
    384: }

    386: PetscErrorCode PetscDrawMovieSave(const char basename[],PetscInt count,const char imext[],PetscInt fps,const char mvext[])
    387: {
    388:   char           input[PETSC_MAX_PATH_LEN];
    389:   char           output[PETSC_MAX_PATH_LEN];
    390:   PetscBool      gifinput;

    397:   if (count < 1) return(0);

    399:   PetscStrcasecmp(imext,".gif",&gifinput);
    400:   PetscDrawMovieCheckFormat(&mvext);
    401:   PetscSNPrintf(input,sizeof(input),"%s/%s_%%d%s",basename,basename,imext);
    402:   PetscSNPrintf(output,sizeof(output),"%s%s",basename,mvext);

    404:   /* use GIFLIB to generate an intermediate GIF animation */
    405: #if defined(PETSC_HAVE_GIFLIB)
    406:   if (gifinput) {
    407:     char gifmovie[PETSC_MAX_PATH_LEN];
    408:     PetscSNPrintf(gifmovie,sizeof(gifmovie),"%s/%s_movie.gif",basename,basename);
    409:     PetscDrawMovieSaveGIF(input,count,gifmovie);
    410:     PetscStrcpy(input,gifmovie);
    411:   }
    412: #endif

    414:   /* use FFmpeg to generate a movie */
    415: #if defined(PETSC_HAVE_POPEN)
    416:   {
    417:     FILE *fd;
    418:     char options[64] = "-loglevel error -y", extraopts[32] = "", framerate[24] = "";
    419:     char command[sizeof(options)+sizeof(extraopts)+sizeof(framerate)+PETSC_MAX_PATH_LEN*2];
    420:     if (fps > 0) {PetscSNPrintf(framerate,sizeof(framerate),"-r %d",(int)fps);}
    421:     if (gifinput) {
    422:       PetscStrlcat(options," -f gif",sizeof(options));
    423:       PetscSNPrintf(extraopts,sizeof(extraopts)," -default_delay %d",(fps > 0) ? 100/(int)fps : 4);
    424:     } else {
    425:       PetscStrlcat(options," -f image2",sizeof(options));
    426:       if (fps > 0) {PetscSNPrintf(extraopts,sizeof(extraopts)," -framerate %d",(int)fps);}
    427:     }
    428:     if (extraopts[0]) {PetscStrlcat(options,extraopts,sizeof(options));}
    429:     PetscSNPrintf(command,sizeof(command),"ffmpeg %s -i \"%s\" %s \"%s\"",options,input,framerate,output);
    430:     PetscPOpen(PETSC_COMM_SELF,NULL,command,"r",&fd);
    431:     PetscPClose(PETSC_COMM_SELF,fd);
    432:   }
    433: #endif
    434:   return(0);
    435: }