texture.cpp

一个自己写的游戏引擎,用DirectX 写成

//--------------------------------------------------
//  Desc: Texture
//  Author: artsylee/2006.10.26
//--------------------------------------------------
#include "../stdafx.h"
#include <windows.h>
#include "Texture.h"
#include "Log.h"
#include "AniFile.h"

#pragma comment (lib, "winmm.lib")

extern LPDIRECT3DDEVICE9 g_pD3DDevice;

//---------------------------------------------------
//				CTexture扩展描述
// 1,加载LoadTexture应释放m_pTexture且在m_IsAutoRelease状态不用加载
// 2,AutoRemove时确保Handle和m_Name信息有效
// 3,使用时(显示)要更新时间
// 4,D3DPOOL_DEFAULT的资源在丢失设备时要重新加载
// 5,自定义格式的自动加载问题(OK,2006_12_16)
// 6,纹理尺寸与mipmap层数.(OK,2006_12_21)
//---------------------------------------------------

//---------------------------------------------------
// 1, 3D用纹理可使用多层mipmap, 但纹理尺寸需要2的N次方
// 2, 2D用的纹理使用1层mipmap即可, AMF格式较好
//---------------------------------------------------
CTexture::CTexture()
{
	m_Ref = 0;
	m_Width = 0;
	m_Height = 0;
	m_MemoryPool = D3DPOOL_MANAGED;
	m_Name = "";
	m_MipmapLevel = 1;
	m_pTexture = NULL;
	m_LastUseTime = 0;
	m_IsAutoRelease = false;
	m_MemoryMode = MM_AUTO;
	m_FileFlag = 0;
}

CTexture::~CTexture()
{
	Destroy();
}

//---------------------------------------------------
// 加载纹理(改变内存模式调用SetMemMode)
//---------------------------------------------------
bool CTexture::LoadTexture(const char *szFilename, int mipLevel /* = 1 */)
{
	if(szFilename == NULL || szFilename[0] == 0)
	{
		WriteLog(INFO_ERROR, "Invalidate parameter!");
		return false;
	}
	
	Destroy();

	HRESULT hr;
	D3DXIMAGE_INFO image_info;
	hr = D3DXGetImageInfoFromFile(szFilename, &image_info);
	if(FAILED(hr))
	{
		WriteLog(INFO_ERROR, "Can't get image info from file[%s]!", szFilename);
		return false;
	}
	hr = D3DXCreateTextureFromFileEx(g_pD3DDevice, szFilename, \
		image_info.Width, image_info.Height, mipLevel, 0, \
	//	image_info.Format, D3DPOOL_MANAGED, D3DX_FILTER_NONE, D3DX_DEFAULT, 
		D3DFMT_A8R8G8B8, D3DPOOL_MANAGED, D3DX_FILTER_NONE, D3DX_DEFAULT, \
		0, &image_info, NULL, &m_pTexture);

	if(FAILED(hr))
	{
		WriteLog(INFO_ERROR, "Can't create texture from file[%s]!", szFilename);
		return false;
	}
	
	m_Ref = 1;
	m_Width	= image_info.Width;
	m_Height = image_info.Height;
	m_MemoryPool = D3DPOOL_MANAGED;
	m_Name = szFilename;
	m_MipmapLevel = mipLevel;
	m_LastUseTime = ::timeGetTime();
    
	// Output Load Texture Success Info
	// WriteLog(INFO_NORMAL, "Load texture[%s]!", szFilename);
	return true;
}

//---------------------------------------------------
// 获取纹理的描述,由TextureManager调用,
// 通过TextureManager接口使用纹理时才加载纹理数据.
//---------------------------------------------------
bool CTexture::LoadTextureDesc(const char *szFilename, int mipLevel /* = 1 */)
{
	if(szFilename == NULL || szFilename[0] == 0)
	{
		WriteLog(INFO_ERROR, "Invalidate parameter!");
		return false;
	}
	Destroy();

	HRESULT hr;
	D3DXIMAGE_INFO image_info;
	hr = D3DXGetImageInfoFromFile(szFilename, &image_info);
	if(FAILED(hr))
	{
		WriteLog(INFO_ERROR, "Can't get image info from file[%s]!", szFilename);
		return false;
	}
	m_Ref = 1;
	m_Width	= image_info.Width;
	m_Height = image_info.Height;
	m_MemoryPool = D3DPOOL_MANAGED;
	m_Name = szFilename;
	m_MipmapLevel = mipLevel;
	m_LastUseTime = ::timeGetTime();
	// 只能是MM_AUTO模式
	m_IsAutoRelease = true;

	return true;
}

