blob: 150ae1f831f0ad408f18e4bd8e7a25f6c2d8c8e9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
// 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.
//
#ifndef DRACO_COMPRESSION_ATTRIBUTES_SEQUENTIAL_INTEGER_ATTRIBUTE_DECODER_H_
#define DRACO_COMPRESSION_ATTRIBUTES_SEQUENTIAL_INTEGER_ATTRIBUTE_DECODER_H_
#include "draco/draco_features.h"
#include "draco/compression/attributes/prediction_schemes/prediction_scheme_decoder.h"
#include "draco/compression/attributes/sequential_attribute_decoder.h"
namespace draco {
// Decoder for attributes encoded with the SequentialIntegerAttributeEncoder.
class SequentialIntegerAttributeDecoder : public SequentialAttributeDecoder {
public:
SequentialIntegerAttributeDecoder();
bool Init(PointCloudDecoder *decoder, int attribute_id) override;
bool TransformAttributeToOriginalFormat(
const std::vector<PointIndex> &point_ids) override;
protected:
bool DecodeValues(const std::vector<PointIndex> &point_ids,
DecoderBuffer *in_buffer) override;
virtual bool DecodeIntegerValues(const std::vector<PointIndex> &point_ids,
DecoderBuffer *in_buffer);
// Returns a prediction scheme that should be used for decoding of the
// integer values.
virtual std::unique_ptr<PredictionSchemeTypedDecoderInterface<int32_t>>
CreateIntPredictionScheme(PredictionSchemeMethod method,
PredictionSchemeTransformType transform_type);
// Returns the number of integer attribute components. In general, this
// can be different from the number of components of the input attribute.
virtual int32_t GetNumValueComponents() const {
return attribute()->num_components();
}
// Called after all integer values are decoded. The implementation should
// use this method to store the values into the attribute.
virtual bool StoreValues(uint32_t num_values);
void PreparePortableAttribute(int num_entries, int num_components);
int32_t *GetPortableAttributeData() {
if (portable_attribute()->size() == 0)
return nullptr;
return reinterpret_cast<int32_t *>(
portable_attribute()->GetAddress(AttributeValueIndex(0)));
}
private:
// Stores decoded values into the attribute with a data type AttributeTypeT.
template <typename AttributeTypeT>
void StoreTypedValues(uint32_t num_values);
std::unique_ptr<PredictionSchemeTypedDecoderInterface<int32_t>>
prediction_scheme_;
};
} // namespace draco
#endif // DRACO_COMPRESSION_ATTRIBUTES_SEQUENTIAL_INTEGER_ATTRIBUTE_DECODER_H_
|