Compare commits
2 Commits
Author | SHA1 | Date | |
---|---|---|---|
92f730457c | |||
90be3fa28d |
@ -1,4 +1,4 @@
|
|||||||
CMAKE_MINIMUM_REQUIRED(VERSION 2.6.0)
|
CMAKE_MINIMUM_REQUIRED(VERSION 2.4.0)
|
||||||
PROJECT(NV)
|
PROJECT(NV)
|
||||||
ENABLE_TESTING()
|
ENABLE_TESTING()
|
||||||
|
|
||||||
@ -16,13 +16,6 @@ MESSAGE(STATUS "Setting optimal options")
|
|||||||
MESSAGE(STATUS " Processor: ${NV_SYSTEM_PROCESSOR}")
|
MESSAGE(STATUS " Processor: ${NV_SYSTEM_PROCESSOR}")
|
||||||
MESSAGE(STATUS " Compiler Flags: ${CMAKE_CXX_FLAGS}")
|
MESSAGE(STATUS " Compiler Flags: ${CMAKE_CXX_FLAGS}")
|
||||||
|
|
||||||
IF(NVTT_SHARED)
|
|
||||||
SET(NVCORE_SHARED TRUE)
|
|
||||||
SET(NVMATH_SHARED TRUE)
|
|
||||||
SET(NVIMAGE_SHARED TRUE)
|
|
||||||
ENDIF(NVTT_SHARED)
|
|
||||||
|
|
||||||
|
|
||||||
ADD_SUBDIRECTORY(src)
|
ADD_SUBDIRECTORY(src)
|
||||||
|
|
||||||
IF(WIN32)
|
IF(WIN32)
|
||||||
|
27
ChangeLog
27
ChangeLog
@ -1,30 +1,3 @@
|
|||||||
NVIDIA Texture Tools version 2.0.6
|
|
||||||
* Fix dll version checking.
|
|
||||||
* Detect CUDA 2.1 and future CUDA versions correctly.
|
|
||||||
* Print CUDA detection message in nvcompress.
|
|
||||||
* Select the fastest CUDA device.
|
|
||||||
* Compile squish with -fPIC. Fixes issue 74.
|
|
||||||
* Fix warnings under gcc 4.3.2.
|
|
||||||
* Fix nvzoom option typo by Frank Richter. Fixes issue 81.
|
|
||||||
* Do not use CUDA to compress small mipmaps. Fixes issue 76.
|
|
||||||
* Compute mipmaps of semi-transparent images correctly.
|
|
||||||
* Shutdown CUDA properly. Fixes issue 83.
|
|
||||||
* Fix pixel format converions. Fixes issue 87.
|
|
||||||
* Update single color compression tables. Fixes issue 85.
|
|
||||||
|
|
||||||
NVIDIA Texture Tools version 2.0.5
|
|
||||||
* Fix error in single color compressor. Fixes issue 66.
|
|
||||||
* Detect mismatch between CUDA runtime and driver, and disable CUDA in that case.
|
|
||||||
* Fix cmake files when compiling NVTT as a shared library.
|
|
||||||
* When linking nvtt dynamically on unix, link all libraries dynamically.
|
|
||||||
* Select fastest CUDA device.
|
|
||||||
|
|
||||||
NVIDIA Texture Tools version 2.0.4
|
|
||||||
* Fix error in RGB format output; reported by jonsoh. See issue 49.
|
|
||||||
* Added support RGB format dithering by jonsoh. Fixes issue 50 and 51.
|
|
||||||
* Prevent infinite loop in indexMirror when width equal 1. Fixes issue 65.
|
|
||||||
* Implement general scale filter, including upsampling.
|
|
||||||
|
|
||||||
NVIDIA Texture Tools version 2.0.3
|
NVIDIA Texture Tools version 2.0.3
|
||||||
* More accurate DXT3 compressor. Fixes issue 38.
|
* More accurate DXT3 compressor. Fixes issue 38.
|
||||||
* Remove legacy compressors. Fix issue 34.
|
* Remove legacy compressors. Fix issue 34.
|
||||||
|
@ -57,7 +57,7 @@ MARK_AS_ADVANCED (CUDA_FOUND CUDA_COMPILER CUDA_RUNTIME_LIBRARY)
|
|||||||
|
|
||||||
|
|
||||||
#SET(CUDA_OPTIONS "-ncfe")
|
#SET(CUDA_OPTIONS "-ncfe")
|
||||||
SET(CUDA_OPTIONS "--host-compilation=C")
|
SET(CUDA_OPTIONS "")
|
||||||
|
|
||||||
IF (CUDA_EMULATION)
|
IF (CUDA_EMULATION)
|
||||||
SET (CUDA_OPTIONS "${CUDA_OPTIONS} -deviceemu")
|
SET (CUDA_OPTIONS "${CUDA_OPTIONS} -deviceemu")
|
||||||
|
2
configure
vendored
2
configure
vendored
@ -53,7 +53,7 @@ echo "-- Configuring nvidia-texture-tools "`cat VERSION`
|
|||||||
|
|
||||||
mkdir -p ./build
|
mkdir -p ./build
|
||||||
cd ./build
|
cd ./build
|
||||||
$CMAKE .. -DNVTT_SHARED=1 -DCMAKE_BUILD_TYPE=$build -DCMAKE_INSTALL_PREFIX=$prefix -G "Unix Makefiles" || exit 1
|
$CMAKE .. -DCMAKE_BUILD_TYPE=$build -DCMAKE_INSTALL_PREFIX=$prefix -G "Unix Makefiles" || exit 1
|
||||||
cd ..
|
cd ..
|
||||||
|
|
||||||
echo ""
|
echo ""
|
||||||
|
@ -281,10 +281,6 @@
|
|||||||
RelativePath="..\..\..\src\nvcore\Debug.cpp"
|
RelativePath="..\..\..\src\nvcore\Debug.cpp"
|
||||||
>
|
>
|
||||||
</File>
|
</File>
|
||||||
<File
|
|
||||||
RelativePath="..\..\..\src\nvcore\Library.cpp"
|
|
||||||
>
|
|
||||||
</File>
|
|
||||||
<File
|
<File
|
||||||
RelativePath="..\..\..\src\nvcore\Memory.cpp"
|
RelativePath="..\..\..\src\nvcore\Memory.cpp"
|
||||||
>
|
>
|
||||||
@ -319,10 +315,6 @@
|
|||||||
RelativePath="..\..\..\src\nvcore\DefsVcWin32.h"
|
RelativePath="..\..\..\src\nvcore\DefsVcWin32.h"
|
||||||
>
|
>
|
||||||
</File>
|
</File>
|
||||||
<File
|
|
||||||
RelativePath="..\..\..\src\nvcore\Library.h"
|
|
||||||
>
|
|
||||||
</File>
|
|
||||||
<File
|
<File
|
||||||
RelativePath="..\..\..\src\nvcore\Memory.h"
|
RelativePath="..\..\..\src\nvcore\Memory.h"
|
||||||
>
|
>
|
||||||
|
@ -355,10 +355,6 @@
|
|||||||
RelativePath="..\..\..\src\nvimage\nvimage.h"
|
RelativePath="..\..\..\src\nvimage\nvimage.h"
|
||||||
>
|
>
|
||||||
</File>
|
</File>
|
||||||
<File
|
|
||||||
RelativePath="..\..\..\src\nvimage\PixelFormat.h"
|
|
||||||
>
|
|
||||||
</File>
|
|
||||||
<File
|
<File
|
||||||
RelativePath="..\..\..\src\nvimage\PsdFile.h"
|
RelativePath="..\..\..\src\nvimage\PsdFile.h"
|
||||||
>
|
>
|
||||||
|
@ -53,8 +53,8 @@ END
|
|||||||
//
|
//
|
||||||
|
|
||||||
VS_VERSION_INFO VERSIONINFO
|
VS_VERSION_INFO VERSIONINFO
|
||||||
FILEVERSION 2,0,6,0
|
FILEVERSION 2,0,3,0
|
||||||
PRODUCTVERSION 2,0,6,0
|
PRODUCTVERSION 2,0,3,0
|
||||||
FILEFLAGSMASK 0x17L
|
FILEFLAGSMASK 0x17L
|
||||||
#ifdef _DEBUG
|
#ifdef _DEBUG
|
||||||
FILEFLAGS 0x1L
|
FILEFLAGS 0x1L
|
||||||
@ -71,12 +71,12 @@ BEGIN
|
|||||||
BEGIN
|
BEGIN
|
||||||
VALUE "CompanyName", "NVIDIA Corporation"
|
VALUE "CompanyName", "NVIDIA Corporation"
|
||||||
VALUE "FileDescription", "NVIDIA Texture Tools Dynamic Link Library"
|
VALUE "FileDescription", "NVIDIA Texture Tools Dynamic Link Library"
|
||||||
VALUE "FileVersion", "2, 0, 6, 0"
|
VALUE "FileVersion", "2, 0, 3, 0"
|
||||||
VALUE "InternalName", "nvtt"
|
VALUE "InternalName", "nvtt"
|
||||||
VALUE "LegalCopyright", "Copyright (C) 2007"
|
VALUE "LegalCopyright", "Copyright (C) 2007"
|
||||||
VALUE "OriginalFilename", "nvtt.dll"
|
VALUE "OriginalFilename", "nvtt.dll"
|
||||||
VALUE "ProductName", "NVIDIA Texture Tools Dynamic Link Library"
|
VALUE "ProductName", "NVIDIA Texture Tools Dynamic Link Library"
|
||||||
VALUE "ProductVersion", "2, 0, 6, 0"
|
VALUE "ProductVersion", "2, 0, 3, 0"
|
||||||
END
|
END
|
||||||
END
|
END
|
||||||
BLOCK "VarFileInfo"
|
BLOCK "VarFileInfo"
|
||||||
|
@ -19,19 +19,13 @@ SET(CORE_SRCS
|
|||||||
TextWriter.h
|
TextWriter.h
|
||||||
TextWriter.cpp
|
TextWriter.cpp
|
||||||
Radix.h
|
Radix.h
|
||||||
Radix.cpp
|
Radix.cpp)
|
||||||
Library.h
|
|
||||||
Library.cpp)
|
|
||||||
|
|
||||||
INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR})
|
INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR})
|
||||||
|
|
||||||
# targets
|
# targets
|
||||||
ADD_DEFINITIONS(-DNVCORE_EXPORTS)
|
ADD_DEFINITIONS(-DNVCORE_EXPORTS)
|
||||||
|
|
||||||
IF(UNIX)
|
|
||||||
SET(LIBS ${LIBS} ${CMAKE_DL_LIBS})
|
|
||||||
ENDIF(UNIX)
|
|
||||||
|
|
||||||
IF(NVCORE_SHARED)
|
IF(NVCORE_SHARED)
|
||||||
ADD_LIBRARY(nvcore SHARED ${CORE_SRCS})
|
ADD_LIBRARY(nvcore SHARED ${CORE_SRCS})
|
||||||
ELSE(NVCORE_SHARED)
|
ELSE(NVCORE_SHARED)
|
||||||
|
@ -824,13 +824,13 @@ namespace nv
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Number of entries in the hash.
|
/// Number of entries in the hash.
|
||||||
int size() const
|
int size()
|
||||||
{
|
{
|
||||||
return entry_count;
|
return entry_count;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Number of entries in the hash.
|
/// Number of entries in the hash.
|
||||||
int count() const
|
int count()
|
||||||
{
|
{
|
||||||
return size();
|
return size();
|
||||||
}
|
}
|
||||||
|
@ -28,7 +28,7 @@
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if NV_OS_LINUX && defined(HAVE_EXECINFO_H)
|
#if NV_OS_LINUX && defined(HAVE_EXECINFO_H)
|
||||||
# include <execinfo.h> // backtrace
|
# include <execinfo.h>
|
||||||
# if NV_CC_GNUC // defined(HAVE_CXXABI_H)
|
# if NV_CC_GNUC // defined(HAVE_CXXABI_H)
|
||||||
# include <cxxabi.h>
|
# include <cxxabi.h>
|
||||||
# endif
|
# endif
|
||||||
@ -39,13 +39,6 @@
|
|||||||
# include <sys/types.h>
|
# include <sys/types.h>
|
||||||
# include <sys/sysctl.h> // sysctl
|
# include <sys/sysctl.h> // sysctl
|
||||||
# include <ucontext.h>
|
# include <ucontext.h>
|
||||||
# undef HAVE_EXECINFO_H
|
|
||||||
# if defined(HAVE_EXECINFO_H) // only after OSX 10.5
|
|
||||||
# include <execinfo.h> // backtrace
|
|
||||||
# if NV_CC_GNUC // defined(HAVE_CXXABI_H)
|
|
||||||
# include <cxxabi.h>
|
|
||||||
# endif
|
|
||||||
# endif
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include <stdexcept> // std::runtime_error
|
#include <stdexcept> // std::runtime_error
|
||||||
@ -135,14 +128,6 @@ namespace
|
|||||||
|
|
||||||
#if defined(HAVE_EXECINFO_H) // NV_OS_LINUX
|
#if defined(HAVE_EXECINFO_H) // NV_OS_LINUX
|
||||||
|
|
||||||
static bool nvHasStackTrace() {
|
|
||||||
#if NV_OS_DARWIN
|
|
||||||
return backtrace != NULL;
|
|
||||||
#else
|
|
||||||
return true;
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
static void nvPrintStackTrace(void * trace[], int size, int start=0) {
|
static void nvPrintStackTrace(void * trace[], int size, int start=0) {
|
||||||
char ** string_array = backtrace_symbols(trace, size);
|
char ** string_array = backtrace_symbols(trace, size);
|
||||||
|
|
||||||
@ -181,36 +166,24 @@ namespace
|
|||||||
|
|
||||||
static void * callerAddress(void * secret)
|
static void * callerAddress(void * secret)
|
||||||
{
|
{
|
||||||
# if NV_OS_DARWIN
|
# if NV_OS_DARWIN && NV_CPU_PPC
|
||||||
# if defined(_STRUCT_MCONTEXT)
|
ucontext_t * ucp = (ucontext_t *)secret;
|
||||||
# if NV_CPU_PPC
|
return (void *) ucp->uc_mcontext->ss.srr0;
|
||||||
ucontext_t * ucp = (ucontext_t *)secret;
|
# elif NV_OS_DARWIN && NV_CPU_X86
|
||||||
return (void *) ucp->uc_mcontext->__ss.__srr0;
|
ucontext_t * ucp = (ucontext_t *)secret;
|
||||||
# elif NV_CPU_X86
|
return (void *) ucp->uc_mcontext->ss.eip;
|
||||||
ucontext_t * ucp = (ucontext_t *)secret;
|
# elif NV_CPU_X86_64
|
||||||
return (void *) ucp->uc_mcontext->__ss.__eip;
|
// #define REG_RIP REG_INDEX(rip) // seems to be 16
|
||||||
# endif
|
ucontext_t * ucp = (ucontext_t *)secret;
|
||||||
# else
|
return (void *)ucp->uc_mcontext.gregs[REG_RIP];
|
||||||
# if NV_CPU_PPC
|
# elif NV_CPU_X86
|
||||||
ucontext_t * ucp = (ucontext_t *)secret;
|
ucontext_t * ucp = (ucontext_t *)secret;
|
||||||
return (void *) ucp->uc_mcontext->ss.srr0;
|
return (void *)ucp->uc_mcontext.gregs[14/*REG_EIP*/];
|
||||||
# elif NV_CPU_X86
|
# elif NV_CPU_PPC
|
||||||
ucontext_t * ucp = (ucontext_t *)secret;
|
ucontext_t * ucp = (ucontext_t *)secret;
|
||||||
return (void *) ucp->uc_mcontext->ss.eip;
|
return (void *) ucp->uc_mcontext.regs->nip;
|
||||||
# endif
|
|
||||||
# endif
|
|
||||||
# else
|
# else
|
||||||
# if NV_CPU_X86_64
|
return NULL;
|
||||||
// #define REG_RIP REG_INDEX(rip) // seems to be 16
|
|
||||||
ucontext_t * ucp = (ucontext_t *)secret;
|
|
||||||
return (void *)ucp->uc_mcontext.gregs[REG_RIP];
|
|
||||||
# elif NV_CPU_X86
|
|
||||||
ucontext_t * ucp = (ucontext_t *)secret;
|
|
||||||
return (void *)ucp->uc_mcontext.gregs[14/*REG_EIP*/];
|
|
||||||
# elif NV_CPU_PPC
|
|
||||||
ucontext_t * ucp = (ucontext_t *)secret;
|
|
||||||
return (void *) ucp->uc_mcontext.regs->nip;
|
|
||||||
# endif
|
|
||||||
# endif
|
# endif
|
||||||
|
|
||||||
// How to obtain the instruction pointers in different platforms, from mlton's source code.
|
// How to obtain the instruction pointers in different platforms, from mlton's source code.
|
||||||
@ -255,18 +228,17 @@ namespace
|
|||||||
}
|
}
|
||||||
|
|
||||||
# if defined(HAVE_EXECINFO_H)
|
# if defined(HAVE_EXECINFO_H)
|
||||||
if (nvHasStackTrace()) // in case of weak linking
|
|
||||||
{
|
|
||||||
void * trace[64];
|
|
||||||
int size = backtrace(trace, 64);
|
|
||||||
|
|
||||||
if (pnt != NULL) {
|
void * trace[64];
|
||||||
// Overwrite sigaction with caller's address.
|
int size = backtrace(trace, 64);
|
||||||
trace[1] = pnt;
|
|
||||||
}
|
|
||||||
|
|
||||||
nvPrintStackTrace(trace, size, 1);
|
if (pnt != NULL) {
|
||||||
|
// Overwrite sigaction with caller's address.
|
||||||
|
trace[1] = pnt;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
nvPrintStackTrace(trace, size, 1);
|
||||||
|
|
||||||
# endif // defined(HAVE_EXECINFO_H)
|
# endif // defined(HAVE_EXECINFO_H)
|
||||||
|
|
||||||
exit(0);
|
exit(0);
|
||||||
@ -401,12 +373,9 @@ namespace
|
|||||||
# endif
|
# endif
|
||||||
|
|
||||||
# if defined(HAVE_EXECINFO_H)
|
# if defined(HAVE_EXECINFO_H)
|
||||||
if (nvHasStackTrace())
|
void * trace[64];
|
||||||
{
|
int size = backtrace(trace, 64);
|
||||||
void * trace[64];
|
nvPrintStackTrace(trace, size, 3);
|
||||||
int size = backtrace(trace, 64);
|
|
||||||
nvPrintStackTrace(trace, size, 2);
|
|
||||||
}
|
|
||||||
# endif
|
# endif
|
||||||
|
|
||||||
// Exit cleanly.
|
// Exit cleanly.
|
||||||
@ -453,12 +422,9 @@ void NV_CDECL nvDebug(const char *msg, ...)
|
|||||||
void debug::dumpInfo()
|
void debug::dumpInfo()
|
||||||
{
|
{
|
||||||
#if !NV_OS_WIN32 && defined(HAVE_SIGNAL_H) && defined(HAVE_EXECINFO_H)
|
#if !NV_OS_WIN32 && defined(HAVE_SIGNAL_H) && defined(HAVE_EXECINFO_H)
|
||||||
if (nvHasStackTrace())
|
void * trace[64];
|
||||||
{
|
int size = backtrace(trace, 64);
|
||||||
void * trace[64];
|
nvPrintStackTrace(trace, size, 1);
|
||||||
int size = backtrace(trace, 64);
|
|
||||||
nvPrintStackTrace(trace, size, 1);
|
|
||||||
}
|
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -2,7 +2,8 @@
|
|||||||
#error "Do not include this file directly."
|
#error "Do not include this file directly."
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include <stdint.h> // uint8_t, int8_t, ...
|
#include <stdlib.h> // uint8_t, int8_t, ...
|
||||||
|
|
||||||
|
|
||||||
// Function linkage
|
// Function linkage
|
||||||
#define DLL_IMPORT
|
#define DLL_IMPORT
|
||||||
|
@ -19,9 +19,7 @@
|
|||||||
|
|
||||||
// Set standard function names.
|
// Set standard function names.
|
||||||
#define snprintf _snprintf
|
#define snprintf _snprintf
|
||||||
#if _MSC_VER < 1500
|
#define vsnprintf _vsnprintf
|
||||||
# define vsnprintf _vsnprintf
|
|
||||||
#endif
|
|
||||||
#define vsscanf _vsscanf
|
#define vsscanf _vsscanf
|
||||||
#define chdir _chdir
|
#define chdir _chdir
|
||||||
#define getcwd _getcwd
|
#define getcwd _getcwd
|
||||||
@ -72,6 +70,8 @@ typedef uint32 uint;
|
|||||||
#pragma warning(disable : 4711) // function selected for automatic inlining
|
#pragma warning(disable : 4711) // function selected for automatic inlining
|
||||||
#pragma warning(disable : 4725) // Pentium fdiv bug
|
#pragma warning(disable : 4725) // Pentium fdiv bug
|
||||||
|
|
||||||
|
#pragma warning(disable : 4345) // behavior change: an object of POD type constructed with an initializer of the form () will be default-initialized
|
||||||
|
|
||||||
#pragma warning(disable : 4786) // Identifier was truncated and cannot be debugged.
|
#pragma warning(disable : 4786) // Identifier was truncated and cannot be debugged.
|
||||||
|
|
||||||
#pragma warning(disable : 4675) // resolved overload was found by argument-dependent lookup
|
#pragma warning(disable : 4675) // resolved overload was found by argument-dependent lookup
|
||||||
|
@ -1,41 +0,0 @@
|
|||||||
|
|
||||||
#include "Library.h"
|
|
||||||
#include "Debug.h"
|
|
||||||
|
|
||||||
#if NV_OS_WIN32
|
|
||||||
#define WIN32_LEAN_AND_MEAN
|
|
||||||
#define VC_EXTRALEAN
|
|
||||||
#include <windows.h>
|
|
||||||
#else
|
|
||||||
#include <dlfcn.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
void * nvLoadLibrary(const char * name)
|
|
||||||
{
|
|
||||||
#if NV_OS_WIN32
|
|
||||||
return (void *)LoadLibraryExA( name, NULL, 0 );
|
|
||||||
#else
|
|
||||||
return dlopen(name, RTLD_LAZY);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
void nvUnloadLibrary(void * handle)
|
|
||||||
{
|
|
||||||
nvDebugCheck(handle != NULL);
|
|
||||||
#if NV_OS_WIN32
|
|
||||||
FreeLibrary((HMODULE)handle);
|
|
||||||
#else
|
|
||||||
dlclose(handle);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
void * nvBindSymbol(void * handle, const char * symbol)
|
|
||||||
{
|
|
||||||
#if NV_OS_WIN32
|
|
||||||
return (void *)GetProcAddress((HMODULE)handle, symbol);
|
|
||||||
#else
|
|
||||||
return (void *)dlsym(handle, symbol);
|
|
||||||
#endif
|
|
||||||
}
|
|
@ -1,50 +0,0 @@
|
|||||||
// This code is in the public domain -- castano@gmail.com
|
|
||||||
|
|
||||||
#ifndef NV_CORE_LIBRARY_H
|
|
||||||
#define NV_CORE_LIBRARY_H
|
|
||||||
|
|
||||||
#include <nvcore/nvcore.h>
|
|
||||||
|
|
||||||
#if NV_OS_WIN32
|
|
||||||
#define LIBRARY_NAME(name) #name ".dll"
|
|
||||||
#elif NV_OS_DARWIN
|
|
||||||
#define NV_LIBRARY_NAME(name) "lib" #name ".dylib"
|
|
||||||
#else
|
|
||||||
#define NV_LIBRARY_NAME(name) "lib" #name ".so"
|
|
||||||
#endif
|
|
||||||
|
|
||||||
NVCORE_API void * nvLoadLibrary(const char * name);
|
|
||||||
NVCORE_API void nvUnloadLibrary(void * lib);
|
|
||||||
NVCORE_API void * nvBindSymbol(void * lib, const char * symbol);
|
|
||||||
|
|
||||||
class NVCORE_CLASS Library
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
Library(const char * name)
|
|
||||||
{
|
|
||||||
handle = nvLoadLibrary(name);
|
|
||||||
}
|
|
||||||
~Library()
|
|
||||||
{
|
|
||||||
if (isValid())
|
|
||||||
{
|
|
||||||
nvUnloadLibrary(handle);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
bool isValid() const
|
|
||||||
{
|
|
||||||
return handle != NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
void * bindSymbol(const char * symbol)
|
|
||||||
{
|
|
||||||
return nvBindSymbol(handle, symbol);
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
void * handle;
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
#endif // NV_CORE_LIBRARY_H
|
|
@ -24,7 +24,7 @@ __forceinline void nvPrefetch(const void * mem)
|
|||||||
#else // NV_CC_MSVC
|
#else // NV_CC_MSVC
|
||||||
|
|
||||||
// do nothing in other case.
|
// do nothing in other case.
|
||||||
#define nvPrefetch(ptr)
|
#define piPrefetch(ptr)
|
||||||
|
|
||||||
#endif // NV_CC_MSVC
|
#endif // NV_CC_MSVC
|
||||||
|
|
||||||
|
@ -43,11 +43,8 @@ public:
|
|||||||
|
|
||||||
/** Delete owned pointer and assign new one. */
|
/** Delete owned pointer and assign new one. */
|
||||||
void operator=( T * p ) {
|
void operator=( T * p ) {
|
||||||
if (p != m_ptr)
|
delete m_ptr;
|
||||||
{
|
m_ptr = p;
|
||||||
delete m_ptr;
|
|
||||||
m_ptr = p;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Member access. */
|
/** Member access. */
|
||||||
@ -252,14 +249,14 @@ public:
|
|||||||
/** -> operator. */
|
/** -> operator. */
|
||||||
BaseClass * operator -> () const
|
BaseClass * operator -> () const
|
||||||
{
|
{
|
||||||
nvCheck( m_ptr != NULL );
|
piCheck( m_ptr != NULL );
|
||||||
return m_ptr;
|
return m_ptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
/** * operator. */
|
/** * operator. */
|
||||||
BaseClass & operator*() const
|
BaseClass & operator*() const
|
||||||
{
|
{
|
||||||
nvCheck( m_ptr != NULL );
|
piCheck( m_ptr != NULL );
|
||||||
return *m_ptr;
|
return *m_ptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -47,25 +47,25 @@ public:
|
|||||||
|
|
||||||
/** @name Stream implementation. */
|
/** @name Stream implementation. */
|
||||||
//@{
|
//@{
|
||||||
virtual void seek( uint pos )
|
virtual void seek( int pos )
|
||||||
{
|
{
|
||||||
nvDebugCheck(m_fp != NULL);
|
nvDebugCheck(m_fp != NULL);
|
||||||
nvDebugCheck(pos < size());
|
nvDebugCheck(pos >= 0 && pos < size());
|
||||||
fseek(m_fp, pos, SEEK_SET);
|
fseek(m_fp, pos, SEEK_SET);
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual uint tell() const
|
virtual int tell() const
|
||||||
{
|
{
|
||||||
nvDebugCheck(m_fp != NULL);
|
nvDebugCheck(m_fp != NULL);
|
||||||
return ftell(m_fp);
|
return ftell(m_fp);
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual uint size() const
|
virtual int size() const
|
||||||
{
|
{
|
||||||
nvDebugCheck(m_fp != NULL);
|
nvDebugCheck(m_fp != NULL);
|
||||||
uint pos = ftell(m_fp);
|
int pos = ftell(m_fp);
|
||||||
fseek(m_fp, 0, SEEK_END);
|
fseek(m_fp, 0, SEEK_END);
|
||||||
uint end = ftell(m_fp);
|
int end = ftell(m_fp);
|
||||||
fseek(m_fp, pos, SEEK_SET);
|
fseek(m_fp, pos, SEEK_SET);
|
||||||
return end;
|
return end;
|
||||||
}
|
}
|
||||||
@ -117,11 +117,11 @@ public:
|
|||||||
/** @name Stream implementation. */
|
/** @name Stream implementation. */
|
||||||
//@{
|
//@{
|
||||||
/// Write data.
|
/// Write data.
|
||||||
virtual uint serialize( void * data, uint len )
|
virtual void serialize( void * data, int len )
|
||||||
{
|
{
|
||||||
nvDebugCheck(data != NULL);
|
nvDebugCheck(data != NULL);
|
||||||
nvDebugCheck(m_fp != NULL);
|
nvDebugCheck(m_fp != NULL);
|
||||||
return (uint)fwrite(data, 1, len, m_fp);
|
fwrite(data, len, 1, m_fp);
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual bool isLoading() const
|
virtual bool isLoading() const
|
||||||
@ -156,11 +156,11 @@ public:
|
|||||||
/** @name Stream implementation. */
|
/** @name Stream implementation. */
|
||||||
//@{
|
//@{
|
||||||
/// Read data.
|
/// Read data.
|
||||||
virtual uint serialize( void * data, uint len )
|
virtual void serialize( void * data, int len )
|
||||||
{
|
{
|
||||||
nvDebugCheck(data != NULL);
|
nvDebugCheck(data != NULL);
|
||||||
nvDebugCheck(m_fp != NULL);
|
nvDebugCheck(m_fp != NULL);
|
||||||
return (uint)fread(data, 1, len, m_fp);
|
fread(data, len, 1, m_fp);
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual bool isLoading() const
|
virtual bool isLoading() const
|
||||||
@ -184,40 +184,33 @@ class NVCORE_CLASS MemoryInputStream : public Stream
|
|||||||
public:
|
public:
|
||||||
|
|
||||||
/// Ctor.
|
/// Ctor.
|
||||||
MemoryInputStream( const uint8 * mem, uint size ) :
|
MemoryInputStream( const uint8 * mem, int size ) :
|
||||||
m_mem(mem), m_ptr(mem), m_size(size) { }
|
m_mem(mem), m_ptr(mem), m_size(size) { }
|
||||||
|
|
||||||
/** @name Stream implementation. */
|
/** @name Stream implementation. */
|
||||||
//@{
|
//@{
|
||||||
/// Read data.
|
/// Read data.
|
||||||
virtual uint serialize( void * data, uint len )
|
virtual void serialize( void * data, int len )
|
||||||
{
|
{
|
||||||
nvDebugCheck(data != NULL);
|
nvDebugCheck(data != NULL);
|
||||||
nvDebugCheck(!isError());
|
nvDebugCheck(!isError());
|
||||||
|
|
||||||
uint left = m_size - tell();
|
|
||||||
if (len > left) len = left;
|
|
||||||
|
|
||||||
memcpy( data, m_ptr, len );
|
memcpy( data, m_ptr, len );
|
||||||
m_ptr += len;
|
m_ptr += len;
|
||||||
|
|
||||||
return len;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual void seek( uint pos )
|
virtual void seek( int pos )
|
||||||
{
|
{
|
||||||
nvDebugCheck(!isError());
|
nvDebugCheck(!isError());
|
||||||
m_ptr = m_mem + pos;
|
m_ptr = m_mem + pos;
|
||||||
nvDebugCheck(!isError());
|
nvDebugCheck(!isError());
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual uint tell() const
|
virtual int tell() const
|
||||||
{
|
{
|
||||||
nvDebugCheck(m_ptr >= m_mem);
|
return int(m_ptr - m_mem);
|
||||||
return uint(m_ptr - m_mem);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual uint size() const
|
virtual int size() const
|
||||||
{
|
{
|
||||||
return m_size;
|
return m_size;
|
||||||
}
|
}
|
||||||
@ -259,7 +252,7 @@ private:
|
|||||||
|
|
||||||
const uint8 * m_mem;
|
const uint8 * m_mem;
|
||||||
const uint8 * m_ptr;
|
const uint8 * m_ptr;
|
||||||
uint m_size;
|
int m_size;
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -293,19 +286,17 @@ public:
|
|||||||
/** @name Stream implementation. */
|
/** @name Stream implementation. */
|
||||||
//@{
|
//@{
|
||||||
/// Read data.
|
/// Read data.
|
||||||
virtual uint serialize( void * data, uint len )
|
virtual void serialize( void * data, int len )
|
||||||
{
|
{
|
||||||
nvDebugCheck(data != NULL);
|
nvDebugCheck(data != NULL);
|
||||||
len = m_s->serialize( data, len );
|
m_s->serialize( data, len );
|
||||||
|
|
||||||
if( m_s->isError() ) {
|
if( m_s->isError() ) {
|
||||||
throw std::exception();
|
throw std::exception();
|
||||||
}
|
}
|
||||||
|
|
||||||
return len;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual void seek( uint pos )
|
virtual void seek( int pos )
|
||||||
{
|
{
|
||||||
m_s->seek( pos );
|
m_s->seek( pos );
|
||||||
|
|
||||||
@ -314,12 +305,12 @@ public:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual uint tell() const
|
virtual int tell() const
|
||||||
{
|
{
|
||||||
return m_s->tell();
|
return m_s->tell();
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual uint size() const
|
virtual int size() const
|
||||||
{
|
{
|
||||||
return m_s->size();
|
return m_s->size();
|
||||||
}
|
}
|
||||||
|
@ -209,11 +209,48 @@ StringBuilder::StringBuilder( const StringBuilder & s ) : m_size(0), m_str(NULL)
|
|||||||
}
|
}
|
||||||
|
|
||||||
/** Copy string. */
|
/** Copy string. */
|
||||||
StringBuilder::StringBuilder( const char * s ) : m_size(0), m_str(NULL)
|
StringBuilder::StringBuilder( const char * s )
|
||||||
{
|
{
|
||||||
copy(s);
|
copy(s);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/** Allocate and copy string. */
|
||||||
|
StringBuilder::StringBuilder( int size_hint, const StringBuilder & s) : m_size(size_hint), m_str(NULL)
|
||||||
|
{
|
||||||
|
nvDebugCheck(m_size > 0);
|
||||||
|
m_str = strAlloc(m_size);
|
||||||
|
copy(s);
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Allocate and format string. */
|
||||||
|
StringBuilder::StringBuilder( const char * fmt, ... ) : m_size(0), m_str(NULL)
|
||||||
|
{
|
||||||
|
nvDebugCheck(fmt != NULL);
|
||||||
|
va_list arg;
|
||||||
|
va_start( arg, fmt );
|
||||||
|
|
||||||
|
format( fmt, arg );
|
||||||
|
|
||||||
|
va_end( arg );
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Allocate and format string. */
|
||||||
|
StringBuilder::StringBuilder( int size_hint, const char * fmt, ... ) : m_size(size_hint), m_str(NULL)
|
||||||
|
{
|
||||||
|
nvDebugCheck(m_size > 0);
|
||||||
|
nvDebugCheck(fmt != NULL);
|
||||||
|
|
||||||
|
m_str = strAlloc(m_size);
|
||||||
|
|
||||||
|
va_list arg;
|
||||||
|
va_start( arg, fmt );
|
||||||
|
|
||||||
|
format( fmt, arg );
|
||||||
|
|
||||||
|
va_end( arg );
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
/** Delete the string. */
|
/** Delete the string. */
|
||||||
StringBuilder::~StringBuilder()
|
StringBuilder::~StringBuilder()
|
||||||
{
|
{
|
||||||
@ -241,7 +278,8 @@ StringBuilder & StringBuilder::format( const char * fmt, ... )
|
|||||||
/** Format a string safely. */
|
/** Format a string safely. */
|
||||||
StringBuilder & StringBuilder::format( const char * fmt, va_list arg )
|
StringBuilder & StringBuilder::format( const char * fmt, va_list arg )
|
||||||
{
|
{
|
||||||
nvDebugCheck(fmt != NULL);
|
nvCheck(fmt != NULL);
|
||||||
|
nvCheck(m_size >= 0);
|
||||||
|
|
||||||
if( m_size == 0 ) {
|
if( m_size == 0 ) {
|
||||||
m_size = 64;
|
m_size = 64;
|
||||||
@ -289,7 +327,8 @@ StringBuilder & StringBuilder::format( const char * fmt, va_list arg )
|
|||||||
/** Append a string. */
|
/** Append a string. */
|
||||||
StringBuilder & StringBuilder::append( const char * s )
|
StringBuilder & StringBuilder::append( const char * s )
|
||||||
{
|
{
|
||||||
nvDebugCheck(s != NULL);
|
nvCheck(s != NULL);
|
||||||
|
nvCheck(m_size >= 0);
|
||||||
|
|
||||||
const uint slen = uint(strlen( s ));
|
const uint slen = uint(strlen( s ));
|
||||||
|
|
||||||
@ -436,6 +475,31 @@ void StringBuilder::reset()
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
Path::Path(const char * fmt, ...)
|
||||||
|
{
|
||||||
|
nvDebugCheck( fmt != NULL );
|
||||||
|
|
||||||
|
va_list arg;
|
||||||
|
va_start( arg, fmt );
|
||||||
|
|
||||||
|
format( fmt, arg );
|
||||||
|
|
||||||
|
va_end( arg );
|
||||||
|
}
|
||||||
|
|
||||||
|
Path::Path(int size_hint, const char * fmt, ...) : StringBuilder(size_hint)
|
||||||
|
{
|
||||||
|
nvDebugCheck( fmt != NULL );
|
||||||
|
|
||||||
|
va_list arg;
|
||||||
|
va_start( arg, fmt );
|
||||||
|
|
||||||
|
format( fmt, arg );
|
||||||
|
|
||||||
|
va_end( arg );
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
/// Get the file name from a path.
|
/// Get the file name from a path.
|
||||||
const char * Path::fileName() const
|
const char * Path::fileName() const
|
||||||
{
|
{
|
||||||
|
@ -14,7 +14,7 @@ namespace nv
|
|||||||
|
|
||||||
uint strHash(const char * str, uint h) NV_PURE;
|
uint strHash(const char * str, uint h) NV_PURE;
|
||||||
|
|
||||||
/// String hash based on Bernstein's hash.
|
/// String hash vased on Bernstein's hash.
|
||||||
inline uint strHash(const char * data, uint h = 5381)
|
inline uint strHash(const char * data, uint h = 5381)
|
||||||
{
|
{
|
||||||
uint i;
|
uint i;
|
||||||
@ -47,6 +47,9 @@ namespace nv
|
|||||||
explicit StringBuilder( int size_hint );
|
explicit StringBuilder( int size_hint );
|
||||||
StringBuilder( const char * str );
|
StringBuilder( const char * str );
|
||||||
StringBuilder( const StringBuilder & );
|
StringBuilder( const StringBuilder & );
|
||||||
|
StringBuilder( int size_hint, const StringBuilder & );
|
||||||
|
StringBuilder( const char * format, ... ) __attribute__((format (printf, 2, 3)));
|
||||||
|
StringBuilder( int size_hint, const char * format, ... ) __attribute__((format (printf, 3, 4)));
|
||||||
|
|
||||||
~StringBuilder();
|
~StringBuilder();
|
||||||
|
|
||||||
@ -119,14 +122,16 @@ namespace nv
|
|||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
/// Path string. @@ This should be called PathBuilder.
|
/// Path string.
|
||||||
class NVCORE_CLASS Path : public StringBuilder
|
class NVCORE_CLASS Path : public StringBuilder
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
Path() : StringBuilder() {}
|
Path() : StringBuilder() {}
|
||||||
explicit Path(int size_hint) : StringBuilder(size_hint) {}
|
explicit Path(int size_hint) : StringBuilder(size_hint) {}
|
||||||
Path(const char * str) : StringBuilder(str) {}
|
Path(const StringBuilder & str) : StringBuilder(str) {}
|
||||||
Path(const Path & path) : StringBuilder(path) {}
|
Path(int size_hint, const StringBuilder & str) : StringBuilder(size_hint, str) {}
|
||||||
|
Path(const char * format, ...) __attribute__((format (printf, 2, 3)));
|
||||||
|
Path(int size_hint, const char * format, ...) __attribute__((format (printf, 3, 4)));
|
||||||
|
|
||||||
const char * fileName() const;
|
const char * fileName() const;
|
||||||
const char * extension() const;
|
const char * extension() const;
|
||||||
@ -208,12 +213,9 @@ namespace nv
|
|||||||
/// Implement value semantics.
|
/// Implement value semantics.
|
||||||
String & operator=( const String & str )
|
String & operator=( const String & str )
|
||||||
{
|
{
|
||||||
if (str.data != data)
|
release();
|
||||||
{
|
data = str.data;
|
||||||
release();
|
addRef();
|
||||||
data = str.data;
|
|
||||||
addRef();
|
|
||||||
}
|
|
||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -41,17 +41,17 @@ public:
|
|||||||
ByteOrder byteOrder() const { return m_byteOrder; }
|
ByteOrder byteOrder() const { return m_byteOrder; }
|
||||||
|
|
||||||
|
|
||||||
/// Serialize the given data.
|
/// Serialize the given data. @@ Should return bytes serialized?
|
||||||
virtual uint serialize( void * data, uint len ) = 0;
|
virtual void serialize( void * data, int len ) = 0;
|
||||||
|
|
||||||
/// Move to the given position in the archive.
|
/// Move to the given position in the archive.
|
||||||
virtual void seek( uint pos ) = 0;
|
virtual void seek( int pos ) = 0;
|
||||||
|
|
||||||
/// Return the current position in the archive.
|
/// Return the current position in the archive.
|
||||||
virtual uint tell() const = 0;
|
virtual int tell() const = 0;
|
||||||
|
|
||||||
/// Return the current size of the archive.
|
/// Return the current size of the archive.
|
||||||
virtual uint size() const = 0;
|
virtual int size() const = 0;
|
||||||
|
|
||||||
/// Determine if there has been any error.
|
/// Determine if there has been any error.
|
||||||
virtual bool isError() const = 0;
|
virtual bool isError() const = 0;
|
||||||
@ -136,13 +136,13 @@ public:
|
|||||||
protected:
|
protected:
|
||||||
|
|
||||||
/// Serialize in the stream byte order.
|
/// Serialize in the stream byte order.
|
||||||
Stream & byteOrderSerialize( void * v, uint len ) {
|
Stream & byteOrderSerialize( void * v, int len ) {
|
||||||
if( m_byteOrder == getSystemByteOrder() ) {
|
if( m_byteOrder == getSystemByteOrder() ) {
|
||||||
serialize( v, len );
|
serialize( v, len );
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
for( uint i = len; i > 0; i-- ) {
|
for( int i=len-1; i>=0; i-- ) {
|
||||||
serialize( (uint8 *)v + i - 1, 1 );
|
serialize( (uint8 *)v + i, 1 );
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return *this;
|
return *this;
|
||||||
|
@ -532,7 +532,7 @@ DDSHeader::DDSHeader()
|
|||||||
|
|
||||||
// Store version information on the reserved header attributes.
|
// Store version information on the reserved header attributes.
|
||||||
this->reserved[9] = MAKEFOURCC('N', 'V', 'T', 'T');
|
this->reserved[9] = MAKEFOURCC('N', 'V', 'T', 'T');
|
||||||
this->reserved[10] = (2 << 16) | (0 << 8) | (6); // major.minor.revision
|
this->reserved[10] = (2 << 16) | (0 << 8) | (3); // major.minor.revision
|
||||||
|
|
||||||
this->pf.size = 32;
|
this->pf.size = 32;
|
||||||
this->pf.flags = 0;
|
this->pf.flags = 0;
|
||||||
@ -989,10 +989,10 @@ void DirectDrawSurface::readLinearImage(Image * img)
|
|||||||
stream->serialize(&c, byteCount);
|
stream->serialize(&c, byteCount);
|
||||||
|
|
||||||
Color32 pixel(0, 0, 0, 0xFF);
|
Color32 pixel(0, 0, 0, 0xFF);
|
||||||
pixel.r = PixelFormat::convert((c & header.pf.rmask) >> rshift, rsize, 8);
|
pixel.r = PixelFormat::convert(c >> rshift, rsize, 8);
|
||||||
pixel.g = PixelFormat::convert((c & header.pf.gmask) >> gshift, gsize, 8);
|
pixel.g = PixelFormat::convert(c >> gshift, gsize, 8);
|
||||||
pixel.b = PixelFormat::convert((c & header.pf.bmask) >> bshift, bsize, 8);
|
pixel.b = PixelFormat::convert(c >> bshift, bsize, 8);
|
||||||
pixel.a = PixelFormat::convert((c & header.pf.amask) >> ashift, asize, 8);
|
pixel.a = PixelFormat::convert(c >> ashift, asize, 8);
|
||||||
|
|
||||||
img->pixel(x, y) = pixel;
|
img->pixel(x, y) = pixel;
|
||||||
}
|
}
|
||||||
|
@ -33,10 +33,11 @@
|
|||||||
* http://www.dspguide.com/ch16.htm
|
* http://www.dspguide.com/ch16.htm
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "Filter.h"
|
|
||||||
|
|
||||||
#include <nvmath/Vector.h> // Vector4
|
|
||||||
#include <nvcore/Containers.h> // swap
|
#include <nvcore/Containers.h> // swap
|
||||||
|
#include <nvmath/nvmath.h> // fabs
|
||||||
|
#include <nvmath/Vector.h> // Vector4
|
||||||
|
#include <nvimage/Filter.h>
|
||||||
|
|
||||||
using namespace nv;
|
using namespace nv;
|
||||||
|
|
||||||
@ -243,7 +244,7 @@ SincFilter::SincFilter(float w) : Filter(w) {}
|
|||||||
|
|
||||||
float SincFilter::evaluate(float x) const
|
float SincFilter::evaluate(float x) const
|
||||||
{
|
{
|
||||||
return sincf(PI * x);
|
return 0.0f;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -503,7 +504,7 @@ void Kernel2::initBlendedSobel(const Vector4 & scale)
|
|||||||
|
|
||||||
for (int i = 0; i < 7; i++) {
|
for (int i = 0; i < 7; i++) {
|
||||||
for (int e = 0; e < 7; e++) {
|
for (int e = 0; e < 7; e++) {
|
||||||
m_data[(i + 1) * 9 + e + 1] += elements[i * 7 + e] * scale.z();
|
m_data[i * 9 + e + 1] += elements[i * 7 + e] * scale.z();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -518,7 +519,7 @@ void Kernel2::initBlendedSobel(const Vector4 & scale)
|
|||||||
|
|
||||||
for (int i = 0; i < 5; i++) {
|
for (int i = 0; i < 5; i++) {
|
||||||
for (int e = 0; e < 5; e++) {
|
for (int e = 0; e < 5; e++) {
|
||||||
m_data[(i + 2) * 9 + e + 2] += elements[i * 5 + e] * scale.y();
|
m_data[i * 9 + e + 2] += elements[i * 5 + e] * scale.y();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -531,7 +532,7 @@ void Kernel2::initBlendedSobel(const Vector4 & scale)
|
|||||||
|
|
||||||
for (int i = 0; i < 3; i++) {
|
for (int i = 0; i < 3; i++) {
|
||||||
for (int e = 0; e < 3; e++) {
|
for (int e = 0; e < 3; e++) {
|
||||||
m_data[(i + 3) * 9 + e + 3] += elements[i * 3 + e] * scale.x();
|
m_data[i * 9 + e + 3] += elements[i * 3 + e] * scale.x();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -540,17 +541,12 @@ void Kernel2::initBlendedSobel(const Vector4 & scale)
|
|||||||
|
|
||||||
PolyphaseKernel::PolyphaseKernel(const Filter & f, uint srcLength, uint dstLength, int samples/*= 32*/)
|
PolyphaseKernel::PolyphaseKernel(const Filter & f, uint srcLength, uint dstLength, int samples/*= 32*/)
|
||||||
{
|
{
|
||||||
|
nvCheck(srcLength >= dstLength); // @@ Upsampling not implemented!
|
||||||
nvDebugCheck(samples > 0);
|
nvDebugCheck(samples > 0);
|
||||||
|
|
||||||
float scale = float(dstLength) / float(srcLength);
|
const float scale = float(dstLength) / float(srcLength);
|
||||||
const float iscale = 1.0f / scale;
|
const float iscale = 1.0f / scale;
|
||||||
|
|
||||||
if (scale > 1) {
|
|
||||||
// Upsampling.
|
|
||||||
samples = 1;
|
|
||||||
scale = 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
m_length = dstLength;
|
m_length = dstLength;
|
||||||
m_width = f.width() * iscale;
|
m_width = f.width() * iscale;
|
||||||
m_windowSize = (int)ceilf(m_width * 2) + 1;
|
m_windowSize = (int)ceilf(m_width * 2) + 1;
|
||||||
|
@ -11,16 +11,16 @@ namespace nv
|
|||||||
class Vector4;
|
class Vector4;
|
||||||
|
|
||||||
/// Base filter class.
|
/// Base filter class.
|
||||||
class NVIMAGE_CLASS Filter
|
class Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
Filter(float width);
|
NVIMAGE_API Filter(float width);
|
||||||
virtual ~Filter();
|
NVIMAGE_API virtual ~Filter();
|
||||||
|
|
||||||
float width() const { return m_width; }
|
NVIMAGE_API float width() const { return m_width; }
|
||||||
float sampleDelta(float x, float scale) const;
|
NVIMAGE_API float sampleDelta(float x, float scale) const;
|
||||||
float sampleBox(float x, float scale, int samples) const;
|
NVIMAGE_API float sampleBox(float x, float scale, int samples) const;
|
||||||
float sampleTriangle(float x, float scale, int samples) const;
|
NVIMAGE_API float sampleTriangle(float x, float scale, int samples) const;
|
||||||
|
|
||||||
virtual float evaluate(float x) const = 0;
|
virtual float evaluate(float x) const = 0;
|
||||||
|
|
||||||
@ -29,56 +29,56 @@ namespace nv
|
|||||||
};
|
};
|
||||||
|
|
||||||
// Box filter.
|
// Box filter.
|
||||||
class NVIMAGE_CLASS BoxFilter : public Filter
|
class BoxFilter : public Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
BoxFilter();
|
NVIMAGE_API BoxFilter();
|
||||||
BoxFilter(float width);
|
NVIMAGE_API BoxFilter(float width);
|
||||||
virtual float evaluate(float x) const;
|
NVIMAGE_API virtual float evaluate(float x) const;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Triangle (bilinear/tent) filter.
|
// Triangle (bilinear/tent) filter.
|
||||||
class NVIMAGE_CLASS TriangleFilter : public Filter
|
class TriangleFilter : public Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
TriangleFilter();
|
NVIMAGE_API TriangleFilter();
|
||||||
TriangleFilter(float width);
|
NVIMAGE_API TriangleFilter(float width);
|
||||||
virtual float evaluate(float x) const;
|
NVIMAGE_API virtual float evaluate(float x) const;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Quadratic (bell) filter.
|
// Quadratic (bell) filter.
|
||||||
class NVIMAGE_CLASS QuadraticFilter : public Filter
|
class QuadraticFilter : public Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
QuadraticFilter();
|
NVIMAGE_API QuadraticFilter();
|
||||||
virtual float evaluate(float x) const;
|
NVIMAGE_API virtual float evaluate(float x) const;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Cubic filter from Thatcher Ulrich.
|
// Cubic filter from Thatcher Ulrich.
|
||||||
class NVIMAGE_CLASS CubicFilter : public Filter
|
class CubicFilter : public Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
CubicFilter();
|
NVIMAGE_API CubicFilter();
|
||||||
virtual float evaluate(float x) const;
|
NVIMAGE_API virtual float evaluate(float x) const;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Cubic b-spline filter from Paul Heckbert.
|
// Cubic b-spline filter from Paul Heckbert.
|
||||||
class NVIMAGE_CLASS BSplineFilter : public Filter
|
class BSplineFilter : public Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
BSplineFilter();
|
NVIMAGE_API BSplineFilter();
|
||||||
virtual float evaluate(float x) const;
|
NVIMAGE_API virtual float evaluate(float x) const;
|
||||||
};
|
};
|
||||||
|
|
||||||
/// Mitchell & Netravali's two-param cubic
|
/// Mitchell & Netravali's two-param cubic
|
||||||
/// @see "Reconstruction Filters in Computer Graphics", SIGGRAPH 88
|
/// @see "Reconstruction Filters in Computer Graphics", SIGGRAPH 88
|
||||||
class NVIMAGE_CLASS MitchellFilter : public Filter
|
class MitchellFilter : public Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
MitchellFilter();
|
NVIMAGE_API MitchellFilter();
|
||||||
virtual float evaluate(float x) const;
|
NVIMAGE_API virtual float evaluate(float x) const;
|
||||||
|
|
||||||
void setParameters(float b, float c);
|
NVIMAGE_API void setParameters(float a, float b);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
float p0, p2, p3;
|
float p0, p2, p3;
|
||||||
@ -86,29 +86,29 @@ namespace nv
|
|||||||
};
|
};
|
||||||
|
|
||||||
// Lanczos3 filter.
|
// Lanczos3 filter.
|
||||||
class NVIMAGE_CLASS LanczosFilter : public Filter
|
class LanczosFilter : public Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
LanczosFilter();
|
NVIMAGE_API LanczosFilter();
|
||||||
virtual float evaluate(float x) const;
|
NVIMAGE_API virtual float evaluate(float x) const;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Sinc filter.
|
// Sinc filter.
|
||||||
class NVIMAGE_CLASS SincFilter : public Filter
|
class SincFilter : public Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
SincFilter(float w);
|
NVIMAGE_API SincFilter(float w);
|
||||||
virtual float evaluate(float x) const;
|
NVIMAGE_API virtual float evaluate(float x) const;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Kaiser filter.
|
// Kaiser filter.
|
||||||
class NVIMAGE_CLASS KaiserFilter : public Filter
|
class KaiserFilter : public Filter
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
KaiserFilter(float w);
|
NVIMAGE_API KaiserFilter(float w);
|
||||||
virtual float evaluate(float x) const;
|
NVIMAGE_API virtual float evaluate(float x) const;
|
||||||
|
|
||||||
void setParameters(float a, float stretch);
|
NVIMAGE_API void setParameters(float a, float stretch);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
float alpha;
|
float alpha;
|
||||||
@ -118,12 +118,12 @@ namespace nv
|
|||||||
|
|
||||||
|
|
||||||
/// A 1D kernel. Used to precompute filter weights.
|
/// A 1D kernel. Used to precompute filter weights.
|
||||||
class NVIMAGE_CLASS Kernel1
|
class Kernel1
|
||||||
{
|
{
|
||||||
NV_FORBID_COPY(Kernel1);
|
NV_FORBID_COPY(Kernel1);
|
||||||
public:
|
public:
|
||||||
Kernel1(const Filter & f, int iscale, int samples = 32);
|
NVIMAGE_API Kernel1(const Filter & f, int iscale, int samples = 32);
|
||||||
~Kernel1();
|
NVIMAGE_API ~Kernel1();
|
||||||
|
|
||||||
float valueAt(uint x) const {
|
float valueAt(uint x) const {
|
||||||
nvDebugCheck(x < (uint)m_windowSize);
|
nvDebugCheck(x < (uint)m_windowSize);
|
||||||
@ -138,7 +138,7 @@ namespace nv
|
|||||||
return m_width;
|
return m_width;
|
||||||
}
|
}
|
||||||
|
|
||||||
void debugPrint();
|
NVIMAGE_API void debugPrint();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
int m_windowSize;
|
int m_windowSize;
|
||||||
@ -148,15 +148,15 @@ namespace nv
|
|||||||
|
|
||||||
|
|
||||||
/// A 2D kernel.
|
/// A 2D kernel.
|
||||||
class NVIMAGE_CLASS Kernel2
|
class Kernel2
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
Kernel2(uint width);
|
NVIMAGE_API Kernel2(uint width);
|
||||||
Kernel2(const Kernel2 & k);
|
NVIMAGE_API Kernel2(const Kernel2 & k);
|
||||||
~Kernel2();
|
NVIMAGE_API ~Kernel2();
|
||||||
|
|
||||||
void normalize();
|
NVIMAGE_API void normalize();
|
||||||
void transpose();
|
NVIMAGE_API void transpose();
|
||||||
|
|
||||||
float valueAt(uint x, uint y) const {
|
float valueAt(uint x, uint y) const {
|
||||||
return m_data[y * m_windowSize + x];
|
return m_data[y * m_windowSize + x];
|
||||||
@ -166,12 +166,12 @@ namespace nv
|
|||||||
return m_windowSize;
|
return m_windowSize;
|
||||||
}
|
}
|
||||||
|
|
||||||
void initLaplacian();
|
NVIMAGE_API void initLaplacian();
|
||||||
void initEdgeDetection();
|
NVIMAGE_API void initEdgeDetection();
|
||||||
void initSobel();
|
NVIMAGE_API void initSobel();
|
||||||
void initPrewitt();
|
NVIMAGE_API void initPrewitt();
|
||||||
|
|
||||||
void initBlendedSobel(const Vector4 & scale);
|
NVIMAGE_API void initBlendedSobel(const Vector4 & scale);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
const uint m_windowSize;
|
const uint m_windowSize;
|
||||||
@ -180,12 +180,12 @@ namespace nv
|
|||||||
|
|
||||||
|
|
||||||
/// A 1D polyphase kernel
|
/// A 1D polyphase kernel
|
||||||
class NVIMAGE_CLASS PolyphaseKernel
|
class PolyphaseKernel
|
||||||
{
|
{
|
||||||
NV_FORBID_COPY(PolyphaseKernel);
|
NV_FORBID_COPY(PolyphaseKernel);
|
||||||
public:
|
public:
|
||||||
PolyphaseKernel(const Filter & f, uint srcLength, uint dstLength, int samples = 32);
|
NVIMAGE_API PolyphaseKernel(const Filter & f, uint srcLength, uint dstLength, int samples = 32);
|
||||||
~PolyphaseKernel();
|
NVIMAGE_API ~PolyphaseKernel();
|
||||||
|
|
||||||
int windowSize() const {
|
int windowSize() const {
|
||||||
return m_windowSize;
|
return m_windowSize;
|
||||||
@ -205,7 +205,7 @@ namespace nv
|
|||||||
return m_data[column * m_windowSize + x];
|
return m_data[column * m_windowSize + x];
|
||||||
}
|
}
|
||||||
|
|
||||||
void debugPrint() const;
|
NVIMAGE_API void debugPrint() const;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
int m_windowSize;
|
int m_windowSize;
|
||||||
|
@ -1,18 +1,16 @@
|
|||||||
// This code is in the public domain -- castanyo@yahoo.es
|
// This code is in the public domain -- castanyo@yahoo.es
|
||||||
|
|
||||||
|
#include <nvcore/Containers.h>
|
||||||
|
#include <nvcore/Ptr.h>
|
||||||
|
|
||||||
|
#include <nvmath/Color.h>
|
||||||
|
|
||||||
#include "FloatImage.h"
|
#include "FloatImage.h"
|
||||||
#include "Filter.h"
|
#include "Filter.h"
|
||||||
#include "Image.h"
|
#include "Image.h"
|
||||||
|
|
||||||
#include <nvmath/Color.h>
|
|
||||||
#include <nvmath/Matrix.h>
|
|
||||||
|
|
||||||
#include <nvcore/Containers.h>
|
|
||||||
#include <nvcore/Ptr.h>
|
|
||||||
|
|
||||||
#include <math.h>
|
#include <math.h>
|
||||||
|
|
||||||
|
|
||||||
using namespace nv;
|
using namespace nv;
|
||||||
|
|
||||||
namespace
|
namespace
|
||||||
@ -142,8 +140,7 @@ Image * FloatImage::createImageGammaCorrect(float gamma/*= 2.2f*/) const
|
|||||||
/// Allocate a 2d float image of the given format and the given extents.
|
/// Allocate a 2d float image of the given format and the given extents.
|
||||||
void FloatImage::allocate(uint c, uint w, uint h)
|
void FloatImage::allocate(uint c, uint w, uint h)
|
||||||
{
|
{
|
||||||
free();
|
nvCheck(m_mem == NULL);
|
||||||
|
|
||||||
m_width = w;
|
m_width = w;
|
||||||
m_height = h;
|
m_height = h;
|
||||||
m_componentNum = c;
|
m_componentNum = c;
|
||||||
@ -154,6 +151,7 @@ void FloatImage::allocate(uint c, uint w, uint h)
|
|||||||
/// Free the image, but don't clear the members.
|
/// Free the image, but don't clear the members.
|
||||||
void FloatImage::free()
|
void FloatImage::free()
|
||||||
{
|
{
|
||||||
|
nvCheck(m_mem != NULL);
|
||||||
nv::mem::free( reinterpret_cast<void *>(m_mem) );
|
nv::mem::free( reinterpret_cast<void *>(m_mem) );
|
||||||
m_mem = NULL;
|
m_mem = NULL;
|
||||||
}
|
}
|
||||||
@ -542,27 +540,73 @@ FloatImage * FloatImage::fastDownSample() const
|
|||||||
return dst_image.release();
|
return dst_image.release();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
/// Downsample applying a 1D kernel separately in each dimension.
|
||||||
|
FloatImage * FloatImage::downSample(const Kernel1 & kernel, WrapMode wm) const
|
||||||
|
{
|
||||||
|
const uint w = max(1, m_width / 2);
|
||||||
|
const uint h = max(1, m_height / 2);
|
||||||
|
|
||||||
|
return downSample(kernel, w, h, wm);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// Downsample applying a 1D kernel separately in each dimension.
|
||||||
|
FloatImage * FloatImage::downSample(const Kernel1 & kernel, uint w, uint h, WrapMode wm) const
|
||||||
|
{
|
||||||
|
nvCheck(!(kernel.windowSize() & 1)); // Make sure that kernel m_width is even.
|
||||||
|
|
||||||
|
AutoPtr<FloatImage> tmp_image( new FloatImage() );
|
||||||
|
tmp_image->allocate(m_componentNum, w, m_height);
|
||||||
|
|
||||||
|
AutoPtr<FloatImage> dst_image( new FloatImage() );
|
||||||
|
dst_image->allocate(m_componentNum, w, h);
|
||||||
|
|
||||||
|
const float xscale = float(m_width) / float(w);
|
||||||
|
const float yscale = float(m_height) / float(h);
|
||||||
|
|
||||||
|
for(uint c = 0; c < m_componentNum; c++) {
|
||||||
|
float * tmp_channel = tmp_image->channel(c);
|
||||||
|
|
||||||
|
for(uint y = 0; y < m_height; y++) {
|
||||||
|
for(uint x = 0; x < w; x++) {
|
||||||
|
|
||||||
|
float sum = this->applyKernelHorizontal(&kernel, uint(x*xscale), y, c, wm);
|
||||||
|
|
||||||
|
const uint tmp_index = tmp_image->index(x, y);
|
||||||
|
tmp_channel[tmp_index] = sum;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
float * dst_channel = dst_image->channel(c);
|
||||||
|
|
||||||
|
for(uint y = 0; y < h; y++) {
|
||||||
|
for(uint x = 0; x < w; x++) {
|
||||||
|
|
||||||
|
float sum = tmp_image->applyKernelVertical(&kernel, uint(x*xscale), uint(y*yscale), c, wm);
|
||||||
|
|
||||||
|
const uint dst_index = dst_image->index(x, y);
|
||||||
|
dst_channel[dst_index] = sum;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return dst_image.release();
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
/// Downsample applying a 1D kernel separately in each dimension.
|
/// Downsample applying a 1D kernel separately in each dimension.
|
||||||
FloatImage * FloatImage::downSample(const Filter & filter, WrapMode wm) const
|
FloatImage * FloatImage::downSample(const Filter & filter, WrapMode wm) const
|
||||||
{
|
{
|
||||||
const uint w = max(1, m_width / 2);
|
const uint w = max(1, m_width / 2);
|
||||||
const uint h = max(1, m_height / 2);
|
const uint h = max(1, m_height / 2);
|
||||||
|
|
||||||
return resize(filter, w, h, wm);
|
return downSample(filter, w, h, wm);
|
||||||
}
|
|
||||||
|
|
||||||
/// Downsample applying a 1D kernel separately in each dimension.
|
|
||||||
FloatImage * FloatImage::downSample(const Filter & filter, WrapMode wm, uint alpha) const
|
|
||||||
{
|
|
||||||
const uint w = max(1, m_width / 2);
|
|
||||||
const uint h = max(1, m_height / 2);
|
|
||||||
|
|
||||||
return resize(filter, w, h, wm, alpha);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// Downsample applying a 1D kernel separately in each dimension.
|
/// Downsample applying a 1D kernel separately in each dimension.
|
||||||
FloatImage * FloatImage::resize(const Filter & filter, uint w, uint h, WrapMode wm) const
|
FloatImage * FloatImage::downSample(const Filter & filter, uint w, uint h, WrapMode wm) const
|
||||||
{
|
{
|
||||||
// @@ Use monophase filters when frac(m_width / w) == 0
|
// @@ Use monophase filters when frac(m_width / w) == 0
|
||||||
|
|
||||||
@ -631,56 +675,10 @@ FloatImage * FloatImage::resize(const Filter & filter, uint w, uint h, WrapMode
|
|||||||
return dst_image.release();
|
return dst_image.release();
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Downsample applying a 1D kernel separately in each dimension.
|
|
||||||
FloatImage * FloatImage::resize(const Filter & filter, uint w, uint h, WrapMode wm, uint alpha) const
|
|
||||||
{
|
|
||||||
nvCheck(alpha < m_componentNum);
|
|
||||||
|
|
||||||
AutoPtr<FloatImage> tmp_image( new FloatImage() );
|
|
||||||
AutoPtr<FloatImage> dst_image( new FloatImage() );
|
|
||||||
|
|
||||||
PolyphaseKernel xkernel(filter, m_width, w, 32);
|
|
||||||
PolyphaseKernel ykernel(filter, m_height, h, 32);
|
|
||||||
|
|
||||||
{
|
|
||||||
tmp_image->allocate(m_componentNum, w, m_height);
|
|
||||||
dst_image->allocate(m_componentNum, w, h);
|
|
||||||
|
|
||||||
Array<float> tmp_column(h);
|
|
||||||
tmp_column.resize(h);
|
|
||||||
|
|
||||||
for (uint c = 0; c < m_componentNum; c++)
|
|
||||||
{
|
|
||||||
float * tmp_channel = tmp_image->channel(c);
|
|
||||||
|
|
||||||
for (uint y = 0; y < m_height; y++) {
|
|
||||||
this->applyKernelHorizontal(xkernel, y, c, alpha, wm, tmp_channel + y * w);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Process all channels before applying vertical kernel to make sure alpha has been computed.
|
|
||||||
|
|
||||||
for (uint c = 0; c < m_componentNum; c++)
|
|
||||||
{
|
|
||||||
float * dst_channel = dst_image->channel(c);
|
|
||||||
|
|
||||||
for (uint x = 0; x < w; x++) {
|
|
||||||
tmp_image->applyKernelVertical(ykernel, x, c, alpha, wm, tmp_column.unsecureBuffer());
|
|
||||||
|
|
||||||
for (uint y = 0; y < h; y++) {
|
|
||||||
dst_channel[y * w + x] = tmp_column[y];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return dst_image.release();
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/// Apply 2D kernel at the given coordinates and return result.
|
/// Apply 2D kernel at the given coordinates and return result.
|
||||||
float FloatImage::applyKernel(const Kernel2 * k, int x, int y, uint c, WrapMode wm) const
|
float FloatImage::applyKernel(const Kernel2 * k, int x, int y, int c, WrapMode wm) const
|
||||||
{
|
{
|
||||||
nvDebugCheck(k != NULL);
|
nvDebugCheck(k != NULL);
|
||||||
|
|
||||||
@ -709,7 +707,7 @@ float FloatImage::applyKernel(const Kernel2 * k, int x, int y, uint c, WrapMode
|
|||||||
|
|
||||||
|
|
||||||
/// Apply 1D vertical kernel at the given coordinates and return result.
|
/// Apply 1D vertical kernel at the given coordinates and return result.
|
||||||
float FloatImage::applyKernelVertical(const Kernel1 * k, int x, int y, uint c, WrapMode wm) const
|
float FloatImage::applyKernelVertical(const Kernel1 * k, int x, int y, int c, WrapMode wm) const
|
||||||
{
|
{
|
||||||
nvDebugCheck(k != NULL);
|
nvDebugCheck(k != NULL);
|
||||||
|
|
||||||
@ -731,7 +729,7 @@ float FloatImage::applyKernelVertical(const Kernel1 * k, int x, int y, uint c, W
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Apply 1D horizontal kernel at the given coordinates and return result.
|
/// Apply 1D horizontal kernel at the given coordinates and return result.
|
||||||
float FloatImage::applyKernelHorizontal(const Kernel1 * k, int x, int y, uint c, WrapMode wm) const
|
float FloatImage::applyKernelHorizontal(const Kernel1 * k, int x, int y, int c, WrapMode wm) const
|
||||||
{
|
{
|
||||||
nvDebugCheck(k != NULL);
|
nvDebugCheck(k != NULL);
|
||||||
|
|
||||||
@ -754,7 +752,7 @@ float FloatImage::applyKernelHorizontal(const Kernel1 * k, int x, int y, uint c,
|
|||||||
|
|
||||||
|
|
||||||
/// Apply 1D vertical kernel at the given coordinates and return result.
|
/// Apply 1D vertical kernel at the given coordinates and return result.
|
||||||
void FloatImage::applyKernelVertical(const PolyphaseKernel & k, int x, uint c, WrapMode wm, float * __restrict output) const
|
void FloatImage::applyKernelVertical(const PolyphaseKernel & k, int x, int c, WrapMode wm, float * output) const
|
||||||
{
|
{
|
||||||
const uint length = k.length();
|
const uint length = k.length();
|
||||||
const float scale = float(length) / float(m_height);
|
const float scale = float(length) / float(m_height);
|
||||||
@ -786,7 +784,7 @@ void FloatImage::applyKernelVertical(const PolyphaseKernel & k, int x, uint c, W
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Apply 1D horizontal kernel at the given coordinates and return result.
|
/// Apply 1D horizontal kernel at the given coordinates and return result.
|
||||||
void FloatImage::applyKernelHorizontal(const PolyphaseKernel & k, int y, uint c, WrapMode wm, float * __restrict output) const
|
void FloatImage::applyKernelHorizontal(const PolyphaseKernel & k, int y, int c, WrapMode wm, float * output) const
|
||||||
{
|
{
|
||||||
const uint length = k.length();
|
const uint length = k.length();
|
||||||
const float scale = float(length) / float(m_width);
|
const float scale = float(length) / float(m_width);
|
||||||
@ -817,93 +815,3 @@ void FloatImage::applyKernelHorizontal(const PolyphaseKernel & k, int y, uint c,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// Apply 1D vertical kernel at the given coordinates and return result.
|
|
||||||
void FloatImage::applyKernelVertical(const PolyphaseKernel & k, int x, uint c, uint a, WrapMode wm, float * __restrict output) const
|
|
||||||
{
|
|
||||||
const uint length = k.length();
|
|
||||||
const float scale = float(length) / float(m_height);
|
|
||||||
const float iscale = 1.0f / scale;
|
|
||||||
|
|
||||||
const float width = k.width();
|
|
||||||
const int windowSize = k.windowSize();
|
|
||||||
|
|
||||||
const float * channel = this->channel(c);
|
|
||||||
const float * alpha = this->channel(a);
|
|
||||||
|
|
||||||
for (uint i = 0; i < length; i++)
|
|
||||||
{
|
|
||||||
const float center = (0.5f + i) * iscale;
|
|
||||||
|
|
||||||
const int left = (int)floorf(center - width);
|
|
||||||
const int right = (int)ceilf(center + width);
|
|
||||||
nvCheck(right - left <= windowSize);
|
|
||||||
|
|
||||||
float norm = 0;
|
|
||||||
float sum = 0;
|
|
||||||
for (int j = 0; j < windowSize; ++j)
|
|
||||||
{
|
|
||||||
const int idx = this->index(x, j+left, wm);
|
|
||||||
|
|
||||||
float w = k.valueAt(i, j) * (alpha[idx] + (1.0f / 256.0f));
|
|
||||||
norm += w;
|
|
||||||
sum += w * channel[idx];
|
|
||||||
}
|
|
||||||
|
|
||||||
output[i] = sum / norm;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Apply 1D horizontal kernel at the given coordinates and return result.
|
|
||||||
void FloatImage::applyKernelHorizontal(const PolyphaseKernel & k, int y, uint c, uint a, WrapMode wm, float * __restrict output) const
|
|
||||||
{
|
|
||||||
const uint length = k.length();
|
|
||||||
const float scale = float(length) / float(m_width);
|
|
||||||
const float iscale = 1.0f / scale;
|
|
||||||
|
|
||||||
const float width = k.width();
|
|
||||||
const int windowSize = k.windowSize();
|
|
||||||
|
|
||||||
const float * channel = this->channel(c);
|
|
||||||
const float * alpha = this->channel(a);
|
|
||||||
|
|
||||||
for (uint i = 0; i < length; i++)
|
|
||||||
{
|
|
||||||
const float center = (0.5f + i) * iscale;
|
|
||||||
|
|
||||||
const int left = (int)floorf(center - width);
|
|
||||||
const int right = (int)ceilf(center + width);
|
|
||||||
nvDebugCheck(right - left <= windowSize);
|
|
||||||
|
|
||||||
float norm = 0.0f;
|
|
||||||
float sum = 0;
|
|
||||||
for (int j = 0; j < windowSize; ++j)
|
|
||||||
{
|
|
||||||
const int idx = this->index(left + j, y, wm);
|
|
||||||
|
|
||||||
float w = k.valueAt(i, j) * (alpha[idx] + (1.0f / 256.0f));
|
|
||||||
norm += w;
|
|
||||||
sum += w * channel[idx];
|
|
||||||
}
|
|
||||||
|
|
||||||
output[i] = sum / norm;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
FloatImage* FloatImage::clone() const
|
|
||||||
{
|
|
||||||
FloatImage* copy = new FloatImage();
|
|
||||||
copy->m_width = m_width;
|
|
||||||
copy->m_height = m_height;
|
|
||||||
copy->m_componentNum = m_componentNum;
|
|
||||||
copy->m_count = m_count;
|
|
||||||
|
|
||||||
if(m_mem)
|
|
||||||
{
|
|
||||||
copy->allocate(m_componentNum, m_width, m_height);
|
|
||||||
memcpy(copy->m_mem, m_mem, m_count * sizeof(float));
|
|
||||||
}
|
|
||||||
|
|
||||||
return copy;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
@ -3,20 +3,12 @@
|
|||||||
#ifndef NV_IMAGE_FLOATIMAGE_H
|
#ifndef NV_IMAGE_FLOATIMAGE_H
|
||||||
#define NV_IMAGE_FLOATIMAGE_H
|
#define NV_IMAGE_FLOATIMAGE_H
|
||||||
|
|
||||||
#include <nvimage/nvimage.h>
|
|
||||||
|
|
||||||
#include <nvmath/Vector.h>
|
|
||||||
|
|
||||||
#include <nvcore/Debug.h>
|
#include <nvcore/Debug.h>
|
||||||
#include <nvcore/Containers.h> // clamp
|
#include <nvcore/Containers.h> // clamp
|
||||||
|
#include <nvimage/nvimage.h>
|
||||||
#include <stdlib.h> // abs
|
|
||||||
|
|
||||||
|
|
||||||
namespace nv
|
namespace nv
|
||||||
{
|
{
|
||||||
class Vector4;
|
|
||||||
class Matrix;
|
|
||||||
class Image;
|
class Image;
|
||||||
class Filter;
|
class Filter;
|
||||||
class Kernel1;
|
class Kernel1;
|
||||||
@ -71,19 +63,17 @@ public:
|
|||||||
|
|
||||||
NVIMAGE_API FloatImage * fastDownSample() const;
|
NVIMAGE_API FloatImage * fastDownSample() const;
|
||||||
NVIMAGE_API FloatImage * downSample(const Filter & filter, WrapMode wm) const;
|
NVIMAGE_API FloatImage * downSample(const Filter & filter, WrapMode wm) const;
|
||||||
NVIMAGE_API FloatImage * downSample(const Filter & filter, WrapMode wm, uint alpha) const;
|
NVIMAGE_API FloatImage * downSample(const Filter & filter, uint w, uint h, WrapMode wm) const;
|
||||||
NVIMAGE_API FloatImage * resize(const Filter & filter, uint w, uint h, WrapMode wm) const;
|
|
||||||
|
|
||||||
NVIMAGE_API FloatImage * resize(const Filter & filter, uint w, uint h, WrapMode wm, uint alpha) const;
|
//NVIMAGE_API FloatImage * downSample(const Kernel1 & filter, WrapMode wm) const;
|
||||||
|
//NVIMAGE_API FloatImage * downSample(const Kernel1 & filter, uint w, uint h, WrapMode wm) const;
|
||||||
//@}
|
//@}
|
||||||
|
|
||||||
NVIMAGE_API float applyKernel(const Kernel2 * k, int x, int y, uint c, WrapMode wm) const;
|
NVIMAGE_API float applyKernel(const Kernel2 * k, int x, int y, int c, WrapMode wm) const;
|
||||||
NVIMAGE_API float applyKernelVertical(const Kernel1 * k, int x, int y, uint c, WrapMode wm) const;
|
NVIMAGE_API float applyKernelVertical(const Kernel1 * k, int x, int y, int c, WrapMode wm) const;
|
||||||
NVIMAGE_API float applyKernelHorizontal(const Kernel1 * k, int x, int y, uint c, WrapMode wm) const;
|
NVIMAGE_API float applyKernelHorizontal(const Kernel1 * k, int x, int y, int c, WrapMode wm) const;
|
||||||
NVIMAGE_API void applyKernelVertical(const PolyphaseKernel & k, int x, uint c, WrapMode wm, float * output) const;
|
NVIMAGE_API void applyKernelVertical(const PolyphaseKernel & k, int x, int c, WrapMode wm, float * output) const;
|
||||||
NVIMAGE_API void applyKernelHorizontal(const PolyphaseKernel & k, int y, uint c, WrapMode wm, float * output) const;
|
NVIMAGE_API void applyKernelHorizontal(const PolyphaseKernel & k, int y, int c, WrapMode wm, float * output) const;
|
||||||
NVIMAGE_API void applyKernelVertical(const PolyphaseKernel & k, int x, uint c, uint a, WrapMode wm, float * output) const;
|
|
||||||
NVIMAGE_API void applyKernelHorizontal(const PolyphaseKernel & k, int y, uint c, uint a, WrapMode wm, float * output) const;
|
|
||||||
|
|
||||||
|
|
||||||
uint width() const { return m_width; }
|
uint width() const { return m_width; }
|
||||||
@ -119,9 +109,6 @@ public:
|
|||||||
float sampleLinearMirror(float x, float y, int c) const;
|
float sampleLinearMirror(float x, float y, int c) const;
|
||||||
//@}
|
//@}
|
||||||
|
|
||||||
|
|
||||||
FloatImage* clone() const;
|
|
||||||
|
|
||||||
public:
|
public:
|
||||||
|
|
||||||
uint index(uint x, uint y) const;
|
uint index(uint x, uint y) const;
|
||||||
@ -239,15 +226,11 @@ inline uint FloatImage::indexRepeat(int x, int y) const
|
|||||||
|
|
||||||
inline uint FloatImage::indexMirror(int x, int y) const
|
inline uint FloatImage::indexMirror(int x, int y) const
|
||||||
{
|
{
|
||||||
if (m_width == 1) x = 0;
|
|
||||||
|
|
||||||
x = abs(x);
|
x = abs(x);
|
||||||
while (x >= m_width) {
|
while (x >= m_width) {
|
||||||
x = abs(m_width + m_width - x - 2);
|
x = abs(m_width + m_width - x - 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (m_height == 1) y = 0;
|
|
||||||
|
|
||||||
y = abs(y);
|
y = abs(y);
|
||||||
while (y >= m_height) {
|
while (y >= m_height) {
|
||||||
y = abs(m_height + m_height - y - 2);
|
y = abs(m_height + m_height - y - 2);
|
||||||
|
@ -21,16 +21,15 @@
|
|||||||
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
||||||
// OTHER DEALINGS IN THE SOFTWARE.
|
// OTHER DEALINGS IN THE SOFTWARE.
|
||||||
|
|
||||||
|
#include <nvcore/Ptr.h>
|
||||||
|
|
||||||
|
#include <nvmath/Color.h>
|
||||||
|
|
||||||
#include <nvimage/NormalMap.h>
|
#include <nvimage/NormalMap.h>
|
||||||
#include <nvimage/Filter.h>
|
#include <nvimage/Filter.h>
|
||||||
#include <nvimage/FloatImage.h>
|
#include <nvimage/FloatImage.h>
|
||||||
#include <nvimage/Image.h>
|
#include <nvimage/Image.h>
|
||||||
|
|
||||||
#include <nvmath/Color.h>
|
|
||||||
|
|
||||||
#include <nvcore/Ptr.h>
|
|
||||||
|
|
||||||
|
|
||||||
using namespace nv;
|
using namespace nv;
|
||||||
|
|
||||||
// Create normal map using the given kernels.
|
// Create normal map using the given kernels.
|
||||||
|
@ -39,7 +39,7 @@ namespace nv
|
|||||||
bool isSupported() const
|
bool isSupported() const
|
||||||
{
|
{
|
||||||
if (version != 1) {
|
if (version != 1) {
|
||||||
nvDebug("*** bad version number %u\n", version);
|
printf("*** bad version number %u\n", version);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
if (channel_count > 4) {
|
if (channel_count > 4) {
|
||||||
|
@ -12,14 +12,10 @@ http://www.efg2.com/Lab/Library/ImageProcessing/DHALF.TXT
|
|||||||
@@ This code needs to be reviewed, I'm not sure it's correct.
|
@@ This code needs to be reviewed, I'm not sure it's correct.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <nvimage/Quantize.h>
|
|
||||||
#include <nvimage/Image.h>
|
|
||||||
#include <nvimage/PixelFormat.h>
|
|
||||||
|
|
||||||
#include <nvmath/Color.h>
|
#include <nvmath/Color.h>
|
||||||
|
|
||||||
#include <nvcore/Containers.h> // swap
|
#include <nvimage/Image.h>
|
||||||
|
#include <nvimage/Quantize.h>
|
||||||
|
|
||||||
using namespace nv;
|
using namespace nv;
|
||||||
|
|
||||||
@ -51,20 +47,94 @@ void nv::Quantize::BinaryAlpha( Image * image, int alpha_threshold /*= 127*/ )
|
|||||||
// Simple quantization.
|
// Simple quantization.
|
||||||
void nv::Quantize::RGB16( Image * image )
|
void nv::Quantize::RGB16( Image * image )
|
||||||
{
|
{
|
||||||
Truncate(image, 5, 6, 5, 8);
|
nvCheck(image != NULL);
|
||||||
|
|
||||||
|
const uint w = image->width();
|
||||||
|
const uint h = image->height();
|
||||||
|
|
||||||
|
for(uint y = 0; y < h; y++) {
|
||||||
|
for(uint x = 0; x < w; x++) {
|
||||||
|
|
||||||
|
Color32 pixel32 = image->pixel(x, y);
|
||||||
|
|
||||||
|
// Convert to 16 bit and back to 32 using regular bit expansion.
|
||||||
|
Color32 pixel16 = toColor32( toColor16(pixel32) );
|
||||||
|
|
||||||
|
// Store color.
|
||||||
|
image->pixel(x, y) = pixel16;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Alpha quantization.
|
// Alpha quantization.
|
||||||
void nv::Quantize::Alpha4( Image * image )
|
void nv::Quantize::Alpha4( Image * image )
|
||||||
{
|
{
|
||||||
Truncate(image, 8, 8, 8, 4);
|
nvCheck(image != NULL);
|
||||||
|
|
||||||
|
const uint w = image->width();
|
||||||
|
const uint h = image->height();
|
||||||
|
|
||||||
|
for(uint y = 0; y < h; y++) {
|
||||||
|
for(uint x = 0; x < w; x++) {
|
||||||
|
|
||||||
|
Color32 pixel = image->pixel(x, y);
|
||||||
|
|
||||||
|
// Convert to 4 bit using regular bit expansion.
|
||||||
|
pixel.a = (pixel.a & 0xF0) | ((pixel.a & 0xF0) >> 4);
|
||||||
|
|
||||||
|
// Store color.
|
||||||
|
image->pixel(x, y) = pixel;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Error diffusion. Floyd Steinberg.
|
// Error diffusion. Floyd Steinberg.
|
||||||
void nv::Quantize::FloydSteinberg_RGB16( Image * image )
|
void nv::Quantize::FloydSteinberg_RGB16( Image * image )
|
||||||
{
|
{
|
||||||
FloydSteinberg(image, 5, 6, 5, 8);
|
nvCheck(image != NULL);
|
||||||
|
|
||||||
|
const uint w = image->width();
|
||||||
|
const uint h = image->height();
|
||||||
|
|
||||||
|
// @@ Use fixed point?
|
||||||
|
Vector3 * row0 = new Vector3[w+2];
|
||||||
|
Vector3 * row1 = new Vector3[w+2];
|
||||||
|
memset(row0, 0, sizeof(Vector3)*(w+2));
|
||||||
|
memset(row1, 0, sizeof(Vector3)*(w+2));
|
||||||
|
|
||||||
|
for(uint y = 0; y < h; y++) {
|
||||||
|
for(uint x = 0; x < w; x++) {
|
||||||
|
|
||||||
|
Color32 pixel32 = image->pixel(x, y);
|
||||||
|
|
||||||
|
// Add error. // @@ We shouldn't clamp here!
|
||||||
|
pixel32.r = clamp(int(pixel32.r) + int(row0[1+x].x()), 0, 255);
|
||||||
|
pixel32.g = clamp(int(pixel32.g) + int(row0[1+x].y()), 0, 255);
|
||||||
|
pixel32.b = clamp(int(pixel32.b) + int(row0[1+x].z()), 0, 255);
|
||||||
|
|
||||||
|
// Convert to 16 bit. @@ Use regular clamp?
|
||||||
|
Color32 pixel16 = toColor32( toColor16(pixel32) );
|
||||||
|
|
||||||
|
// Store color.
|
||||||
|
image->pixel(x, y) = pixel16;
|
||||||
|
|
||||||
|
// Compute new error.
|
||||||
|
Vector3 diff(float(pixel32.r - pixel16.r), float(pixel32.g - pixel16.g), float(pixel32.b - pixel16.b));
|
||||||
|
|
||||||
|
// Propagate new error.
|
||||||
|
row0[1+x+1] += 7.0f / 16.0f * diff;
|
||||||
|
row1[1+x-1] += 3.0f / 16.0f * diff;
|
||||||
|
row1[1+x+0] += 5.0f / 16.0f * diff;
|
||||||
|
row1[1+x+1] += 1.0f / 16.0f * diff;
|
||||||
|
}
|
||||||
|
|
||||||
|
swap(row0, row1);
|
||||||
|
memset(row1, 0, sizeof(Vector3)*(w+2));
|
||||||
|
}
|
||||||
|
|
||||||
|
delete [] row0;
|
||||||
|
delete [] row1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -118,90 +188,34 @@ void nv::Quantize::FloydSteinberg_BinaryAlpha( Image * image, int alpha_threshol
|
|||||||
|
|
||||||
// Error diffusion. Floyd Steinberg.
|
// Error diffusion. Floyd Steinberg.
|
||||||
void nv::Quantize::FloydSteinberg_Alpha4( Image * image )
|
void nv::Quantize::FloydSteinberg_Alpha4( Image * image )
|
||||||
{
|
|
||||||
FloydSteinberg(image, 8, 8, 8, 4);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void nv::Quantize::Truncate(Image * image, uint rsize, uint gsize, uint bsize, uint asize)
|
|
||||||
{
|
{
|
||||||
nvCheck(image != NULL);
|
nvCheck(image != NULL);
|
||||||
|
|
||||||
const uint w = image->width();
|
const uint w = image->width();
|
||||||
const uint h = image->height();
|
const uint h = image->height();
|
||||||
|
|
||||||
|
// @@ Use fixed point?
|
||||||
|
float * row0 = new float[(w+2)];
|
||||||
|
float * row1 = new float[(w+2)];
|
||||||
|
memset(row0, 0, sizeof(float)*(w+2));
|
||||||
|
memset(row1, 0, sizeof(float)*(w+2));
|
||||||
|
|
||||||
for(uint y = 0; y < h; y++) {
|
for(uint y = 0; y < h; y++) {
|
||||||
for(uint x = 0; x < w; x++) {
|
for(uint x = 0; x < w; x++) {
|
||||||
|
|
||||||
Color32 pixel = image->pixel(x, y);
|
Color32 pixel = image->pixel(x, y);
|
||||||
|
|
||||||
// Convert to our desired size, and reconstruct.
|
// Add error.
|
||||||
pixel.r = PixelFormat::convert(pixel.r, 8, rsize);
|
int alpha = int(pixel.a) + int(row0[1+x]);
|
||||||
pixel.r = PixelFormat::convert(pixel.r, rsize, 8);
|
|
||||||
|
|
||||||
pixel.g = PixelFormat::convert(pixel.g, 8, gsize);
|
// Convert to 4 bit using regular bit expansion.
|
||||||
pixel.g = PixelFormat::convert(pixel.g, gsize, 8);
|
pixel.a = (pixel.a & 0xF0) | ((pixel.a & 0xF0) >> 4);
|
||||||
|
|
||||||
pixel.b = PixelFormat::convert(pixel.b, 8, bsize);
|
|
||||||
pixel.b = PixelFormat::convert(pixel.b, bsize, 8);
|
|
||||||
|
|
||||||
pixel.a = PixelFormat::convert(pixel.a, 8, asize);
|
|
||||||
pixel.a = PixelFormat::convert(pixel.a, asize, 8);
|
|
||||||
|
|
||||||
// Store color.
|
// Store color.
|
||||||
image->pixel(x, y) = pixel;
|
image->pixel(x, y) = pixel;
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// Error diffusion. Floyd Steinberg.
|
|
||||||
void nv::Quantize::FloydSteinberg(Image * image, uint rsize, uint gsize, uint bsize, uint asize)
|
|
||||||
{
|
|
||||||
nvCheck(image != NULL);
|
|
||||||
|
|
||||||
const uint w = image->width();
|
|
||||||
const uint h = image->height();
|
|
||||||
|
|
||||||
Vector4 * row0 = new Vector4[w+2];
|
|
||||||
Vector4 * row1 = new Vector4[w+2];
|
|
||||||
memset(row0, 0, sizeof(Vector4)*(w+2));
|
|
||||||
memset(row1, 0, sizeof(Vector4)*(w+2));
|
|
||||||
|
|
||||||
for (uint y = 0; y < h; y++) {
|
|
||||||
for (uint x = 0; x < w; x++) {
|
|
||||||
|
|
||||||
Color32 pixel = image->pixel(x, y);
|
|
||||||
|
|
||||||
// Add error.
|
|
||||||
pixel.r = clamp(int(pixel.r) + int(row0[1+x].x()), 0, 255);
|
|
||||||
pixel.g = clamp(int(pixel.g) + int(row0[1+x].y()), 0, 255);
|
|
||||||
pixel.b = clamp(int(pixel.b) + int(row0[1+x].z()), 0, 255);
|
|
||||||
pixel.a = clamp(int(pixel.a) + int(row0[1+x].w()), 0, 255);
|
|
||||||
|
|
||||||
int r = pixel.r;
|
|
||||||
int g = pixel.g;
|
|
||||||
int b = pixel.b;
|
|
||||||
int a = pixel.a;
|
|
||||||
|
|
||||||
// Convert to our desired size, and reconstruct.
|
|
||||||
r = PixelFormat::convert(r, 8, rsize);
|
|
||||||
r = PixelFormat::convert(r, rsize, 8);
|
|
||||||
|
|
||||||
g = PixelFormat::convert(g, 8, gsize);
|
|
||||||
g = PixelFormat::convert(g, gsize, 8);
|
|
||||||
|
|
||||||
b = PixelFormat::convert(b, 8, bsize);
|
|
||||||
b = PixelFormat::convert(b, bsize, 8);
|
|
||||||
|
|
||||||
a = PixelFormat::convert(a, 8, asize);
|
|
||||||
a = PixelFormat::convert(a, asize, 8);
|
|
||||||
|
|
||||||
// Store color.
|
|
||||||
image->pixel(x, y) = Color32(r, g, b, a);
|
|
||||||
|
|
||||||
// Compute new error.
|
// Compute new error.
|
||||||
Vector4 diff(float(int(pixel.r) - r), float(int(pixel.g) - g), float(int(pixel.b) - b), float(int(pixel.a) - a));
|
float diff = float(alpha - pixel.a);
|
||||||
|
|
||||||
// Propagate new error.
|
// Propagate new error.
|
||||||
row0[1+x+1] += 7.0f / 16.0f * diff;
|
row0[1+x+1] += 7.0f / 16.0f * diff;
|
||||||
@ -211,9 +225,10 @@ void nv::Quantize::FloydSteinberg(Image * image, uint rsize, uint gsize, uint bs
|
|||||||
}
|
}
|
||||||
|
|
||||||
swap(row0, row1);
|
swap(row0, row1);
|
||||||
memset(row1, 0, sizeof(Vector4)*(w+2));
|
memset(row1, 0, sizeof(float)*(w+2));
|
||||||
}
|
}
|
||||||
|
|
||||||
delete [] row0;
|
delete [] row0;
|
||||||
delete [] row1;
|
delete [] row1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -3,9 +3,6 @@
|
|||||||
#ifndef NV_IMAGE_QUANTIZE_H
|
#ifndef NV_IMAGE_QUANTIZE_H
|
||||||
#define NV_IMAGE_QUANTIZE_H
|
#define NV_IMAGE_QUANTIZE_H
|
||||||
|
|
||||||
#include <nvimage/nvimage.h>
|
|
||||||
|
|
||||||
|
|
||||||
namespace nv
|
namespace nv
|
||||||
{
|
{
|
||||||
class Image;
|
class Image;
|
||||||
@ -20,9 +17,6 @@ namespace nv
|
|||||||
void FloydSteinberg_BinaryAlpha(Image * img, int alpha_threshold = 127);
|
void FloydSteinberg_BinaryAlpha(Image * img, int alpha_threshold = 127);
|
||||||
void FloydSteinberg_Alpha4(Image * img);
|
void FloydSteinberg_Alpha4(Image * img);
|
||||||
|
|
||||||
void Truncate(Image * image, uint rsize, uint gsize, uint bsize, uint asize);
|
|
||||||
void FloydSteinberg(Image * image, uint rsize, uint gsize, uint bsize, uint asize);
|
|
||||||
|
|
||||||
// @@ Add palette quantization algorithms!
|
// @@ Add palette quantization algorithms!
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -108,7 +108,7 @@ public:
|
|||||||
float area() const
|
float area() const
|
||||||
{
|
{
|
||||||
const Vector3 d = extents();
|
const Vector3 d = extents();
|
||||||
return 8.0f * (d.x()*d.y() + d.x()*d.z() + d.y()*d.z());
|
return 4.0f * (d.x()*d.y() + d.x()*d.z() + d.y()*d.z());
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the volume of the box.
|
/// Get the volume of the box.
|
||||||
@ -118,14 +118,6 @@ public:
|
|||||||
return 8.0f * (d.x() * d.y() * d.z());
|
return 8.0f * (d.x() * d.y() * d.z());
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Return true if the box contains the given point.
|
|
||||||
bool contains(Vector3::Arg p) const
|
|
||||||
{
|
|
||||||
return
|
|
||||||
m_mins.x() < p.x() && m_mins.y() < p.y() && m_mins.z() < p.z() &&
|
|
||||||
m_maxs.x() > p.x() && m_maxs.y() > p.y() && m_maxs.z() > p.z();
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|
||||||
Vector3 m_mins;
|
Vector3 m_mins;
|
||||||
@ -133,6 +125,15 @@ private:
|
|||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
/// Point inside box test.
|
||||||
|
inline bool pointInsideBox(const Box & b, Vector3::Arg p) const
|
||||||
|
{
|
||||||
|
return (m_mins.x() < p.x() && m_mins.y() < p.y() && m_mins.z() < p.z() &&
|
||||||
|
m_maxs.x() > p.x() && m_maxs.y() > p.y() && m_maxs.z() > p.z());
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
} // nv namespace
|
} // nv namespace
|
||||||
|
|
||||||
|
@ -48,37 +48,19 @@
|
|||||||
#define IS_NEGATIVE_FLOAT(x) (IR(x)&SIGN_BITMASK)
|
#define IS_NEGATIVE_FLOAT(x) (IR(x)&SIGN_BITMASK)
|
||||||
*/
|
*/
|
||||||
|
|
||||||
inline double sqrt_assert(const double f)
|
inline float sqrt_assert(const float f)
|
||||||
{
|
|
||||||
nvDebugCheck(f >= 0.0f);
|
|
||||||
return sqrt(f);
|
|
||||||
}
|
|
||||||
|
|
||||||
inline float sqrtf_assert(const float f)
|
|
||||||
{
|
{
|
||||||
nvDebugCheck(f >= 0.0f);
|
nvDebugCheck(f >= 0.0f);
|
||||||
return sqrtf(f);
|
return sqrtf(f);
|
||||||
}
|
}
|
||||||
|
|
||||||
inline double acos_assert(const double f)
|
inline float acos_assert(const float f)
|
||||||
{
|
|
||||||
nvDebugCheck(f >= -1.0f && f <= 1.0f);
|
|
||||||
return acos(f);
|
|
||||||
}
|
|
||||||
|
|
||||||
inline float acosf_assert(const float f)
|
|
||||||
{
|
{
|
||||||
nvDebugCheck(f >= -1.0f && f <= 1.0f);
|
nvDebugCheck(f >= -1.0f && f <= 1.0f);
|
||||||
return acosf(f);
|
return acosf(f);
|
||||||
}
|
}
|
||||||
|
|
||||||
inline double asin_assert(const double f)
|
inline float asin_assert(const float f)
|
||||||
{
|
|
||||||
nvDebugCheck(f >= -1.0f && f <= 1.0f);
|
|
||||||
return asin(f);
|
|
||||||
}
|
|
||||||
|
|
||||||
inline float asinf_assert(const float f)
|
|
||||||
{
|
{
|
||||||
nvDebugCheck(f >= -1.0f && f <= 1.0f);
|
nvDebugCheck(f >= -1.0f && f <= 1.0f);
|
||||||
return asinf(f);
|
return asinf(f);
|
||||||
@ -86,11 +68,11 @@ inline float asinf_assert(const float f)
|
|||||||
|
|
||||||
// Replace default functions with asserting ones.
|
// Replace default functions with asserting ones.
|
||||||
#define sqrt sqrt_assert
|
#define sqrt sqrt_assert
|
||||||
#define sqrtf sqrtf_assert
|
#define sqrtf sqrt_assert
|
||||||
#define acos acos_assert
|
#define acos acos_assert
|
||||||
#define acosf acosf_assert
|
#define acosf acos_assert
|
||||||
#define asin asin_assert
|
#define asin asin_assert
|
||||||
#define asinf asinf_assert
|
#define asinf asin_assert
|
||||||
|
|
||||||
#if NV_OS_WIN32
|
#if NV_OS_WIN32
|
||||||
#include <float.h>
|
#include <float.h>
|
||||||
@ -154,11 +136,6 @@ inline float lerp(float f0, float f1, float t)
|
|||||||
return f0 * s + f1 * t;
|
return f0 * s + f1 * t;
|
||||||
}
|
}
|
||||||
|
|
||||||
inline float square(float f)
|
|
||||||
{
|
|
||||||
return f * f;
|
|
||||||
}
|
|
||||||
|
|
||||||
} // nv
|
} // nv
|
||||||
|
|
||||||
#endif // NV_MATH_H
|
#endif // NV_MATH_H
|
||||||
|
@ -44,8 +44,7 @@ INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR})
|
|||||||
ADD_DEFINITIONS(-DNVTT_EXPORTS)
|
ADD_DEFINITIONS(-DNVTT_EXPORTS)
|
||||||
|
|
||||||
IF(NVTT_SHARED)
|
IF(NVTT_SHARED)
|
||||||
ADD_DEFINITIONS(-DNVTT_SHARED=1)
|
ADD_LIBRARY(nvtt SHARED ${DXT_SRCS})
|
||||||
ADD_LIBRARY(nvtt SHARED ${NVTT_SRCS})
|
|
||||||
ELSE(NVTT_SHARED)
|
ELSE(NVTT_SHARED)
|
||||||
ADD_LIBRARY(nvtt ${NVTT_SRCS})
|
ADD_LIBRARY(nvtt ${NVTT_SRCS})
|
||||||
ENDIF(NVTT_SHARED)
|
ENDIF(NVTT_SHARED)
|
||||||
|
@ -123,7 +123,7 @@ void nv::compressRGB(const Image * image, const OutputOptions::Private & outputO
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Zero padding.
|
// Zero padding.
|
||||||
for (uint x = w * byteCount; x < pitch; x++)
|
for (uint x = w; x < pitch; x++)
|
||||||
{
|
{
|
||||||
*(dst + x) = 0;
|
*(dst + x) = 0;
|
||||||
}
|
}
|
||||||
|
@ -34,7 +34,6 @@
|
|||||||
#include <nvimage/Filter.h>
|
#include <nvimage/Filter.h>
|
||||||
#include <nvimage/Quantize.h>
|
#include <nvimage/Quantize.h>
|
||||||
#include <nvimage/NormalMap.h>
|
#include <nvimage/NormalMap.h>
|
||||||
#include <nvimage/PixelFormat.h>
|
|
||||||
|
|
||||||
#include "Compressor.h"
|
#include "Compressor.h"
|
||||||
#include "InputOptions.h"
|
#include "InputOptions.h"
|
||||||
@ -211,10 +210,6 @@ Compressor::Compressor() : m(*new Compressor::Private())
|
|||||||
|
|
||||||
if (m.cudaEnabled)
|
if (m.cudaEnabled)
|
||||||
{
|
{
|
||||||
// Select fastest CUDA device.
|
|
||||||
int device = cuda::getFastestDevice();
|
|
||||||
cuda::setDevice(device);
|
|
||||||
|
|
||||||
m.cuda = new CudaCompressor();
|
m.cuda = new CudaCompressor();
|
||||||
|
|
||||||
if (!m.cuda->isValid())
|
if (!m.cuda->isValid())
|
||||||
@ -228,7 +223,6 @@ Compressor::Compressor() : m(*new Compressor::Private())
|
|||||||
Compressor::~Compressor()
|
Compressor::~Compressor()
|
||||||
{
|
{
|
||||||
delete &m;
|
delete &m;
|
||||||
cuda::exit();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -242,10 +236,6 @@ void Compressor::enableCudaAcceleration(bool enable)
|
|||||||
|
|
||||||
if (m.cudaEnabled && m.cuda == NULL)
|
if (m.cudaEnabled && m.cuda == NULL)
|
||||||
{
|
{
|
||||||
// Select fastest CUDA device.
|
|
||||||
int device = cuda::getFastestDevice();
|
|
||||||
cuda::setDevice(device);
|
|
||||||
|
|
||||||
m.cuda = new CudaCompressor();
|
m.cuda = new CudaCompressor();
|
||||||
|
|
||||||
if (!m.cuda->isValid())
|
if (!m.cuda->isValid())
|
||||||
@ -580,7 +570,7 @@ void Compressor::Private::scaleMipmap(Mipmap & mipmap, const InputOptions::Priva
|
|||||||
|
|
||||||
// Resize image.
|
// Resize image.
|
||||||
BoxFilter boxFilter;
|
BoxFilter boxFilter;
|
||||||
mipmap.setImage(mipmap.asFloatImage()->resize(boxFilter, w, h, (FloatImage::WrapMode)inputOptions.wrapMode));
|
mipmap.setImage(mipmap.asFloatImage()->downSample(boxFilter, w, h, (FloatImage::WrapMode)inputOptions.wrapMode));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -627,6 +617,13 @@ void Compressor::Private::quantizeMipmap(Mipmap & mipmap, const CompressionOptio
|
|||||||
{
|
{
|
||||||
nvDebugCheck(mipmap.asFixedImage() != NULL);
|
nvDebugCheck(mipmap.asFixedImage() != NULL);
|
||||||
|
|
||||||
|
if (compressionOptions.enableColorDithering)
|
||||||
|
{
|
||||||
|
if (compressionOptions.format >= Format_DXT1 && compressionOptions.format <= Format_DXT5)
|
||||||
|
{
|
||||||
|
Quantize::FloydSteinberg_RGB16(mipmap.asMutableFixedImage());
|
||||||
|
}
|
||||||
|
}
|
||||||
if (compressionOptions.binaryAlpha)
|
if (compressionOptions.binaryAlpha)
|
||||||
{
|
{
|
||||||
if (compressionOptions.enableAlphaDithering)
|
if (compressionOptions.enableAlphaDithering)
|
||||||
@ -638,50 +635,19 @@ void Compressor::Private::quantizeMipmap(Mipmap & mipmap, const CompressionOptio
|
|||||||
Quantize::BinaryAlpha(mipmap.asMutableFixedImage(), compressionOptions.alphaThreshold);
|
Quantize::BinaryAlpha(mipmap.asMutableFixedImage(), compressionOptions.alphaThreshold);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
else
|
||||||
if (compressionOptions.enableColorDithering || compressionOptions.enableAlphaDithering)
|
|
||||||
{
|
{
|
||||||
uint rsize = 8;
|
|
||||||
uint gsize = 8;
|
|
||||||
uint bsize = 8;
|
|
||||||
uint asize = 8;
|
|
||||||
|
|
||||||
if (compressionOptions.enableColorDithering)
|
|
||||||
{
|
|
||||||
if (compressionOptions.format >= Format_DXT1 && compressionOptions.format <= Format_DXT5)
|
|
||||||
{
|
|
||||||
rsize = 5;
|
|
||||||
gsize = 6;
|
|
||||||
bsize = 5;
|
|
||||||
}
|
|
||||||
else if (compressionOptions.format == Format_RGB)
|
|
||||||
{
|
|
||||||
uint rshift, gshift, bshift;
|
|
||||||
PixelFormat::maskShiftAndSize(compressionOptions.rmask, &rshift, &rsize);
|
|
||||||
PixelFormat::maskShiftAndSize(compressionOptions.gmask, &gshift, &gsize);
|
|
||||||
PixelFormat::maskShiftAndSize(compressionOptions.bmask, &bshift, &bsize);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (compressionOptions.enableAlphaDithering)
|
if (compressionOptions.enableAlphaDithering)
|
||||||
{
|
{
|
||||||
if (compressionOptions.format == Format_DXT3)
|
if (compressionOptions.format == Format_DXT3)
|
||||||
{
|
{
|
||||||
asize = 4;
|
Quantize::Alpha4(mipmap.asMutableFixedImage());
|
||||||
}
|
}
|
||||||
else if (compressionOptions.format == Format_RGB)
|
else if (compressionOptions.format == Format_DXT1a)
|
||||||
{
|
{
|
||||||
uint ashift;
|
Quantize::BinaryAlpha(mipmap.asMutableFixedImage(), compressionOptions.alphaThreshold);
|
||||||
PixelFormat::maskShiftAndSize(compressionOptions.amask, &ashift, &asize);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (compressionOptions.binaryAlpha)
|
|
||||||
{
|
|
||||||
asize = 8; // Already quantized.
|
|
||||||
}
|
|
||||||
|
|
||||||
Quantize::FloydSteinberg(mipmap.asMutableFixedImage(), rsize, gsize, bsize, asize);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -698,7 +664,6 @@ bool Compressor::Private::compressMipmap(const Mipmap & mipmap, const InputOptio
|
|||||||
SlowCompressor slow;
|
SlowCompressor slow;
|
||||||
slow.setImage(image, inputOptions.alphaMode);
|
slow.setImage(image, inputOptions.alphaMode);
|
||||||
|
|
||||||
const bool useCuda = cudaEnabled && image->width() * image->height() >= 512;
|
|
||||||
|
|
||||||
if (compressionOptions.format == Format_RGBA || compressionOptions.format == Format_RGB)
|
if (compressionOptions.format == Format_RGBA || compressionOptions.format == Format_RGB)
|
||||||
{
|
{
|
||||||
@ -727,7 +692,7 @@ bool Compressor::Private::compressMipmap(const Mipmap & mipmap, const InputOptio
|
|||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
if (useCuda)
|
if (cudaEnabled)
|
||||||
{
|
{
|
||||||
nvDebugCheck(cudaSupported);
|
nvDebugCheck(cudaSupported);
|
||||||
cuda->setImage(image, inputOptions.alphaMode);
|
cuda->setImage(image, inputOptions.alphaMode);
|
||||||
@ -747,7 +712,7 @@ bool Compressor::Private::compressMipmap(const Mipmap & mipmap, const InputOptio
|
|||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
if (useCuda)
|
if (cudaEnabled)
|
||||||
{
|
{
|
||||||
nvDebugCheck(cudaSupported);
|
nvDebugCheck(cudaSupported);
|
||||||
/*cuda*/slow.compressDXT1a(compressionOptions, outputOptions);
|
/*cuda*/slow.compressDXT1a(compressionOptions, outputOptions);
|
||||||
@ -766,7 +731,7 @@ bool Compressor::Private::compressMipmap(const Mipmap & mipmap, const InputOptio
|
|||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
if (useCuda)
|
if (cudaEnabled)
|
||||||
{
|
{
|
||||||
nvDebugCheck(cudaSupported);
|
nvDebugCheck(cudaSupported);
|
||||||
cuda->setImage(image, inputOptions.alphaMode);
|
cuda->setImage(image, inputOptions.alphaMode);
|
||||||
@ -786,7 +751,7 @@ bool Compressor::Private::compressMipmap(const Mipmap & mipmap, const InputOptio
|
|||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
if (useCuda)
|
if (cudaEnabled)
|
||||||
{
|
{
|
||||||
nvDebugCheck(cudaSupported);
|
nvDebugCheck(cudaSupported);
|
||||||
cuda->setImage(image, inputOptions.alphaMode);
|
cuda->setImage(image, inputOptions.alphaMode);
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -191,9 +191,6 @@ __device__ void loadColorBlock(const uint * image, float3 colors[16], float3 sum
|
|||||||
|
|
||||||
*sameColor = (axis == make_float3(0, 0, 0));
|
*sameColor = (axis == make_float3(0, 0, 0));
|
||||||
|
|
||||||
// Single color compressor needs unweighted colors.
|
|
||||||
if (*sameColor) colors[idx] = rawColors[idx];
|
|
||||||
|
|
||||||
dps[idx] = dot(rawColors[idx], axis);
|
dps[idx] = dot(rawColors[idx], axis);
|
||||||
|
|
||||||
#if __DEVICE_EMULATION__
|
#if __DEVICE_EMULATION__
|
||||||
|
@ -37,7 +37,7 @@
|
|||||||
|
|
||||||
|
|
||||||
#if defined HAVE_CUDA
|
#if defined HAVE_CUDA
|
||||||
#include <cuda_runtime_api.h>
|
#include <cuda_runtime.h>
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include <time.h>
|
#include <time.h>
|
||||||
|
@ -148,7 +148,7 @@ inline __device__ bool singleColor(const float3 * colors)
|
|||||||
bool sameColor = false;
|
bool sameColor = false;
|
||||||
for (int i = 0; i < 16; i++)
|
for (int i = 0; i < 16; i++)
|
||||||
{
|
{
|
||||||
sameColor &= (colors[i] == colors[0]);
|
sameColor &= (colors[idx] == colors[0]);
|
||||||
}
|
}
|
||||||
return sameColor;
|
return sameColor;
|
||||||
#else
|
#else
|
||||||
|
@ -22,18 +22,15 @@
|
|||||||
// OTHER DEALINGS IN THE SOFTWARE.
|
// OTHER DEALINGS IN THE SOFTWARE.
|
||||||
|
|
||||||
#include <nvcore/Debug.h>
|
#include <nvcore/Debug.h>
|
||||||
#include <nvcore/Library.h>
|
|
||||||
#include "CudaUtils.h"
|
#include "CudaUtils.h"
|
||||||
|
|
||||||
#if defined HAVE_CUDA
|
#if defined HAVE_CUDA
|
||||||
#include <cuda.h>
|
#include <cuda_runtime.h>
|
||||||
#include <cuda_runtime_api.h>
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
using namespace nv;
|
using namespace nv;
|
||||||
using namespace cuda;
|
using namespace cuda;
|
||||||
|
|
||||||
/* @@ Move this to win32 utils or somewhere else.
|
|
||||||
#if NV_OS_WIN32
|
#if NV_OS_WIN32
|
||||||
|
|
||||||
#define WINDOWS_LEAN_AND_MEAN
|
#define WINDOWS_LEAN_AND_MEAN
|
||||||
@ -55,93 +52,31 @@ static bool isWow32()
|
|||||||
{
|
{
|
||||||
LPFN_ISWOW64PROCESS fnIsWow64Process = (LPFN_ISWOW64PROCESS)GetProcAddress(GetModuleHandle("kernel32"), "IsWow64Process");
|
LPFN_ISWOW64PROCESS fnIsWow64Process = (LPFN_ISWOW64PROCESS)GetProcAddress(GetModuleHandle("kernel32"), "IsWow64Process");
|
||||||
|
|
||||||
BOOL bIsWow64 = FALSE;
|
BOOL bIsWow64 = FALSE;
|
||||||
|
|
||||||
if (NULL != fnIsWow64Process)
|
if (NULL != fnIsWow64Process)
|
||||||
{
|
{
|
||||||
if (!fnIsWow64Process(GetCurrentProcess(), &bIsWow64))
|
if (!fnIsWow64Process(GetCurrentProcess(), &bIsWow64))
|
||||||
{
|
{
|
||||||
// Assume 32 bits.
|
// Assume 32 bits.
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return !bIsWow64;
|
return !bIsWow64;
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
*/
|
|
||||||
|
|
||||||
|
|
||||||
static bool isCudaDriverAvailable(int version)
|
|
||||||
{
|
|
||||||
#if defined HAVE_CUDA
|
|
||||||
#if NV_OS_WIN32
|
|
||||||
Library nvcuda("nvcuda.dll");
|
|
||||||
#else
|
|
||||||
Library nvcuda(NV_LIBRARY_NAME(cuda));
|
|
||||||
#endif
|
|
||||||
|
|
||||||
if (!nvcuda.isValid())
|
|
||||||
{
|
|
||||||
nvDebug("*** CUDA driver not found.\n");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (version >= 2000)
|
|
||||||
{
|
|
||||||
void * address = nvcuda.bindSymbol("cuStreamCreate");
|
|
||||||
if (address == NULL) {
|
|
||||||
nvDebug("*** CUDA driver version < 2.0.\n");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (version >= 2010)
|
|
||||||
{
|
|
||||||
void * address = nvcuda.bindSymbol("cuModuleLoadDataEx");
|
|
||||||
if (address == NULL) {
|
|
||||||
nvDebug("*** CUDA driver version < 2.1.\n");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (version >= 2020)
|
|
||||||
{
|
|
||||||
typedef CUresult (CUDAAPI * PFCU_DRIVERGETVERSION)(int * version);
|
|
||||||
|
|
||||||
PFCU_DRIVERGETVERSION driverGetVersion = (PFCU_DRIVERGETVERSION)nvcuda.bindSymbol("cuDriverGetVersion");
|
|
||||||
if (driverGetVersion == NULL) {
|
|
||||||
nvDebug("*** CUDA driver version < 2.2.\n");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
int driverVersion;
|
|
||||||
CUresult err = driverGetVersion(&driverVersion);
|
|
||||||
if (err != CUDA_SUCCESS) {
|
|
||||||
nvDebug("*** Error querying driver version: '%s'.\n", cudaGetErrorString((cudaError_t)err));
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return driverVersion >= version;
|
|
||||||
}
|
|
||||||
#endif // HAVE_CUDA
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
/// Determine if CUDA is available.
|
/// Determine if CUDA is available.
|
||||||
bool nv::cuda::isHardwarePresent()
|
bool nv::cuda::isHardwarePresent()
|
||||||
{
|
{
|
||||||
#if defined HAVE_CUDA
|
#if defined HAVE_CUDA
|
||||||
// Make sure that CUDA driver matches CUDA runtime.
|
#if NV_OS_WIN32
|
||||||
if (!isCudaDriverAvailable(CUDART_VERSION))
|
//if (isWindowsVista()) return false;
|
||||||
{
|
//if (isWindowsVista() || !isWow32()) return false;
|
||||||
nvDebug("CUDA driver not available for CUDA runtime %d\n", CUDART_VERSION);
|
#endif
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
int count = deviceCount();
|
int count = deviceCount();
|
||||||
if (count == 1)
|
if (count == 1)
|
||||||
{
|
{
|
||||||
@ -154,9 +89,9 @@ bool nv::cuda::isHardwarePresent()
|
|||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// @@ Make sure that warp size == 32
|
// @@ Make sure that warp size == 32
|
||||||
|
}
|
||||||
|
|
||||||
return count > 0;
|
return count > 0;
|
||||||
#else
|
#else
|
||||||
@ -180,60 +115,14 @@ int nv::cuda::deviceCount()
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int nv::cuda::getFastestDevice()
|
|
||||||
{
|
|
||||||
int max_gflops_device = 0;
|
|
||||||
#if defined HAVE_CUDA
|
|
||||||
int max_gflops = 0;
|
|
||||||
|
|
||||||
const int device_count = deviceCount();
|
|
||||||
int current_device = 0;
|
|
||||||
while (current_device < device_count)
|
|
||||||
{
|
|
||||||
cudaDeviceProp device_properties;
|
|
||||||
cudaGetDeviceProperties(&device_properties, current_device);
|
|
||||||
int gflops = device_properties.multiProcessorCount * device_properties.clockRate;
|
|
||||||
|
|
||||||
if (device_properties.major != -1 && device_properties.minor != -1)
|
|
||||||
{
|
|
||||||
if( gflops > max_gflops )
|
|
||||||
{
|
|
||||||
max_gflops = gflops;
|
|
||||||
max_gflops_device = current_device;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
current_device++;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
return max_gflops_device;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
/// Activate the given devices.
|
/// Activate the given devices.
|
||||||
bool nv::cuda::setDevice(int i)
|
bool nv::cuda::setDevice(int i)
|
||||||
{
|
{
|
||||||
nvCheck(i < deviceCount());
|
nvCheck(i < deviceCount());
|
||||||
#if defined HAVE_CUDA
|
#if defined HAVE_CUDA
|
||||||
cudaError_t result = cudaSetDevice(i);
|
cudaError_t result = cudaSetDevice(i);
|
||||||
|
|
||||||
if (result != cudaSuccess) {
|
|
||||||
nvDebug("*** CUDA Error: %s\n", cudaGetErrorString(result));
|
|
||||||
}
|
|
||||||
|
|
||||||
return result == cudaSuccess;
|
return result == cudaSuccess;
|
||||||
#else
|
#else
|
||||||
return false;
|
return false;
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
void nv::cuda::exit()
|
|
||||||
{
|
|
||||||
#if defined HAVE_CUDA
|
|
||||||
cudaError_t result = cudaThreadExit();
|
|
||||||
|
|
||||||
if (result != cudaSuccess) {
|
|
||||||
nvDebug("*** CUDA Error: %s\n", cudaGetErrorString(result));
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
@ -31,9 +31,7 @@ namespace nv
|
|||||||
{
|
{
|
||||||
bool isHardwarePresent();
|
bool isHardwarePresent();
|
||||||
int deviceCount();
|
int deviceCount();
|
||||||
int getFastestDevice();
|
|
||||||
bool setDevice(int i);
|
bool setDevice(int i);
|
||||||
void exit();
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} // nv namespace
|
} // nv namespace
|
||||||
|
@ -1,8 +1,13 @@
|
|||||||
PROJECT(squish)
|
PROJECT(squish)
|
||||||
|
ENABLE_TESTING()
|
||||||
|
|
||||||
INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR})
|
INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR})
|
||||||
|
|
||||||
SET(SQUISH_SRCS
|
SET(SQUISH_SRCS
|
||||||
|
# alpha.cpp
|
||||||
|
# alpha.h
|
||||||
|
# clusterfit.cpp
|
||||||
|
# clusterfit.h
|
||||||
fastclusterfit.cpp
|
fastclusterfit.cpp
|
||||||
fastclusterfit.h
|
fastclusterfit.h
|
||||||
weightedclusterfit.cpp
|
weightedclusterfit.cpp
|
||||||
@ -16,13 +21,32 @@ SET(SQUISH_SRCS
|
|||||||
config.h
|
config.h
|
||||||
maths.cpp
|
maths.cpp
|
||||||
maths.h
|
maths.h
|
||||||
|
# rangefit.cpp
|
||||||
|
# rangefit.h
|
||||||
|
# singlecolourfit.cpp
|
||||||
|
# singlecolourfit.h
|
||||||
|
# singlecolourlookup.inl
|
||||||
|
# squish.cpp
|
||||||
|
# squish.h
|
||||||
simd.h
|
simd.h
|
||||||
simd_sse.h
|
simd_sse.h
|
||||||
simd_ve.h)
|
simd_ve.h)
|
||||||
|
|
||||||
ADD_LIBRARY(squish STATIC ${SQUISH_SRCS})
|
ADD_LIBRARY(squish STATIC ${SQUISH_SRCS})
|
||||||
|
|
||||||
IF(CMAKE_COMPILER_IS_GNUCXX)
|
# libpng
|
||||||
SET_TARGET_PROPERTIES(squish PROPERTIES COMPILE_FLAGS -fPIC)
|
#FIND_PACKAGE(PNG)
|
||||||
ENDIF(CMAKE_COMPILER_IS_GNUCXX)
|
|
||||||
|
#IF(PNG_FOUND)
|
||||||
|
# INCLUDE_DIRECTORIES(${PNG_INCLUDE_DIR})
|
||||||
|
# ADD_EXECUTABLE(squishpng extra/squishpng.cpp)
|
||||||
|
# TARGET_LINK_LIBRARIES(squishpng squish ${PNG_LIBRARY})
|
||||||
|
#ENDIF(PNG_FOUND)
|
||||||
|
|
||||||
|
##ADD_EXECUTABLE(squishgen extra/squishgen.cpp)
|
||||||
|
|
||||||
|
#ADD_EXECUTABLE(squishtest extra/squishtest.cpp)
|
||||||
|
#TARGET_LINK_LIBRARIES(squishtest squish)
|
||||||
|
|
||||||
|
#ADD_TEST(SQUISHTEST squishtest)
|
||||||
|
|
||||||
|
@ -373,6 +373,7 @@ int main(int argc, char *argv[])
|
|||||||
inputOptions.setMipmapGeneration(false);
|
inputOptions.setMipmapGeneration(false);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
nvtt::CompressionOptions compressionOptions;
|
nvtt::CompressionOptions compressionOptions;
|
||||||
compressionOptions.setFormat(format);
|
compressionOptions.setFormat(format);
|
||||||
if (fast)
|
if (fast)
|
||||||
@ -408,16 +409,6 @@ int main(int argc, char *argv[])
|
|||||||
nvtt::Compressor compressor;
|
nvtt::Compressor compressor;
|
||||||
compressor.enableCudaAcceleration(!nocuda);
|
compressor.enableCudaAcceleration(!nocuda);
|
||||||
|
|
||||||
printf("CUDA acceleration ");
|
|
||||||
if (compressor.isCudaAccelerationEnabled())
|
|
||||||
{
|
|
||||||
printf("ENABLED\n\n");
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
printf("DISABLED\n\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
outputHandler.setTotal(compressor.estimateSize(inputOptions, compressionOptions));
|
outputHandler.setTotal(compressor.estimateSize(inputOptions, compressionOptions));
|
||||||
outputHandler.setDisplayProgress(!silent);
|
outputHandler.setDisplayProgress(!silent);
|
||||||
|
|
||||||
|
@ -84,7 +84,7 @@ struct Error
|
|||||||
{
|
{
|
||||||
mabse /= samples;
|
mabse /= samples;
|
||||||
mse /= samples;
|
mse /= samples;
|
||||||
rmse = sqrtf(mse);
|
rmse = sqrt(mse);
|
||||||
psnr = (rmse == 0) ? 999.0f : 20.0f * log10(255.0f / rmse);
|
psnr = (rmse == 0) ? 999.0f : 20.0f * log10(255.0f / rmse);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -134,7 +134,7 @@ struct NormalError
|
|||||||
{
|
{
|
||||||
ade /= samples;
|
ade /= samples;
|
||||||
mse /= samples * 3;
|
mse /= samples * 3;
|
||||||
rmse = sqrtf(mse);
|
rmse = sqrt(mse);
|
||||||
psnr = (rmse == 0) ? 999.0f : 20.0f * log10(255.0f / rmse);
|
psnr = (rmse == 0) ? 999.0f : 20.0f * log10(255.0f / rmse);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -73,12 +73,10 @@ int main(int argc, char *argv[])
|
|||||||
|
|
||||||
float scale = 0.5f;
|
float scale = 0.5f;
|
||||||
float gamma = 2.2f;
|
float gamma = 2.2f;
|
||||||
nv::AutoPtr<nv::Filter> filter;
|
nv::Filter * filter = NULL;
|
||||||
nv::Path input;
|
nv::Path input;
|
||||||
nv::Path output;
|
nv::Path output;
|
||||||
|
|
||||||
nv::FloatImage::WrapMode wrapMode = nv::FloatImage::WrapMode_Mirror;
|
|
||||||
|
|
||||||
// Parse arguments.
|
// Parse arguments.
|
||||||
for (int i = 1; i < argc; i++)
|
for (int i = 1; i < argc; i++)
|
||||||
{
|
{
|
||||||
@ -110,18 +108,9 @@ int main(int argc, char *argv[])
|
|||||||
else if (strcmp("lanczos", argv[i]) == 0) filter = new nv::LanczosFilter();
|
else if (strcmp("lanczos", argv[i]) == 0) filter = new nv::LanczosFilter();
|
||||||
else if (strcmp("kaiser", argv[i]) == 0) {
|
else if (strcmp("kaiser", argv[i]) == 0) {
|
||||||
filter = new nv::KaiserFilter(3);
|
filter = new nv::KaiserFilter(3);
|
||||||
((nv::KaiserFilter *)filter.ptr())->setParameters(4.0f, 1.0f);
|
((nv::KaiserFilter *)filter)->setParameters(4.0f, 1.0f);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else if (strcmp("-w", argv[i]) == 0)
|
|
||||||
{
|
|
||||||
if (i+1 == argc) break;
|
|
||||||
i++;
|
|
||||||
|
|
||||||
if (strcmp("mirror", argv[i]) == 0) wrapMode = nv::FloatImage::WrapMode_Mirror;
|
|
||||||
else if (strcmp("repeat", argv[i]) == 0) wrapMode = nv::FloatImage::WrapMode_Repeat;
|
|
||||||
else if (strcmp("clamp", argv[i]) == 0) wrapMode = nv::FloatImage::WrapMode_Clamp;
|
|
||||||
}
|
|
||||||
else if (argv[i][0] != '-')
|
else if (argv[i][0] != '-')
|
||||||
{
|
{
|
||||||
input = argv[i];
|
input = argv[i];
|
||||||
@ -151,10 +140,6 @@ int main(int argc, char *argv[])
|
|||||||
printf(" * mitchell\n");
|
printf(" * mitchell\n");
|
||||||
printf(" * lanczos\n");
|
printf(" * lanczos\n");
|
||||||
printf(" * kaiser\n");
|
printf(" * kaiser\n");
|
||||||
printf(" -w mode One of the following: (default = 'mirror')\n");
|
|
||||||
printf(" * mirror\n");
|
|
||||||
printf(" * repeat\n");
|
|
||||||
printf(" * clamp\n");
|
|
||||||
|
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
@ -170,14 +155,15 @@ int main(int argc, char *argv[])
|
|||||||
nv::FloatImage fimage(&image);
|
nv::FloatImage fimage(&image);
|
||||||
fimage.toLinear(0, 3, gamma);
|
fimage.toLinear(0, 3, gamma);
|
||||||
|
|
||||||
nv::AutoPtr<nv::FloatImage> fresult(fimage.resize(*filter, uint(image.width() * scale), uint(image.height() * scale), wrapMode));
|
nv::AutoPtr<nv::FloatImage> fresult(fimage.downSample(*filter, uint(image.width() * scale), uint(image.height() * scale), nv::FloatImage::WrapMode_Mirror));
|
||||||
|
|
||||||
nv::AutoPtr<nv::Image> result(fresult->createImageGammaCorrect(gamma));
|
nv::AutoPtr<nv::Image> result(fresult->createImageGammaCorrect(gamma));
|
||||||
result->setFormat(nv::Image::Format_ARGB);
|
|
||||||
|
|
||||||
nv::StdOutputStream stream(output);
|
nv::StdOutputStream stream(output);
|
||||||
nv::ImageIO::saveTGA(stream, result.ptr()); // @@ Add generic save function. Add support for png too.
|
nv::ImageIO::saveTGA(stream, result.ptr()); // @@ Add generic save function. Add support for png too.
|
||||||
|
|
||||||
|
delete filter;
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user