//---------------------------------------------------
// 从AMF加载纹理
//---------------------------------------------------
bool CTexture::LoadAMFTexture(const char *szFilename, DWORD index /* = 0 */, int mipLevel /* = 1 */)
{
	if(szFilename == NULL || szFilename[0] == 0)
	{
		WriteLog(INFO_ERROR, "Invalidate parameter!");
		return false;
	}
	Destroy();
	if(!ReLoadAMFTexture(szFilename, index, mipLevel))
	{
		WriteLog(INFO_ERROR, "Can't Load AMF[%s]!", szFilename);
		return false;
	}

	m_Ref = 1;
	m_MemoryPool = D3DPOOL_MANAGED;
	m_Name = szFilename;
	m_MipmapLevel = mipLevel;
	m_LastUseTime = ::timeGetTime();
	// AMF格式
	m_FileFlag = 0xff000000 | index;
	return true;
}

//---------------------------------------------------
// 从AMF加载纹理的描述,由TextureManager调用
//---------------------------------------------------
bool CTexture::LoadAMFTextureDesc(const char *szFilename, DWORD index /* =0 */, int mipLevel /* = 1 */)
{
	if(szFilename == NULL || szFilename[0] == 0)
	{
		WriteLog(INFO_ERROR, "Invalidate parameter!");
		return false;
	}
	Destroy();

	CAniFile anifile;
	FILE *fp = fopen(szFilename, "rb");
	if(fp==NULL)
	{
		return false;
	}
	AMFHeader header;
	if(!anifile.ReadHeader(fp, &header))
	{
		return false;
	}

	if(index>=header.m_FrameCount)
	{
		return false;
	}
	AMFIndex picIndex;
	if(!anifile.ReadIndex(fp, index, &picIndex))
	{
		return false;
	}
	m_Width	= picIndex.m_PictureWidth;
	m_Height = picIndex.m_PictureHeight;
	
	m_Ref = 1;
	m_MemoryPool = D3DPOOL_MANAGED;
	m_Name = szFilename;
	m_MipmapLevel = mipLevel;
	m_LastUseTime = ::timeGetTime();
	// 只能是MM_AUTO模式
	m_IsAutoRelease = true;
	// AMF格式
	m_FileFlag = 0xff000000 | index;

	fclose(fp);

	return true;
}

//---------------------------------------------------
// 重新加载AMF
//---------------------------------------------------
bool CTexture::ReLoadAMFTexture(const char *szFilename, DWORD index, int mipLevel /* = 1 */)
{
	if(szFilename == NULL || szFilename[0] == 0)
	{
		WriteLog(INFO_ERROR, "Invalidate parameter!");
		return false;
	}
	CAniFile anifile;
	FILE *fp = fopen(szFilename, "rb");
	if(fp==NULL)
	{
		return false;
	}
	AMFHeader header;
	if(!anifile.ReadHeader(fp, &header))
	{
		return false;
	}
	
	if(index>=header.m_FrameCount)
	{
		return false;
	}

	AMFIndex picIndex;
	if(!anifile.ReadIndex(fp, index, &picIndex))
	{
		return false;
	}
	HRESULT hr;
	hr = D3DXCreateTexture(g_pD3DDevice, picIndex.m_PictureWidth, picIndex.m_PictureHeight, mipLevel, 0, D3DFMT_A8R8G8B8, D3DPOOL_MANAGED, &m_pTexture);
	if(FAILED(hr))
	{
		WriteLog(INFO_ERROR, "ERROR! Can't create texture.");
		return false;
	}
	m_Width	= picIndex.m_PictureWidth;
	m_Height = picIndex.m_PictureHeight;
	if(mipLevel==1)
	{
		// 非mipmap方式加载
		D3DLOCKED_RECT lrDst;
		m_pTexture->LockRect(0, &lrDst, 0, 0);
		if(!anifile.ReadData(fp, index, (void*)(lrDst.pBits)))
		{
			m_pTexture->UnlockRect(0);
			return false;
		}
		m_pTexture->UnlockRect(0);
	}
	else
	{	// mipmap方式加载
		DWORD* pData = new DWORD[m_Width*m_Height];
		if(pData==NULL)
		{
			return false;
		}
		if(!anifile.ReadData(fp, index, (void*)pData))
		{
			return false;
		}
		D3DLOCKED_RECT lrDst;
		m_pTexture->LockRect(0, &lrDst, 0, 0);
		DWORD* pDst = (DWORD*)(lrDst.pBits);
		for(int i=0; i<m_Height; i++)
		{
			for(int j=0; j<m_Width; j++)
			{
				pDst[i*(lrDst.Pitch/4)+j] = pData[i*m_Width+j];
			}
		}
		m_pTexture->UnlockRect(0);
		delete [] pData;
		pData = NULL;
		D3DXFilterTexture(m_pTexture, 0, 0, D3DX_DEFAULT);
	}

	fclose(fp);
	return true;
}

