synfig-core/trunk/src/modules/mod_bmp/trgt_bmp.cpp

   1 /* === S Y N F I G ========================================================= */
   2 /*!     \file trgt_bmp.cpp
   3 **      \brief Bitmap Target
   4 **
   5 **      $Id$
   6 **
   7 **      \legal
   8 **      Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley
   9 **      Copyright (c) 2007 Chris Moore
  10 **
  11 **      This package is free software; you can redistribute it and/or
  12 **      modify it under the terms of the GNU General Public License as
  13 **      published by the Free Software Foundation; either version 2 of
  14 **      the License, or (at your option) any later version.
  15 **
  16 **      This package is distributed in the hope that it will be useful,
  17 **      but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 **      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19 **      General Public License for more details.
  20 **      \endlegal
  21 */
  22 /* ========================================================================= */
  23
  24 /* === H E A D E R S ======================================================= */
  25
  26 #define SYNFIG_NO_ANGLE
  27
  28 #ifdef USING_PCH
  29 #       include "pch.h"
  30 #else
  31 #ifdef HAVE_CONFIG_H
  32 #       include <config.h>
  33 #endif
  34
  35 #include "trgt_bmp.h"
  36 #include <synfig/general.h>
  37
  38 #include <cstdio>
  39 #include <algorithm>
  40 #include <functional>
  41 #endif
  42
  43 /* === U S I N G =========================================================== */
  44
  45 using namespace synfig;
  46 using namespace std;
  47 using namespace etl;
  48
  49 /* === I N F O ============================================================= */
  50
  51 SYNFIG_TARGET_INIT(bmp);
  52 SYNFIG_TARGET_SET_NAME(bmp,"bmp");
  53 SYNFIG_TARGET_SET_EXT(bmp,"bmp");
  54 SYNFIG_TARGET_SET_VERSION(bmp,"0.1");
  55 SYNFIG_TARGET_SET_CVS_ID(bmp,"$Id$");
  56
  57 /* === C L A S S E S & S T R U C T S ======================================= */
  58 namespace synfig {
  59
  60 struct BITMAPFILEHEADER
  61 {
  62         unsigned char   bfType[2];
  63         unsigned long   bfSize;
  64         unsigned short  bfReserved1;
  65         unsigned short  bfReserved2;
  66         unsigned long   bfOffsetBits;
  67 };
  68
  69 struct BITMAPINFOHEADER
  70 {
  71         unsigned long   biSize;
  72         long                    biWidth;
  73         long                    biHeight;
  74         unsigned short  biPlanes;
  75         unsigned short  biBitCount;
  76         unsigned long   biCompression;
  77         unsigned long   biSizeImage;
  78         long                    biXPelsPerMeter;
  79         long                    biYPelsPerMeter;
  80         unsigned long   biClrUsed;
  81         unsigned long   biClrImportant;
  82 };
  83
  84 }
  85 /* === M E T H O D S ======================================================= */
  86
  87 #ifdef WORDS_BIGENDIAN
  88 inline long little_endian(const long &x)
  89 {
  90         long ret;
  91         char *big_e=(char *)&ret;
  92         char *lit_e=(char *)&x;
  93         big_e[0]=lit_e[3];
  94         big_e[1]=lit_e[2];
  95         big_e[2]=lit_e[1];
  96         big_e[3]=lit_e[0];
  97         return ret;
  98 }
  99 inline short little_endian_short(const short &x)
 100 {
 101         short ret;
 102         char *big_e=(char *)&ret;
 103         char *lit_e=(char *)&x;
 104         big_e[0]=lit_e[1];
 105         big_e[1]=lit_e[0];
 106         return ret;
 107 }
 108 #else
 109 #define little_endian(x)        (x)
 110 #define little_endian_short(x)  (x)
 111 #endif
 112
 113 bmp::bmp(const char *Filename)
 114 {
 115         file=NULL;
 116         filename=Filename;
 117         multi_image=false;
 118         buffer=0;
 119         color_buffer=0;
 120         set_remove_alpha();
 121
 122 }
 123
 124 bmp::~bmp()
 125 {
 126         if(file)
 127                 fclose(file);
 128         file=NULL;
 129         delete [] buffer;
 130         delete [] color_buffer;
 131 }
 132
 133 bool
 134 bmp::set_rend_desc(RendDesc *given_desc)
 135 {
 136         pf=PF_BGR;
 137
 138     // Flip the image upside down,
 139         // because bitmaps are upside down.
 140     given_desc->set_flags(0);
 141         Point tl=given_desc->get_tl();
 142         Point br=given_desc->get_br();
 143         Point::value_type tmp;
 144         tmp=tl[1];
 145         tl[1]=br[1];
 146         br[1]=tmp;
 147         given_desc->set_tl(tl);
 148         given_desc->set_br(br);
 149
 150         desc=*given_desc;
 151         if(desc.get_frame_end()-desc.get_frame_start()>0)
 152         {
 153                 multi_image=true;
 154                 imagecount=desc.get_frame_start();
 155         }
 156         else
 157                 multi_image=false;
 158
 159     return true;
 160 }
 161
 162 void
 163 bmp::end_frame()
 164 {
 165         if(file)
 166                 fclose(file);
 167         delete [] color_buffer;
 168         color_buffer=0;
 169         file=NULL;
 170         imagecount++;
 171 }
 172
 173 bool
 174 bmp::start_frame(synfig::ProgressCallback *callback)
 175 {
 176         int w=desc.get_w(),h=desc.get_h();
 177
 178         rowspan=4*((w*(channels(pf)*8)+31)/32);
 179         if(multi_image)
 180         {
 181                 String newfilename(filename_sans_extension(filename) +
 182                                                    etl::strprintf(".%04d",imagecount) +
 183                                                    filename_extension(filename));
 184                 file=fopen(newfilename.c_str(),POPEN_BINARY_WRITE_TYPE);
 185                 if(callback)callback->task(newfilename+_(" (animated)"));
 186         }
 187         else
 188         {
 189                 file=fopen(filename.c_str(),POPEN_BINARY_WRITE_TYPE);
 190                 if(callback)callback->task(filename);
 191         }
 192
 193         if(!file)
 194         {
 195                 if(callback)callback->error(_("Unable to open file"));
 196                 else synfig::error(_("Unable to open file"));
 197                 return false;
 198         }
 199
 200         synfig::BITMAPFILEHEADER fileheader;
 201         synfig::BITMAPINFOHEADER infoheader;
 202
 203         fileheader.bfType[0]='B';
 204         fileheader.bfType[1]='M';
 205         fileheader.bfSize=little_endian(sizeof(synfig::BITMAPFILEHEADER)+sizeof(synfig::BITMAPINFOHEADER)+rowspan*h);
 206         fileheader.bfReserved1=0;
 207         fileheader.bfReserved2=0;
 208         fileheader.bfOffsetBits=little_endian(sizeof(synfig::BITMAPFILEHEADER)+sizeof(synfig::BITMAPINFOHEADER)-2);
 209
 210         infoheader.biSize=little_endian(40);
 211         infoheader.biWidth=little_endian(w);
 212         infoheader.biHeight=little_endian(h);
 213         infoheader.biPlanes=little_endian_short((short)1);
 214         infoheader.biBitCount=little_endian_short((short)(channels(pf)*8));
 215         infoheader.biCompression=little_endian(0);
 216         infoheader.biSizeImage=little_endian(0);
 217         infoheader.biXPelsPerMeter=little_endian((int)rend_desc().get_x_res());
 218         infoheader.biYPelsPerMeter=little_endian((int)rend_desc().get_y_res()); // pels per meter...?
 219         infoheader.biClrUsed=little_endian(0);
 220         infoheader.biClrImportant=little_endian(0);
 221
 222         fprintf(file,"BM");
 223
 224         if(!fwrite(&fileheader.bfSize,sizeof(synfig::BITMAPFILEHEADER)-4,1,file))
 225         {
 226                 if(callback)callback->error(_("Unable to write file header to file"));
 227                 else synfig::error(_("Unable to write file header to file"));
 228                 return false;
 229         }
 230
 231         if(!fwrite(&infoheader,sizeof(synfig::BITMAPINFOHEADER),1,file))
 232         {
 233                 if(callback)callback->error(_("Unable to write info header"));
 234                 else synfig::error(_("Unable to write info header"));
 235                 return false;
 236         }
 237
 238         delete [] buffer;
 239         buffer=new unsigned char[rowspan];
 240
 241         delete [] color_buffer;
 242         color_buffer=new Color[desc.get_w()];
 243
 244         return true;
 245 }
 246
 247 Color *
 248 bmp::start_scanline(int /*scanline*/)
 249 {
 250         return color_buffer;
 251 }
 252
 253 bool
 254 bmp::end_scanline()
 255 {
 256         if(!file)
 257                 return false;
 258
 259         convert_color_format(buffer, color_buffer, desc.get_w(), pf, gamma());
 260
 261         if(!fwrite(buffer,1,rowspan,file))
 262                 return false;
 263
 264         return true;
 265 }