feat: Exposing both encode and decode speed for better fine-tuning

google · Mar 3, 2021 · 2ceedf7 · 2ceedf7
1 parent edc12fd
commit 2ceedf7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/draco/unity/draco_unity_enc_plugin.cc b/src/draco/unity/draco_unity_enc_plugin.cc
@@ -33,8 +33,9 @@ namespace draco {
       delete encoder;
   }
 
-  void dracoEncoderSetCompressionSpeed(DracoEncoder *encoder, uint32_t speedLevel) {
-      encoder->speed = speedLevel;
+  void dracoEncoderSetCompressionSpeed(DracoEncoder *encoder, uint32_t encodingSpeed, uint32_t decodingSpeed) {
+      encoder->encodingSpeed = encodingSpeed;
+      encoder->decodingSpeed = decodingSpeed;
   }
 
   void dracoEncoderSetQuantizationBits(DracoEncoder *encoder, uint32_t position, uint32_t normal, uint32_t uv, uint32_t color, uint32_t generic)
@@ -49,7 +50,7 @@ namespace draco {
   bool dracoEncoderEncode(DracoEncoder *encoder, uint8_t preserveTriangleOrder)
   {
       draco::Encoder dracoEncoder;
-      dracoEncoder.SetSpeedOptions(encoder->speed, encoder->speed);
+      dracoEncoder.SetSpeedOptions(encoder->encodingSpeed, encoder->decodingSpeed);
       dracoEncoder.SetAttributeQuantization(draco::GeometryAttribute::POSITION, encoder->quantization.position);
       dracoEncoder.SetAttributeQuantization(draco::GeometryAttribute::NORMAL, encoder->quantization.normal);
       dracoEncoder.SetAttributeQuantization(draco::GeometryAttribute::TEX_COORD, encoder->quantization.uv);

diff --git a/src/draco/unity/draco_unity_enc_plugin.h b/src/draco/unity/draco_unity_enc_plugin.h
@@ -41,7 +41,8 @@ namespace draco {
       uint32_t encodedIndices;
       std::vector<std::unique_ptr<draco::DataBuffer>> buffers;
       draco::EncoderBuffer encoderBuffer;
-      uint32_t speed = 0;
+      uint32_t encodingSpeed = 0;
+      uint32_t decodingSpeed = 0;
       std::size_t rawSize = 0;
       struct
       {
@@ -60,7 +61,7 @@ extern "C" {
 
   EXPORT_API DracoEncoder * dracoEncoderCreate(uint32_t vertexCount);
   void EXPORT_API dracoEncoderRelease(DracoEncoder *encoder);
-  void EXPORT_API dracoEncoderSetCompressionSpeed(DracoEncoder *encoder, uint32_t speedLevel);
+  void EXPORT_API dracoEncoderSetCompressionSpeed(DracoEncoder *encoder, uint32_t encodingSpeed, uint32_t decodingSpeed);
   void EXPORT_API dracoEncoderSetQuantizationBits(DracoEncoder *encoder, uint32_t position, uint32_t normal, uint32_t uv, uint32_t color, uint32_t generic);
   bool EXPORT_API dracoEncoderEncode(DracoEncoder *encoder, uint8_t preserveTriangleOrder);
   uint32_t EXPORT_API dracoEncoderGetEncodedVertexCount(DracoEncoder *encoder);