ImageLoaders.cpp

Go to the documentation of this file.

#include "ImageLoaders.h"
#include "MercuryTextureManager.h"
#include "MercuryFiles.h"
#include "MercuryLog.h"

//for PNG Loading
#if defined(WIN32)
#  include <png.h>
#  if defined(_MSC_VER)
#    pragma comment(lib, "libpng.lib")
#  endif
#  pragma warning(disable: 4611) /* interaction between '_setjmp' and C++ object destruction is non-portable */
#else
#  include <png.h>
#endif

#define abort_( x ) { LOG.Warn(x); return NULL; }

void PNGRead( png_struct *png, png_byte *p, png_size_t size )
{
    MercuryFile* f = (MercuryFile*)png->io_ptr;

    int got = f->Read( p, size );

    if( got == -1 )
        png_error( png, "Error reading from file");
//  else if( got != size )
//      png_error( png, "Unexpected EOF" );
}


RawImageData* LoadPNG( MercuryFile * fp )
{
    png_structp png_ptr;
    png_infop info_ptr;
    int number_of_passes;
    png_bytep* row_pointers;
    png_byte color_type;
    png_byte bit_depth;
    RawImageData* image;
    unsigned char header[8];    // 8 is the maximum size that can be checked

    //open file and test for it being a png 
    if (!fp)
        abort_("[read_png_file] File %s could not be opened for reading");
    fp->Read(header, 8 );
    if (png_sig_cmp(header, 0, 8))
        abort_("[read_png_file] File %s is not recognized as a PNG file");


    //initialize stuff 
    png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);

    if (!png_ptr)
        abort_("[read_png_file] png_create_read_struct failed");

    info_ptr = png_create_info_struct(png_ptr);
    if (!info_ptr)
        abort_("[read_png_file] png_create_info_struct failed");

    if (setjmp(png_jmpbuf(png_ptr)))
        abort_("[read_png_file] Error during init_io");

    png_set_sig_bytes(png_ptr, 8);
    png_set_read_fn( png_ptr, fp, PNGRead );

    png_read_info(png_ptr, info_ptr);

    image = new RawImageData;

    image->attrs.m_width = info_ptr->width;
    image->attrs.m_height = info_ptr->height;
    color_type = info_ptr->color_type;
    bit_depth = info_ptr->bit_depth;

//  if ( color_type & PNG_COLOR_MASK_PALETTE )
//  {
//      SAFE_DELETE(image);
//      abort_("Cannot open paletted PNG files.");
//  }
    if (color_type == PNG_COLOR_TYPE_PALETTE)
        png_set_palette_to_rgb(png_ptr);

    number_of_passes = png_set_interlace_handling(png_ptr);
    png_read_update_info(png_ptr, info_ptr);

    // read file 
    if (setjmp(png_jmpbuf(png_ptr)))
    {
        SAFE_DELETE(image);
        abort_("[read_png_file] Error during read_image");
    }

    row_pointers = (png_bytep*) malloc(sizeof(png_bytep) * image->attrs.m_height);
    unsigned int y;
    for ( y=0; y < (unsigned)image->attrs.m_height; y++)
        row_pointers[y] = (png_byte*) malloc(info_ptr->rowbytes);

    png_read_image(png_ptr, row_pointers);

    png_read_end( png_ptr, info_ptr );
    png_destroy_read_struct( &png_ptr, &info_ptr, NULL );

    png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1);

    if (color_type & PNG_COLOR_MASK_ALPHA)
        image->attrs.m_ColorByteType = RGBA;    
    else
        image->attrs.m_ColorByteType = RGB; 

//  SAFE_DELETE(texture->m_data);
    
    image->data = new unsigned char[sizeof(unsigned char) * image->attrs.m_height * image->attrs.m_width *  4];

    switch (image->attrs.m_ColorByteType)
    {
    case RGBA:
        for ( y=0; y < (unsigned)image->attrs.m_height; ++y) {
            png_byte* row = row_pointers[y];
            for (int x = 0; x < image->attrs.m_width; ++x) {
                png_byte* ptr = &(row[x*4]);
                image->data[(x + y * image->attrs.m_width) * 4] = ptr[0];
                image->data[(x + y * image->attrs.m_width) * 4 + 1] = ptr[1];
                image->data[(x + y * image->attrs.m_width) * 4 + 2] = ptr[2];
                image->data[(x + y * image->attrs.m_width) * 4 + 3] = ptr[3];
            }   
        }
        break;
    case RGB:
        for ( y=0; y < (unsigned)image->attrs.m_height; y++) {
            png_byte* row = row_pointers[y];
            for (int x=0; x<image->attrs.m_width; x++) {
                png_byte* ptr = &(row[x * 3]);
                image->data[(x + y * image->attrs.m_width) * 3] = ptr[0];
                image->data[(x + y * image->attrs.m_width) * 3 + 1] = ptr[1];
                image->data[(x + y * image->attrs.m_width) * 3 + 2] = ptr[2];
            }   
        }
        break;
    default:
        LOG.Warn("Invalid color byte type for PNG.");
    }

    for ( y=0; y < (unsigned)image->attrs.m_height; y++)
        SAFE_DELETE(row_pointers[y]);
    SAFE_DELETE(row_pointers);

