mirror of
https://git.mirrors.martin98.com/https://github.com/google/draco
synced 2025-07-26 16:11:59 +08:00
73bb3c8530
- Improved compression for triangular meshes (~10%) - Added WebAssembly decoder - Code cleanup + robustness fixes
119 lines
3.8 KiB
C++
119 lines
3.8 KiB
C++
// Copyright 2016 The Draco 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.
|
|
//
|
|
#include "compression/point_cloud/point_cloud_decoder.h"
|
|
|
|
namespace draco {
|
|
|
|
PointCloudDecoder::PointCloudDecoder()
|
|
: point_cloud_(nullptr),
|
|
buffer_(nullptr),
|
|
version_major_(0),
|
|
version_minor_(0) {}
|
|
|
|
bool PointCloudDecoder::DecodeHeader(DecoderBuffer *buffer,
|
|
DracoHeader *out_header) {
|
|
// TODO(ostava): Add error codes for better error reporting.
|
|
if (!buffer->Decode(out_header->draco_string, 5))
|
|
return false;
|
|
if (memcmp(out_header->draco_string, "DRACO", 5) != 0)
|
|
return false; // Wrong file format?
|
|
if (!buffer->Decode(&(out_header->version_major)))
|
|
return false;
|
|
if (!buffer->Decode(&(out_header->version_minor)))
|
|
return false;
|
|
if (!buffer->Decode(&(out_header->encoder_type)))
|
|
return false;
|
|
if (!buffer->Decode(&(out_header->encoder_method)))
|
|
return false;
|
|
if (!buffer->Decode(&(out_header->flags)))
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
bool PointCloudDecoder::Decode(DecoderBuffer *in_buffer,
|
|
PointCloud *out_point_cloud) {
|
|
buffer_ = in_buffer;
|
|
point_cloud_ = out_point_cloud;
|
|
DracoHeader header;
|
|
if (!DecodeHeader(buffer_, &header))
|
|
return false;
|
|
// Sanity check that we are really using the right decoder (mostly for cases
|
|
// where the Decode method was called manually outside of our main API.
|
|
if (header.encoder_type != GetGeometryType())
|
|
return false;
|
|
// TODO(ostava): We should check the method as well, but currently decoders
|
|
// don't expose the decoding method id.
|
|
version_major_ = header.version_major;
|
|
version_minor_ = header.version_minor;
|
|
|
|
// Check for version compatibility.
|
|
if (version_major_ < 1 || version_major_ > kDracoBitstreamVersionMajor)
|
|
return false;
|
|
if (version_major_ == kDracoBitstreamVersionMajor &&
|
|
version_minor_ > kDracoBitstreamVersionMinor)
|
|
return false;
|
|
|
|
if (!InitializeDecoder())
|
|
return false;
|
|
if (!DecodeGeometryData())
|
|
return false;
|
|
if (!DecodePointAttributes())
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
bool PointCloudDecoder::DecodePointAttributes() {
|
|
uint8_t num_attributes_decoders;
|
|
if (!buffer_->Decode(&num_attributes_decoders))
|
|
return false;
|
|
// Create all attribute decoders. This is implementation specific and the
|
|
// derived classes can use any data encoded in the
|
|
// PointCloudEncoder::EncodeAttributesEncoderIdentifier() call.
|
|
for (int i = 0; i < num_attributes_decoders; ++i) {
|
|
if (!CreateAttributesDecoder(i))
|
|
return false;
|
|
}
|
|
|
|
// Initialize all attributes decoders. No data is decoded here.
|
|
for (auto &att_dec : attributes_decoders_) {
|
|
if (!att_dec->Initialize(this, point_cloud_))
|
|
return false;
|
|
}
|
|
|
|
// Decode any data needed by the attribute decoders.
|
|
for (int i = 0; i < num_attributes_decoders; ++i) {
|
|
if (!attributes_decoders_[i]->DecodeAttributesDecoderData(buffer_))
|
|
return false;
|
|
}
|
|
|
|
// Decode the actual attributes using the created attribute decoders.
|
|
if (!DecodeAllAttributes())
|
|
return false;
|
|
|
|
if (!OnAttributesDecoded())
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
bool PointCloudDecoder::DecodeAllAttributes() {
|
|
for (auto &att_dec : attributes_decoders_) {
|
|
if (!att_dec->DecodeAttributes(buffer_))
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
} // namespace draco
|