drewcassidy/quicktex
1
0
Fork 0
mirror of https://github.com/drewcassidy/quicktex.git synced 2024-06-11 06:04:05 +00:00
Code Issues Projects Releases Activity

Rearrange bindings

Browse Source
This commit is contained in:
Andrew Cassidy 2021-03-10 03:49:06 -08:00
parent 04d11112d4
commit 83d547dd8e
4 changed files with 46 additions and 85 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
src/rgbcx/bindings/BlockEncoder.cpp
View File

@ -1,64 +0,0 @@
/* Python-rgbcx Texture Compression Library
Copyright (C) 2021 Andrew Cassidy <drewcassidy@me.com>
Partially derived from rgbcx.h written by Richard Geldreich <richgel99@gmail.com>
and licenced under the public domain
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "../BlockEncoder.h"
#include <pybind11/pybind11.h>
#include <stdexcept>
#include "../bitwiseEnums.h"
#define STRINGIFY(x) #x
#define MACRO_STRINGIFY(x) STRINGIFY(x)
namespace py = pybind11;
namespace rgbcx::bindings {
py::bytes EncodeImage(const BlockEncoder &self, py::bytes decoded, unsigned image_width, unsigned image_height) {
if (image_width % self.BlockWidth() != 0) throw std::invalid_argument("Width is not an even multiple of block_width");
if (image_height % self.BlockHeight() != 0) throw std::invalid_argument("Height is not an even multiple of block_height");
if (image_width == 0 || image_height == 0) throw std::invalid_argument("Image has zero size");
size_t size = image_width * image_height;
size_t block_size = (size / (self.BlockHeight() * self.BlockWidth())) * self.BlockSize();
size_t color_size = size * sizeof(Color);
std::string encoded_str = std::string(block_size, 0);
std::string decoded_str = (std::string)decoded; // decoded data is copied here, unfortunately
if (decoded_str.size() != color_size) throw std::invalid_argument("Incompatible data: image width and height do not match the size of the decoded image");
self.EncodeImage(reinterpret_cast<uint8_t *>(encoded_str.data()), reinterpret_cast<Color *>(decoded_str.data()), image_width, image_height);
auto bytes = py::bytes(encoded_str); // encoded data is copied here, unfortunately
return bytes;
}
void InitBlockEncoder(py::module_ &m) {
py::class_<BlockEncoder> block_encoder(m, "BlockEncoder");
block_encoder.def("encode_image", &EncodeImage);
block_encoder.def_property_readonly("block_size", &BlockEncoder::BlockSize);
block_encoder.def_property_readonly("block_width", &BlockEncoder::BlockWidth);
block_encoder.def_property_readonly("block_height", &BlockEncoder::BlockHeight);
}
} // namespace rgbcx::bindings

14
src/rgbcx/bindings/BlockDecoder.cpp → src/rgbcx/bindings/Decoders.cpp
View File

@ -17,12 +17,12 @@
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "../BlockDecoder.h"
#include <pybind11/pybind11.h>
#include <stdexcept>
#include "../BC1/BC1Decoder.h"
#include "../BlockDecoder.h"
#include "../bitwiseEnums.h"
#define STRINGIFY(x) #x
@ -52,13 +52,21 @@ py::bytes DecodeImage(const BlockDecoder &self, py::bytes encoded, unsigned imag
return bytes;
}
void InitBlockDecoder(py::module_ &m) {
void InitDecoders(py::module_ &m) {
// BlockDecoder
py::class_<BlockDecoder> block_decoder(m, "BlockDecoder");
block_decoder.def("decode_image", &DecodeImage);
block_decoder.def_property_readonly("block_size", &BlockDecoder::BlockSize);
block_decoder.def_property_readonly("block_width", &BlockDecoder::BlockWidth);
block_decoder.def_property_readonly("block_height", &BlockDecoder::BlockHeight);
// BC1Decoder
py::class_<BC1Decoder> bc1_decoder(m, "BC1Decoder", block_decoder);
bc1_decoder.def(py::init<Interpolator::Type, bool>(), py::arg("interpolator") = Interpolator::Type::Ideal, py::arg("write_alpha") = false);
bc1_decoder.def_property_readonly("interpolator_type", &BC1Decoder::GetInterpolatorType);
bc1_decoder.def_readwrite("write_alpha", &BC1Decoder::write_alpha);
}
} // namespace rgbcx::bindings

43
src/rgbcx/bindings/BC1.cpp → src/rgbcx/bindings/Encoders.cpp
View File

@ -19,9 +19,9 @@
#include <pybind11/pybind11.h>
#include "../BC1/BC1Decoder.h"
#include <stdexcept>
#include "../BC1/BC1Encoder.h"
#include "../BlockDecoder.h"
#include "../BlockEncoder.h"
#include "../bitwiseEnums.h"
@ -31,9 +31,35 @@
namespace py = pybind11;
namespace rgbcx::bindings {
void InitBC1(py::module_ &m) {
auto block_encoder = py::type::of<BlockEncoder>();
auto block_decoder = py::type::of<BlockDecoder>();
py::bytes EncodeImage(const BlockEncoder &self, py::bytes decoded, unsigned image_width, unsigned image_height) {
if (image_width % self.BlockWidth() != 0) throw std::invalid_argument("Width is not an even multiple of block_width");
if (image_height % self.BlockHeight() != 0) throw std::invalid_argument("Height is not an even multiple of block_height");
if (image_width == 0 || image_height == 0) throw std::invalid_argument("Image has zero size");
size_t size = image_width * image_height;
size_t block_size = (size / (self.BlockHeight() * self.BlockWidth())) * self.BlockSize();
size_t color_size = size * sizeof(Color);
std::string encoded_str = std::string(block_size, 0);
std::string decoded_str = (std::string)decoded; // decoded data is copied here, unfortunately
if (decoded_str.size() != color_size) throw std::invalid_argument("Incompatible data: image width and height do not match the size of the decoded image");
self.EncodeImage(reinterpret_cast<uint8_t *>(encoded_str.data()), reinterpret_cast<Color *>(decoded_str.data()), image_width, image_height);
auto bytes = py::bytes(encoded_str); // encoded data is copied here, unfortunately
return bytes;
}
void InitEncoders(py::module_ &m) {
// BlockEncoder
py::class_<BlockEncoder> block_encoder(m, "BlockEncoder");
block_encoder.def("encode_image", &EncodeImage);
block_encoder.def_property_readonly("block_size", &BlockEncoder::BlockSize);
block_encoder.def_property_readonly("block_width", &BlockEncoder::BlockWidth);
block_encoder.def_property_readonly("block_height", &BlockEncoder::BlockHeight);
// BC1Encoder
py::class_<BC1Encoder> bc1_encoder(m, "BC1Encoder", block_encoder);
@ -78,13 +104,6 @@ void InitBC1(py::module_ &m) {
.value("Faster", BC1Encoder::ErrorMode::Faster)
.value("Check2", BC1Encoder::ErrorMode::Check2)
.value("Full", BC1Encoder::ErrorMode::Full);
// BC1Decoder
py::class_<BC1Decoder> bc1_decoder(m, "BC1Decoder", block_decoder);
bc1_decoder.def(py::init<Interpolator::Type, bool>(), py::arg("interpolator") = Interpolator::Type::Ideal, py::arg("write_alpha") = false);
bc1_decoder.def_property_readonly("interpolator_type", &BC1Decoder::GetInterpolatorType);
bc1_decoder.def_readwrite("write_alpha", &BC1Decoder::write_alpha);
}
} // namespace rgbcx::bindings

10
src/rgbcx/bindings/Module.cpp
View File

@ -24,9 +24,8 @@
namespace py = pybind11;
namespace rgbcx::bindings {
void InitBlockEncoder(py::module_ &m);
void InitBlockDecoder(py::module_ &m);
void InitBC1(py::module_ &m);
void InitEncoders(py::module_ &m);
void InitDecoders(py::module_ &m);
PYBIND11_MODULE(_rgbcx, m) {
m.doc() = "More Stuff";
@ -38,9 +37,8 @@ PYBIND11_MODULE(_rgbcx, m) {
.value("Nvidia", IType::Nvidia)
.value("AMD", IType::AMD);
InitBlockEncoder(m);
InitBlockDecoder(m);
InitBC1(m);
InitEncoders(m);
InitDecoders(m);
}
} // namespace rgbcx::bindings