You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
4.8 KiB
C++
96 lines
4.8 KiB
C++
/* 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 <pybind11/pybind11.h>
|
|
|
|
#include <array>
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <stdexcept>
|
|
#include <string>
|
|
|
|
#include "../../BlockDecoder.h"
|
|
#include "../../BlockEncoder.h"
|
|
#include "../Interpolator.h"
|
|
#include "BC1Decoder.h"
|
|
#include "BC1Encoder.h"
|
|
|
|
namespace py = pybind11;
|
|
namespace quicktex::bindings {
|
|
|
|
using namespace quicktex::s3tc;
|
|
using namespace quicktex::s3tc ;
|
|
|
|
void InitBC1(py::module_ &s3tc) {
|
|
auto bc1 = s3tc.def_submodule("_bc1", "BC1 encoding/decoding module");
|
|
auto block_encoder = py::type::of<BlockEncoder>();
|
|
auto block_decoder = py::type::of<BlockDecoder>();
|
|
|
|
// BC1Encoder
|
|
py::class_<BC1Encoder> bc1_encoder(bc1, "BC1Encoder", block_encoder);
|
|
|
|
bc1_encoder.def(py::init<Interpolator::Type, unsigned, bool, bool>(), py::arg("interpolator") = Interpolator::Type::Ideal, py::arg("level") = 5,
|
|
py::arg("use_3color") = true, py::arg("use_3color_black") = true);
|
|
bc1_encoder.def("set_level", &BC1Encoder::SetLevel);
|
|
bc1_encoder.def_property_readonly("interpolator_type", &BC1Encoder::GetInterpolatorType);
|
|
bc1_encoder.def_property("flags", &BC1Encoder::GetFlags, &BC1Encoder::SetFlags);
|
|
bc1_encoder.def_property("error_mode", &BC1Encoder::GetErrorMode, &BC1Encoder::SetErrorMode);
|
|
bc1_encoder.def_property("endpoint_mode", &BC1Encoder::GetEndpointMode, &BC1Encoder::SetEndpointMode);
|
|
bc1_encoder.def_property("search_rounds", &BC1Encoder::GetSearchRounds, &BC1Encoder::SetSearchRounds);
|
|
bc1_encoder.def_property("orderings_4", &BC1Encoder::GetOrderings4, &BC1Encoder::SetOrderings4);
|
|
bc1_encoder.def_property("orderings_3", &BC1Encoder::GetOrderings3, &BC1Encoder::SetOrderings3);
|
|
|
|
using Flags = BC1Encoder::Flags;
|
|
py::enum_<Flags>(bc1_encoder, "Flags", py::arithmetic())
|
|
.value("UseLikelyTotalOrderings", Flags::UseLikelyTotalOrderings)
|
|
.value("TwoLeastSquaresPasses", Flags::TwoLeastSquaresPasses)
|
|
.value("Use3ColorBlocksForBlackPixels", Flags::Use3ColorBlocksForBlackPixels)
|
|
.value("Use3ColorBlocks", Flags::Use3ColorBlocks)
|
|
.value("Iterative", Flags::Iterative)
|
|
.value("Use6PowerIters", Flags::Use6PowerIters)
|
|
.value("Exhaustive", Flags::Exhaustive)
|
|
.value("TryAllInitialEndpoints", Flags::TryAllInitialEndpoints)
|
|
.def("__invert__", [](Flags f1) { return ~unsigned(f1); })
|
|
.def("__and__", [](Flags f1, Flags f2) { return unsigned(f1) & unsigned(f2); })
|
|
.def("__rand__", [](Flags f1, Flags f2) { return unsigned(f1) & unsigned(f2); })
|
|
.def("__or__", [](Flags f1, Flags f2) { return unsigned(f1) | unsigned(f2); })
|
|
.def("__ror__", [](Flags f1, Flags f2) { return unsigned(f1) | unsigned(f2); })
|
|
.def("__xor__", [](Flags f1, Flags f2) { return unsigned(f1) ^ unsigned(f2); })
|
|
.def("__rxor__", [](Flags f1, Flags f2) { return unsigned(f2) ^ unsigned(f1); });
|
|
|
|
py::enum_<BC1Encoder::EndpointMode>(bc1_encoder, "EndpointMode")
|
|
.value("LeastSquares", BC1Encoder::EndpointMode::LeastSquares)
|
|
.value("BoundingBox", BC1Encoder::EndpointMode::BoundingBox)
|
|
.value("BoundingBoxInt", BC1Encoder::EndpointMode::BoundingBoxInt)
|
|
.value("PCA", BC1Encoder::EndpointMode::PCA);
|
|
|
|
py::enum_<BC1Encoder::ErrorMode>(bc1_encoder, "ErrorMode")
|
|
.value("None", BC1Encoder::ErrorMode::None)
|
|
.value("Faster", BC1Encoder::ErrorMode::Faster)
|
|
.value("Check2", BC1Encoder::ErrorMode::Check2)
|
|
.value("Full", BC1Encoder::ErrorMode::Full);
|
|
|
|
// BC1Decoder
|
|
py::class_<BC1Decoder> bc1_decoder(bc1, "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 quicktex::bindings
|