PfmWriter.C

Go to the documentation of this file.
00001 /*!@file Raster/PfmWriter.C Write pfm image files */
00002 
00003 // //////////////////////////////////////////////////////////////////// //
00004 // The iLab Neuromorphic Vision C++ Toolkit - Copyright (C) 2000-2005   //
00005 // by the University of Southern California (USC) and the iLab at USC.  //
00006 // See http://iLab.usc.edu for information about this project.          //
00007 // //////////////////////////////////////////////////////////////////// //
00008 // Major portions of the iLab Neuromorphic Vision Toolkit are protected //
00009 // under the U.S. patent ``Computation of Intrinsic Perceptual Saliency //
00010 // in Visual Environments, and Applications'' by Christof Koch and      //
00011 // Laurent Itti, California Institute of Technology, 2001 (patent       //
00012 // pending; application number 09/912,225 filed July 23, 2001; see      //
00013 // http://pair.uspto.gov/cgi-bin/final/home.pl for current status).     //
00014 // //////////////////////////////////////////////////////////////////// //
00015 // This file is part of the iLab Neuromorphic Vision C++ Toolkit.       //
00016 //                                                                      //
00017 // The iLab Neuromorphic Vision C++ Toolkit is free software; you can   //
00018 // redistribute it and/or modify it under the terms of the GNU General  //
00019 // Public License as published by the Free Software Foundation; either  //
00020 // version 2 of the License, or (at your option) any later version.     //
00021 //                                                                      //
00022 // The iLab Neuromorphic Vision C++ Toolkit is distributed in the hope  //
00023 // that it will be useful, but WITHOUT ANY WARRANTY; without even the   //
00024 // implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR      //
00025 // PURPOSE.  See the GNU General Public License for more details.       //
00026 //                                                                      //
00027 // You should have received a copy of the GNU General Public License    //
00028 // along with the iLab Neuromorphic Vision C++ Toolkit; if not, write   //
00029 // to the Free Software Foundation, Inc., 59 Temple Place, Suite 330,   //
00030 // Boston, MA 02111-1307 USA.                                           //
00031 // //////////////////////////////////////////////////////////////////// //
00032 //
00033 // Primary maintainer for this file: Rob Peters <rjpeters at usc dot edu>
00034 // $HeadURL: svn://isvn.usc.edu/software/invt/trunk/saliency/src/Raster/PfmWriter.C $
00035 // $Id: PfmWriter.C 6956 2006-08-08 18:20:23Z rjpeters $
00036 //
00037 
00038 #ifndef RASTER_PFMWRITER_C_DEFINED
00039 #define RASTER_PFMWRITER_C_DEFINED
00040 
00041 #include "Raster/PfmWriter.H"
00042 
00043 #include "Image/Image.H"
00044 #include "Image/Pixels.H"
00045 #include "Raster/GenericFrame.H"
00046 
00047 #include <fstream>
00048 
00049 using std::deque;
00050 using std::string;
00051 
00052 PfmWriter::PfmWriter() {}
00053 
00054 PfmWriter::~PfmWriter() {}
00055 
00056 string PfmWriter::writeFrame(const GenericFrame& image,
00057                              const string& fname)
00058 {
00059   PfmWriter::writeFloat(image.asFloat(), fname);
00060 
00061   return fname;
00062 }
00063 
00064 void PfmWriter::writeFloat(const Image<float>& image,
00065                            const string& fname,
00066                            const deque<string>& tagName,
00067                            const deque<string>& tagValue)
00068 {
00069   if (tagName.size() != tagValue.size())
00070     LFATAL("the number of tag names and tag values must be equal");
00071 
00072   std::ofstream ofs(fname.c_str(), std::ios::binary);
00073   if (!ofs.is_open())
00074     LFATAL("Couldn't open PFM file '%s' for writing.", fname.c_str());
00075 
00076   ofs << "PF\n" << image.getWidth() << ' ' << image.getHeight() << '\n';
00077 
00078   // write extra tags if supplied
00079   if (tagName.size() > 0)
00080     {
00081       deque<string>::const_iterator tagNameItr  = tagName.begin();
00082       deque<string>::const_iterator tagValueItr = tagValue.begin();
00083       while (tagNameItr != tagName.end())
00084         {
00085           ofs << '!' << *tagNameItr << '\n'
00086               << *tagValueItr << '\n';
00087           ++tagNameItr; ++tagValueItr;
00088         }
00089     }
00090 
00091   // 1.0 is our "max gray" value for the PFM format:
00092   ofs << "1.0\n";
00093 
00094   ofs.write(reinterpret_cast<const char*>(image.getArrayPtr()),
00095             image.getSize() * sizeof(float));
00096   if (ofs.fail())
00097     LFATAL("Output stream failure while writing '%s'.", fname.c_str());
00098 }
00099 
00100 void PfmWriter::writeFloat(const Image<float>& image,
00101                            const string& fname)
00102 {
00103   deque<string> nothing;
00104   PfmWriter::writeFloat(image, fname, nothing, nothing);
00105 }
00106 
00107 // ######################################################################
00108 /* So things look consistent in everyone's emacs... */
00109 /* Local Variables: */
00110 /* mode: c++ */
00111 /* indent-tabs-mode: nil */
00112 /* End: */
00113 
00114 #endif // RASTER_PFMWRITER_C_DEFINED
Generated on Sun May 8 08:41:16 2011 for iLab Neuromorphic Vision Toolkit by  doxygen 1.6.3