X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/9ab6ee851ebc3200fa7d40381a06d27b4fac61d1..abd474ea63667f727940a009cc3e0b23ba9f418f:/src/common/imagpnm.cpp?ds=sidebyside diff --git a/src/common/imagpnm.cpp b/src/common/imagpnm.cpp index 28bf08eba3..731220450b 100644 --- a/src/common/imagpnm.cpp +++ b/src/common/imagpnm.cpp @@ -1,5 +1,5 @@ ///////////////////////////////////////////////////////////////////////////// -// Name: imagpnm.cpp +// Name: src/common/imagpnm.cpp // Purpose: wxImage PNM handler // Author: Sylvain Bougnoux // RCS-ID: $Id$ @@ -7,57 +7,49 @@ // Licence: wxWindows licence ///////////////////////////////////////////////////////////////////////////// -/* - We don't put pragma implement in this file because it is already present in - src/common/image.cpp -*/ - // For compilers that support precompilation, includes "wx.h". #include "wx/wxprec.h" #ifdef __BORLANDC__ -#pragma hdrstop + #pragma hdrstop #endif +#if wxUSE_IMAGE && wxUSE_PNM + +#include "wx/imagpnm.h" + #ifndef WX_PRECOMP -# include "wx/setup.h" + #include "wx/intl.h" + #include "wx/log.h" #endif -#if wxUSE_PNM - -#include "wx/image.h" -#include "wx/log.h" #include "wx/txtstrm.h" //----------------------------------------------------------------------------- // wxBMPHandler //----------------------------------------------------------------------------- -#if !USE_SHARED_LIBRARIES IMPLEMENT_DYNAMIC_CLASS(wxPNMHandler,wxImageHandler) -#endif #if wxUSE_STREAMS void Skip_Comment(wxInputStream &stream) { - wxString line; - wxTextInputStream text_stream(stream); + wxTextInputStream text_stream(stream); - if (stream.Peek()==_T('#')) + if (stream.Peek()==wxT('#')) { - text_stream >> line; - Skip_Comment(stream); + text_stream.ReadLine(); + Skip_Comment(stream); } } -bool wxPNMHandler::LoadFile( wxImage *image, wxInputStream& stream, bool WXUNUSED(verbose) ) +bool wxPNMHandler::LoadFile( wxImage *image, wxInputStream& stream, bool verbose, int WXUNUSED(index) ) { wxUint32 width, height; wxUint16 maxval; - wxString line; char c(0); - + image->Destroy(); /* @@ -68,26 +60,26 @@ bool wxPNMHandler::LoadFile( wxImage *image, wxInputStream& stream, bool WXUNUSE wxTextInputStream text_stream(buf_stream); Skip_Comment(buf_stream); - if (buf_stream.GetC()==_T('P')) c=buf_stream.GetC(); + if (buf_stream.GetC()==wxT('P')) c=buf_stream.GetC(); switch (c) - { - case _T('2'): - wxLogError(_T("Loading Grey Ascii PNM image is not yet implemented.")); - return FALSE; - case _T('5'): - wxLogError(_T("Loading Grey Raw PNM image is not yet implemented.")); - return FALSE; - case _T('3'): case _T('6'): break; - default : - wxLogError(_T("Loading PNM image : file not recognized.")); - return FALSE; - } - - text_stream >> line; // for the \n + { + case wxT('2'): // ASCII Grey + case wxT('3'): // ASCII RGB + case wxT('5'): // RAW Grey + case wxT('6'): break; + default: + if (verbose) + { + wxLogError(_("PNM: File format is not recognized.")); + } + return false; + } + + text_stream.ReadLine(); // for the \n Skip_Comment(buf_stream); text_stream >> width >> height ; - Skip_Comment(buf_stream); + Skip_Comment(buf_stream); text_stream >> maxval; //cout << line << " " << width << " " << height << " " << maxval << endl; @@ -95,67 +87,130 @@ bool wxPNMHandler::LoadFile( wxImage *image, wxInputStream& stream, bool WXUNUSE unsigned char *ptr = image->GetData(); if (!ptr) { - wxLogError( _T("Cannot allocate RAM for RGB data in PNM file.") ); - return FALSE; + if (verbose) + { + wxLogError( _("PNM: Couldn't allocate memory.") ); + } + return false; + } + + + if (c=='2') // Ascii GREY + { + wxUint32 value, size=width*height; + for (wxUint32 i=0; iSetMask( FALSE ); - - return (buf_stream.LastError()==wxStream_NOERROR || buf_stream.LastError()==wxStream_EOF); + if ( c=='6' ) // Raw RGB + { + buf_stream.Read(ptr, 3*width*height); + if ( maxval != 255 ) + { + for ( unsigned i = 0; i < 3*width*height; i++ ) + ptr[i] = (255 * ptr[i])/maxval; + } + } + + image->SetMask( false ); + + const wxStreamError err = buf_stream.GetLastError(); + return err == wxSTREAM_NO_ERROR || err == wxSTREAM_EOF; } bool wxPNMHandler::SaveFile( wxImage *image, wxOutputStream& stream, bool WXUNUSED(verbose) ) { wxTextOutputStream text_stream(stream); - - //text_stream << "P6" << endl - //<< image->GetWidth() << " " << image->GetHeight() << endl + + //text_stream << "P6" << endl + //<< image->GetWidth() << " " << image->GetHeight() << endl //<< "255" << endl; - text_stream << "P6\n" << image->GetWidth() << " " << image->GetHeight() << "\n255\n"; + text_stream << wxT("P6\n") << image->GetWidth() << wxT(" ") << image->GetHeight() << wxT("\n255\n"); stream.Write(image->GetData(),3*image->GetWidth()*image->GetHeight()); - return (stream.LastError()==wxStream_NOERROR); + return stream.IsOk(); } -bool wxPNMHandler::CanRead( wxInputStream& stream ) +bool wxPNMHandler::DoCanRead( wxInputStream& stream ) { - off_t pos=stream.TellI(); - Skip_Comment(stream); - if (stream.GetC()==_T('P')) - switch (stream.GetC()) - { - case _T('3'): case _T('6'): - stream.SeekI(pos); - return TRUE; - } + // it's ok to modify the stream position here + if ( stream.GetC() == 'P' ) + { + switch ( stream.GetC() ) + { + case '2': // ASCII Grey + case '3': // ASCII RGB + case '5': // RAW Grey + case '6': // RAW RGB + return true; + } + } - stream.SeekI(pos); - return FALSE; + return false; } #endif // wxUSE_STREAMS -#endif // wxUSE_PNM +#endif // wxUSE_IMAGE && wxUSE_PNM