ad81651f00
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@4950 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
164 lines
4.1 KiB
C++
164 lines
4.1 KiB
C++
/////////////////////////////////////////////////////////////////////////////
|
|
// Name: imagpnm.cpp
|
|
// Purpose: wxImage PNM handler
|
|
// Author: Sylvain Bougnoux
|
|
// RCS-ID: $Id$
|
|
// Copyright: (c) Sylvain Bougnoux
|
|
// 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
|
|
#endif
|
|
|
|
#ifndef WX_PRECOMP
|
|
# include "wx/setup.h"
|
|
#endif
|
|
|
|
#if wxUSE_PNM
|
|
|
|
#include "wx/image.h"
|
|
#include "wx/log.h"
|
|
#include "wx/intl.h"
|
|
#include "wx/txtstrm.h"
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// wxBMPHandler
|
|
//-----------------------------------------------------------------------------
|
|
|
|
IMPLEMENT_DYNAMIC_CLASS(wxPNMHandler,wxImageHandler)
|
|
|
|
#if wxUSE_STREAMS
|
|
|
|
void Skip_Comment(wxInputStream &stream)
|
|
{
|
|
wxTextInputStream text_stream(stream);
|
|
|
|
if (stream.Peek()==wxT('#'))
|
|
{
|
|
text_stream.ReadLine();
|
|
Skip_Comment(stream);
|
|
}
|
|
}
|
|
|
|
bool wxPNMHandler::LoadFile( wxImage *image, wxInputStream& stream, bool verbose, int WXUNUSED(index) )
|
|
{
|
|
wxUint32 width, height;
|
|
wxUint16 maxval;
|
|
char c(0);
|
|
|
|
image->Destroy();
|
|
|
|
/*
|
|
* Read the PNM header
|
|
*/
|
|
|
|
wxBufferedInputStream buf_stream(stream);
|
|
wxTextInputStream text_stream(buf_stream);
|
|
|
|
Skip_Comment(buf_stream);
|
|
if (buf_stream.GetC()==wxT('P')) c=buf_stream.GetC();
|
|
|
|
switch (c)
|
|
{
|
|
case wxT('2'):
|
|
if (verbose) wxLogError(_("Loading Grey Ascii PNM image is not yet implemented."));
|
|
return FALSE;
|
|
case wxT('5'):
|
|
if (verbose) wxLogError(_("Loading Grey Raw PNM image is not yet implemented."));
|
|
return FALSE;
|
|
case wxT('3'):
|
|
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);
|
|
text_stream >> maxval;
|
|
|
|
//cout << line << " " << width << " " << height << " " << maxval << endl;
|
|
image->Create( width, height );
|
|
unsigned char *ptr = image->GetData();
|
|
if (!ptr)
|
|
{
|
|
if (verbose)
|
|
wxLogError( _("PNM: Couldn't allocate memory.") );
|
|
return FALSE;
|
|
}
|
|
|
|
if (c=='3') // Ascii RBG
|
|
{
|
|
wxUint32 value, size=3*width*height;
|
|
for (wxUint32 i=0; i<size; ++i)
|
|
{
|
|
//this is very slow !!!
|
|
//I wonder how we can make any better ?
|
|
value=text_stream.Read32();
|
|
*ptr++=(unsigned char)value;
|
|
|
|
if (buf_stream.LastError()!=wxSTREAM_NOERROR)
|
|
{
|
|
if (verbose) wxLogError(_("PNM: File seems truncated."));
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
if (c=='6') // Raw RGB
|
|
buf_stream.Read( ptr, 3*width*height );
|
|
|
|
image->SetMask( FALSE );
|
|
|
|
return (buf_stream.LastError()==wxStream_NOERROR || buf_stream.LastError()==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
|
|
//<< "255" << endl;
|
|
text_stream << "P6\n" << image->GetWidth() << " " << image->GetHeight() << "\n255\n";
|
|
stream.Write(image->GetData(),3*image->GetWidth()*image->GetHeight());
|
|
|
|
return (stream.LastError()==wxStream_NOERROR);
|
|
}
|
|
|
|
bool wxPNMHandler::DoCanRead( wxInputStream& stream )
|
|
{
|
|
off_t pos = stream.TellI();
|
|
|
|
Skip_Comment(stream);
|
|
|
|
if ( stream.GetC() == 'P' )
|
|
{
|
|
switch (stream.GetC())
|
|
{
|
|
case '3':
|
|
case '6':
|
|
stream.SeekI(pos);
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
stream.SeekI(pos);
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
#endif // wxUSE_STREAMS
|
|
|
|
#endif // wxUSE_PNM
|