You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
320 lines
11 KiB
320 lines
11 KiB
//========= Copyright Valve Corporation, All rights reserved. ============// |
|
// |
|
// Purpose: |
|
// |
|
// $NoKeywords: $ |
|
//=============================================================================// |
|
|
|
#ifdef _WIN32 |
|
#include <direct.h> |
|
#endif |
|
#include "mathlib/mathlib.h" |
|
#include "bitmap/tgawriter.h" |
|
#include "tier1/strtools.h" |
|
#include "vtf/vtf.h" |
|
#include "tier1/utlbuffer.h" |
|
#include "tier0/dbg.h" |
|
#include "tier0/icommandline.h" |
|
#include "tier1/utlbuffer.h" |
|
#include "tier2/tier2.h" |
|
#include "filesystem.h" |
|
|
|
|
|
//----------------------------------------------------------------------------- |
|
// HDRFIXME: move this somewhere else. |
|
//----------------------------------------------------------------------------- |
|
static void PFMWrite( float *pFloatImage, const char *pFilename, int width, int height ) |
|
{ |
|
FILE *fp; |
|
fp = fopen( pFilename, "wb" ); |
|
fprintf( fp, "PF\n%d %d\n-1.000000\n", width, height ); |
|
int i; |
|
for( i = height-1; i >= 0; i-- ) |
|
{ |
|
float *pRow = &pFloatImage[3 * width * i]; |
|
fwrite( pRow, width * sizeof( float ) * 3, 1, fp ); |
|
} |
|
fclose( fp ); |
|
} |
|
|
|
SpewRetval_t VTF2TGAOutputFunc( SpewType_t spewType, char const *pMsg ) |
|
{ |
|
printf( "%s", pMsg ); |
|
fflush( stdout ); |
|
|
|
if (spewType == SPEW_ERROR) |
|
return SPEW_ABORT; |
|
return (spewType == SPEW_ASSERT) ? SPEW_DEBUGGER : SPEW_CONTINUE; |
|
} |
|
|
|
static void Usage( void ) |
|
{ |
|
Error( "Usage: vtf2tga -i <input vtf> [-o <output tga>] [-mip]\n" ); |
|
exit( -1 ); |
|
} |
|
|
|
int main( int argc, char **argv ) |
|
{ |
|
SpewOutputFunc( VTF2TGAOutputFunc ); |
|
CommandLine()->CreateCmdLine( argc, argv ); |
|
MathLib_Init( 2.2f, 2.2f, 0.0f, 1.0f, false, false, false, false ); |
|
InitDefaultFileSystem(); |
|
|
|
const char *pVTFFileName = CommandLine()->ParmValue( "-i" ); |
|
const char *pTGAFileName = CommandLine()->ParmValue( "-o" ); |
|
bool bGenerateMipLevels = CommandLine()->CheckParm( "-mip" ) != NULL; |
|
if ( !pVTFFileName ) |
|
{ |
|
Usage(); |
|
} |
|
|
|
if ( !pTGAFileName ) |
|
{ |
|
pTGAFileName = pVTFFileName; |
|
} |
|
|
|
char pCurrentDirectory[MAX_PATH]; |
|
if ( _getcwd( pCurrentDirectory, sizeof(pCurrentDirectory) ) == NULL ) |
|
{ |
|
fprintf( stderr, "Unable to get the current directory\n" ); |
|
return -1; |
|
} |
|
Q_StripTrailingSlash( pCurrentDirectory ); |
|
|
|
char pBuf[MAX_PATH]; |
|
if ( !Q_IsAbsolutePath( pTGAFileName ) ) |
|
{ |
|
Q_snprintf( pBuf, sizeof(pBuf), "%s\\%s", pCurrentDirectory, pTGAFileName ); |
|
} |
|
else |
|
{ |
|
Q_strncpy( pBuf, pTGAFileName, sizeof(pBuf) ); |
|
} |
|
Q_FixSlashes( pBuf ); |
|
|
|
char pOutFileNameBase[MAX_PATH]; |
|
Q_StripExtension( pBuf, pOutFileNameBase, MAX_PATH ); |
|
|
|
char pActualVTFFileName[MAX_PATH]; |
|
Q_strncpy( pActualVTFFileName, pVTFFileName, MAX_PATH ); |
|
if ( !Q_strstr( pActualVTFFileName, ".vtf" ) ) |
|
{ |
|
Q_strcat( pActualVTFFileName, ".vtf", MAX_PATH ); |
|
} |
|
|
|
FILE *vtfFp = fopen( pActualVTFFileName, "rb" ); |
|
if( !vtfFp ) |
|
{ |
|
Error( "Can't open %s\n", pActualVTFFileName ); |
|
exit( -1 ); |
|
} |
|
|
|
fseek( vtfFp, 0, SEEK_END ); |
|
int srcVTFLength = ftell( vtfFp ); |
|
fseek( vtfFp, 0, SEEK_SET ); |
|
|
|
CUtlBuffer buf; |
|
buf.EnsureCapacity( srcVTFLength ); |
|
int nBytesRead = fread( buf.Base(), 1, srcVTFLength, vtfFp ); |
|
fclose( vtfFp ); |
|
buf.SeekPut( CUtlBuffer::SEEK_HEAD, nBytesRead ); |
|
|
|
IVTFTexture *pTex = CreateVTFTexture(); |
|
if (!pTex->Unserialize( buf )) |
|
{ |
|
Error( "*** Error reading in .VTF file %s\n", pActualVTFFileName ); |
|
exit(-1); |
|
} |
|
|
|
Msg( "vtf width: %d\n", pTex->Width() ); |
|
Msg( "vtf height: %d\n", pTex->Height() ); |
|
Msg( "vtf numFrames: %d\n", pTex->FrameCount() ); |
|
|
|
Msg( "TEXTUREFLAGS_POINTSAMPLE=%s\n", ( pTex->Flags() & TEXTUREFLAGS_POINTSAMPLE ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_TRILINEAR=%s\n", ( pTex->Flags() & TEXTUREFLAGS_TRILINEAR ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_CLAMPS=%s\n", ( pTex->Flags() & TEXTUREFLAGS_CLAMPS ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_CLAMPT=%s\n", ( pTex->Flags() & TEXTUREFLAGS_CLAMPT ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_CLAMPU=%s\n", ( pTex->Flags() & TEXTUREFLAGS_CLAMPU ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_BORDER=%s\n", ( pTex->Flags() & TEXTUREFLAGS_BORDER ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_ANISOTROPIC=%s\n", ( pTex->Flags() & TEXTUREFLAGS_ANISOTROPIC ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_HINT_DXT5=%s\n", ( pTex->Flags() & TEXTUREFLAGS_HINT_DXT5 ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_SRGB=%s\n", ( pTex->Flags() & TEXTUREFLAGS_SRGB ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_NORMAL=%s\n", ( pTex->Flags() & TEXTUREFLAGS_NORMAL ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_NOMIP=%s\n", ( pTex->Flags() & TEXTUREFLAGS_NOMIP ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_NOLOD=%s\n", ( pTex->Flags() & TEXTUREFLAGS_NOLOD ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_ALL_MIPS=%s\n", ( pTex->Flags() & TEXTUREFLAGS_ALL_MIPS ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_PROCEDURAL=%s\n", ( pTex->Flags() & TEXTUREFLAGS_PROCEDURAL ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_ONEBITALPHA=%s\n", ( pTex->Flags() & TEXTUREFLAGS_ONEBITALPHA ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_EIGHTBITALPHA=%s\n", ( pTex->Flags() & TEXTUREFLAGS_EIGHTBITALPHA ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_ENVMAP=%s\n", ( pTex->Flags() & TEXTUREFLAGS_ENVMAP ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_RENDERTARGET=%s\n", ( pTex->Flags() & TEXTUREFLAGS_RENDERTARGET ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_DEPTHRENDERTARGET=%s\n", ( pTex->Flags() & TEXTUREFLAGS_DEPTHRENDERTARGET ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_NODEBUGOVERRIDE=%s\n", ( pTex->Flags() & TEXTUREFLAGS_NODEBUGOVERRIDE ) ? "true" : "false" ); |
|
Msg( "TEXTUREFLAGS_SINGLECOPY=%s\n", ( pTex->Flags() & TEXTUREFLAGS_SINGLECOPY ) ? "true" : "false" ); |
|
|
|
Vector vecReflectivity = pTex->Reflectivity(); |
|
Msg( "vtf reflectivity: %f %f %f\n", vecReflectivity[0], vecReflectivity[1], vecReflectivity[2] ); |
|
Msg( "transparency: " ); |
|
if( pTex->Flags() & TEXTUREFLAGS_EIGHTBITALPHA ) |
|
{ |
|
Msg( "eightbitalpha\n" ); |
|
} |
|
else if( pTex->Flags() & TEXTUREFLAGS_ONEBITALPHA ) |
|
{ |
|
Msg( "onebitalpha\n" ); |
|
} |
|
else |
|
{ |
|
Msg( "noalpha\n" ); |
|
} |
|
ImageFormat srcFormat = pTex->Format(); |
|
Msg( "vtf format: %s\n", ImageLoader::GetName( srcFormat ) ); |
|
|
|
int iTGANameLen = Q_strlen( pOutFileNameBase ); |
|
|
|
int iFaceCount = pTex->FaceCount(); |
|
int nFrameCount = pTex->FrameCount(); |
|
bool bIsCubeMap = pTex->IsCubeMap(); |
|
|
|
int iLastMipLevel = bGenerateMipLevels ? pTex->MipCount() - 1 : 0; |
|
for( int iFrame = 0; iFrame < nFrameCount; ++iFrame ) |
|
{ |
|
for ( int iMipLevel = 0; iMipLevel <= iLastMipLevel; ++iMipLevel ) |
|
{ |
|
int iWidth, iHeight, iDepth; |
|
pTex->ComputeMipLevelDimensions( iMipLevel, &iWidth, &iHeight, &iDepth ); |
|
|
|
for (int iCubeFace = 0; iCubeFace < iFaceCount; ++iCubeFace) |
|
{ |
|
for ( int z = 0; z < iDepth; ++z ) |
|
{ |
|
// Construct output filename |
|
char *pTempNameBuf = new char[iTGANameLen + 13];//(char *)stackalloc( iTGANameLen + 13 ); |
|
Q_strncpy( pTempNameBuf, pOutFileNameBase, iTGANameLen + 1 ); |
|
char *pExt = Q_strrchr( pTempNameBuf, '.' ); |
|
if ( pExt ) |
|
{ |
|
pExt = 0; |
|
} |
|
|
|
if ( bIsCubeMap ) |
|
{ |
|
Assert( pTex->Depth() == 1 ); // shouldn't this be 1 instead of 0? |
|
static const char *pCubeFaceName[7] = { "rt", "lf", "bk", "ft", "up", "dn", "sph" }; |
|
Q_strcat( pTempNameBuf, pCubeFaceName[iCubeFace], iTGANameLen + 13 ); |
|
} |
|
|
|
if ( nFrameCount > 1 ) |
|
{ |
|
char pTemp[4]; |
|
Q_snprintf( pTemp, 4, "%03d", iFrame ); |
|
Q_strcat( pTempNameBuf, pTemp, iTGANameLen + 13 ); |
|
} |
|
|
|
if ( iLastMipLevel != 0 ) |
|
{ |
|
char pTemp[8]; |
|
Q_snprintf( pTemp, 8, "_mip%d", iMipLevel ); |
|
Q_strcat( pTempNameBuf, pTemp, iTGANameLen + 13 ); |
|
} |
|
|
|
if ( pTex->Depth() > 1 ) |
|
{ |
|
char pTemp[6]; |
|
Q_snprintf( pTemp, 6, "_z%03d", z ); |
|
Q_strcat( pTempNameBuf, pTemp, iTGANameLen + 13 ); |
|
} |
|
|
|
if( srcFormat == IMAGE_FORMAT_RGBA16161616F ) |
|
{ |
|
Q_strcat( pTempNameBuf, ".pfm", iTGANameLen + 13 ); |
|
} |
|
else |
|
{ |
|
Q_strcat( pTempNameBuf, ".tga", iTGANameLen + 13 ); |
|
} |
|
|
|
unsigned char *pSrcImage = pTex->ImageData( iFrame, iCubeFace, iMipLevel, 0, 0, z ); |
|
|
|
ImageFormat dstFormat; |
|
if( srcFormat == IMAGE_FORMAT_RGBA16161616F ) |
|
{ |
|
dstFormat = IMAGE_FORMAT_RGB323232F; |
|
} |
|
else |
|
{ |
|
if( ImageLoader::IsTransparent( srcFormat ) || (srcFormat == IMAGE_FORMAT_ATI1N ) || (srcFormat == IMAGE_FORMAT_ATI2N )) |
|
{ |
|
dstFormat = IMAGE_FORMAT_BGRA8888; |
|
} |
|
else |
|
{ |
|
dstFormat = IMAGE_FORMAT_BGR888; |
|
} |
|
} |
|
// dstFormat = IMAGE_FORMAT_RGBA8888; |
|
// dstFormat = IMAGE_FORMAT_RGB888; |
|
// dstFormat = IMAGE_FORMAT_BGRA8888; |
|
// dstFormat = IMAGE_FORMAT_BGR888; |
|
// dstFormat = IMAGE_FORMAT_BGRA5551; |
|
// dstFormat = IMAGE_FORMAT_BGR565; |
|
// dstFormat = IMAGE_FORMAT_BGRA4444; |
|
// printf( "dstFormat: %s\n", ImageLoader::GetName( dstFormat ) ); |
|
unsigned char *pDstImage = new unsigned char[ImageLoader::GetMemRequired( iWidth, iHeight, 1, dstFormat, false )]; |
|
if( !ImageLoader::ConvertImageFormat( pSrcImage, srcFormat, |
|
pDstImage, dstFormat, iWidth, iHeight, 0, 0 ) ) |
|
{ |
|
Error( "Error converting from %s to %s\n", |
|
ImageLoader::GetName( srcFormat ), ImageLoader::GetName( dstFormat ) ); |
|
exit( -1 ); |
|
} |
|
|
|
if( dstFormat != IMAGE_FORMAT_RGB323232F ) |
|
{ |
|
if( ImageLoader::IsTransparent( dstFormat ) && ( dstFormat != IMAGE_FORMAT_RGBA8888 ) ) |
|
{ |
|
unsigned char *tmpImage = pDstImage; |
|
pDstImage = new unsigned char[ImageLoader::GetMemRequired( iWidth, iHeight, 1, IMAGE_FORMAT_RGBA8888, false )]; |
|
if( !ImageLoader::ConvertImageFormat( tmpImage, dstFormat, pDstImage, IMAGE_FORMAT_RGBA8888, |
|
iWidth, iHeight, 0, 0 ) ) |
|
{ |
|
Error( "Error converting from %s to %s\n", |
|
ImageLoader::GetName( dstFormat ), ImageLoader::GetName( IMAGE_FORMAT_RGBA8888 ) ); |
|
} |
|
dstFormat = IMAGE_FORMAT_RGBA8888; |
|
} |
|
else if( !ImageLoader::IsTransparent( dstFormat ) && ( dstFormat != IMAGE_FORMAT_RGB888 ) ) |
|
{ |
|
unsigned char *tmpImage = pDstImage; |
|
pDstImage = new unsigned char[ImageLoader::GetMemRequired( iWidth, iHeight, 1, IMAGE_FORMAT_RGB888, false )]; |
|
if( !ImageLoader::ConvertImageFormat( tmpImage, dstFormat, pDstImage, IMAGE_FORMAT_RGB888, |
|
iWidth, iHeight, 0, 0 ) ) |
|
{ |
|
Error( "Error converting from %s to %s\n", |
|
ImageLoader::GetName( dstFormat ), ImageLoader::GetName( IMAGE_FORMAT_RGB888 ) ); |
|
} |
|
dstFormat = IMAGE_FORMAT_RGB888; |
|
} |
|
|
|
CUtlBuffer outBuffer; |
|
TGAWriter::WriteToBuffer( pDstImage, outBuffer, iWidth, iHeight, |
|
dstFormat, dstFormat ); |
|
if ( !g_pFullFileSystem->WriteFile( pTempNameBuf, NULL, outBuffer ) ) |
|
{ |
|
fprintf( stderr, "unable to write %s\n", pTempNameBuf ); |
|
} |
|
} |
|
else |
|
{ |
|
PFMWrite( ( float * )pDstImage, pTempNameBuf, iWidth, iHeight ); |
|
} |
|
|
|
delete[] pTempNameBuf; |
|
} |
|
} |
|
} |
|
} |
|
|
|
// leak leak leak leak leak, leak leak, leak leak (Blue Danube) |
|
return 0; |
|
}
|
|
|