]> git.saurik.com Git - wxWidgets.git/blob - src/common/imagpnm.cpp
cleanup - reformat
[wxWidgets.git] / 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