5fbeb8d34b
Added support for RGBA pixel format dithering. Fixes issue 50 and issue 51. Add PixelFormat.h to vc8 project.
462 lines
11 KiB
C++
462 lines
11 KiB
C++
// Copyright NVIDIA Corporation 2007 -- Ignacio Castano <icastano@nvidia.com>
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person
|
|
// obtaining a copy of this software and associated documentation
|
|
// files (the "Software"), to deal in the Software without
|
|
// restriction, including without limitation the rights to use,
|
|
// copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the
|
|
// Software is furnished to do so, subject to the following
|
|
// conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be
|
|
// included in all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
|
// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
|
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
|
// OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
#include <nvcore/StrLib.h>
|
|
#include <nvcore/StdStream.h>
|
|
|
|
#include <nvimage/Image.h>
|
|
#include <nvimage/DirectDrawSurface.h>
|
|
|
|
#include <nvtt/nvtt.h>
|
|
|
|
#include "cmdline.h"
|
|
|
|
#include <time.h> // clock
|
|
|
|
//#define WINDOWS_LEAN_AND_MEAN
|
|
//#include <windows.h> // TIMER
|
|
|
|
|
|
struct MyOutputHandler : public nvtt::OutputHandler
|
|
{
|
|
MyOutputHandler(const char * name) : total(0), progress(0), percentage(0), stream(new nv::StdOutputStream(name)) {}
|
|
virtual ~MyOutputHandler() { delete stream; }
|
|
|
|
void setTotal(int64 t)
|
|
{
|
|
total = t + 128;
|
|
}
|
|
void setDisplayProgress(bool b)
|
|
{
|
|
verbose = b;
|
|
}
|
|
|
|
virtual void beginImage(int size, int width, int height, int depth, int face, int miplevel)
|
|
{
|
|
// ignore.
|
|
}
|
|
|
|
// Output data.
|
|
virtual bool writeData(const void * data, int size)
|
|
{
|
|
nvDebugCheck(stream != NULL);
|
|
stream->serialize(const_cast<void *>(data), size);
|
|
|
|
progress += size;
|
|
int p = int((100 * progress) / total);
|
|
if (verbose && p != percentage)
|
|
{
|
|
nvCheck(p >= 0);
|
|
|
|
percentage = p;
|
|
printf("\r%d%%", percentage);
|
|
fflush(stdout);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
int64 total;
|
|
int64 progress;
|
|
int percentage;
|
|
bool verbose;
|
|
nv::StdOutputStream * stream;
|
|
};
|
|
|
|
struct MyErrorHandler : public nvtt::ErrorHandler
|
|
{
|
|
virtual void error(nvtt::Error e)
|
|
{
|
|
nvDebugBreak();
|
|
}
|
|
};
|
|
|
|
|
|
|
|
|
|
// Set color to normal map conversion options.
|
|
void setColorToNormalMap(nvtt::InputOptions & inputOptions)
|
|
{
|
|
inputOptions.setNormalMap(false);
|
|
inputOptions.setConvertToNormalMap(true);
|
|
inputOptions.setHeightEvaluation(1.0f/3.0f, 1.0f/3.0f, 1.0f/3.0f, 0.0f);
|
|
//inputOptions.setNormalFilter(1.0f, 0, 0, 0);
|
|
//inputOptions.setNormalFilter(0.0f, 0, 0, 1);
|
|
inputOptions.setGamma(1.0f, 1.0f);
|
|
inputOptions.setNormalizeMipmaps(true);
|
|
}
|
|
|
|
// Set options for normal maps.
|
|
void setNormalMap(nvtt::InputOptions & inputOptions)
|
|
{
|
|
inputOptions.setNormalMap(true);
|
|
inputOptions.setConvertToNormalMap(false);
|
|
inputOptions.setGamma(1.0f, 1.0f);
|
|
inputOptions.setNormalizeMipmaps(true);
|
|
}
|
|
|
|
// Set options for color maps.
|
|
void setColorMap(nvtt::InputOptions & inputOptions)
|
|
{
|
|
inputOptions.setNormalMap(false);
|
|
inputOptions.setConvertToNormalMap(false);
|
|
inputOptions.setGamma(2.2f, 2.2f);
|
|
inputOptions.setNormalizeMipmaps(false);
|
|
}
|
|
|
|
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
MyAssertHandler assertHandler;
|
|
MyMessageHandler messageHandler;
|
|
|
|
bool normal = false;
|
|
bool color2normal = false;
|
|
bool wrapRepeat = false;
|
|
bool noMipmaps = false;
|
|
bool fast = false;
|
|
bool nocuda = false;
|
|
bool silent = false;
|
|
bool bc1n = false;
|
|
nvtt::Format format = nvtt::Format_BC1;
|
|
|
|
const char * externalCompressor = NULL;
|
|
|
|
nv::Path input;
|
|
nv::Path output;
|
|
|
|
|
|
// Parse arguments.
|
|
for (int i = 1; i < argc; i++)
|
|
{
|
|
// Input options.
|
|
if (strcmp("-color", argv[i]) == 0)
|
|
{
|
|
}
|
|
else if (strcmp("-normal", argv[i]) == 0)
|
|
{
|
|
normal = true;
|
|
}
|
|
else if (strcmp("-tonormal", argv[i]) == 0)
|
|
{
|
|
color2normal = true;
|
|
}
|
|
else if (strcmp("-clamp", argv[i]) == 0)
|
|
{
|
|
}
|
|
else if (strcmp("-repeat", argv[i]) == 0)
|
|
{
|
|
wrapRepeat = true;
|
|
}
|
|
else if (strcmp("-nomips", argv[i]) == 0)
|
|
{
|
|
noMipmaps = true;
|
|
}
|
|
|
|
// Compression options.
|
|
else if (strcmp("-fast", argv[i]) == 0)
|
|
{
|
|
fast = true;
|
|
}
|
|
else if (strcmp("-nocuda", argv[i]) == 0)
|
|
{
|
|
nocuda = true;
|
|
}
|
|
else if (strcmp("-rgb", argv[i]) == 0)
|
|
{
|
|
format = nvtt::Format_RGB;
|
|
}
|
|
else if (strcmp("-bc1", argv[i]) == 0)
|
|
{
|
|
format = nvtt::Format_BC1;
|
|
}
|
|
else if (strcmp("-bc1n", argv[i]) == 0)
|
|
{
|
|
format = nvtt::Format_BC1;
|
|
bc1n = true;
|
|
}
|
|
else if (strcmp("-bc1a", argv[i]) == 0)
|
|
{
|
|
format = nvtt::Format_BC1a;
|
|
}
|
|
else if (strcmp("-bc2", argv[i]) == 0)
|
|
{
|
|
format = nvtt::Format_BC2;
|
|
}
|
|
else if (strcmp("-bc3", argv[i]) == 0)
|
|
{
|
|
format = nvtt::Format_BC3;
|
|
}
|
|
else if (strcmp("-bc3n", argv[i]) == 0)
|
|
{
|
|
format = nvtt::Format_BC3n;
|
|
}
|
|
else if (strcmp("-bc4", argv[i]) == 0)
|
|
{
|
|
format = nvtt::Format_BC4;
|
|
}
|
|
else if (strcmp("-bc5", argv[i]) == 0)
|
|
{
|
|
format = nvtt::Format_BC5;
|
|
}
|
|
|
|
// Undocumented option. Mainly used for testing.
|
|
else if (strcmp("-ext", argv[i]) == 0)
|
|
{
|
|
if (i+1 < argc && argv[i+1][0] != '-') {
|
|
externalCompressor = argv[i+1];
|
|
i++;
|
|
}
|
|
}
|
|
|
|
// Misc options
|
|
else if (strcmp("-silent", argv[i]) == 0)
|
|
{
|
|
silent = true;
|
|
}
|
|
|
|
else if (argv[i][0] != '-')
|
|
{
|
|
input = argv[i];
|
|
|
|
if (i+1 < argc && argv[i+1][0] != '-') {
|
|
output = argv[i+1];
|
|
}
|
|
else
|
|
{
|
|
output.copy(input.str());
|
|
output.stripExtension();
|
|
output.append(".dds");
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
printf("NVIDIA Texture Tools - Copyright NVIDIA Corporation 2007\n\n");
|
|
|
|
if (input.isNull())
|
|
{
|
|
printf("usage: nvcompress [options] infile [outfile]\n\n");
|
|
|
|
printf("Input options:\n");
|
|
printf(" -color \tThe input image is a color map (default).\n");
|
|
printf(" -normal \tThe input image is a normal map.\n");
|
|
printf(" -tonormal\tConvert input to normal map.\n");
|
|
printf(" -clamp \tClamp wrapping mode (default).\n");
|
|
printf(" -repeat \tRepeat wrapping mode.\n");
|
|
printf(" -nomips \tDisable mipmap generation.\n\n");
|
|
|
|
printf("Compression options:\n");
|
|
printf(" -fast \tFast compression.\n");
|
|
printf(" -nocuda \tDo not use cuda compressor.\n");
|
|
printf(" -rgb \tRGBA format\n");
|
|
printf(" -bc1 \tBC1 format (DXT1)\n");
|
|
printf(" -bc1n \tBC1 normal map format (DXT1nm)\n");
|
|
printf(" -bc1a \tBC1 format with binary alpha (DXT1a)\n");
|
|
printf(" -bc2 \tBC2 format (DXT3)\n");
|
|
printf(" -bc3 \tBC3 format (DXT5)\n");
|
|
printf(" -bc3n \tBC3 normal map format (DXT5nm)\n");
|
|
printf(" -bc4 \tBC4 format (ATI1)\n");
|
|
printf(" -bc5 \tBC5 format (3Dc/ATI2)\n\n");
|
|
|
|
return 1;
|
|
}
|
|
|
|
// @@ Make sure input file exists.
|
|
|
|
// Set input options.
|
|
nvtt::InputOptions inputOptions;
|
|
|
|
if (nv::strCaseCmp(input.extension(), ".dds") == 0)
|
|
{
|
|
// Load surface.
|
|
nv::DirectDrawSurface dds(input);
|
|
if (!dds.isValid())
|
|
{
|
|
fprintf(stderr, "The file '%s' is not a valid DDS file.\n", input.str());
|
|
return 1;
|
|
}
|
|
|
|
if (!dds.isSupported() || dds.isTexture3D())
|
|
{
|
|
fprintf(stderr, "The file '%s' is not a supported DDS file.\n", input.str());
|
|
return 1;
|
|
}
|
|
|
|
uint faceCount;
|
|
if (dds.isTexture2D())
|
|
{
|
|
inputOptions.setTextureLayout(nvtt::TextureType_2D, dds.width(), dds.height());
|
|
faceCount = 1;
|
|
}
|
|
else
|
|
{
|
|
nvDebugCheck(dds.isTextureCube());
|
|
inputOptions.setTextureLayout(nvtt::TextureType_Cube, dds.width(), dds.height());
|
|
faceCount = 6;
|
|
}
|
|
|
|
uint mipmapCount = dds.mipmapCount();
|
|
|
|
nv::Image mipmap;
|
|
|
|
for (uint f = 0; f < faceCount; f++)
|
|
{
|
|
for (uint m = 0; m < mipmapCount; m++)
|
|
{
|
|
dds.mipmap(&mipmap, f, m);
|
|
|
|
inputOptions.setMipmapData(mipmap.pixels(), mipmap.width(), mipmap.height(), 1, f, m);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Regular image.
|
|
nv::Image image;
|
|
if (!image.load(input))
|
|
{
|
|
fprintf(stderr, "The file '%s' is not a supported image type.\n", input.str());
|
|
return 1;
|
|
}
|
|
|
|
inputOptions.setTextureLayout(nvtt::TextureType_2D, image.width(), image.height());
|
|
inputOptions.setMipmapData(image.pixels(), image.width(), image.height());
|
|
}
|
|
|
|
if (wrapRepeat)
|
|
{
|
|
inputOptions.setWrapMode(nvtt::WrapMode_Repeat);
|
|
}
|
|
else
|
|
{
|
|
inputOptions.setWrapMode(nvtt::WrapMode_Clamp);
|
|
}
|
|
|
|
if (normal)
|
|
{
|
|
setNormalMap(inputOptions);
|
|
}
|
|
else if (color2normal)
|
|
{
|
|
setColorToNormalMap(inputOptions);
|
|
}
|
|
else
|
|
{
|
|
setColorMap(inputOptions);
|
|
}
|
|
|
|
if (noMipmaps)
|
|
{
|
|
inputOptions.setMipmapGeneration(false);
|
|
}
|
|
|
|
nvtt::CompressionOptions compressionOptions;
|
|
compressionOptions.setFormat(format);
|
|
if (fast)
|
|
{
|
|
compressionOptions.setQuality(nvtt::Quality_Fastest);
|
|
}
|
|
else
|
|
{
|
|
compressionOptions.setQuality(nvtt::Quality_Normal);
|
|
//compressionOptions.setQuality(nvtt::Quality_Production);
|
|
//compressionOptions.setQuality(nvtt::Quality_Highest);
|
|
}
|
|
|
|
if (bc1n)
|
|
{
|
|
compressionOptions.setColorWeights(1, 1, 0);
|
|
}
|
|
|
|
if (externalCompressor != NULL)
|
|
{
|
|
compressionOptions.setExternalCompressor(externalCompressor);
|
|
}
|
|
|
|
if (format == nvtt::Format_RGB)
|
|
{
|
|
compressionOptions.setQuantization(true, false, false);
|
|
//compressionOptions.setPixelFormat(16, 0xF000, 0x0F00, 0x00F0, 0x000F);
|
|
compressionOptions.setPixelFormat(16,
|
|
0x0F00,
|
|
0x00F0,
|
|
0x000F,
|
|
0xF000);
|
|
// 0x003F0000,
|
|
// 0x00003F00,
|
|
// 0x0000003F,
|
|
// 0x3F000000);
|
|
}
|
|
|
|
|
|
MyErrorHandler errorHandler;
|
|
MyOutputHandler outputHandler(output);
|
|
if (outputHandler.stream->isError())
|
|
{
|
|
fprintf(stderr, "Error opening '%s' for writting\n", output.str());
|
|
return 1;
|
|
}
|
|
|
|
nvtt::Compressor compressor;
|
|
compressor.enableCudaAcceleration(!nocuda);
|
|
|
|
outputHandler.setTotal(compressor.estimateSize(inputOptions, compressionOptions));
|
|
outputHandler.setDisplayProgress(!silent);
|
|
|
|
nvtt::OutputOptions outputOptions;
|
|
//outputOptions.setFileName(output);
|
|
outputOptions.setOutputHandler(&outputHandler);
|
|
outputOptions.setErrorHandler(&errorHandler);
|
|
|
|
// printf("Press ENTER.\n");
|
|
// fflush(stdout);
|
|
// getchar();
|
|
|
|
/* LARGE_INTEGER temp;
|
|
QueryPerformanceFrequency((LARGE_INTEGER*) &temp);
|
|
double freq = ((double) temp.QuadPart) / 1000.0;
|
|
|
|
LARGE_INTEGER start_time;
|
|
QueryPerformanceCounter((LARGE_INTEGER*) &start_time);
|
|
*/
|
|
clock_t start = clock();
|
|
|
|
compressor.process(inputOptions, compressionOptions, outputOptions);
|
|
/*
|
|
LARGE_INTEGER end_time;
|
|
QueryPerformanceCounter((LARGE_INTEGER*) &end_time);
|
|
|
|
float diff_time = (float) (((double) end_time.QuadPart - (double) start_time.QuadPart) / freq);
|
|
printf("\rtime taken: %.3f seconds\n", diff_time/1000);
|
|
*/
|
|
|
|
clock_t end = clock();
|
|
printf("\rtime taken: %.3f seconds\n", float(end-start) / CLOCKS_PER_SEC);
|
|
|
|
return 0;
|
|
}
|
|
|