TexConv/TextureFileCreater.cpp

122 lines
3.4 KiB
C++

#include"TextureFileCreater.h"
#include<hgl/filesystem/FileSystem.h>
namespace
{
/**
* 截取完整路径中的路径名和文件名
* @param pathname 拆分后的路径名
* @param filename 拆分后的文件名
* @param fullname 拆分前的完整路径文件名
*/
template<typename T>
inline bool SplitFilename(BaseString<T> &pathname,BaseString<T> &filename,const BaseString<T> &fullname)
{
if(fullname.Length()<=1)
return false;
const T spear_char[] = { '/','\\' };
const int pos=fullname.FindRightChar(spear_char);
if(pos==-1)
return(false);
pathname.Strcpy(fullname,pos);
filename.Strcpy(fullname.c_str()+pos+1);
return(true);
}
template<typename T>
inline BaseString<T> ReplaceExtName(const BaseString<T> &old_name,const BaseString<T> &new_extname,const T split_char='.')
{
if(old_name.Length()<=1)
return(BaseString<T>::charOf(split_char)+new_extname);
const int pos=old_name.FindRightChar(split_char);
if(pos==-1)
return old_name.SubString(0,pos+1)+new_extname;
else
return old_name+BaseString<T>(split_char)+new_extname;
}
}//namespace
bool ToILType(ILuint &type,const uint8 bits,const ColorDataType cdt)
{
constexpr ILuint target_type[3][(uint)ColorDataType::END-1]=
{
//UNORM SNORM UINT SINT, USCALE,SSCALE, UFLOAT SFLOAT
{IL_UNSIGNED_BYTE, IL_BYTE, IL_UNSIGNED_BYTE, IL_BYTE, 0,0, 0, 0},
{IL_UNSIGNED_SHORT, IL_SHORT, IL_UNSIGNED_SHORT, IL_SHORT, 0,0, IL_HALF, IL_HALF},
{IL_UNSIGNED_INT, IL_INT, IL_UNSIGNED_INT, IL_INT, 0,0, IL_FLOAT, IL_FLOAT}
};
if(bits<=8 )type=target_type[0][(uint)cdt-1];else
if(bits<=16 )type=target_type[1][(uint)cdt-1];else
if(bits<=32 )type=target_type[2][(uint)cdt-1];else
return(false);
return(type);
}
TextureFileCreater::TextureFileCreater(const PixelFormat *pf,ILImage *img)
{
fmt=pf;
image=img;
pixel_total=image->pixel_total();
pixel_bytes=pf->GetPixelBytes();
total_bytes=pixel_total*pixel_bytes;
dos=nullptr;
}
TextureFileCreater::~TextureFileCreater()
{
SAFE_CLEAR(dos);
}
bool TextureFileCreater::WriteFileHeader(const OSString &old_filename)
{
OSString pn,fn;
SplitFilename<os_char>(pn,fn,old_filename);
filename=ReplaceExtName<os_char>(old_filename,OS_TEXT(".Tex2D"));
if(!fos.CreateTrunc(filename))
return(false);
dos=new io::LEDataOutputStream(&fos);
dos->Write("Tex2D\x1A",6);
dos->WriteUint8(2); //版本
dos->WriteBool(false); //是否有mipmaps
dos->WriteUint32(image->width());
dos->WriteUint32(image->height());
dos->WriteUint8(fmt->channels); //颜色通道数
dos->WriteUint8((uint8 *)fmt->color,fmt->channels); //颜色标记
dos->WriteUint8(fmt->bits,fmt->channels); //颜色位数
dos->WriteUint8((uint8)fmt->type); //数据类型
return(true);
}
bool TextureFileCreater::Write(void *data)
{
return(dos->Write(data,total_bytes)==total_bytes);
}
void TextureFileCreater::Close()
{
SAFE_CLEAR(dos);
fos.Close();
}
void TextureFileCreater::Delete()
{
Close();
filesystem::FileDelete(filename);
}