mirror of
https://port.numenaute.org/aleajactaest/khanat-opennel-code.git
synced 2024-12-21 00:18:43 +00:00
445 lines
12 KiB
C++
445 lines
12 KiB
C++
// lightmap_optimizer
|
|
// ------------------
|
|
// the goal is to regroup lightmap of a level into lightmap with a higher level
|
|
|
|
#include "nel/misc/common.h"
|
|
#include "nel/misc/file.h"
|
|
#include "nel/misc/bitmap.h"
|
|
#include "nel/misc/log.h"
|
|
#include "nel/misc/path.h"
|
|
#include "nel/misc/uv.h"
|
|
|
|
//#include "windows.h"
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
using namespace std;
|
|
using namespace NLMISC;
|
|
|
|
// ***************************************************************************
|
|
//char sExeDir[MAX_PATH];
|
|
std::string sExeDir;
|
|
NLMISC::CApplicationContext _ApplicationContext;
|
|
|
|
void outString (const string &sText)
|
|
{
|
|
std::string sCurDir = CPath::getCurrentPath();
|
|
CPath::setCurrentPath(sExeDir.c_str());
|
|
//char sCurDir[MAX_PATH];
|
|
//GetCurrentDirectory (MAX_PATH, sCurDir);
|
|
//SetCurrentDirectory (sExeDir);
|
|
NLMISC::createDebug ();
|
|
NLMISC::InfoLog->displayRaw(sText.c_str());
|
|
//SetCurrentDirectory (sCurDir);
|
|
CPath::setCurrentPath(sCurDir.c_str());
|
|
}
|
|
|
|
// ***************************************************************************
|
|
// test every 4 pixels for 2 reason: DXTC and speed
|
|
const uint32 posStep= 4;
|
|
|
|
// ***************************************************************************
|
|
// Try all position to put pSrc in pDst
|
|
bool tryAllPos (NLMISC::CBitmap *pSrc, NLMISC::CBitmap *pDst, sint32 &x, sint32 &y)
|
|
{
|
|
uint32 i, j;
|
|
CObjectVector<uint8> &rSrcPix = pSrc->getPixels();
|
|
CObjectVector<uint8> &rDstPix = pDst->getPixels();
|
|
|
|
// Recalculate real size of the source (without padding to power of 2)
|
|
uint32 nSrcWidth = pSrc->getWidth(), nSrcHeight = pSrc->getHeight();
|
|
|
|
if (nSrcWidth > pDst->getWidth() ) return false;
|
|
if (nSrcHeight > pDst->getHeight() ) return false;
|
|
|
|
// For all position test if the Src plane can be put in
|
|
for (j = 0; j <= (pDst->getHeight() - nSrcHeight); j+= posStep)
|
|
for (i = 0; i <= (pDst->getWidth() - nSrcWidth); i+= posStep)
|
|
{
|
|
x = i; y = j;
|
|
|
|
uint32 a, b;
|
|
bool bCanPut = true;
|
|
for (b = 0; b < nSrcHeight; ++b)
|
|
{
|
|
for (a = 0; a < nSrcWidth; ++a)
|
|
{
|
|
if (rDstPix[4*((x+a)+(y+b)*pDst->getWidth())+3] != 0)
|
|
{
|
|
bCanPut = false;
|
|
break;
|
|
}
|
|
}
|
|
if (bCanPut == false)
|
|
break;
|
|
}
|
|
if (bCanPut)
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// ***************************************************************************
|
|
void putPixel(uint8 *dst, uint8 *src, bool alphaTransfert)
|
|
{
|
|
dst[0] = src[0];
|
|
dst[1] = src[1];
|
|
dst[2] = src[2];
|
|
if (alphaTransfert)
|
|
dst[3] = src[3];
|
|
else
|
|
dst[3] = 255;
|
|
}
|
|
|
|
// ***************************************************************************
|
|
bool putIn (NLMISC::CBitmap *pSrc, NLMISC::CBitmap *pDst, sint32 x, sint32 y, bool alphaTransfert=true)
|
|
{
|
|
uint32 a, b;
|
|
|
|
uint8 *rSrcPix = &pSrc->getPixels()[0];
|
|
uint8 *rDstPix = &pDst->getPixels()[0];
|
|
|
|
uint wSrc= pSrc->getWidth();
|
|
uint hSrc= pSrc->getHeight();
|
|
for (b = 0; b < hSrc; ++b)
|
|
for (a = 0; a < wSrc; ++a)
|
|
{
|
|
if (rDstPix[4*((x+a)+(y+b)*pDst->getWidth())+3] != 0)
|
|
return false;
|
|
// write
|
|
putPixel(rDstPix + 4*((x+a)+(y+b)*pDst->getWidth()), rSrcPix+ 4*(a+b*pSrc->getWidth()), alphaTransfert);
|
|
}
|
|
|
|
// DXTC compression optim: fill last column block and last row block of 4 pixels with block color (don't let black or undefined)
|
|
uint wSrc4= 4*((wSrc+3)/4);
|
|
uint hSrc4= 4*((hSrc+3)/4);
|
|
// expand on W
|
|
if(wSrc<wSrc4)
|
|
{
|
|
for(a=wSrc;a<wSrc4;a++)
|
|
{
|
|
for(b=0;b<hSrc4;b++)
|
|
{
|
|
putPixel(rDstPix + 4*((x+a)+(y+b)*pDst->getWidth()), rDstPix + 4*((x+wSrc-1)+(y+b)*pDst->getWidth()), alphaTransfert);
|
|
}
|
|
}
|
|
}
|
|
// expand on H
|
|
if(hSrc<hSrc4)
|
|
{
|
|
for(b=hSrc;b<hSrc4;b++)
|
|
{
|
|
for(a=0;a<wSrc4;a++)
|
|
{
|
|
putPixel(rDstPix + 4*((x+a)+(y+b)*pDst->getWidth()), rDstPix + 4*((x+a)+(y+hSrc-1)*pDst->getWidth()), alphaTransfert);
|
|
}
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
// ***************************************************************************
|
|
string getBaseName (const string &fullname)
|
|
{
|
|
string sTmp2 = "";
|
|
int pos = fullname.rfind('_');
|
|
for (int j = 0; j <= pos; ++j)
|
|
sTmp2 += fullname[j];
|
|
return sTmp2;
|
|
}
|
|
|
|
// ***************************************************************************
|
|
// resize the bitmap to the next power of 2 and preserve content
|
|
void enlargeCanvas (NLMISC::CBitmap &b)
|
|
{
|
|
sint32 nNewWidth = b.getWidth(), nNewHeight = b.getHeight();
|
|
if (nNewWidth > nNewHeight)
|
|
nNewHeight *= 2;
|
|
else
|
|
nNewWidth *= 2;
|
|
|
|
NLMISC::CBitmap b2;
|
|
b2.resize (nNewWidth, nNewHeight, NLMISC::CBitmap::RGBA);
|
|
|
|
CObjectVector<uint8> &rPixelBitmap = b2.getPixels(0);
|
|
for (sint32 i = 0; i < nNewWidth*nNewHeight*4; ++i)
|
|
rPixelBitmap[i] = 0;
|
|
|
|
putIn (&b, &b2, 0, 0);
|
|
b = b2;
|
|
}
|
|
|
|
// ***************************************************************************
|
|
// main
|
|
// ***************************************************************************
|
|
int main(int nNbArg, char **ppArgs)
|
|
{
|
|
//GetCurrentDirectory (MAX_PATH, sExeDir);
|
|
sExeDir = CPath::getCurrentPath();
|
|
|
|
if (nNbArg < 3)
|
|
{
|
|
outString ("ERROR : Wrong number of arguments\n");
|
|
outString ("USAGE : build_interface [-s<existing_uv_txt_name>] <out_tga_name> <path_maps1> [path_maps2] [path_maps3] ....\n");
|
|
outString (" -s : build a subset of an existing interface definition while preserving the existing texture ids,");
|
|
outString (" to support freeing up VRAM by switching to the subset without rebuilding the entire interface\n");
|
|
return -1;
|
|
}
|
|
|
|
// build as a subset of existing interface
|
|
bool buildSubset = false;
|
|
string existingUVfilename;
|
|
list<string> inputDirs;
|
|
for ( uint i=1; (sint)i<nNbArg; ++i )
|
|
{
|
|
if ( ppArgs[i][0] == '-' )
|
|
{
|
|
switch ( ppArgs[i][1] )
|
|
{
|
|
case 'S':
|
|
case 's':
|
|
buildSubset = true;
|
|
existingUVfilename = string( ppArgs[i]+2 );
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
inputDirs.push_back(ppArgs[i]);
|
|
}
|
|
|
|
string fmtName;
|
|
uint iNumDirs = inputDirs.size();
|
|
if( iNumDirs )
|
|
{
|
|
fmtName = inputDirs.front();
|
|
inputDirs.pop_front();
|
|
--iNumDirs;
|
|
}
|
|
vector<string> AllMapNames;
|
|
list<string>::iterator it = inputDirs.begin();
|
|
list<string>::iterator itEnd = inputDirs.end();
|
|
while( it != itEnd )
|
|
{
|
|
string sDir = *it++;
|
|
if( !CFile::isDirectory(sDir) )
|
|
{
|
|
outString (string("ERROR : directory ") + sDir + " does not exist\n");
|
|
return -1;
|
|
}
|
|
CPath::getPathContent(sDir, false, false, true, AllMapNames);
|
|
}
|
|
|
|
vector<NLMISC::CBitmap*> AllMaps;
|
|
sint32 i, j;
|
|
|
|
// Load all maps
|
|
sint32 mapSize = (sint32)AllMapNames.size();
|
|
AllMaps.resize( mapSize );
|
|
for( i = 0; i < mapSize; ++i )
|
|
{
|
|
try
|
|
{
|
|
NLMISC::CBitmap *pBtmp = new NLMISC::CBitmap;
|
|
NLMISC::CIFile inFile;
|
|
inFile.open( AllMapNames[i] );
|
|
pBtmp->load(inFile);
|
|
AllMaps[i] = pBtmp;
|
|
}
|
|
catch (NLMISC::Exception &e)
|
|
{
|
|
outString (string("ERROR :") + e.what());
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
// Sort all maps by decreasing size
|
|
for (i = 0; i < mapSize-1; ++i)
|
|
for (j = i+1; j < mapSize; ++j)
|
|
{
|
|
NLMISC::CBitmap *pBI = AllMaps[i];
|
|
NLMISC::CBitmap *pBJ = AllMaps[j];
|
|
if ((pBI->getWidth()*pBI->getHeight()) < (pBJ->getWidth()*pBJ->getHeight()))
|
|
{
|
|
NLMISC::CBitmap *pBTmp = AllMaps[i];
|
|
AllMaps[i] = AllMaps[j];
|
|
AllMaps[j] = pBTmp;
|
|
|
|
string sTmp = AllMapNames[i];
|
|
AllMapNames[i] = AllMapNames[j];
|
|
AllMapNames[j] = sTmp;
|
|
}
|
|
}
|
|
|
|
// Place all maps into the global texture
|
|
NLMISC::CBitmap GlobalTexture, GlobalMask;
|
|
GlobalTexture.resize (1, 1, NLMISC::CBitmap::RGBA);
|
|
GlobalMask.resize (1, 1, NLMISC::CBitmap::RGBA);
|
|
CObjectVector<uint8> &rPixelBitmap = GlobalTexture.getPixels(0);
|
|
rPixelBitmap[0] = rPixelBitmap[1] = rPixelBitmap[2] = rPixelBitmap[3] = 0;
|
|
CObjectVector<uint8> &rPixelMask = GlobalMask.getPixels(0);
|
|
rPixelMask[0] = rPixelMask[1] = rPixelMask[2] = rPixelMask[3] = 0;
|
|
vector<NLMISC::CUV> UVMin, UVMax;
|
|
UVMin.resize (mapSize, NLMISC::CUV(0.0f, 0.0f));
|
|
UVMax.resize (mapSize, NLMISC::CUV(0.0f, 0.0f));
|
|
for (i = 0; i < mapSize; ++i)
|
|
{
|
|
sint32 x, y;
|
|
while (!tryAllPos(AllMaps[i], &GlobalMask, x, y))
|
|
{
|
|
// Enlarge global texture
|
|
enlargeCanvas (GlobalTexture);
|
|
enlargeCanvas (GlobalMask);
|
|
}
|
|
putIn (AllMaps[i], &GlobalTexture, x, y);
|
|
putIn (AllMaps[i], &GlobalMask, x, y, false);
|
|
UVMin[i].U = (float)x;
|
|
UVMin[i].V = (float)y;
|
|
UVMax[i].U = (float)x+AllMaps[i]->getWidth();
|
|
UVMax[i].V = (float)y+AllMaps[i]->getHeight();
|
|
|
|
/* // Do not remove this is usefull for debugging
|
|
{
|
|
NLMISC::COFile outTga;
|
|
string fmtName = ppArgs[1];
|
|
if (fmtName.rfind('.') == string::npos)
|
|
fmtName += ".tga";
|
|
if (outTga.open(fmtName))
|
|
{
|
|
GlobalTexture.writeTGA (outTga, 32);
|
|
outTga.close();
|
|
}
|
|
}
|
|
{
|
|
NLMISC::COFile outTga;
|
|
string fmtName = ppArgs[1];
|
|
if (fmtName.rfind('.') == string::npos)
|
|
fmtName += "_msk.tga";
|
|
else
|
|
fmtName = fmtName.substr(0,fmtName.rfind('.')) + "_msk.tga";
|
|
if (outTga.open(fmtName))
|
|
{
|
|
GlobalMask.writeTGA (outTga, 32);
|
|
outTga.close();
|
|
}
|
|
}*/
|
|
|
|
|
|
}
|
|
|
|
// Convert UV from pixel to ratio
|
|
for (i = 0; i < mapSize; ++i)
|
|
{
|
|
UVMin[i].U = UVMin[i].U / (float)GlobalTexture.getWidth();
|
|
UVMin[i].V = UVMin[i].V / (float)GlobalTexture.getHeight();
|
|
UVMax[i].U = UVMax[i].U / (float)GlobalTexture.getWidth();
|
|
UVMax[i].V = UVMax[i].V / (float)GlobalTexture.getHeight();
|
|
}
|
|
|
|
// Write global texture file
|
|
//SetCurrentDirectory (sExeDir);
|
|
CPath::setCurrentPath(sExeDir.c_str());
|
|
|
|
NLMISC::COFile outTga;
|
|
if (fmtName.rfind('.') == string::npos)
|
|
fmtName += ".tga";
|
|
if (outTga.open(fmtName))
|
|
{
|
|
GlobalTexture.writeTGA (outTga, 32);
|
|
outTga.close();
|
|
outString (string("Writing tga file : ") + fmtName + "\n");
|
|
}
|
|
else
|
|
{
|
|
outString (string("ERROR: Cannot write tga file : ") + fmtName + "\n");
|
|
}
|
|
|
|
// Write UV text file
|
|
if( !buildSubset )
|
|
{
|
|
fmtName = fmtName.substr(0, fmtName.rfind('.'));
|
|
fmtName += ".txt";
|
|
FILE *f = fopen (fmtName.c_str(), "wt");
|
|
if (f != NULL)
|
|
{
|
|
for (i = 0; i < mapSize; ++i)
|
|
{
|
|
// get the string whitout path
|
|
string fileName= CFile::getFilename(AllMapNames[i]);
|
|
fprintf (f, "%s %.12f %.12f %.12f %.12f\n", fileName.c_str(), UVMin[i].U, UVMin[i].V,
|
|
UVMax[i].U, UVMax[i].V);
|
|
}
|
|
fclose (f);
|
|
outString (string("Writing UV file : ") + fmtName + "\n");
|
|
}
|
|
else
|
|
{
|
|
outString (string("ERROR: Cannot write UV file : ") + fmtName + "\n");
|
|
}
|
|
}
|
|
else // build as a subset
|
|
{
|
|
// Load existing uv file
|
|
CIFile iFile;
|
|
string filename = CPath::lookup (existingUVfilename, false);
|
|
if( (filename == "") || (!iFile.open(filename)) )
|
|
{
|
|
outString (string("ERROR : could not open file ") + existingUVfilename + "\n");
|
|
return -1;
|
|
}
|
|
|
|
// Write subset UV text file
|
|
fmtName = fmtName.substr(0, fmtName.rfind('.'));
|
|
fmtName += ".txt";
|
|
FILE *f = fopen (fmtName.c_str(), "wt");
|
|
if (f == NULL)
|
|
{
|
|
outString (string("ERROR: Cannot write UV file : ") + fmtName + "\n");
|
|
// fclose (iFile);
|
|
return -1;
|
|
}
|
|
|
|
char bufTmp[256], tgaName[256];
|
|
string sTGAname;
|
|
float uvMinU, uvMinV, uvMaxU, uvMaxV;
|
|
while (!iFile.eof())
|
|
{
|
|
iFile.getline (bufTmp, 256);
|
|
sscanf (bufTmp, "%s %f %f %f %f", tgaName, &uvMinU, &uvMinV, &uvMaxU, &uvMaxV);
|
|
|
|
sTGAname = tgaName;
|
|
sTGAname = strlwr(sTGAname);
|
|
string findTGAName;
|
|
for (i = 0; i < mapSize; ++i)
|
|
{
|
|
// get the string whitout path
|
|
findTGAName = CFile::getFilename(AllMapNames[i]);
|
|
findTGAName = strlwr(findTGAName);
|
|
if( findTGAName == sTGAname )
|
|
break;
|
|
}
|
|
|
|
if( i == mapSize )
|
|
{
|
|
// not present in subset: offset existing uv's to (0,0), preserving size
|
|
fprintf (f, "%s %.12f %.12f %.12f %.12f\n", sTGAname.c_str(), 0.0f, 0.0f, uvMaxU - uvMinU, uvMaxV - uvMinV);
|
|
}
|
|
else
|
|
{
|
|
// present in subset: use new uv's
|
|
fprintf (f, "%s %.12f %.12f %.12f %.12f\n", sTGAname.c_str(), UVMin[i].U, UVMin[i].V,
|
|
UVMax[i].U, UVMax[i].V);
|
|
}
|
|
}
|
|
// fclose (iFile);
|
|
fclose (f);
|
|
outString (string("Writing UV file : ") + fmtName + "\n");
|
|
}
|
|
|
|
return 0;
|
|
}
|