home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / wxos2233.zip / wxOS2-2_3_3.zip / wxWindows-2.3.3 / src / common / imagpnm.cpp < prev    next >
C/C++ Source or Header  |  2002-06-04  |  4KB  |  159 lines
  1. /////////////////////////////////////////////////////////////////////////////
  2. // Name:        imagpnm.cpp
  3. // Purpose:     wxImage PNM handler
  4. // Author:      Sylvain Bougnoux
  5. // RCS-ID:      $Id: imagpnm.cpp,v 1.19 2002/05/22 23:14:47 VZ Exp $
  6. // Copyright:   (c) Sylvain Bougnoux
  7. // Licence:     wxWindows licence
  8. /////////////////////////////////////////////////////////////////////////////
  9.  
  10. #ifdef __GNUG__
  11. #pragma implementation "imagpnm.h"
  12. #endif
  13.  
  14. // For compilers that support precompilation, includes "wx.h".
  15. #include "wx/wxprec.h"
  16.  
  17. #ifdef __BORLANDC__
  18. #pragma hdrstop
  19. #endif
  20.  
  21. #ifndef WX_PRECOMP
  22. #  include "wx/setup.h"
  23. #endif
  24.  
  25. #if wxUSE_IMAGE && wxUSE_PNM
  26.  
  27. #include "wx/imagpnm.h"
  28. #include "wx/log.h"
  29. #include "wx/intl.h"
  30. #include "wx/txtstrm.h"
  31.  
  32. //-----------------------------------------------------------------------------
  33. // wxBMPHandler
  34. //-----------------------------------------------------------------------------
  35.  
  36. IMPLEMENT_DYNAMIC_CLASS(wxPNMHandler,wxImageHandler)
  37.  
  38. #if wxUSE_STREAMS
  39.  
  40. void Skip_Comment(wxInputStream &stream)
  41. {
  42.   wxTextInputStream text_stream(stream);
  43.  
  44.   if (stream.Peek()==wxT('#'))
  45.     {
  46.       text_stream.ReadLine();
  47.       Skip_Comment(stream);
  48.     }
  49. }
  50.  
  51. bool wxPNMHandler::LoadFile( wxImage *image, wxInputStream& stream, bool verbose, int WXUNUSED(index) )
  52. {
  53.     wxUint32  width, height;
  54.     wxUint16  maxval;
  55.     char      c(0);
  56.  
  57.     image->Destroy();
  58.  
  59.     /*
  60.      * Read the PNM header
  61.      */
  62.  
  63.     wxBufferedInputStream buf_stream(stream);
  64.     wxTextInputStream text_stream(buf_stream);
  65.  
  66.     Skip_Comment(buf_stream);
  67.     if (buf_stream.GetC()==wxT('P')) c=buf_stream.GetC();
  68.  
  69.     switch (c)
  70.     {
  71.         case wxT('2'):
  72.             if (verbose) wxLogError(_("Loading Grey Ascii PNM image is not yet implemented."));
  73.             return FALSE;
  74.         case wxT('5'):
  75.             if (verbose) wxLogError(_("Loading Grey Raw PNM image is not yet implemented."));
  76.             return FALSE;
  77.         case wxT('3'): 
  78.     case wxT('6'): break;
  79.         default:
  80.             if (verbose) wxLogError(_("PNM: File format is not recognized."));
  81.             return FALSE;
  82.     }
  83.  
  84.     text_stream.ReadLine(); // for the \n
  85.     Skip_Comment(buf_stream);
  86.     text_stream >> width >> height ;
  87.     Skip_Comment(buf_stream);
  88.     text_stream >> maxval;
  89.  
  90.     //cout << line << " " << width << " " << height << " " << maxval << endl;
  91.     image->Create( width, height );
  92.     unsigned char *ptr = image->GetData();
  93.     if (!ptr)
  94.     {
  95.         if (verbose)
  96.            wxLogError( _("PNM: Couldn't allocate memory.") );
  97.         return FALSE;
  98.     }
  99.  
  100.    if (c=='3') // Ascii RBG
  101.       {
  102.         wxUint32 value, size=3*width*height;
  103.         for (wxUint32 i=0; i<size; ++i)
  104.           {
  105.             //this is very slow !!!
  106.             //I wonder how we can make any better ?
  107.             value=text_stream.Read32();
  108.             *ptr++=(unsigned char)value;
  109.  
  110.             if (buf_stream.LastError()!=wxSTREAM_NOERROR)
  111.               {
  112.                 if (verbose) wxLogError(_("PNM: File seems truncated."));
  113.                 return FALSE;
  114.               }
  115.           }
  116.       }
  117.     if (c=='6') // Raw RGB
  118.       buf_stream.Read( ptr, 3*width*height );
  119.  
  120.     image->SetMask( FALSE );
  121.  
  122.     return (buf_stream.LastError()==wxStream_NOERROR || buf_stream.LastError()==wxStream_EOF);
  123. }
  124.  
  125. bool wxPNMHandler::SaveFile( wxImage *image, wxOutputStream& stream, bool WXUNUSED(verbose) )
  126. {
  127.     wxTextOutputStream text_stream(stream);
  128.  
  129.     //text_stream << "P6" << endl
  130.     //<< image->GetWidth() << " " << image->GetHeight() << endl
  131.     //<< "255" << endl;
  132.     text_stream << "P6\n" << image->GetWidth() << " " << image->GetHeight() << "\n255\n";
  133.     stream.Write(image->GetData(),3*image->GetWidth()*image->GetHeight());
  134.  
  135.     return (stream.LastError()==wxStream_NOERROR);
  136. }
  137.  
  138. bool wxPNMHandler::DoCanRead( wxInputStream& stream )
  139. {
  140.     Skip_Comment(stream);
  141.  
  142.     if ( stream.GetC() == 'P' )
  143.     {
  144.         switch (stream.GetC())
  145.         {
  146.             case '3':
  147.             case '6':
  148.                 return TRUE;
  149.         }
  150.     }
  151.  
  152.     return FALSE;
  153. }
  154.  
  155.  
  156. #endif // wxUSE_STREAMS
  157.  
  158. #endif // wxUSE_PNM
  159.