---

ATGAImage.cpp

Go to the documentation of this file.

/*
CS Senior Project 2003
Team : Leftfield
Project : ModernWarfare
Members :
- Russ Christensen              <rchriste@cs.utah.edu>
- Todd Smith                    <tcsmith@cs.utah.edu>
- Usit Duongsaa                 <duongsaa@cs.utah.edu>

Copyright 2003 Russ Christensen, Usit Duongsaa, and Todd Smith. 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. 
THIS SOFTWARE IS PROVIDED BY RUSS CHRISTENSEN, USIT DUONGSAA, AND TODD SMITH ``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 RUSS, USIT, TODD OR OTHER 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.
*/

#include "ATGAImage.h"

// Construction/Destruction

CImage::CImage()
{
        m_ubImageData = NULL;
        m_dwHeight = 0;
        m_dwWidth = 0;
        m_ubBpp = 0;
        m_ubAlphaValue = 128;
        m_bAlpha = false;
}

CImage::~CImage()
{
/*      if (m_ubImageData != NULL)
                delete [] m_ubImageData;
        m_ubImageData = NULL;*/
}

void CImage::LoadBmp(const string& strFileName)
{
        
}

void CImage::LoadTga(const string& strFileName)
{
        FILE* pFile;
        tgaHeader Header;

        pFile = fopen(strFileName.data(), "rb");

        if (!pFile)
        {
                return;
        }

        // Load header
        fread (&Header, sizeof(tgaHeader), 1, pFile);

        m_dwWidth = Header.Width;
        m_dwHeight = Header.Height;
        m_ubBpp = Header.BitsPerPixel / 8;

        // if we want an alpha channel for the 24 bit graphic and make it 32 bit 
        int ColorDepth = m_ubBpp;

        if (m_bAlpha)
                ColorDepth++;

        int nImageSize = m_dwWidth * m_dwHeight * ColorDepth;

        m_ubImageData = new UBYTE [nImageSize];
        if (m_ubImageData == NULL)
        {
                fclose(pFile);
                return;
        }

        // Is the data read equal to the amount we wanted?
        if (nImageSize != 
                fread (m_ubImageData, sizeof(UBYTE), nImageSize, pFile) && Header.DataTypeCode == 2)
        {
                fclose(pFile);
                return;
        }

        if (Header.DataTypeCode == 2)   // uncompressed data
                DecodeT2Tga24(Header, nImageSize);

        else if (Header.DataTypeCode == 10)   // RLE compressed
                DecodeT10Tga24(Header, nImageSize);

        // close file
        fclose(pFile);
}

UBYTE* CImage::GetImageData()
{
        return m_ubImageData;
}


UBYTE CImage::GetBPP()
{
        return m_ubBpp;
}

UINT CImage::GetHeight()
{
        return m_dwHeight;
}
        
UINT CImage::GetWidth()
{
        return m_dwWidth;
}


void CImage::DecodeT2Tga24(const tgaHeader& Header, const int& nImageSize)
{
        UBYTE ColorSwap;

        // BGAR to ARGB
        for (int i = 0; i < nImageSize; i+=m_ubBpp)
        {
                ColorSwap = m_ubImageData[i];
                m_ubImageData[i] = m_ubImageData[i + 2];
                m_ubImageData[i + 2] = ColorSwap;
        }
}

void CImage::DecodeT10Tga24(const tgaHeader& Header, int& nImageSize)
{
        UBYTE p[4];

        int tempImageSize;

        if (m_bAlpha)
                        tempImageSize = Header.Width * Header.Height * 4;
                else
                        tempImageSize = nImageSize;

        UBYTE * TempData = new UBYTE [nImageSize];
        if (TempData != NULL)
        {
                memcpy(TempData, m_ubImageData, sizeof(UBYTE) * tempImageSize);
        }
        
        int nReadCnt = 0;

        UBYTE ubPacketInfo;
        short PacketType;
        short PixelCount;

        for (int i = 0; i < tempImageSize; /*i +=m_ubBpp */)
        {
                ubPacketInfo = TempData[nReadCnt];
                nReadCnt++;

                // Check to see if it's encoded RLE or not
                PacketType = 0x80 & ubPacketInfo;

                // Get number of pixels in run - can go beyond current line
                //short PixelCount2 = ( 0x007f & ubPacketInfo );
                PixelCount = ubPacketInfo;

                if (PixelCount < 128)
                        PixelCount++;
                else
                        PixelCount -= 127;

                if (PacketType)
                {
                        // read the colors swap BGAR to RGBA
                        p[3] = TempData[nReadCnt + 0]; // B
                        p[2] = TempData[nReadCnt + 1]; // G
                        p[1] = TempData[nReadCnt + 2]; // R

                        nReadCnt += 3;

                        if (m_ubBpp == 4)
                                p[0] = TempData[nReadCnt + 3];  //change when done
                        else if (m_bAlpha)
                        {
                                p[0] = m_ubAlphaValue;          // Alpha channel color change to variable
                        }

                        
                        while(PixelCount--)
                        {
                                // read the colors swap BGAR to RGBA
                                m_ubImageData[i++] = p[1];
                                m_ubImageData[i++] = p[2];
                                m_ubImageData[i++] = p[3];

                                if (m_ubBpp == 4)
                                        m_ubImageData[i++] = p[0];
                                else if (m_bAlpha)
                                        m_ubImageData[i++] = p[0];              // Alpha channel color change to variable*/

                        }
                }
                else
                {
                        while(PixelCount--)
                        {
                                p[3] = TempData[nReadCnt + 0]; // B
                                p[2] = TempData[nReadCnt + 1]; // G
                                p[1] = TempData[nReadCnt + 2]; // R

                                nReadCnt += 3;
                                m_ubImageData[i++] = p[1];
                                m_ubImageData[i++] = p[2];
                                m_ubImageData[i++] = p[3];

                                if (m_ubBpp == 4)
                                {
                                        m_ubImageData[i++] = TempData[nReadCnt];
                                
                                        nReadCnt++;
                                }
                                else if (m_bAlpha)
                                {
                                        m_ubImageData[i++] = m_ubAlphaValue;            // Alpha channel color change to variable
                                }
                        }
                }       
        }

        delete [] TempData;
}

void CImage::SetAlpha(UBYTE AlphaValue)
{
        m_ubAlphaValue = AlphaValue;
}

void CImage::bAlphaChannel(bool bAlpha)
{
        m_bAlpha = bAlpha;
}

void CImage::LoadRaw(const string &strName, int nSize)
{
        FILE *pFile = NULL;
        m_ubBpp = 1;

        // Open The File In Read / Binary Mode.
        pFile = fopen(strName.data(), "rb");

        // Check To See If We Found The File And Could Open It
        if (pFile == NULL)
        {
                // Display Error Message And Stop The Function
                MessageBox(NULL, "Can't Find The Height Map!", "Error", MB_OK);
                return;
        }

        m_ubImageData = new UBYTE [nSize];
        fread(m_ubImageData, 1, nSize, pFile);

        // After We Read The Data, It's A Good Idea To Check If Everything Read Fine
        int result = ferror(pFile);

        // Check If We Received An Error
        if (result)
        {
                MessageBox(NULL, "Failed To Get Data!", "Error", MB_OK);
        }

        // Close The File.*/
        fclose(pFile);
}

---