src/common/imagpnm.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        imagpnm.cpp
   3 // Purpose:     wxImage PNM handler
   4 // Author:      Sylvain Bougnoux
   5 // RCS-ID:      $Id$
   6 // Copyright:   (c) Sylvain Bougnoux
   7 // Licence:     wxWindows licence
   8 /////////////////////////////////////////////////////////////////////////////
   9
  10 // For compilers that support precompilation, includes "wx.h".
  11 #include "wx/wxprec.h"
  12
  13 #ifdef __BORLANDC__
  14 #pragma hdrstop
  15 #endif
  16
  17 #ifndef WX_PRECOMP
  18 #  include "wx/setup.h"
  19 #endif
  20
  21 #if wxUSE_IMAGE && wxUSE_PNM
  22
  23 #include "wx/imagpnm.h"
  24 #include "wx/log.h"
  25 #include "wx/intl.h"
  26 #include "wx/txtstrm.h"
  27
  28 //-----------------------------------------------------------------------------
  29 // wxBMPHandler
  30 //-----------------------------------------------------------------------------
  31
  32 IMPLEMENT_DYNAMIC_CLASS(wxPNMHandler,wxImageHandler)
  33
  34 #if wxUSE_STREAMS
  35
  36 void Skip_Comment(wxInputStream &stream)
  37 {
  38     wxTextInputStream text_stream(stream);
  39
  40     if (stream.Peek()==wxT('#'))
  41     {
  42         text_stream.ReadLine();
  43         Skip_Comment(stream);
  44     }
  45 }
  46
  47 bool wxPNMHandler::LoadFile( wxImage *image, wxInputStream& stream, bool verbose, int WXUNUSED(index) )
  48 {
  49     wxUint32  width, height;
  50     wxUint16  maxval;
  51     char      c(0);
  52
  53     image->Destroy();
  54
  55     /*
  56      * Read the PNM header
  57      */
  58
  59     wxBufferedInputStream buf_stream(stream);
  60     wxTextInputStream text_stream(buf_stream);
  61
  62     Skip_Comment(buf_stream);
  63     if (buf_stream.GetC()==wxT('P')) c=buf_stream.GetC();
  64
  65     switch (c)
  66     {
  67         case wxT('2'): // ASCII Grey
  68         case wxT('3'): // ASCII RGB
  69         case wxT('5'): // RAW Grey
  70         case wxT('6'): break;
  71         default:
  72             if (verbose) wxLogError(_("PNM: File format is not recognized."));
  73             return false;
  74     }
  75
  76     text_stream.ReadLine(); // for the \n
  77     Skip_Comment(buf_stream);
  78     text_stream >> width >> height ;
  79     Skip_Comment(buf_stream);
  80     text_stream >> maxval;
  81
  82     //cout << line << " " << width << " " << height << " " << maxval << endl;
  83     image->Create( width, height );
  84     unsigned char *ptr = image->GetData();
  85     if (!ptr)
  86     {
  87         if (verbose)
  88            wxLogError( _("PNM: Couldn't allocate memory.") );
  89         return false;
  90     }
  91
  92
  93     if (c=='2') // Ascii GREY
  94     {
  95         wxUint32 value, size=width*height;
  96         for (wxUint32 i=0; i<size; ++i)
  97         {
  98             value=text_stream.Read32();
  99             *ptr++=(unsigned char)value; // R
 100             *ptr++=(unsigned char)value; // G
 101             *ptr++=(unsigned char)value; // B
 102             if ( !buf_stream )
 103             {
 104                 if (verbose) wxLogError(_("PNM: File seems truncated."));
 105                 return false;
 106             }
 107         }
 108     }
 109     if (c=='3') // Ascii RBG
 110     {
 111         wxUint32 value, size=3*width*height;
 112         for (wxUint32 i=0; i<size; ++i)
 113           {
 114             //this is very slow !!!
 115             //I wonder how we can make any better ?
 116             value=text_stream.Read32();
 117             *ptr++=(unsigned char)value;
 118
 119             if ( !buf_stream )
 120               {
 121                 if (verbose) wxLogError(_("PNM: File seems truncated."));
 122                 return false;
 123               }
 124           }
 125     }
 126     if (c=='5') // Raw GREY
 127     {
 128         wxUint32 size=width*height;
 129         unsigned char value;
 130         for (wxUint32 i=0; i<size; ++i)
 131         {
 132             buf_stream.Read(&value,1);
 133             *ptr++=value; // R
 134             *ptr++=value; // G
 135             *ptr++=value; // B
 136             if ( !buf_stream )
 137             {
 138                 if (verbose) wxLogError(_("PNM: File seems truncated."));
 139                 return false;
 140             }
 141         }
 142     }
 143     if (c=='6') // Raw RGB
 144       buf_stream.Read( ptr, 3*width*height );
 145
 146     image->SetMask( false );
 147
 148     const wxStreamError err = buf_stream.GetLastError();
 149     return err == wxSTREAM_NO_ERROR || err == wxSTREAM_EOF;
 150 }
 151
 152 bool wxPNMHandler::SaveFile( wxImage *image, wxOutputStream& stream, bool WXUNUSED(verbose) )
 153 {
 154     wxTextOutputStream text_stream(stream);
 155
 156     //text_stream << "P6" << endl
 157     //<< image->GetWidth() << " " << image->GetHeight() << endl
 158     //<< "255" << endl;
 159     text_stream << wxT("P6\n") << image->GetWidth() << wxT(" ") << image->GetHeight() << wxT("\n255\n");
 160     stream.Write(image->GetData(),3*image->GetWidth()*image->GetHeight());
 161
 162     return stream.IsOk();
 163 }
 164
 165 bool wxPNMHandler::DoCanRead( wxInputStream& stream )
 166 {
 167     Skip_Comment(stream);
 168
 169     if ( stream.GetC() == 'P' )
 170     {
 171         switch ( stream.GetC() )
 172         {
 173             case '2': // ASCII Grey
 174             case '3': // ASCII RGB
 175             case '5': // RAW Grey
 176             case '6': // RAW RGB
 177                 return true;
 178         }
 179     }
 180
 181     return false;
 182 }
 183
 184
 185 #endif // wxUSE_STREAMS
 186
 187 #endif // wxUSE_PNM