//  texture->CorrectSize();
//  texture->CreateCache(); 

    return image;
}

RawImageData* LoadBMP( MercuryFile * file )
{
    int offset;
    char* tmp = new char[sizeof(int)];
    int compression = 0;
    int length;
    int rawlength = 0;
    int bitsapix = 0;
    unsigned char b[3];
    unsigned int res_x, res_y;

    LOG.Info( "BMP Load Start" );
    if (file==NULL)
    {
        LOG.Warn("Could not open BMP (null file pointer)");
        SAFE_DELETE(tmp);
        return NULL;
    }

    //Get the type of file and test
    memset(tmp, 0, 4);
    file->Read(tmp, sizeof(char) * 2);
    MString type(tmp);

    if (type != "BM")
    {
        LOG.Warn("\"" + file->GetName() + "\" not a valid BMP");
        SAFE_DELETE(tmp);
        return NULL;
    }
    //Offset of bitmap data.
    file->Seek(10);
    file->Read(tmp, 4);
    memcpy(&offset, tmp, 4);
    TO_ENDIAN( offset );

    RawImageData* image = new RawImageData;

    //width & width
    file->Seek(18);
    file->Read(tmp, sizeof(int));
    memcpy(&image->attrs.m_width, tmp, sizeof(int));
    TO_ENDIAN( image->attrs.m_width );
    file->Read(tmp, sizeof(int));
    memcpy(&image->attrs.m_height, tmp, sizeof(int));
    TO_ENDIAN( image->attrs.m_height );

    //bits per pixel
    memset(tmp, 0, sizeof(int));
    file->Seek(28);
    file->Read(tmp, sizeof(int));
    memcpy(&bitsapix, tmp, sizeof(int));
    TO_ENDIAN( bitsapix );

    if (bitsapix != 24)
    {
        LOG.Warn("\"" + file->GetName() + "\" is not 24b/pix");
        SAFE_DELETE(tmp);
        SAFE_DELETE(image);
        return NULL;
    }

    //compression
    file->Seek(30);
    file->Read(tmp, sizeof(int));
    memcpy(&compression, tmp, sizeof(int));
    TO_ENDIAN(compression);

    if (compression != 0)
    {
        LOG.Warn("\"" + file->GetName() + "\" uses compression (not supported)");
        SAFE_DELETE(tmp);
        SAFE_DELETE(image);
        return NULL;
    }

    //pix/m X
    memset(tmp, 0, sizeof(int));
    file->Seek(38);
    file->Read(tmp, sizeof(int));
    memcpy(&res_x, tmp, sizeof(int));
    TO_ENDIAN(res_x);

    //pix/m Y
    memset(tmp, 0, sizeof(int));
    file->Seek(42);
    file->Read(tmp, sizeof(int));
    memcpy(&res_y, tmp, sizeof(int));
    TO_ENDIAN(res_y);

    if (res_x > 0)
    {
        image->attrs.m_dpi_x = M2DPI(res_x);
    }

    if (res_y > 0)
    {
        image->attrs.m_dpi_y = M2DPI(res_y);
    }

    //Get the file length
    length = file->Length();
    rawlength = (length) - (offset-1); //Remember to subtract 1 from the offset.

    //Allocate space
    SAFE_DELETE(image->data);
    image->data = new unsigned char[rawlength];
    memset(image->data, 0, rawlength);

    //Get raw data and convert BGR->RGB
    file->Seek(offset);
    for (unsigned int x = 0; !file->Eof(); x += 3)
    {
        memset(b, 0, sizeof(unsigned char) * 3);
        file->Read((char*)&b, sizeof(unsigned char) * 3);

        image->data[x] = b[2];
        image->data[x+1] = b[1];
        image->data[x+2] = b[0];
    }
    image->attrs.m_ColorByteType = RGB;
    SAFE_DELETE(tmp);
    LOG.Info( "BMP Load End" );
    return image;
}

RUN_STATEMENT_AT_BOOT( include_png, IMAGEREADERREGISTER.AddDecoder( LoadPNG, "‰PN" ); )
RUN_STATEMENT_AT_BOOT( include_bmp, IMAGEREADERREGISTER.AddDecoder( LoadBMP, "BM8" ); )


/* 
 * Copyright (c) 2004 Glenn Maynard
 * (c) 2006 Joshua Allen
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or
 * without modification, are permitted provided that the following
 * conditions are met:
 *  -   Redistributions of source code must retain the above
 *      copyright notice, this list of conditions and the following disclaimer.
 *  -   Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in
 *      the documentation and/or other materials provided with the distribution.
 *  -   Neither the name of the Mercury Engine nor the names of its
 *      contributors may be used to endorse or promote products derived from
 *      this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
 * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

---

Hosted by