//---------------------------------------------------
// 创建纹理(主要用于动画帧, 字体纹理, 这样方式不支持mipmap, 
// 且不支持MM_AUTO模式)
//---------------------------------------------------
bool CTexture::CreateTexture(int width, int height, D3DFORMAT colorFmt, D3DPOOL mem /* = D3DPOOL_MANAGED */)
{
	Destroy();
	HRESULT hr;
	hr = D3DXCreateTexture(g_pD3DDevice, width, height, 1, 0, colorFmt, mem, &m_pTexture);
	if(FAILED(hr))
	{
		WriteLog(INFO_ERROR, "ERROR! Can't create texture.");
		return false;
	}
	m_Ref = 1;
	m_Width	= width;
	m_Height = height;
	m_MemoryPool = mem;
	m_Name = "";
	m_LastUseTime = ::timeGetTime();
	// 只能是MM_KEEPINMEMORY模式
	m_MemoryMode = MM_KEEPINMEMORY;

	return true;
}

//---------------------------------------------------
// 创建纹理(主要用于3D Compose纹理, 支持mipmap, 
// 但不支持MM_AUTO模式)
//---------------------------------------------------
bool CTexture::CreateMipTexture(int width, int height, D3DFORMAT colorFmt, D3DPOOL mem /* = D3DPOOL_MANAGED */)
{
	Destroy();
	HRESULT hr;
	hr = D3DXCreateTexture(g_pD3DDevice, width, height, 0, 0, colorFmt, mem, &m_pTexture);
	if(FAILED(hr))
	{
		WriteLog(INFO_ERROR, "ERROR! Can't create texture.");
		return false;
	}
	m_Ref = 1;
	m_Width	= width;
	m_Height = height;
	m_MemoryPool = mem;
	m_Name = "";
	m_LastUseTime = ::timeGetTime();
	// 只能是MM_KEEPINMEMORY模式
	m_MemoryMode = MM_KEEPINMEMORY;

	return true;
}

//---------------------------------------------------
// 自动加载图片数据
//---------------------------------------------------
bool CTexture::AutoLoad(void)
{
	if(m_IsAutoRelease && !m_Name.empty())
	{
		if(m_FileFlag & 0xff000000)
		{
			if(!ReLoadAMFTexture(m_Name.c_str(), m_FileFlag & 0x00ffffff, m_MipmapLevel))
			{
				return false;
			}
		}
		else
		{
			HRESULT hr;
			D3DXIMAGE_INFO image_info;
			hr = D3DXGetImageInfoFromFile(m_Name.c_str(), &image_info);
			if(FAILED(hr))
			{
				WriteLog(INFO_ERROR, "Auto get image info fail[%s]!", m_Name.c_str());
				return false;
			}
			hr = D3DXCreateTextureFromFileEx(g_pD3DDevice, m_Name.c_str(), \
				image_info.Width, image_info.Height, m_MipmapLevel, 0, \
				//	image_info.Format, D3DPOOL_MANAGED, D3DX_FILTER_NONE, D3DX_DEFAULT, 
				D3DFMT_A8R8G8B8, D3DPOOL_MANAGED, D3DX_FILTER_NONE, D3DX_DEFAULT, \
				0, &image_info, NULL, &m_pTexture);

			if(FAILED(hr))
			{
				WriteLog(INFO_ERROR, "Auto load texture fail[%s]!", m_Name.c_str());
				return false;
			}
		}
		
		m_IsAutoRelease = false;
	}
	return true;
}

//---------------------------------------------------
// 自动释放图片数据
//---------------------------------------------------
int CTexture::AutoRemove(void)
{
	if(m_pTexture)
	{
		m_pTexture->Release();
		m_pTexture = NULL;
	}
	m_IsAutoRelease = true;
	return m_Ref;
}

int CTexture::AddRef(void)
{
	return ++m_Ref;
}

int CTexture::Release(void)
{
	--m_Ref;
	if(m_Ref <= 0)
	{
		Destroy();
	}
	return m_Ref;
}

void CTexture::Destroy(void)
{
	if(m_pTexture)
	{
		m_pTexture->Release();
		m_pTexture = NULL;
	}
	if(!m_Name.empty())
	{
		m_Name.clear();
	}
	m_MipmapLevel = 1;
	m_FileFlag = 0;
	m_IsAutoRelease = false;
}

// end of this file 2006.10.26 artsylee