From 552b8440b17dbc754f3fe854a6e157d3bc13cf84 Mon Sep 17 00:00:00 2001 From: Pavel Krajcevski Date: Thu, 23 Jan 2014 18:24:11 -0500 Subject: [PATCH] Add KTX writer. --- IO/CMakeLists.txt | 2 + IO/src/ImageWriterKTX.cpp | 148 ++++++++++++++++++++++++++++++++++++++ IO/src/ImageWriterKTX.h | 71 ++++++++++++++++++ 3 files changed, 221 insertions(+) create mode 100644 IO/src/ImageWriterKTX.cpp create mode 100644 IO/src/ImageWriterKTX.h diff --git a/IO/CMakeLists.txt b/IO/CMakeLists.txt index 414a8dc..49bc4de 100644 --- a/IO/CMakeLists.txt +++ b/IO/CMakeLists.txt @@ -90,6 +90,8 @@ SET( HEADERS ${HEADERS} "src/ImageLoaderTGA.h" ) # Add KTX loaders SET( SOURCES ${SOURCES} "src/ImageLoaderKTX.cpp" ) SET( HEADERS ${HEADERS} "src/ImageLoaderKTX.h" ) +SET( SOURCES ${SOURCES} "src/ImageWriterKTX.cpp" ) +SET( HEADERS ${HEADERS} "src/ImageWriterKTX.h" ) FIND_PACKAGE( OpenGL ) IF(OPENGL_FOUND) diff --git a/IO/src/ImageWriterKTX.cpp b/IO/src/ImageWriterKTX.cpp new file mode 100644 index 0000000..1a6a371 --- /dev/null +++ b/IO/src/ImageWriterKTX.cpp @@ -0,0 +1,148 @@ +/* FasTC + * Copyright (c) 2014 University of North Carolina at Chapel Hill. + * All rights reserved. + * + * Permission to use, copy, modify, and distribute this software and its + * documentation for educational, research, and non-profit purposes, without + * fee, and without a written agreement is hereby granted, provided that the + * above copyright notice, this paragraph, and the following four paragraphs + * appear in all copies. + * + * Permission to incorporate this software into commercial products may be + * obtained by contacting the authors or the Office of Technology Development + * at the University of North Carolina at Chapel Hill . + * + * This software program and documentation are copyrighted by the University of + * North Carolina at Chapel Hill. The software program and documentation are + * supplied "as is," without any accompanying services from the University of + * North Carolina at Chapel Hill or the authors. The University of North + * Carolina at Chapel Hill and the authors do not warrant that the operation of + * the program will be uninterrupted or error-free. The end-user understands + * that the program was developed for research purposes and is advised not to + * rely exclusively on the program for any reason. + * + * IN NO EVENT SHALL THE UNIVERSITY OF NORTH CAROLINA AT CHAPEL HILL OR THE + * AUTHORS BE LIABLE TO ANY PARTY FOR DIRECT, INDIRECT, SPECIAL, INCIDENTAL, + * OR CONSEQUENTIAL DAMAGES, INCLUDING LOST PROFITS, ARISING OUT OF THE USE OF + * THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF THE UNIVERSITY OF NORTH CAROLINA + * AT CHAPEL HILL OR THE AUTHORS HAVE BEEN ADVISED OF THE POSSIBILITY OF SUCH + * DAMAGE. + * + * THE UNIVERSITY OF NORTH CAROLINA AT CHAPEL HILL AND THE AUTHORS SPECIFICALLY + * DISCLAIM ANY WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE AND ANY + * STATUTORY WARRANTY OF NON-INFRINGEMENT. THE SOFTWARE PROVIDED HEREUNDER IS ON + * AN "AS IS" BASIS, AND THE UNIVERSITY OF NORTH CAROLINA AT CHAPEL HILL AND + * THE AUTHORS HAVE NO OBLIGATIONS TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, + * ENHANCEMENTS, OR MODIFICATIONS. + * + * Please send all BUG REPORTS to . + * + * The authors may be contacted via: + * + * Pavel Krajcevski + * Dept of Computer Science + * 201 S Columbia St + * Frederick P. Brooks, Jr. Computer Science Bldg + * Chapel Hill, NC 27599-3175 + * USA + * + * + */ + +#include "ImageWriterKTX.h" + +#include +#include +#include + +#include "Image.h" +#include "Pixel.h" + +#include "CompressedImage.h" +#include "GLDefines.h" + +ImageWriterKTX::ImageWriterKTX(FasTC::Image<> &im) + : ImageWriter(im.GetWidth(), im.GetHeight(), NULL) + , m_Image(im) +{ } + +class ByteWriter { + private: + uint8 *m_Head; + uint32 m_BytesWritten; + public: + ByteWriter(uint8 *dst) : m_Head(dst), m_BytesWritten(0) { } + + uint32 GetBytesWritten() const { return m_BytesWritten; } + + void Write(const uint32 v) { + memcpy(m_Head, &v, 4); + m_Head += 4; + m_BytesWritten += 4; + } + + void Write(const void *src, const uint32 nBytes) { + memcpy(m_Head, src, nBytes); + m_Head += nBytes; + m_BytesWritten += nBytes; + } +}; + +bool ImageWriterKTX::WriteImage() { + ByteWriter wtr (m_RawFileData); + + const uint8 kIdentifier[12] = { + 0xAB, 0x4B, 0x54, 0x58, 0x20, 0x31, 0x31, 0xBB, 0x0D, 0x0A, 0x1A, 0x0A + }; + wtr.Write(kIdentifier, 12); + wtr.Write(0x04030201); + + const char *orientationKey = "KTXorientation"; + const char *orientationValue = "S=r,T=u"; + const uint32 kvSz = + strlen(orientationKey) + 1 // key + + strlen(orientationValue) + 1; // value + uint32 tkvSz = kvSz + 4; // total kv size + tkvSz = (tkvSz + 3) & ~0x3; // 4-byte aligned + + CompressedImage *ci = dynamic_cast(&m_Image); + if(ci && ci->GetFormat() == FasTC::eCompressionFormat_BPTC) { + wtr.Write(0); // glType + wtr.Write(1); // glTypeSize + wtr.Write(GL_RGBA); // glFormat + wtr.Write(GL_COMPRESSED_RGBA_BPTC_UNORM); // glInternalFormat + wtr.Write(GL_RGBA); // glBaseFormat + } else { + wtr.Write(GL_BYTE); // glType + wtr.Write(1); // glTypeSize + wtr.Write(GL_RGBA); // glFormat + wtr.Write(GL_RGBA8); // glInternalFormat + wtr.Write(GL_RGBA); // glBaseFormat + } + + wtr.Write(m_Width); // pixelWidth + wtr.Write(m_Height); // pixelHeight + wtr.Write(0); // pixelDepth + wtr.Write(0); // numberOfArrayElements + wtr.Write(1); // numberOfFaces + wtr.Write(1); // numberOfMipmapLevels + wtr.Write(tkvSz); // total key value size + wtr.Write(kvSz); // key value size + wtr.Write(orientationKey, strlen(orientationKey) + 1); // key + wtr.Write(orientationValue, strlen(orientationValue) + 1); // value + wtr.Write(orientationKey, tkvSz - kvSz); // padding + + if(ci && ci->GetFormat() == FasTC::eCompressionFormat_BPTC) { + static const uint32 kImageSize = m_Width * m_Height; + wtr.Write(kImageSize); // imageSize + wtr.Write(ci->GetCompressedData(), kImageSize); // imagedata... + } else { + static const uint32 kImageSize = m_Width * m_Height * 4; + wtr.Write(kImageSize); // imageSize + wtr.Write(m_Image.GetPixels(), kImageSize); // imagedata... + } + + m_RawFileDataSz = wtr.GetBytesWritten(); + return true; +} diff --git a/IO/src/ImageWriterKTX.h b/IO/src/ImageWriterKTX.h new file mode 100644 index 0000000..2bc3d37 --- /dev/null +++ b/IO/src/ImageWriterKTX.h @@ -0,0 +1,71 @@ +/* FasTC + * Copyright (c) 2014 University of North Carolina at Chapel Hill. + * All rights reserved. + * + * Permission to use, copy, modify, and distribute this software and its + * documentation for educational, research, and non-profit purposes, without + * fee, and without a written agreement is hereby granted, provided that the + * above copyright notice, this paragraph, and the following four paragraphs + * appear in all copies. + * + * Permission to incorporate this software into commercial products may be + * obtained by contacting the authors or the Office of Technology Development + * at the University of North Carolina at Chapel Hill . + * + * This software program and documentation are copyrighted by the University of + * North Carolina at Chapel Hill. The software program and documentation are + * supplied "as is," without any accompanying services from the University of + * North Carolina at Chapel Hill or the authors. The University of North + * Carolina at Chapel Hill and the authors do not warrant that the operation of + * the program will be uninterrupted or error-free. The end-user understands + * that the program was developed for research purposes and is advised not to + * rely exclusively on the program for any reason. + * + * IN NO EVENT SHALL THE UNIVERSITY OF NORTH CAROLINA AT CHAPEL HILL OR THE + * AUTHORS BE LIABLE TO ANY PARTY FOR DIRECT, INDIRECT, SPECIAL, INCIDENTAL, + * OR CONSEQUENTIAL DAMAGES, INCLUDING LOST PROFITS, ARISING OUT OF THE USE OF + * THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF THE UNIVERSITY OF NORTH CAROLINA + * AT CHAPEL HILL OR THE AUTHORS HAVE BEEN ADVISED OF THE POSSIBILITY OF SUCH + * DAMAGE. + * + * THE UNIVERSITY OF NORTH CAROLINA AT CHAPEL HILL AND THE AUTHORS SPECIFICALLY + * DISCLAIM ANY WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE AND ANY + * STATUTORY WARRANTY OF NON-INFRINGEMENT. THE SOFTWARE PROVIDED HEREUNDER IS ON + * AN "AS IS" BASIS, AND THE UNIVERSITY OF NORTH CAROLINA AT CHAPEL HILL AND + * THE AUTHORS HAVE NO OBLIGATIONS TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, + * ENHANCEMENTS, OR MODIFICATIONS. + * + * Please send all BUG REPORTS to . + * + * The authors may be contacted via: + * + * Pavel Krajcevski + * Dept of Computer Science + * 201 S Columbia St + * Frederick P. Brooks, Jr. Computer Science Bldg + * Chapel Hill, NC 27599-3175 + * USA + * + * + */ + +#ifndef _IMAGE_WRITER_KTX_H_ +#define _IMAGE_WRITER_KTX_H_ + +#include "ImageWriter.h" +#include "ImageFwd.h" + +// Forward Declare +class ImageWriterKTX : public ImageWriter { + public: + ImageWriterKTX(FasTC::Image<> &); + virtual ~ImageWriterKTX() { } + + virtual bool WriteImage(); + + private: + FasTC::Image<> &m_Image; +}; + +#endif // _IMAGE_LOADER_H_