synfig-core/tags/stable/src/modules/mod_openexr/trgt_openexr.cpp

   1 /* === S Y N F I G ========================================================= */
   2 /*!     \file trgt_openexr.cpp
   3 **      \brief exr_trgt Target Module
   4 **
   5 **      $Id$
   6 **
   7 **      \legal
   8 **      Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley
   9 **
  10 **      This package is free software; you can redistribute it and/or
  11 **      modify it under the terms of the GNU General Public License as
  12 **      published by the Free Software Foundation; either version 2 of
  13 **      the License, or (at your option) any later version.
  14 **
  15 **      This package is distributed in the hope that it will be useful,
  16 **      but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 **      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18 **      General Public License for more details.
  19 **      \endlegal
  20 **
  21 ** === N O T E S ===========================================================
  22 **
  23 ** ========================================================================= */
  24
  25 /* === H E A D E R S ======================================================= */
  26
  27 #define SYNFIG_TARGET
  28
  29 #ifdef USING_PCH
  30 #       include "pch.h"
  31 #else
  32 #ifdef HAVE_CONFIG_H
  33 #       include <config.h>
  34 #endif
  35
  36 #include "trgt_openexr.h"
  37 #include <ETL/stringf>
  38 #include <cstdio>
  39 #include <algorithm>
  40 #include <functional>
  41 #endif
  42
  43 /* === M A C R O S ========================================================= */
  44
  45 using namespace synfig;
  46 using namespace std;
  47 using namespace etl;
  48
  49 /* === G L O B A L S ======================================================= */
  50
  51 SYNFIG_TARGET_INIT(exr_trgt);
  52 SYNFIG_TARGET_SET_NAME(exr_trgt,"openexr");
  53 SYNFIG_TARGET_SET_EXT(exr_trgt,"exr");
  54 SYNFIG_TARGET_SET_VERSION(exr_trgt,"1.0.4");
  55 SYNFIG_TARGET_SET_CVS_ID(exr_trgt,"$Id$");
  56
  57 /* === M E T H O D S ======================================================= */
  58
  59 bool
  60 exr_trgt::ready()
  61 {
  62         return (bool)exr_file;
  63 }
  64
  65 exr_trgt::exr_trgt(const char *Filename):
  66         multi_image(false),
  67         imagecount(0),
  68         filename(Filename),
  69         exr_file(0)
  70 {
  71         buffer=0;
  72 #ifndef USE_HALF_TYPE
  73         buffer_color=0;
  74 #endif
  75
  76         // OpenEXR uses linear gamma
  77         gamma().set_gamma(1.0);
  78 }
  79
  80 exr_trgt::~exr_trgt()
  81 {
  82         if(exr_file)
  83                 delete exr_file;
  84
  85         if(buffer) delete [] buffer;
  86 #ifndef USE_HALF_TYPE
  87         if(buffer_color) delete [] buffer_color;
  88 #endif
  89 }
  90
  91 bool
  92 exr_trgt::set_rend_desc(RendDesc *given_desc)
  93 {
  94         //given_desc->set_pixel_format(PixelFormat((int)PF_RAW_COLOR));
  95         desc=*given_desc;
  96         imagecount=desc.get_frame_start();
  97         if(desc.get_frame_end()-desc.get_frame_start()>0)
  98                 multi_image=true;
  99         else
 100                 multi_image=false;
 101         return true;
 102 }
 103
 104 bool
 105 exr_trgt::start_frame(synfig::ProgressCallback *cb)
 106 {
 107         int w=desc.get_w(),h=desc.get_h();
 108
 109         String frame_name;
 110
 111         if(exr_file)
 112                 delete exr_file;
 113         if(multi_image)
 114         {
 115                 String
 116                         newfilename(filename),
 117                         ext(find(filename.begin(),filename.end(),'.'),filename.end());
 118                 newfilename.erase(find(newfilename.begin(),newfilename.end(),'.'),newfilename.end());
 119
 120                 newfilename+=etl::strprintf("%04d",imagecount)+ext;
 121                 frame_name=newfilename;
 122                 if(cb)cb->task(newfilename);
 123         }
 124         else
 125         {
 126                 frame_name=filename;
 127                 if(cb)cb->task(filename);
 128         }
 129         exr_file=new Imf::RgbaOutputFile(frame_name.c_str(),w,h,Imf::WRITE_RGBA,desc.get_pixel_aspect());
 130 #ifndef USE_HALF_TYPE
 131         if(buffer_color) delete [] buffer_color;
 132         buffer_color=new Color[w];
 133 #endif
 134         //if(buffer) delete [] buffer;
 135         //buffer=new Imf::Rgba[w];
 136         out_surface.set_wh(w,h);
 137
 138         return true;
 139 }
 140
 141 void
 142 exr_trgt::end_frame()
 143 {
 144         if(exr_file)
 145         {
 146                 exr_file->setFrameBuffer(out_surface[0],1,desc.get_w());
 147                 exr_file->writePixels(desc.get_h());
 148
 149                 delete exr_file;
 150         }
 151
 152         exr_file=0;
 153
 154         imagecount++;
 155 }
 156
 157 Color *
 158 exr_trgt::start_scanline(int i)
 159 {
 160         scanline=i;
 161 #ifndef USE_HALF_TYPE
 162         return reinterpret_cast<Color *>(buffer_color);
 163 #else
 164         return reinterpret_cast<Color *>(out_surface[scanline]);
 165 //      return reinterpret_cast<unsigned char *>(buffer);
 166 #endif
 167 }
 168
 169 bool
 170 exr_trgt::end_scanline()
 171 {
 172         if(!ready())
 173                 return false;
 174
 175 #ifndef USE_HALF_TYPE
 176         int i;
 177         for(i=0;i<desc.get_w();i++)
 178         {
 179 //              Imf::Rgba &rgba=buffer[i];
 180                 Imf::Rgba &rgba=out_surface[scanline][i];
 181                 Color &color=buffer_color[i];
 182                 rgba.r=color.get_r();
 183                 rgba.g=color.get_g();
 184                 rgba.b=color.get_b();
 185                 rgba.a=color.get_a();
 186         }
 187 #endif
 188
 189     //exr_file->setFrameBuffer(buffer,1,desc.get_w());
 190         //exr_file->writePixels(1);
 191
 192         return true;
 193 }