src/common/imagpnm.cpp

/////////////////////////////////////////////////////////////////////////////
// Name:        src/common/imagpnm.cpp
// Purpose:     wxImage PNM handler
// Author:      Sylvain Bougnoux
// RCS-ID:      $Id: imagpnm.cpp 46311 2007-06-03 22:14:32Z VZ $
// Copyright:   (c) Sylvain Bougnoux
// Licence:     wxWindows licence
/////////////////////////////////////////////////////////////////////////////

// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"

#ifdef __BORLANDC__
    #pragma hdrstop
#endif

#if wxUSE_IMAGE && wxUSE_PNM

#include "wx/imagpnm.h"

#ifndef WX_PRECOMP
    #include "wx/intl.h"
    #include "wx/log.h"
#endif

#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'): // 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);
    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=='2') // Ascii GREY
    {
        wxUint32 value, size=width*height;
        for (wxUint32 i=0; i<size; ++i)
        {
            value=text_stream.Read32();
            if ( maxval != 255 )
                value = (255 * value)/maxval;
            *ptr++=(unsigned char)value; // R
            *ptr++=(unsigned char)value; // G
            *ptr++=(unsigned char)value; // B
            if ( !buf_stream )
            {
                if (verbose) wxLogError(_("PNM: File seems truncated."));
                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();
            if ( maxval != 255 )
                value = (255 * value)/maxval;
            *ptr++=(unsigned char)value;

            if ( !buf_stream )
              {
                if (verbose) wxLogError(_("PNM: File seems truncated."));
                return false;
              }
          }
    }
    if (c=='5') // Raw GREY
    {
        wxUint32 size=width*height;
        unsigned char value;
        for (wxUint32 i=0; i<size; ++i)
        {
            buf_stream.Read(&value,1);
            if ( maxval != 255 )
                value = (255 * value)/maxval;
            *ptr++=value; // R
            *ptr++=value; // G
            *ptr++=value; // B
            if ( !buf_stream )
            {
                if (verbose) wxLogError(_("PNM: File seems truncated."));
                return false;
            }
        }
    }

    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
    //<< "255" << endl;
    text_stream << wxT("P6\n") << image->GetWidth() << wxT(" ") << image->GetHeight() << wxT("\n255\n");
    stream.Write(image->GetData(),3*image->GetWidth()*image->GetHeight());

    return stream.IsOk();
}

bool wxPNMHandler::DoCanRead( wxInputStream& stream )
{
    Skip_Comment(stream);

    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;
        }
    }

    return false;
}


#endif // wxUSE_STREAMS

#endif // wxUSE_IMAGE && wxUSE_PNM
1	william	31	/////////////////////////////////////////////////////////////////////////////
2			// Name: src/common/imagpnm.cpp
3			// Purpose: wxImage PNM handler
4			// Author: Sylvain Bougnoux
5			// RCS-ID: $Id: imagpnm.cpp 46311 2007-06-03 22:14:32Z VZ $
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			#if wxUSE_IMAGE && wxUSE_PNM
18
19			#include "wx/imagpnm.h"
20
21			#ifndef WX_PRECOMP
22			#include "wx/intl.h"
23			#include "wx/log.h"
24			#endif
25
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			if ( maxval != 255 )
100			value = (255 * value)/maxval;
101			*ptr++=(unsigned char)value; // R
102			*ptr++=(unsigned char)value; // G
103			*ptr++=(unsigned char)value; // B
104			if ( !buf_stream )
105			{
106			if (verbose) wxLogError(_("PNM: File seems truncated."));
107			return false;
108			}
109			}
110			}
111			if (c=='3') // Ascii RBG
112			{
113			wxUint32 value, size=3widthheight;
114			for (wxUint32 i=0; i<size; ++i)
115			{
116			//this is very slow !!!
117			//I wonder how we can make any better ?
118			value=text_stream.Read32();
119			if ( maxval != 255 )
120			value = (255 * value)/maxval;
121			*ptr++=(unsigned char)value;
122
123			if ( !buf_stream )
124			{
125			if (verbose) wxLogError(_("PNM: File seems truncated."));
126			return false;
127			}
128			}
129			}
130			if (c=='5') // Raw GREY
131			{
132			wxUint32 size=width*height;
133			unsigned char value;
134			for (wxUint32 i=0; i<size; ++i)
135			{
136			buf_stream.Read(&value,1);
137			if ( maxval != 255 )
138			value = (255 * value)/maxval;
139			*ptr++=value; // R
140			*ptr++=value; // G
141			*ptr++=value; // B
142			if ( !buf_stream )
143			{
144			if (verbose) wxLogError(_("PNM: File seems truncated."));
145			return false;
146			}
147			}
148			}
149
150			if ( c=='6' ) // Raw RGB
151			{
152			buf_stream.Read(ptr, 3widthheight);
153			if ( maxval != 255 )
154			{
155			for ( unsigned i = 0; i < 3widthheight; i++ )
156			ptr[i] = (255 * ptr[i])/maxval;
157			}
158			}
159
160			image->SetMask( false );
161
162			const wxStreamError err = buf_stream.GetLastError();
163			return err == wxSTREAM_NO_ERROR \|\| err == wxSTREAM_EOF;
164			}
165
166			bool wxPNMHandler::SaveFile( wxImage *image, wxOutputStream& stream, bool WXUNUSED(verbose) )
167			{
168			wxTextOutputStream text_stream(stream);
169
170			//text_stream << "P6" << endl
171			//<< image->GetWidth() << " " << image->GetHeight() << endl
172			//<< "255" << endl;
173			text_stream << wxT("P6\n") << image->GetWidth() << wxT(" ") << image->GetHeight() << wxT("\n255\n");
174			stream.Write(image->GetData(),3image->GetWidth()image->GetHeight());
175
176			return stream.IsOk();
177			}
178
179			bool wxPNMHandler::DoCanRead( wxInputStream& stream )
180			{
181			Skip_Comment(stream);
182
183			if ( stream.GetC() == 'P' )
184			{
185			switch ( stream.GetC() )
186			{
187			case '2': // ASCII Grey
188			case '3': // ASCII RGB
189			case '5': // RAW Grey
190			case '6': // RAW RGB
191			return true;
192			}
193			}
194
195			return false;
196			}
197
198
199			#endif // wxUSE_STREAMS
200
201			#endif // wxUSE_IMAGE && wxUSE_PNM