png_bitmap.cpp
上传用户:riyaled888
上传日期:2009-03-27
资源大小:7338k
文件大小:4k
源码类别:

多媒体

开发平台:

MultiPlatform

  1. /*****************************************************************************
  2.  * png_bitmap.cpp
  3.  *****************************************************************************
  4.  * Copyright (C) 2003 VideoLAN
  5.  * $Id: png_bitmap.cpp 6990 2004-03-06 23:52:16Z asmax $
  6.  *
  7.  * Authors: Cyril Deguet     <asmax@via.ecp.fr>
  8.  *          Olivier Teuli鑢e <ipkiss@via.ecp.fr>
  9.  *
  10.  * This program is free software; you can redistribute it and/or modify
  11.  * it under the terms of the GNU General Public License as published by
  12.  * the Free Software Foundation; either version 2 of the License, or
  13.  * (at your option) any later version.
  14.  *
  15.  * This program is distributed in the hope that it will be useful,
  16.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18.  * GNU General Public License for more details.
  19.  *
  20.  * You should have received a copy of the GNU General Public License
  21.  * along with this program; if not, write to the Free Software
  22.  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  23.  *****************************************************************************/
  24. #include <png.h>
  25. #include "png_bitmap.hpp"
  26. PngBitmap::PngBitmap( intf_thread_t *pIntf, string fileName,
  27.                       uint32_t aColor ):
  28.     GenericBitmap( pIntf ), m_width( 0 ), m_height( 0 ), m_pData( NULL )
  29. {
  30.     // Open the PNG file
  31.     FILE *pFile = fopen( fileName.c_str(), "rb" );
  32.     if( pFile == NULL )
  33.     {
  34.         msg_Err( getIntf(), "Cannot open bitmap %s", fileName.c_str() );
  35.         return;
  36.     }
  37.     // Create the PNG structures
  38.     png_structp pReadStruct = png_create_read_struct( PNG_LIBPNG_VER_STRING,
  39.         NULL, NULL, NULL );
  40.     if ( pReadStruct == NULL )
  41.     {
  42.         msg_Err( getIntf(), "Failed to create PNG read struct" );
  43.         return;
  44.     }
  45.     png_infop pInfo = png_create_info_struct( pReadStruct );
  46.     if( pInfo == NULL )
  47.     {
  48.         png_destroy_read_struct( &pReadStruct, NULL, NULL );
  49.         msg_Err( getIntf(), "Failed to create PNG info struct" );
  50.         return;
  51.     }
  52.     png_infop pEndInfo = png_create_info_struct( pReadStruct );
  53.     if( pEndInfo == NULL )
  54.     {
  55.         png_destroy_read_struct( &pReadStruct, NULL, NULL );
  56.         msg_Err( getIntf(), "Failed to create PNG end info struct" );
  57.         return;
  58.     }
  59.     // Initialize the PNG reader
  60.     png_init_io( pReadStruct, pFile );
  61.     // Read the image header
  62.     png_read_info( pReadStruct, pInfo );
  63.     m_width = png_get_image_width( pReadStruct, pInfo );
  64.     m_height = png_get_image_height( pReadStruct, pInfo );
  65.     int depth = png_get_bit_depth( pReadStruct, pInfo );
  66.     int colorType = png_get_color_type( pReadStruct, pInfo );
  67.     // Convert paletted images to RGB
  68.     if( colorType == PNG_COLOR_TYPE_PALETTE )
  69.     {
  70.         png_set_palette_to_rgb( pReadStruct );
  71.     }
  72.     // Strip to 8 bits per channel
  73.     if( depth == 16 )
  74.     {
  75.         png_set_strip_16( pReadStruct );
  76.     }
  77.     // 4 bytes per pixel
  78.     if( !(colorType & PNG_COLOR_MASK_ALPHA ) )
  79.     {
  80.         png_set_filler( pReadStruct, 0xff, PNG_FILLER_AFTER );
  81.     }
  82.     // Invert colors
  83.     if( colorType & PNG_COLOR_MASK_COLOR )
  84.     {
  85.         png_set_bgr( pReadStruct );
  86.     }
  87.     png_read_update_info( pReadStruct, pInfo );
  88.     // Allocate memory for the buffers
  89.     m_pData = new uint8_t[m_height * m_width * 4];
  90.     uint8_t** pRows = new uint8_t*[m_height];
  91.     for( int i = 0; i < m_height; i++ )
  92.     {
  93.         pRows[i] = m_pData + (i * m_width * 4);
  94.     }
  95.     // Read the image
  96.     png_read_image( pReadStruct, pRows );
  97.     png_read_end( pReadStruct, pEndInfo );
  98.     // Compute the alpha layer
  99.     uint8_t *pData = m_pData;
  100.     for( int y = 0; y < m_height; y++ )
  101.     {
  102.         for( int x = 0; x < m_width; x++ )
  103.         {
  104.             uint32_t b = (uint32_t)*(pData++);
  105.             uint32_t g = (uint32_t)*(pData++);
  106.             uint32_t r = (uint32_t)*(pData++);
  107.             // Transparent pixel ?
  108.             if( aColor == (r<<16 | g<<8 | b) )
  109.             {
  110.                 *pData = 0;
  111.             }
  112.             pData++;
  113.         }
  114.     }
  115.     // Free the structures
  116.     png_destroy_read_struct( &pReadStruct, &pInfo, &pEndInfo );
  117.     delete[] pRows;
  118.     // Close the file
  119.     fclose( pFile );
  120. }
  121. PngBitmap::~PngBitmap()
  122. {
  123.     if( m_pData )
  124.     {
  125.         delete[] m_pData;
  126.     }
  127. }
  128. uint8_t *PngBitmap::getData() const
  129. {
  130.     if( m_pData == NULL )
  131.     {
  132.         msg_Warn( getIntf(), "PngBitmap::getData() returns NULL" );
  133.     }
  134.     return m_pData;
  135. }