391 lines
12 KiB
C++
391 lines
12 KiB
C++
|
/*
|
||
|
|
* Copyright 2015 The Etc2Comp Authors.
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
* you may not use this file except in compliance with the License.
|
||
|
|
* You may obtain a copy of the License at
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
* See the License for the specific language governing permissions and
|
||
|
|
* limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifdef _WIN32
|
||
|
|
#define _CRT_SECURE_NO_WARNINGS (1)
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include "EtcConfig.h"
|
||
|
|
|
||
|
|
|
||
|
|
#include "EtcFile.h"
|
||
|
|
|
||
|
|
#include "EtcFileHeader.h"
|
||
|
|
#include "EtcColor.h"
|
||
|
|
#include "Etc.h"
|
||
|
|
#include "EtcBlock4x4EncodingBits.h"
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <assert.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
|
||
|
|
using namespace Etc;
|
||
|
|
|
||
|
|
// ----------------------------------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
File::File(const char *a_pstrFilename, Format a_fileformat, Image::Format a_imageformat,
|
||
|
|
unsigned char *a_paucEncodingBits, unsigned int a_uiEncodingBitsBytes,
|
||
|
|
unsigned int a_uiSourceWidth, unsigned int a_uiSourceHeight,
|
||
|
|
unsigned int a_uiExtendedWidth, unsigned int a_uiExtendedHeight)
|
||
|
|
{
|
||
|
|
if (a_pstrFilename == nullptr)
|
||
|
|
{
|
||
|
|
m_pstrFilename = const_cast<char *>("");
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
m_pstrFilename = new char[strlen(a_pstrFilename) + 1];
|
||
|
|
strcpy(m_pstrFilename, a_pstrFilename);
|
||
|
|
}
|
||
|
|
|
||
|
|
m_fileformat = a_fileformat;
|
||
|
|
if (m_fileformat == Format::INFER_FROM_FILE_EXTENSION)
|
||
|
|
{
|
||
|
|
// ***** TODO: add this later *****
|
||
|
|
m_fileformat = Format::KTX;
|
||
|
|
}
|
||
|
|
|
||
|
|
m_imageformat = a_imageformat;
|
||
|
|
|
||
|
|
m_uiNumMipmaps = 1;
|
||
|
|
m_pMipmapImages = new RawImage[m_uiNumMipmaps];
|
||
|
|
m_pMipmapImages[0].paucEncodingBits = std::shared_ptr<unsigned char>(a_paucEncodingBits, [](unsigned char *p) { delete[] p; } );
|
||
|
|
m_pMipmapImages[0].uiEncodingBitsBytes = a_uiEncodingBitsBytes;
|
||
|
|
m_pMipmapImages[0].uiExtendedWidth = a_uiExtendedWidth;
|
||
|
|
m_pMipmapImages[0].uiExtendedHeight = a_uiExtendedHeight;
|
||
|
|
|
||
|
|
m_uiSourceWidth = a_uiSourceWidth;
|
||
|
|
m_uiSourceHeight = a_uiSourceHeight;
|
||
|
|
|
||
|
|
switch (m_fileformat)
|
||
|
|
{
|
||
|
|
case Format::PKM:
|
||
|
|
m_pheader = new FileHeader_Pkm(this);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case Format::KTX:
|
||
|
|
m_pheader = new FileHeader_Ktx(this);
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
assert(0);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// ----------------------------------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
File::File(const char *a_pstrFilename, Format a_fileformat, Image::Format a_imageformat,
|
||
|
|
unsigned int a_uiNumMipmaps, RawImage *a_pMipmapImages,
|
||
|
|
unsigned int a_uiSourceWidth, unsigned int a_uiSourceHeight)
|
||
|
|
{
|
||
|
|
if (a_pstrFilename == nullptr)
|
||
|
|
{
|
||
|
|
m_pstrFilename = const_cast<char *>("");
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
m_pstrFilename = new char[strlen(a_pstrFilename) + 1];
|
||
|
|
strcpy(m_pstrFilename, a_pstrFilename);
|
||
|
|
}
|
||
|
|
|
||
|
|
m_fileformat = a_fileformat;
|
||
|
|
if (m_fileformat == Format::INFER_FROM_FILE_EXTENSION)
|
||
|
|
{
|
||
|
|
// ***** TODO: add this later *****
|
||
|
|
m_fileformat = Format::KTX;
|
||
|
|
}
|
||
|
|
|
||
|
|
m_imageformat = a_imageformat;
|
||
|
|
|
||
|
|
m_uiNumMipmaps = a_uiNumMipmaps;
|
||
|
|
m_pMipmapImages = new RawImage[m_uiNumMipmaps];
|
||
|
|
|
||
|
|
for(unsigned int mip = 0; mip < m_uiNumMipmaps; mip++)
|
||
|
|
{
|
||
|
|
m_pMipmapImages[mip] = a_pMipmapImages[mip];
|
||
|
|
}
|
||
|
|
|
||
|
|
m_uiSourceWidth = a_uiSourceWidth;
|
||
|
|
m_uiSourceHeight = a_uiSourceHeight;
|
||
|
|
|
||
|
|
switch (m_fileformat)
|
||
|
|
{
|
||
|
|
case Format::PKM:
|
||
|
|
m_pheader = new FileHeader_Pkm(this);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case Format::KTX:
|
||
|
|
m_pheader = new FileHeader_Ktx(this);
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
assert(0);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// ----------------------------------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
File::File(const char *a_pstrFilename, Format a_fileformat)
|
||
|
|
{
|
||
|
|
if (a_pstrFilename == nullptr)
|
||
|
|
{
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
m_pstrFilename = new char[strlen(a_pstrFilename) + 1];
|
||
|
|
strcpy(m_pstrFilename, a_pstrFilename);
|
||
|
|
}
|
||
|
|
|
||
|
|
m_fileformat = a_fileformat;
|
||
|
|
if (m_fileformat == Format::INFER_FROM_FILE_EXTENSION)
|
||
|
|
{
|
||
|
|
// ***** TODO: add this later *****
|
||
|
|
m_fileformat = Format::KTX;
|
||
|
|
}
|
||
|
|
|
||
|
|
FILE *pfile = fopen(m_pstrFilename, "rb");
|
||
|
|
if (pfile == nullptr)
|
||
|
|
{
|
||
|
|
printf("ERROR: Couldn't open %s", m_pstrFilename);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
fseek(pfile, 0, SEEK_END);
|
||
|
|
unsigned int fileSize = ftell(pfile);
|
||
|
|
fseek(pfile, 0, SEEK_SET);
|
||
|
|
size_t szResult;
|
||
|
|
|
||
|
|
m_pheader = new FileHeader_Ktx(this);
|
||
|
|
szResult = fread( ((FileHeader_Ktx*)m_pheader)->GetData(), 1, sizeof(FileHeader_Ktx::Data), pfile);
|
||
|
|
assert(szResult > 0);
|
||
|
|
|
||
|
|
m_uiNumMipmaps = 1;
|
||
|
|
m_pMipmapImages = new RawImage[m_uiNumMipmaps];
|
||
|
|
|
||
|
|
if (((FileHeader_Ktx*)m_pheader)->GetData()->m_u32BytesOfKeyValueData > 0)
|
||
|
|
fseek(pfile, ((FileHeader_Ktx*)m_pheader)->GetData()->m_u32BytesOfKeyValueData, SEEK_CUR);
|
||
|
|
szResult = fread(&m_pMipmapImages->uiEncodingBitsBytes, 1, sizeof(unsigned int), pfile);
|
||
|
|
assert(szResult > 0);
|
||
|
|
|
||
|
|
m_pMipmapImages->paucEncodingBits = std::shared_ptr<unsigned char>(new unsigned char[m_pMipmapImages->uiEncodingBitsBytes], [](unsigned char *p) { delete[] p; } );
|
||
|
|
assert(ftell(pfile) + m_pMipmapImages->uiEncodingBitsBytes <= fileSize);
|
||
|
|
szResult = fread(m_pMipmapImages->paucEncodingBits.get(), 1, m_pMipmapImages->uiEncodingBitsBytes, pfile);
|
||
|
|
assert(szResult == m_pMipmapImages->uiEncodingBitsBytes);
|
||
|
|
|
||
|
|
uint32_t uiInternalFormat = ((FileHeader_Ktx*)m_pheader)->GetData()->m_u32GlInternalFormat;
|
||
|
|
uint32_t uiBaseInternalFormat = ((FileHeader_Ktx*)m_pheader)->GetData()->m_u32GlBaseInternalFormat;
|
||
|
|
|
||
|
|
if (uiInternalFormat == (uint32_t)FileHeader_Ktx::InternalFormat::ETC1_RGB8 && uiBaseInternalFormat == (uint32_t)FileHeader_Ktx::BaseInternalFormat::ETC1_RGB8)
|
||
|
|
{
|
||
|
|
m_imageformat = Image::Format::ETC1;
|
||
|
|
}
|
||
|
|
else if (uiInternalFormat == (uint32_t)FileHeader_Ktx::InternalFormat::ETC2_RGB8 && uiBaseInternalFormat == (uint32_t)FileHeader_Ktx::BaseInternalFormat::ETC2_RGB8)
|
||
|
|
{
|
||
|
|
m_imageformat = Image::Format::RGB8;
|
||
|
|
}
|
||
|
|
else if (uiInternalFormat == (uint32_t)FileHeader_Ktx::InternalFormat::ETC2_RGB8A1 && uiBaseInternalFormat == (uint32_t)FileHeader_Ktx::BaseInternalFormat::ETC2_RGB8A1)
|
||
|
|
{
|
||
|
|
m_imageformat = Image::Format::RGB8A1;
|
||
|
|
}
|
||
|
|
else if (uiInternalFormat == (uint32_t)FileHeader_Ktx::InternalFormat::ETC2_RGBA8 && uiBaseInternalFormat == (uint32_t)FileHeader_Ktx::BaseInternalFormat::ETC2_RGBA8)
|
||
|
|
{
|
||
|
|
m_imageformat = Image::Format::RGBA8;
|
||
|
|
}
|
||
|
|
else if (uiInternalFormat == (uint32_t)FileHeader_Ktx::InternalFormat::ETC2_R11 && uiBaseInternalFormat == (uint32_t)FileHeader_Ktx::BaseInternalFormat::ETC2_R11)
|
||
|
|
{
|
||
|
|
m_imageformat = Image::Format::R11;
|
||
|
|
}
|
||
|
|
else if (uiInternalFormat == (uint32_t)FileHeader_Ktx::InternalFormat::ETC2_SIGNED_R11 && uiBaseInternalFormat == (uint32_t)FileHeader_Ktx::BaseInternalFormat::ETC2_R11)
|
||
|
|
{
|
||
|
|
m_imageformat = Image::Format::SIGNED_R11;
|
||
|
|
}
|
||
|
|
else if (uiInternalFormat == (uint32_t)FileHeader_Ktx::InternalFormat::ETC2_RG11 && uiBaseInternalFormat == (uint32_t)FileHeader_Ktx::BaseInternalFormat::ETC2_RG11)
|
||
|
|
{
|
||
|
|
m_imageformat = Image::Format::RG11;
|
||
|
|
}
|
||
|
|
else if (uiInternalFormat == (uint32_t)FileHeader_Ktx::InternalFormat::ETC2_SIGNED_RG11 && uiBaseInternalFormat == (uint32_t)FileHeader_Ktx::BaseInternalFormat::ETC2_RG11)
|
||
|
|
{
|
||
|
|
m_imageformat = Image::Format::SIGNED_RG11;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
m_imageformat = Image::Format::UNKNOWN;
|
||
|
|
}
|
||
|
|
|
||
|
|
m_uiSourceWidth = ((FileHeader_Ktx*)m_pheader)->GetData()->m_u32PixelWidth;
|
||
|
|
m_uiSourceHeight = ((FileHeader_Ktx*)m_pheader)->GetData()->m_u32PixelHeight;
|
||
|
|
m_pMipmapImages->uiExtendedWidth = Image::CalcExtendedDimension((unsigned short)m_uiSourceWidth);
|
||
|
|
m_pMipmapImages->uiExtendedHeight = Image::CalcExtendedDimension((unsigned short)m_uiSourceHeight);
|
||
|
|
|
||
|
|
unsigned int uiBlocks = m_pMipmapImages->uiExtendedWidth * m_pMipmapImages->uiExtendedHeight / 16;
|
||
|
|
Block4x4EncodingBits::Format encodingbitsformat = Image::DetermineEncodingBitsFormat(m_imageformat);
|
||
|
|
unsigned int expectedbytes = uiBlocks * Block4x4EncodingBits::GetBytesPerBlock(encodingbitsformat);
|
||
|
|
assert(expectedbytes == m_pMipmapImages->uiEncodingBitsBytes);
|
||
|
|
|
||
|
|
fclose(pfile);
|
||
|
|
}
|
||
|
|
|
||
|
|
File::~File()
|
||
|
|
{
|
||
|
|
if (m_pMipmapImages != nullptr)
|
||
|
|
{
|
||
|
|
delete [] m_pMipmapImages;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(m_pstrFilename != nullptr)
|
||
|
|
{
|
||
|
|
delete[] m_pstrFilename;
|
||
|
|
m_pstrFilename = nullptr;
|
||
|
|
}
|
||
|
|
if (m_pheader != nullptr)
|
||
|
|
{
|
||
|
|
delete m_pheader;
|
||
|
|
m_pheader = nullptr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void File::UseSingleBlock(int a_iPixelX, int a_iPixelY)
|
||
|
|
{
|
||
|
|
if (a_iPixelX <= -1 || a_iPixelY <= -1)
|
||
|
|
return;
|
||
|
|
if (a_iPixelX >(int) m_uiSourceWidth)
|
||
|
|
{
|
||
|
|
//if we are using a ktx thats the size of a single block or less
|
||
|
|
//then make sure we use the 4x4 image as the single block
|
||
|
|
if (m_uiSourceWidth <= 4)
|
||
|
|
{
|
||
|
|
a_iPixelX = 0;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
printf("blockAtHV: H coordinate out of range, capped to image width\n");
|
||
|
|
a_iPixelX = m_uiSourceWidth - 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (a_iPixelY >(int) m_uiSourceHeight)
|
||
|
|
{
|
||
|
|
//if we are using a ktx thats the size of a single block or less
|
||
|
|
//then make sure we use the 4x4 image as the single block
|
||
|
|
if (m_uiSourceHeight <= 4)
|
||
|
|
{
|
||
|
|
a_iPixelY= 0;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
printf("blockAtHV: V coordinate out of range, capped to image height\n");
|
||
|
|
a_iPixelY = m_uiSourceHeight - 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
unsigned int origWidth = m_uiSourceWidth;
|
||
|
|
unsigned int origHeight = m_uiSourceHeight;
|
||
|
|
|
||
|
|
m_uiSourceWidth = 4;
|
||
|
|
m_uiSourceHeight = 4;
|
||
|
|
|
||
|
|
Block4x4EncodingBits::Format encodingbitsformat = Image::DetermineEncodingBitsFormat(m_imageformat);
|
||
|
|
unsigned int uiEncodingBitsBytesPerBlock = Block4x4EncodingBits::GetBytesPerBlock(encodingbitsformat);
|
||
|
|
|
||
|
|
int numMipmaps = 1;
|
||
|
|
RawImage* pMipmapImages = new RawImage[numMipmaps];
|
||
|
|
pMipmapImages[0].uiExtendedWidth = Image::CalcExtendedDimension((unsigned short)m_uiSourceWidth);
|
||
|
|
pMipmapImages[0].uiExtendedHeight = Image::CalcExtendedDimension((unsigned short)m_uiSourceHeight);
|
||
|
|
pMipmapImages[0].uiEncodingBitsBytes = 0;
|
||
|
|
pMipmapImages[0].paucEncodingBits = std::shared_ptr<unsigned char>(new unsigned char[uiEncodingBitsBytesPerBlock], [](unsigned char *p) { delete[] p; });
|
||
|
|
|
||
|
|
//block position in pixels
|
||
|
|
// remove the bottom 2 bits to get the block coordinates
|
||
|
|
unsigned int iBlockPosX = (a_iPixelX & 0xFFFFFFFC);
|
||
|
|
unsigned int iBlockPosY = (a_iPixelY & 0xFFFFFFFC);
|
||
|
|
|
||
|
|
int numXBlocks = (origWidth / 4);
|
||
|
|
int numYBlocks = (origHeight / 4);
|
||
|
|
|
||
|
|
|
||
|
|
// block location
|
||
|
|
//int iBlockX = (a_iPixelX % 4) == 0 ? a_iPixelX / 4.0f : (a_iPixelX / 4) + 1;
|
||
|
|
//int iBlockY = (a_iPixelY % 4) == 0 ? a_iPixelY / 4.0f : (a_iPixelY / 4) + 1;
|
||
|
|
//m_paucEncodingBits += ((iBlockY * numXBlocks) + iBlockX) * uiEncodingBitsBytesPerBlock;
|
||
|
|
|
||
|
|
|
||
|
|
unsigned int num = numXBlocks*numYBlocks;
|
||
|
|
unsigned int uiH = 0, uiV = 0;
|
||
|
|
unsigned char* pEncodingBits = m_pMipmapImages[0].paucEncodingBits.get();
|
||
|
|
for (unsigned int uiBlock = 0; uiBlock < num; uiBlock++)
|
||
|
|
{
|
||
|
|
if (uiH == iBlockPosX && uiV == iBlockPosY)
|
||
|
|
{
|
||
|
|
memcpy(pMipmapImages[0].paucEncodingBits.get(),pEncodingBits, uiEncodingBitsBytesPerBlock);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
pEncodingBits += uiEncodingBitsBytesPerBlock;
|
||
|
|
uiH += 4;
|
||
|
|
|
||
|
|
if (uiH >= origWidth)
|
||
|
|
{
|
||
|
|
uiH = 0;
|
||
|
|
uiV += 4;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
delete [] m_pMipmapImages;
|
||
|
|
m_pMipmapImages = pMipmapImages;
|
||
|
|
}
|
||
|
|
// ----------------------------------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
void File::Write()
|
||
|
|
{
|
||
|
|
|
||
|
|
FILE *pfile = fopen(m_pstrFilename, "wb");
|
||
|
|
if (pfile == nullptr)
|
||
|
|
{
|
||
|
|
printf("Error: couldn't open Etc file (%s)\n", m_pstrFilename);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
m_pheader->Write(pfile);
|
||
|
|
|
||
|
|
for(unsigned int mip = 0; mip < m_uiNumMipmaps; mip++)
|
||
|
|
{
|
||
|
|
if(m_fileformat == Format::KTX)
|
||
|
|
{
|
||
|
|
// Write u32 image size
|
||
|
|
uint32_t u32ImageSize = m_pMipmapImages[mip].uiEncodingBitsBytes;
|
||
|
|
uint32_t szBytesWritten = fwrite(&u32ImageSize, 1, sizeof(u32ImageSize), pfile);
|
||
|
|
assert(szBytesWritten == sizeof(u32ImageSize));
|
||
|
|
}
|
||
|
|
|
||
|
|
unsigned int iResult = (int)fwrite(m_pMipmapImages[mip].paucEncodingBits.get(), 1, m_pMipmapImages[mip].uiEncodingBitsBytes, pfile);
|
||
|
|
if (iResult != m_pMipmapImages[mip].uiEncodingBitsBytes)
|
||
|
|
{
|
||
|
|
printf("Error: couldn't write Etc file (%s)\n", m_pstrFilename);
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fclose(pfile);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// ----------------------------------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
|