Compare commits
2 commits
main
...
feature/ex
Author | SHA1 | Date | |
---|---|---|---|
17f5e802c5 | |||
4c9f5802dd |
1 changed files with 192 additions and 119 deletions
311
Source/Main.cpp
311
Source/Main.cpp
|
@ -41,20 +41,20 @@ const std::map<std::string, std::tuple<std::string, int, vk::Format>> formats =
|
||||||
{"BC7_SRGB", {"8 bit RGBA - Good general purpose. 16 bytes per block.", 16, vk::Format::eBc7SrgbBlock}}
|
{"BC7_SRGB", {"8 bit RGBA - Good general purpose. 16 bytes per block.", 16, vk::Format::eBc7SrgbBlock}}
|
||||||
};
|
};
|
||||||
|
|
||||||
const std::string usage = "[cube|array] <input> [input2, input3...] <output> <format> [fast|normal|slow|veryslow]";
|
const std::string usage = "[cube|array] <input> [input2, input3...] <output> <format> [fast|normal|slow|veryslow] [explicitmips]";
|
||||||
|
|
||||||
int main(int argc, char ** argv)
|
int main(int argc, char ** argv)
|
||||||
{
|
{
|
||||||
ISPCInit();
|
ISPCInit();
|
||||||
|
|
||||||
if (argc < 4) {
|
if (argc < 4) {
|
||||||
std::cout << "Usage: " << argv[0] << " " << usage << std::endl;
|
std::cerr << "Usage: " << argv[0] << " " << usage << std::endl;
|
||||||
std::cout << "Formats:" << std::endl;
|
std::cerr << "Formats:" << std::endl;
|
||||||
for (auto & formatName : formatOrder) {
|
for (auto & formatName : formatOrder) {
|
||||||
auto format = formats.at(formatName);
|
auto format = formats.at(formatName);
|
||||||
std::cout << " " << formatName << " - " << std::get<0>(format) << std::endl;
|
std::cerr << " " << formatName << " - " << std::get<0>(format) << std::endl;
|
||||||
}
|
}
|
||||||
return 1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int numInputs = argc - 3;
|
int numInputs = argc - 3;
|
||||||
|
@ -72,10 +72,20 @@ int main(int argc, char ** argv)
|
||||||
std::string speedString(argv[argc - 1]);
|
std::string speedString(argv[argc - 1]);
|
||||||
std::string formatString;
|
std::string formatString;
|
||||||
int speed = 2;
|
int speed = 2;
|
||||||
|
bool explicitmips = false;
|
||||||
|
|
||||||
|
int formatPosition = -1;
|
||||||
|
|
||||||
|
if (speedString == "explicitmips") {
|
||||||
|
speedString = std::string(argv[argc - 2]);
|
||||||
|
numInputs -= 1;
|
||||||
|
formatPosition -= 1;
|
||||||
|
explicitmips = true;
|
||||||
|
}
|
||||||
|
|
||||||
if (speedString == "fast" || speedString == "normal" || speedString == "slow" || speedString == "veryslow") {
|
if (speedString == "fast" || speedString == "normal" || speedString == "slow" || speedString == "veryslow") {
|
||||||
formatString = std::string(argv[argc - 2]);
|
|
||||||
numInputs -= 1;
|
numInputs -= 1;
|
||||||
|
formatPosition -= 1;
|
||||||
|
|
||||||
if (speedString == "slow") {
|
if (speedString == "slow") {
|
||||||
speed = 1;
|
speed = 1;
|
||||||
|
@ -86,29 +96,41 @@ int main(int argc, char ** argv)
|
||||||
} else {
|
} else {
|
||||||
speed = 2;
|
speed = 2;
|
||||||
}
|
}
|
||||||
} else {
|
|
||||||
formatString = std::string(argv[argc - 1]);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
formatString = std::string(argv[argc + formatPosition]);
|
||||||
|
|
||||||
if (numInputs < 1) {
|
if (numInputs < 1) {
|
||||||
std::cout << "Usage: " << argv[0] << usage << " " << std::endl;
|
std::cerr << "Usage: " << argv[0] << usage << " " << std::endl;
|
||||||
return 1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (option == "cube" && numInputs != 6) {
|
if (explicitmips) {
|
||||||
std::cout << "Cube maps must have 6 inputs." << std::endl;
|
if (option == "cube" && numInputs % 6 != 0) {
|
||||||
return 1;
|
std::cerr << "Cube maps must have a multiple of 6 inputs." << std::endl;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if (option == "cube" && numInputs != 6) {
|
||||||
|
std::cerr << "Cube maps must have 6 inputs." << std::endl;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (option == "array" && numInputs < 2) {
|
if (option == "array" && numInputs < 2) {
|
||||||
std::cout << "Array maps must have at least 2 inputs." << std::endl;
|
std::cerr << "Array maps must have at least 2 inputs." << std::endl;
|
||||||
return 1;
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (option == "array" && explicitmips) {
|
||||||
|
std::cerr << "Explicit mips are not supported for arrays." << std::endl;
|
||||||
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (formats.find(formatString) == formats.end()) {
|
if (formats.find(formatString) == formats.end()) {
|
||||||
std::cout << "Invalid format: " << formatString << std::endl;
|
std::cerr << "Invalid format: " << formatString << std::endl;
|
||||||
std::cout << usage << std::endl;
|
std::cerr << usage << std::endl;
|
||||||
std::cout << "Formats:" << std::endl;
|
std::cerr << "Formats:" << std::endl;
|
||||||
for (auto & formatName : formatOrder) {
|
for (auto & formatName : formatOrder) {
|
||||||
auto format = formats.at(formatName);
|
auto format = formats.at(formatName);
|
||||||
std::cout << " " << formatName << " - " << std::get<0>(format) << std::endl;
|
std::cout << " " << formatName << " - " << std::get<0>(format) << std::endl;
|
||||||
|
@ -188,94 +210,142 @@ int main(int argc, char ** argv)
|
||||||
ldrLevelBlocks.resize(numInputs);
|
ldrLevelBlocks.resize(numInputs);
|
||||||
}
|
}
|
||||||
|
|
||||||
uint32_t levelCount;
|
unsigned int levelCount;
|
||||||
|
int numFaces = 1;
|
||||||
|
|
||||||
for (int input = 0; input < numInputs; input++) {
|
int topLevelWidth, topLevelHeight;
|
||||||
int level = 0;
|
|
||||||
|
|
||||||
std::cout << "Loading/scaling " << input << ": " << inputs[input] << std::endl;
|
if (option == "cube") {
|
||||||
|
numFaces = 6;
|
||||||
|
}
|
||||||
|
|
||||||
if (hdr) {
|
if (explicitmips) {
|
||||||
hdrBufferA = stbi_loadf(inputs[input].c_str(), &width, &height, &channels, forcedChannels);
|
levelCount = numInputs / numFaces;
|
||||||
if (hdrBufferA == nullptr) {
|
|
||||||
std::cout << "Failed to load image: " << inputs[input] << std::endl;
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
hdrBufferB = new float[width * height * forcedChannels];
|
for (int face = 0; face < numFaces; face++) {
|
||||||
hdrBufferMain = hdrBufferA;
|
int levelWidth = 0, levelHeight = 0;
|
||||||
hdrBufferOther = hdrBufferB;
|
|
||||||
} else {
|
|
||||||
ldrBufferA = stbi_load(inputs[input].c_str(), &width, &height, &channels, forcedChannels);
|
|
||||||
if (ldrBufferA == nullptr) {
|
|
||||||
std::cout << "Failed to load image: " << inputs[input] << std::endl;
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
ldrBufferB = new unsigned char[width * height * forcedChannels];
|
for (unsigned int level = 0; level < levelCount; level++) {
|
||||||
ldrBufferMain = ldrBufferA;
|
std::cout << "Loading/scaling face " << face << ", level " << level << ", " << inputs[face * levelCount + level] << std::endl;
|
||||||
ldrBufferOther = ldrBufferB;
|
|
||||||
}
|
|
||||||
|
|
||||||
int oldWidth = width;
|
if (hdr) {
|
||||||
int oldHeight = height;
|
hdrBufferA = stbi_loadf(inputs[face * levelCount + level].c_str(), &width, &height, &channels, forcedChannels);
|
||||||
|
} else {
|
||||||
|
ldrBufferA = stbi_load(inputs[face * levelCount + level].c_str(), &width, &height, &channels, forcedChannels);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (level == 0) {
|
||||||
|
levelWidth = width;
|
||||||
|
levelHeight = height;
|
||||||
|
topLevelWidth = width;
|
||||||
|
topLevelHeight = height;
|
||||||
|
} else {
|
||||||
|
if (width != levelWidth || height != levelHeight) {
|
||||||
|
std::cout << "Input " << inputs[face * levelCount + level] << " for face " << face << ", level " << level << "dimensions " << width << "x" << height << " do not match expected " << levelWidth << "x" << levelHeight << std::endl;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (hdr) {
|
||||||
|
hdrLevels[face].push_back(std::vector<float>(hdrBufferA, hdrBufferA + width * height * forcedChannels));
|
||||||
|
stbi_image_free(hdrBufferA);
|
||||||
|
} else {
|
||||||
|
ldrLevels[face].push_back(std::vector<uint8_t>(ldrBufferA, ldrBufferA + width * height * forcedChannels));
|
||||||
|
stbi_image_free(ldrBufferA);
|
||||||
|
}
|
||||||
|
|
||||||
levelCount = 1;
|
|
||||||
{
|
|
||||||
int levelWidth = width;
|
|
||||||
int levelHeight = height;
|
|
||||||
while (levelWidth > 1 || levelHeight > 1) {
|
|
||||||
levelWidth = std::max(1, (int)floorf((float)levelWidth / 2));
|
levelWidth = std::max(1, (int)floorf((float)levelWidth / 2));
|
||||||
levelHeight = std::max(1, (int)floorf((float)levelHeight / 2));
|
levelHeight = std::max(1, (int)floorf((float)levelHeight / 2));
|
||||||
levelCount++;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
while(1) {
|
|
||||||
if (hdr) {
|
|
||||||
hdrLevels[input].push_back(std::vector<float>(hdrBufferMain, hdrBufferMain + oldWidth * oldHeight * forcedChannels));
|
|
||||||
} else {
|
|
||||||
ldrLevels[input].push_back(std::vector<uint8_t>(ldrBufferMain, ldrBufferMain + oldWidth * oldHeight * forcedChannels));
|
|
||||||
}
|
|
||||||
|
|
||||||
if (oldWidth == 1 && oldHeight == 1) {
|
width = topLevelWidth;
|
||||||
break;
|
height = topLevelHeight;
|
||||||
}
|
} else {
|
||||||
|
for (int input = 0; input < numInputs; input++) {
|
||||||
|
int level = 0;
|
||||||
|
|
||||||
int newWidth = std::max(1, (int)floorf((float)oldWidth / 2));
|
std::cout << "Loading/scaling " << input << ": " << inputs[input] << std::endl;
|
||||||
int newHeight = std::max(1, (int)floorf((float)oldHeight / 2));
|
|
||||||
|
|
||||||
stbir_colorspace colorspace = srgb ? STBIR_COLORSPACE_SRGB : STBIR_COLORSPACE_LINEAR;
|
|
||||||
int alphaChannel = channels == 4 ? 3 : STBIR_ALPHA_CHANNEL_NONE;
|
|
||||||
|
|
||||||
if (hdr) {
|
if (hdr) {
|
||||||
int rv = stbir_resize_float_generic(hdrBufferMain, oldWidth, oldHeight, 0, hdrBufferOther, newWidth, newHeight, 0, forcedChannels, alphaChannel, 0, STBIR_EDGE_CLAMP, STBIR_FILTER_MITCHELL, colorspace, nullptr);
|
hdrBufferA = stbi_loadf(inputs[input].c_str(), &width, &height, &channels, forcedChannels);
|
||||||
if (rv != 1) {
|
if (hdrBufferA == nullptr) {
|
||||||
std::cerr << "Error resizing" << std::endl;
|
std::cout << "Failed to load image: " << inputs[input] << std::endl;
|
||||||
|
return 1;
|
||||||
}
|
}
|
||||||
std::swap(hdrBufferMain, hdrBufferOther);
|
|
||||||
|
hdrBufferB = new float[width * height * forcedChannels];
|
||||||
|
hdrBufferMain = hdrBufferA;
|
||||||
|
hdrBufferOther = hdrBufferB;
|
||||||
} else {
|
} else {
|
||||||
int rv = stbir_resize_uint8_generic(ldrBufferMain, oldWidth, oldHeight, 0, ldrBufferOther, newWidth, newHeight, 0, forcedChannels, alphaChannel, 0, STBIR_EDGE_CLAMP, STBIR_FILTER_MITCHELL, colorspace, nullptr);
|
ldrBufferA = stbi_load(inputs[input].c_str(), &width, &height, &channels, forcedChannels);
|
||||||
if (rv != 1) {
|
if (ldrBufferA == nullptr) {
|
||||||
std::cerr << "Error resizing" << std::endl;
|
std::cout << "Failed to load image: " << inputs[input] << std::endl;
|
||||||
|
return 1;
|
||||||
}
|
}
|
||||||
std::swap(ldrBufferMain, ldrBufferOther);
|
|
||||||
|
ldrBufferB = new unsigned char[width * height * forcedChannels];
|
||||||
|
ldrBufferMain = ldrBufferA;
|
||||||
|
ldrBufferOther = ldrBufferB;
|
||||||
}
|
}
|
||||||
|
|
||||||
oldWidth = newWidth;
|
int oldWidth = width;
|
||||||
oldHeight = newHeight;
|
int oldHeight = height;
|
||||||
level++;
|
|
||||||
}
|
|
||||||
|
|
||||||
oldWidth = width;
|
levelCount = 1;
|
||||||
oldHeight = height;
|
{
|
||||||
level = 0;
|
int levelWidth = width;
|
||||||
if (hdr) {
|
int levelHeight = height;
|
||||||
free(hdrBufferA);
|
while (levelWidth > 1 || levelHeight > 1) {
|
||||||
delete[] hdrBufferB;
|
levelWidth = std::max(1, (int)floorf((float)levelWidth / 2));
|
||||||
} else {
|
levelHeight = std::max(1, (int)floorf((float)levelHeight / 2));
|
||||||
free(ldrBufferA);
|
levelCount++;
|
||||||
delete[] ldrBufferB;
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
while(1) {
|
||||||
|
if (hdr) {
|
||||||
|
hdrLevels[input].push_back(std::vector<float>(hdrBufferMain, hdrBufferMain + oldWidth * oldHeight * forcedChannels));
|
||||||
|
} else {
|
||||||
|
ldrLevels[input].push_back(std::vector<uint8_t>(ldrBufferMain, ldrBufferMain + oldWidth * oldHeight * forcedChannels));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (oldWidth == 1 && oldHeight == 1) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
int newWidth = std::max(1, (int)floorf((float)oldWidth / 2));
|
||||||
|
int newHeight = std::max(1, (int)floorf((float)oldHeight / 2));
|
||||||
|
|
||||||
|
stbir_colorspace colorspace = srgb ? STBIR_COLORSPACE_SRGB : STBIR_COLORSPACE_LINEAR;
|
||||||
|
int alphaChannel = channels == 4 ? 3 : STBIR_ALPHA_CHANNEL_NONE;
|
||||||
|
|
||||||
|
if (hdr) {
|
||||||
|
int rv = stbir_resize_float_generic(hdrBufferMain, oldWidth, oldHeight, 0, hdrBufferOther, newWidth, newHeight, 0, forcedChannels, alphaChannel, 0, STBIR_EDGE_CLAMP, STBIR_FILTER_MITCHELL, colorspace, nullptr);
|
||||||
|
if (rv != 1) {
|
||||||
|
std::cerr << "Error resizing" << std::endl;
|
||||||
|
}
|
||||||
|
std::swap(hdrBufferMain, hdrBufferOther);
|
||||||
|
} else {
|
||||||
|
int rv = stbir_resize_uint8_generic(ldrBufferMain, oldWidth, oldHeight, 0, ldrBufferOther, newWidth, newHeight, 0, forcedChannels, alphaChannel, 0, STBIR_EDGE_CLAMP, STBIR_FILTER_MITCHELL, colorspace, nullptr);
|
||||||
|
if (rv != 1) {
|
||||||
|
std::cerr << "Error resizing" << std::endl;
|
||||||
|
}
|
||||||
|
std::swap(ldrBufferMain, ldrBufferOther);
|
||||||
|
}
|
||||||
|
|
||||||
|
oldWidth = newWidth;
|
||||||
|
oldHeight = newHeight;
|
||||||
|
level++;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (hdr) {
|
||||||
|
free(hdrBufferA);
|
||||||
|
delete[] hdrBufferB;
|
||||||
|
} else {
|
||||||
|
free(ldrBufferA);
|
||||||
|
delete[] ldrBufferB;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -316,19 +386,19 @@ int main(int argc, char ** argv)
|
||||||
std::tuple<std::string, int, vk::Format> format = formats.find(formatString)->second;
|
std::tuple<std::string, int, vk::Format> format = formats.find(formatString)->second;
|
||||||
size_t blockSize = std::get<1>(format);
|
size_t blockSize = std::get<1>(format);
|
||||||
|
|
||||||
for (int input = 0; input < numInputs; input++) {
|
for (int face = 0; face < numFaces; face++) {
|
||||||
if (numInputs > 1) {
|
if (numInputs > 1) {
|
||||||
if (option == "cube") {
|
if (option == "cube") {
|
||||||
std::cout << "Face " << input << std::endl;
|
std::cout << "Face " << face << std::endl;
|
||||||
} else {
|
} else {
|
||||||
std::cout << "Layer " << input << std::endl;
|
std::cout << "Layer " << face << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (hdr) {
|
if (hdr) {
|
||||||
hdrLevelBlocks[input].resize(levelCount);
|
hdrLevelBlocks[face].resize(levelCount);
|
||||||
} else {
|
} else {
|
||||||
ldrLevelBlocks[input].resize(levelCount);
|
ldrLevelBlocks[face].resize(levelCount);
|
||||||
}
|
}
|
||||||
int level = 0;
|
int level = 0;
|
||||||
|
|
||||||
|
@ -340,7 +410,7 @@ int main(int argc, char ** argv)
|
||||||
unsigned int blocksHeight = (oldHeight + 3) / 4;
|
unsigned int blocksHeight = (oldHeight + 3) / 4;
|
||||||
|
|
||||||
if (hdr) {
|
if (hdr) {
|
||||||
hdrLevelBlocks[input][level].resize(blocksWidth * blocksHeight);
|
hdrLevelBlocks[face][level].resize(blocksWidth * blocksHeight);
|
||||||
|
|
||||||
for (unsigned int y = 0; y < blocksHeight; y++) {
|
for (unsigned int y = 0; y < blocksHeight; y++) {
|
||||||
for (unsigned int x = 0; x < blocksWidth; x++) {
|
for (unsigned int x = 0; x < blocksWidth; x++) {
|
||||||
|
@ -352,7 +422,7 @@ int main(int argc, char ** argv)
|
||||||
unsigned int clampedX = std::min(pixelX, (unsigned int)oldWidth - 1);
|
unsigned int clampedX = std::min(pixelX, (unsigned int)oldWidth - 1);
|
||||||
|
|
||||||
for (int channel = 0; channel < copyChannels; channel++) {
|
for (int channel = 0; channel < copyChannels; channel++) {
|
||||||
float value = hdrLevels[input][level][(clampedY * oldWidth + clampedX) * forcedChannels + channel];
|
float value = hdrLevels[face][level][(clampedY * oldWidth + clampedX) * forcedChannels + channel];
|
||||||
if (value < 0.0f) {
|
if (value < 0.0f) {
|
||||||
value = 0.0f;
|
value = 0.0f;
|
||||||
}
|
}
|
||||||
|
@ -366,11 +436,11 @@ int main(int argc, char ** argv)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
hdrLevelBlocks[input][level][blocksWidth * y + x] = block;
|
hdrLevelBlocks[face][level][blocksWidth * y + x] = block;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
ldrLevelBlocks[input][level].resize(blocksWidth * blocksHeight);
|
ldrLevelBlocks[face][level].resize(blocksWidth * blocksHeight);
|
||||||
|
|
||||||
for (unsigned int y = 0; y < blocksHeight; y++) {
|
for (unsigned int y = 0; y < blocksHeight; y++) {
|
||||||
for (unsigned int x = 0; x < blocksWidth; x++) {
|
for (unsigned int x = 0; x < blocksWidth; x++) {
|
||||||
|
@ -382,12 +452,12 @@ int main(int argc, char ** argv)
|
||||||
unsigned int clampedX = std::min(pixelX, (unsigned int)oldWidth - 1);
|
unsigned int clampedX = std::min(pixelX, (unsigned int)oldWidth - 1);
|
||||||
|
|
||||||
for (int channel = 0; channel < copyChannels; channel++) {
|
for (int channel = 0; channel < copyChannels; channel++) {
|
||||||
block[((pixelY % 4) * 4 + (pixelX % 4)) * copyChannels + channel] = ldrLevels[input][level][(clampedY * oldWidth + clampedX) * forcedChannels + channel];
|
block[((pixelY % 4) * 4 + (pixelX % 4)) * copyChannels + channel] = ldrLevels[face][level][(clampedY * oldWidth + clampedX) * forcedChannels + channel];
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
ldrLevelBlocks[input][level][blocksWidth * y + x] = block;
|
ldrLevelBlocks[face][level][blocksWidth * y + x] = block;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -399,16 +469,19 @@ int main(int argc, char ** argv)
|
||||||
oldWidth = std::max(1, (int)floorf((float)oldWidth / 2));
|
oldWidth = std::max(1, (int)floorf((float)oldWidth / 2));
|
||||||
oldHeight = std::max(1, (int)floorf((float)oldHeight / 2));
|
oldHeight = std::max(1, (int)floorf((float)oldHeight / 2));
|
||||||
level++;
|
level++;
|
||||||
|
|
||||||
|
if (level >= levelCount) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/* Compress */
|
/* Compress */
|
||||||
levelBlocksCompressed[input].resize(levelCount);
|
levelBlocksCompressed[face].resize(levelCount);
|
||||||
for (unsigned int l = 0; l < levelCount; l++) {
|
for (unsigned int l = 0; l < levelCount; l++) {
|
||||||
if (hdr) {
|
if (hdr) {
|
||||||
levelBlocksCompressed[input][l].resize(hdrLevelBlocks[input][l].size() * blockSize);
|
levelBlocksCompressed[face][l].resize(hdrLevelBlocks[face][l].size() * blockSize);
|
||||||
} else {
|
} else {
|
||||||
levelBlocksCompressed[input][l].resize(ldrLevelBlocks[input][l].size() * blockSize);
|
levelBlocksCompressed[face][l].resize(ldrLevelBlocks[face][l].size() * blockSize);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -423,49 +496,49 @@ int main(int argc, char ** argv)
|
||||||
for (unsigned int l = 0; l < levelCount; l++) {
|
for (unsigned int l = 0; l < levelCount; l++) {
|
||||||
unsigned int blocksPerThread;
|
unsigned int blocksPerThread;
|
||||||
if (hdr) {
|
if (hdr) {
|
||||||
blocksPerThread = hdrLevelBlocks[input][l].size() / numThreads;
|
blocksPerThread = hdrLevelBlocks[face][l].size() / numThreads;
|
||||||
} else {
|
} else {
|
||||||
blocksPerThread = ldrLevelBlocks[input][l].size() / numThreads;
|
blocksPerThread = ldrLevelBlocks[face][l].size() / numThreads;
|
||||||
}
|
}
|
||||||
unsigned int startBlock = t * blocksPerThread;
|
unsigned int startBlock = t * blocksPerThread;
|
||||||
unsigned int endBlock = startBlock + blocksPerThread;
|
unsigned int endBlock = startBlock + blocksPerThread;
|
||||||
|
|
||||||
if (hdr) {
|
if (hdr) {
|
||||||
if (t == numThreads - 1) {
|
if (t == numThreads - 1) {
|
||||||
endBlock = hdrLevelBlocks[input][l].size();
|
endBlock = hdrLevelBlocks[face][l].size();
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if (t == numThreads - 1) {
|
if (t == numThreads - 1) {
|
||||||
endBlock = ldrLevelBlocks[input][l].size();
|
endBlock = ldrLevelBlocks[face][l].size();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for (unsigned int b = startBlock; b < endBlock; b++) {
|
for (unsigned int b = startBlock; b < endBlock; b++) {
|
||||||
if (formatString == "BC6H") {
|
if (formatString == "BC6H") {
|
||||||
rgba_surface surface;
|
rgba_surface surface;
|
||||||
surface.ptr = (uint8_t *)hdrLevelBlocks[input][l][b].data();
|
surface.ptr = (uint8_t *)hdrLevelBlocks[face][l][b].data();
|
||||||
surface.width = 4;
|
surface.width = 4;
|
||||||
surface.height = 4;
|
surface.height = 4;
|
||||||
surface.stride = copyChannels * 4 * 2;
|
surface.stride = copyChannels * 4 * 2;
|
||||||
|
|
||||||
CompressBlocksBC6H(&surface, &levelBlocksCompressed[input][l][b * blockSize], &bc6henc);
|
CompressBlocksBC6H(&surface, &levelBlocksCompressed[face][l][b * blockSize], &bc6henc);
|
||||||
} else {
|
} else {
|
||||||
rgba_surface surface;
|
rgba_surface surface;
|
||||||
surface.ptr = ldrLevelBlocks[input][l][b].data();
|
surface.ptr = ldrLevelBlocks[face][l][b].data();
|
||||||
surface.width = 4;
|
surface.width = 4;
|
||||||
surface.height = 4;
|
surface.height = 4;
|
||||||
surface.stride = copyChannels * 4;
|
surface.stride = copyChannels * 4;
|
||||||
|
|
||||||
if (formatString == "BC1" || formatString == "BC1_SRGB") {
|
if (formatString == "BC1" || formatString == "BC1_SRGB") {
|
||||||
CompressBlocksBC1(&surface, &levelBlocksCompressed[input][l][b * blockSize]);
|
CompressBlocksBC1(&surface, &levelBlocksCompressed[face][l][b * blockSize]);
|
||||||
} else if (formatString == "BC3" || formatString == "BC3_SRGB") {
|
} else if (formatString == "BC3" || formatString == "BC3_SRGB") {
|
||||||
CompressBlocksBC3(&surface, &levelBlocksCompressed[input][l][b * blockSize]);
|
CompressBlocksBC3(&surface, &levelBlocksCompressed[face][l][b * blockSize]);
|
||||||
} else if (formatString == "BC4") {
|
} else if (formatString == "BC4") {
|
||||||
CompressBlocksBC4(&surface, &levelBlocksCompressed[input][l][b * blockSize]);
|
CompressBlocksBC4(&surface, &levelBlocksCompressed[face][l][b * blockSize]);
|
||||||
} else if (formatString == "BC5") {
|
} else if (formatString == "BC5") {
|
||||||
CompressBlocksBC5(&surface, &levelBlocksCompressed[input][l][b * blockSize]);
|
CompressBlocksBC5(&surface, &levelBlocksCompressed[face][l][b * blockSize]);
|
||||||
} else if (formatString == "BC7" || formatString == "BC7_SRGB") {
|
} else if (formatString == "BC7" || formatString == "BC7_SRGB") {
|
||||||
CompressBlocksBC7(&surface, &levelBlocksCompressed[input][l][b * blockSize], &bc7enc);
|
CompressBlocksBC7(&surface, &levelBlocksCompressed[face][l][b * blockSize], &bc7enc);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -476,9 +549,9 @@ int main(int argc, char ** argv)
|
||||||
if (l == maxLevel) {
|
if (l == maxLevel) {
|
||||||
float progress;
|
float progress;
|
||||||
if (hdr) {
|
if (hdr) {
|
||||||
progress = (float)completedBlocks[l] / hdrLevelBlocks[input][l].size();
|
progress = (float)completedBlocks[l] / hdrLevelBlocks[face][l].size();
|
||||||
} else {
|
} else {
|
||||||
progress = (float)completedBlocks[l] / ldrLevelBlocks[input][l].size();
|
progress = (float)completedBlocks[l] / ldrLevelBlocks[face][l].size();
|
||||||
}
|
}
|
||||||
int barWidth = 70;
|
int barWidth = 70;
|
||||||
|
|
||||||
|
@ -503,9 +576,9 @@ int main(int argc, char ** argv)
|
||||||
}
|
}
|
||||||
|
|
||||||
if (hdr) {
|
if (hdr) {
|
||||||
hdrLevelBlocks[input].clear();
|
hdrLevelBlocks[face].clear();
|
||||||
} else {
|
} else {
|
||||||
ldrLevelBlocks[input].clear();
|
ldrLevelBlocks[face].clear();
|
||||||
}
|
}
|
||||||
|
|
||||||
int barWidth = 70;
|
int barWidth = 70;
|
||||||
|
@ -612,13 +685,13 @@ int main(int argc, char ** argv)
|
||||||
fh.seekp(levelOffsetBytePosition + (std::ofstream::pos_type)(level * 24));
|
fh.seekp(levelOffsetBytePosition + (std::ofstream::pos_type)(level * 24));
|
||||||
uint64_t byteOffset = levelBytePosition;
|
uint64_t byteOffset = levelBytePosition;
|
||||||
fh.write((char *)&byteOffset, sizeof(byteOffset));
|
fh.write((char *)&byteOffset, sizeof(byteOffset));
|
||||||
uint64_t byteLength = levelBlocksCompressed[0][level].size() * numInputs;
|
uint64_t byteLength = levelBlocksCompressed[0][level].size() * faceCount;
|
||||||
fh.write((char *)&byteLength, sizeof(byteLength));
|
fh.write((char *)&byteLength, sizeof(byteLength));
|
||||||
fh.write((char *)&byteLength, sizeof(byteLength));
|
fh.write((char *)&byteLength, sizeof(byteLength));
|
||||||
fh.seekp(levelBytePosition);
|
fh.seekp(levelBytePosition);
|
||||||
|
|
||||||
for (int input = 0; input < numInputs; input++) {
|
for (int face = 0; face < faceCount; face++) {
|
||||||
fh.write((char *)levelBlocksCompressed[input][level].data(), levelBlocksCompressed[input][level].size());
|
fh.write((char *)levelBlocksCompressed[face][level].data(), levelBlocksCompressed[face][level].size());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue