synfig-core/trunk/src/modules/mod_png/trgt_png.cpp

   1 /* === S Y N F I G ========================================================= */
   2 /*!     \file trgt_png.cpp
   3 **      \brief png_trgt Target Module
   4 **
   5 **      \legal
   6 **      Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley
   7 **
   8 **      This package is free software; you can redistribute it and/or
   9 **      modify it under the terms of the GNU General Public License as
  10 **      published by the Free Software Foundation; either version 2 of
  11 **      the License, or (at your option) any later version.
  12 **
  13 **      This package 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 GNU
  16 **      General Public License for more details.
  17 **      \endlegal
  18 **
  19 ** === N O T E S ===========================================================
  20 **
  21 ** ========================================================================= */
  22
  23 /* === H E A D E R S ======================================================= */
  24
  25 #define SYNFIG_TARGET
  26
  27 #ifdef USING_PCH
  28 #       include "pch.h"
  29 #else
  30 #ifdef HAVE_CONFIG_H
  31 #       include <config.h>
  32 #endif
  33
  34 #include "trgt_png.h"
  35 #include <png.h>
  36 #include <ETL/stringf>
  37 #include <cstdio>
  38 #include <algorithm>
  39 #include <functional>
  40 #include <ETL/misc>
  41
  42 #endif
  43
  44 /* === M A C R O S ========================================================= */
  45
  46 using namespace synfig;
  47 using namespace std;
  48 using namespace etl;
  49
  50 /* === G L O B A L S ======================================================= */
  51
  52 SYNFIG_TARGET_INIT(png_trgt);
  53 SYNFIG_TARGET_SET_NAME(png_trgt,"png");
  54 SYNFIG_TARGET_SET_EXT(png_trgt,"png");
  55 SYNFIG_TARGET_SET_VERSION(png_trgt,"0.1");
  56 SYNFIG_TARGET_SET_CVS_ID(png_trgt,"$Id$");
  57
  58 /* === M E T H O D S ======================================================= */
  59
  60 void
  61 png_trgt::png_out_error(png_struct *png_data,const char *msg)
  62 {
  63         png_trgt *me=(png_trgt*)png_data->error_ptr;
  64         synfig::error(strprintf("png_trgt: error: %s",msg));
  65         me->ready=false;
  66 }
  67
  68 void
  69 png_trgt::png_out_warning(png_struct *png_data,const char *msg)
  70 {
  71         png_trgt *me=(png_trgt*)png_data->error_ptr;
  72         synfig::warning(strprintf("png_trgt: warning: %s",msg));
  73         me->ready=false;
  74 }
  75
  76
  77 //Target *png_trgt::New(const char *filename){  return new png_trgt(filename);}
  78
  79 png_trgt::png_trgt(const char *Filename)
  80 {
  81         file=NULL;
  82         filename=Filename;
  83         buffer=NULL;
  84         ready=false;
  85         color_buffer=0;
  86 }
  87
  88 png_trgt::~png_trgt()
  89 {
  90         if(file)
  91                 fclose(file);
  92         file=NULL;
  93         delete [] buffer;
  94         delete [] color_buffer;
  95 }
  96
  97 bool
  98 png_trgt::set_rend_desc(RendDesc *given_desc)
  99 {
 100         //given_desc->set_pixel_format(PixelFormat((int)PF_RGB|(int)PF_A));
 101         desc=*given_desc;
 102         imagecount=desc.get_frame_start();
 103         if(desc.get_frame_end()-desc.get_frame_start()>0)
 104                 multi_image=true;
 105         else
 106                 multi_image=false;
 107         return true;
 108 }
 109
 110 void
 111 png_trgt::end_frame()
 112 {
 113         if(ready && file)
 114         {
 115                 png_write_end(png_ptr,info_ptr);
 116                 png_destroy_write_struct(&png_ptr, &info_ptr);
 117         }
 118
 119         if(file && file!=stdout)
 120                 fclose(file);
 121         file=NULL;
 122         imagecount++;
 123         ready=false;
 124 }
 125
 126 bool
 127 png_trgt::start_frame(synfig::ProgressCallback *callback)
 128 {
 129         int w=desc.get_w(),h=desc.get_h();
 130
 131         if(file && file!=stdout)
 132                 fclose(file);
 133         if(filename=="-")
 134         {
 135                 if(callback)callback->task(strprintf("(stdout) %d",imagecount).c_str());
 136                 file=stdout;
 137         }
 138         else if(multi_image)
 139         {
 140                 String
 141                         newfilename(filename),
 142                         ext(find(filename.begin(),filename.end(),'.'),filename.end());
 143                 newfilename.erase(find(newfilename.begin(),newfilename.end(),'.'),newfilename.end());
 144
 145                 newfilename+=etl::strprintf("%04d",imagecount)+ext;
 146                 file=fopen(newfilename.c_str(),"wb");
 147                 if(callback)callback->task(newfilename);
 148         }
 149         else
 150         {
 151                 file=fopen(filename.c_str(),"wb");
 152                 if(callback)callback->task(filename);
 153         }
 154
 155         if(!file)
 156                 return false;
 157
 158         delete [] buffer;
 159         buffer=new unsigned char[4*w];
 160
 161         delete [] color_buffer;
 162         color_buffer=new Color[w];
 163
 164         png_ptr=png_create_write_struct(PNG_LIBPNG_VER_STRING, (png_voidp)this,png_out_error, png_out_warning);
 165         if (!png_ptr)
 166         {
 167                 synfig::error("Unable to setup PNG struct");
 168                 fclose(file);
 169                 return false;
 170         }
 171
 172         info_ptr= png_create_info_struct(png_ptr);
 173         if (!info_ptr)
 174         {
 175                 synfig::error("Unable to setup PNG info struct");
 176                 fclose(file);
 177                 png_destroy_write_struct(&png_ptr,(png_infopp)NULL);
 178                 return false;
 179         }
 180
 181         if (setjmp(png_jmpbuf(png_ptr)))
 182         {
 183                 synfig::error("Unable to setup longjump");
 184                 png_destroy_write_struct(&png_ptr, &info_ptr);
 185                 fclose(file);
 186                 return false;
 187         }
 188         png_init_io(png_ptr,file);
 189         png_set_filter(png_ptr,0,PNG_FILTER_NONE);
 190
 191         setjmp(png_jmpbuf(png_ptr));
 192         png_set_IHDR(png_ptr,info_ptr,w,h,8,PNG_COLOR_TYPE_RGBA,PNG_INTERLACE_NONE,PNG_COMPRESSION_TYPE_DEFAULT,PNG_FILTER_TYPE_DEFAULT);
 193
 194         // Write the gamma
 195         //png_set_gAMA(png_ptr, info_ptr,1.0/gamma().get_gamma());
 196         png_set_gAMA(png_ptr, info_ptr,gamma().get_gamma());
 197
 198         // Write the physical size
 199         png_set_pHYs(png_ptr,info_ptr,round_to_int(desc.get_x_res()),round_to_int(desc.get_y_res()),PNG_RESOLUTION_METER);
 200
 201         // Output any text info along with the file
 202         png_text comments[]=
 203         {
 204                 { PNG_TEXT_COMPRESSION_NONE, "Title", const_cast<char *>(get_canvas()->get_name().c_str()) },
 205                 { PNG_TEXT_COMPRESSION_NONE, "Description", const_cast<char *>(get_canvas()->get_description().c_str()) },
 206 //              { PNG_TEXT_COMPRESSION_NONE, "Copyright", "(c) 2004 Voria Studios, LLC" },
 207                 { PNG_TEXT_COMPRESSION_NONE, "Software", "SYNFIG" },
 208         };
 209         png_set_text(png_ptr,info_ptr,comments,sizeof(comments)/sizeof(png_text));
 210
 211         png_write_info_before_PLTE(png_ptr, info_ptr);
 212         png_write_info(png_ptr, info_ptr);
 213         ready=true;
 214         return true;
 215 }
 216
 217 Color *
 218 png_trgt::start_scanline(int scanline)
 219 {
 220         return color_buffer;
 221 }
 222
 223 bool
 224 png_trgt::end_scanline()
 225 {
 226         if(!file || !ready)
 227                 return false;
 228
 229         convert_color_format(buffer, color_buffer, desc.get_w(), PF_RGB|PF_A, gamma());
 230
 231         setjmp(png_jmpbuf(png_ptr));
 232         png_write_row(png_ptr,buffer);
 233
 234         return true;
 235 }