Updating corrupt file

[wxWidgets.git] / src / common / imagpnm.cpp

/////////////////////////////////////////////////////////////////////////////
// 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

#include "wx/image.h"
#include "wx/log.h"
#include "wx/txtstrm.h"

#ifdef __WXMSW__
#include <windows.h>
#endif

//-----------------------------------------------------------------------------
// wxBMPHandler
//-----------------------------------------------------------------------------

#if !USE_SHARED_LIBRARIES
IMPLEMENT_DYNAMIC_CLASS(wxPNMHandler,wxImageHandler)
#endif

#if wxUSE_STREAMS

//#include <stream.h> // for cout

void Skip_Comment(wxInputStream &stream)
{
  wxString line;
  wxTextInputStream text_stream(stream);

  if (stream.Peek()==_T('#')) 
    {
      text_stream >> line;
      Skip_Comment(stream);
    }
}

bool wxPNMHandler::LoadFile( wxImage *image, wxInputStream& stream, bool WXUNUSED(verbose) )
{
    wxUint32  width, height;
    wxUint16  maxval;
    wxString  line;
    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()==_T('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
    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)
    {
        wxLogError( _T("Cannot allocate RAM for RGB data in PNM file.") );
	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)
	      {
		wxLogError(_T("Loading PNM image : 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::CanRead( 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;
	}

    stream.SeekI(pos);
    return FALSE;
}


#endif // wxUSE_STREAMS