Skip to content
GitLab
Commit 46efadca authored by David Flynn's avatar David Flynn
Browse files

attr: refactor quantisation state and detail into Quantizer class 

In order to simplify the introduction of quantisation using the
multiplicative inverse, the existing quantiser state is refactored
into a Quantizer class that represents the current step size.
parent b40f9396
Hide whitespace changes
Inline Side-by-side
tmc3/AttributeDecoder.cpp
View file @ 46efadca
......@@ -181,7 +181,7 @@ AttributeDecoder::decode(
{
int abhSize;
AttributeBrickHeader abh = parseAbh(attr_aps, payload, &abhSize);
Quantizers qstep = deriveQuantSteps(attr_aps, abh);
Quantizers quant = deriveQuantizers(attr_aps, abh);
PCCResidualsDecoder decoder;
decoder.start(payload.data() + abhSize, payload.size() - abhSize);
......@@ -189,29 +189,29 @@ AttributeDecoder::decode(
if (attr_desc.attr_num_dimensions == 1) {
switch (attr_aps.attr_encoding) {
case AttributeEncoding::kRAHTransform:
decodeReflectancesRaht(attr_desc, attr_aps, qstep, decoder, pointCloud);
decodeReflectancesRaht(attr_desc, attr_aps, quant, decoder, pointCloud);
break;
case AttributeEncoding::kPredictingTransform:
decodeReflectancesPred(attr_desc, attr_aps, qstep, decoder, pointCloud);
decodeReflectancesPred(attr_desc, attr_aps, quant, decoder, pointCloud);
break;
case AttributeEncoding::kLiftingTransform:
decodeReflectancesLift(attr_desc, attr_aps, qstep, decoder, pointCloud);
decodeReflectancesLift(attr_desc, attr_aps, quant, decoder, pointCloud);
break;
}
} else if (attr_desc.attr_num_dimensions == 3) {
switch (attr_aps.attr_encoding) {
case AttributeEncoding::kRAHTransform:
decodeColorsRaht(attr_desc, attr_aps, qstep, decoder, pointCloud);
decodeColorsRaht(attr_desc, attr_aps, quant, decoder, pointCloud);
break;
case AttributeEncoding::kPredictingTransform:
decodeColorsPred(attr_desc, attr_aps, qstep, decoder, pointCloud);
decodeColorsPred(attr_desc, attr_aps, quant, decoder, pointCloud);
break;
case AttributeEncoding::kLiftingTransform:
decodeColorsLift(attr_desc, attr_aps, qstep, decoder, pointCloud);
decodeColorsLift(attr_desc, attr_aps, quant, decoder, pointCloud);
break;
}
} else {
......@@ -263,7 +263,7 @@ void
AttributeDecoder::decodeReflectancesPred(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud)
{
......@@ -283,7 +283,6 @@ AttributeDecoder::decodeReflectancesPred(
for (size_t predictorIndex = 0; predictorIndex < pointCount;
++predictorIndex) {
auto& predictor = predictors[predictorIndex];
const int64_t qs = qstep[0];
computeReflectancePredictionWeights(
aps, pointCloud, indexesLOD, predictor, decoder);
const uint32_t pointIndex = indexesLOD[predictorIndex];
......@@ -298,8 +297,7 @@ AttributeDecoder::decodeReflectancesPred(
const int64_t quantPredAttValue =
predictor.predictReflectance(pointCloud, indexesLOD);
const int64_t delta = divExp2RoundHalfUp(
PCCInverseQuantization(UIntToInt(attValue0), qs),
kFixedPointAttributeShift);
quant[0].scale(UIntToInt(attValue0)), kFixedPointAttributeShift);
const int64_t reconstructedQuantAttValue = quantPredAttValue + delta;
reflectance = uint16_t(
PCCClip(reconstructedQuantAttValue, int64_t(0), maxReflectance));
......@@ -350,7 +348,7 @@ void
AttributeDecoder::decodeColorsPred(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud)
{
......@@ -370,8 +368,6 @@ AttributeDecoder::decodeColorsPred(
for (size_t predictorIndex = 0; predictorIndex < pointCount;
++predictorIndex) {
auto& predictor = predictors[predictorIndex];
const int64_t qs = qstep[0];
const int64_t qs2 = qstep[1];
computeColorPredictionWeights(
aps, pointCloud, indexesLOD, predictor, decoder);
if (zero_cnt > 0) {
......@@ -387,8 +383,7 @@ AttributeDecoder::decodeColorsPred(
predictor.predictColor(pointCloud, indexesLOD);
const int64_t quantPredAttValue = predictedColor[0];
const int64_t delta = divExp2RoundHalfUp(
PCCInverseQuantization(UIntToInt(values[0]), qs),
kFixedPointAttributeShift);
quant[0].scale(UIntToInt(values[0])), kFixedPointAttributeShift);
const int64_t reconstructedQuantAttValue = quantPredAttValue + delta;
int64_t clipMax = (1 << desc.attr_bitdepth) - 1;
color[0] =
......@@ -396,8 +391,7 @@ AttributeDecoder::decodeColorsPred(
for (size_t k = 1; k < 3; ++k) {
const int64_t quantPredAttValue = predictedColor[k];
const int64_t delta = divExp2RoundHalfUp(
PCCInverseQuantization(UIntToInt(values[k]), qs2),
kFixedPointAttributeShift);
quant[1].scale(UIntToInt(values[k])), kFixedPointAttributeShift);
const int64_t reconstructedQuantAttValue = quantPredAttValue + delta;
color[k] =
uint8_t(PCCClip(reconstructedQuantAttValue, int64_t(0), clipMax));
......@@ -411,7 +405,7 @@ void
AttributeDecoder::decodeReflectancesRaht(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud)
{
......@@ -447,7 +441,7 @@ AttributeDecoder::decodeReflectancesRaht(
int* attributes = new int[attribCount * voxelCount];
regionAdaptiveHierarchicalInverseTransform(
aps.raht_prediction_enabled_flag, qstep, mortonCode, attributes,
aps.raht_prediction_enabled_flag, quant, mortonCode, attributes,
attribCount, voxelCount, coefficients);
const int64_t maxReflectance = (1 << desc.attr_bitdepth) - 1;
......@@ -471,7 +465,7 @@ void
AttributeDecoder::decodeColorsRaht(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud)
{
......@@ -511,7 +505,7 @@ AttributeDecoder::decodeColorsRaht(
int* attributes = new int[attribCount * voxelCount];
regionAdaptiveHierarchicalInverseTransform(
aps.raht_prediction_enabled_flag, qstep, mortonCode, attributes,
aps.raht_prediction_enabled_flag, quant, mortonCode, attributes,
attribCount, voxelCount, coefficients);
const int clipMax = (1 << desc.attr_bitdepth) - 1;
......@@ -538,7 +532,7 @@ void
AttributeDecoder::decodeColorsLift(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud)
{
......@@ -574,17 +568,15 @@ AttributeDecoder::decodeColorsLift(
decoder.decode(values);
zero_cnt = decoder.decodeZeroCnt(pointCount);
}
const int64_t qs = qstep[0] << (kFixedPointWeightShift / 2);
const int64_t qs2 = qstep[1] << (kFixedPointWeightShift / 2);
// + kFixedPointAttributeShift ???
const int64_t quantWeight = weights[predictorIndex];
auto& color = colors[predictorIndex];
const int64_t delta = UIntToInt(values[0]);
const int64_t reconstructedDelta = PCCInverseQuantization(delta, qs);
const int64_t reconstructedDelta = quant[0].scale(delta);
color[0] = reconstructedDelta / quantWeight;
for (size_t d = 1; d < 3; ++d) {
const int64_t delta = UIntToInt(values[d]);
const int64_t reconstructedDelta = PCCInverseQuantization(delta, qs2);
const int64_t reconstructedDelta = quant[1].scale(delta);
color[d] = reconstructedDelta / quantWeight;
}
}
......@@ -615,7 +607,7 @@ void
AttributeDecoder::decodeReflectancesLift(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud)
{
......@@ -652,11 +644,10 @@ AttributeDecoder::decodeReflectancesLift(
detail = decoder.decode();
zero_cnt = decoder.decodeZeroCnt(pointCount);
}
const int64_t qs = qstep[0] << (kFixedPointWeightShift / 2);
const int64_t quantWeight = weights[predictorIndex];
auto& reflectance = reflectances[predictorIndex];
const int64_t delta = UIntToInt(detail);
const int64_t reconstructedDelta = PCCInverseQuantization(delta, qs);
const int64_t reconstructedDelta = quant[0].scale(delta);
reflectance = reconstructedDelta / quantWeight;
}
......
tmc3/AttributeDecoder.h
View file @ 46efadca
......@@ -64,42 +64,42 @@ protected:
void decodeReflectancesLift(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud);
void decodeColorsLift(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud);
void decodeReflectancesPred(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud);
void decodeColorsPred(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud);
void decodeReflectancesRaht(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud);
void decodeColorsRaht(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCResidualsDecoder& decoder,
PCCPointSet3& pointCloud);
......
tmc3/AttributeEncoder.cpp
View file @ 46efadca
......@@ -333,7 +333,7 @@ AttributeEncoder::encode(
PCCPointSet3& pointCloud,
PayloadBuffer* payload)
{
Quantizers qstep = deriveQuantSteps(attr_aps, abh);
Quantizers quant = deriveQuantizers(attr_aps, abh);
PCCResidualsEncoder encoder;
encoder.start(int(pointCloud.getPointCount()));
......@@ -342,29 +342,29 @@ AttributeEncoder::encode(
switch (attr_aps.attr_encoding) {
case AttributeEncoding::kRAHTransform:
encodeReflectancesTransformRaht(
desc, attr_aps, qstep, pointCloud, encoder);
desc, attr_aps, quant, pointCloud, encoder);
break;
case AttributeEncoding::kPredictingTransform:
encodeReflectancesPred(desc, attr_aps, qstep, pointCloud, encoder);
encodeReflectancesPred(desc, attr_aps, quant, pointCloud, encoder);
break;
case AttributeEncoding::kLiftingTransform:
encodeReflectancesLift(desc, attr_aps, qstep, pointCloud, encoder);
encodeReflectancesLift(desc, attr_aps, quant, pointCloud, encoder);
break;
}
} else if (desc.attr_num_dimensions == 3) {
switch (attr_aps.attr_encoding) {
case AttributeEncoding::kRAHTransform:
encodeColorsTransformRaht(desc, attr_aps, qstep, pointCloud, encoder);
encodeColorsTransformRaht(desc, attr_aps, quant, pointCloud, encoder);
break;
case AttributeEncoding::kPredictingTransform:
encodeColorsPred(desc, attr_aps, qstep, pointCloud, encoder);
encodeColorsPred(desc, attr_aps, quant, pointCloud, encoder);
break;
case AttributeEncoding::kLiftingTransform:
encodeColorsLift(desc, attr_aps, qstep, pointCloud, encoder);
encodeColorsLift(desc, attr_aps, quant, pointCloud, encoder);
break;
}
} else {
......@@ -383,12 +383,12 @@ int64_t
AttributeEncoder::computeReflectanceResidual(
const uint64_t reflectance,
const uint64_t predictedReflectance,
const int64_t qs)
const Quantizer& quant)
{
const int64_t quantAttValue = reflectance;
const int64_t quantPredAttValue = predictedReflectance;
const int64_t delta = PCCQuantization(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift, qs);
const int64_t delta = quant.quantize(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift);
return IntToUInt(delta);
}
......@@ -404,7 +404,7 @@ AttributeEncoder::computeReflectancePredictionWeights(
PCCPredictor& predictor,
PCCResidualsEncoder& encoder,
PCCResidualsEntropyEstimator& context,
const int64_t qs)
const Quantizer& quant)
{
predictor.computeWeights();
predictor.maxDiff = 0;
......@@ -431,10 +431,10 @@ AttributeEncoder::computeReflectancePredictionWeights(
predictor.predMode = 0;
uint64_t attrPred = predictor.predictReflectance(pointCloud, indexesLOD);
int64_t attrResidualQuant =
computeReflectanceResidual(attrValue, attrPred, qs);
computeReflectanceResidual(attrValue, attrPred, quant);
double best_score = attrResidualQuant
+ kAttrPredLambdaR * (qs >> kFixedPointAttributeShift);
+ kAttrPredLambdaR * (quant.stepSize() >> kFixedPointAttributeShift);
for (int i = 0; i < predictor.neighborCount; i++) {
if (i == aps.max_num_direct_predictors)
......@@ -443,11 +443,12 @@ AttributeEncoder::computeReflectancePredictionWeights(
attrPred = pointCloud.getReflectance(
indexesLOD[predictor.neighbors[i].predictorIndex]);
attrResidualQuant =
computeReflectanceResidual(attrValue, attrPred, qs);
computeReflectanceResidual(attrValue, attrPred, quant);
double idxBits = i + (i == aps.max_num_direct_predictors - 1 ? 1 : 2);
double score = attrResidualQuant
+ idxBits * kAttrPredLambdaR * (qs >> kFixedPointAttributeShift);
+ idxBits * kAttrPredLambdaR
* (quant.stepSize() >> kFixedPointAttributeShift);
if (score < best_score) {
best_score = score;
......@@ -466,7 +467,7 @@ void
AttributeEncoder::encodeReflectancesPred(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCPointSet3& pointCloud,
PCCResidualsEncoder& encoder)
{
......@@ -492,10 +493,9 @@ AttributeEncoder::encodeReflectancesPred(
for (size_t predictorIndex = 0; predictorIndex < pointCount;
++predictorIndex) {
auto& predictor = predictors[predictorIndex];
const int64_t qs = qstep[0];
computeReflectancePredictionWeights(
aps, pointCloud, indexesLOD, predictorIndex, predictor, encoder, context,
qs);
quant[0]);
const uint32_t pointIndex = indexesLOD[predictorIndex];
const uint64_t reflectance = pointCloud.getReflectance(pointIndex);
......@@ -503,11 +503,11 @@ AttributeEncoder::encodeReflectancesPred(
predictor.predictReflectance(pointCloud, indexesLOD);
const int64_t quantAttValue = reflectance;
const int64_t quantPredAttValue = predictedReflectance;
const int64_t delta = PCCQuantization(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift, qs);
const int64_t delta = quant[0].quantize(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift);
const uint32_t attValue0 = uint32_t(IntToUInt(long(delta)));
const int64_t reconstructedDelta = divExp2RoundHalfUp(
PCCInverseQuantization(delta, qs), kFixedPointAttributeShift);
const int64_t reconstructedDelta =
divExp2RoundHalfUp(quant[0].scale(delta), kFixedPointAttributeShift);
const int64_t reconstructedQuantAttValue =
quantPredAttValue + reconstructedDelta;
const uint16_t reconstructedReflectance =
......@@ -552,20 +552,19 @@ Vec3<int64_t>
AttributeEncoder::computeColorResiduals(
const Vec3<uint8_t> color,
const Vec3<uint8_t> predictedColor,
const int64_t qs,
const int64_t qs2)
const Quantizers& quant)
{
Vec3<int64_t> residuals;
const int64_t quantAttValue = color[0];
const int64_t quantPredAttValue = predictedColor[0];
const int64_t delta = PCCQuantization(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift, qs);
const int64_t delta = quant[0].quantize(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift);
residuals[0] = IntToUInt(delta);
for (size_t k = 1; k < 3; ++k) {
const int64_t quantAttValue = color[k];
const int64_t quantPredAttValue = predictedColor[k];
const int64_t delta = PCCQuantization(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift, qs2);
const int64_t delta = quant[1].quantize(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift);
residuals[k] = IntToUInt(delta);
}
return residuals;
......@@ -582,8 +581,7 @@ AttributeEncoder::computeColorPredictionWeights(
PCCPredictor& predictor,
PCCResidualsEncoder& encoder,
PCCResidualsEntropyEstimator& context,
const int64_t qs,
const int64_t qs2)
const Quantizers& quant)
{
predictor.computeWeights();
predictor.maxDiff = 0;
......@@ -615,11 +613,12 @@ AttributeEncoder::computeColorPredictionWeights(
predictor.predMode = 0;
Vec3<uint8_t> attrPred = predictor.predictColor(pointCloud, indexesLOD);
Vec3<int64_t> attrResidualQuant =
computeColorResiduals(attrValue, attrPred, qs, qs2);
computeColorResiduals(attrValue, attrPred, quant);
double best_score = attrResidualQuant[0] + attrResidualQuant[1]
+ attrResidualQuant[2]
+ kAttrPredLambdaC * (double)(qs >> kFixedPointAttributeShift);
+ kAttrPredLambdaC
* (double)(quant[0].stepSize() >> kFixedPointAttributeShift);
for (int i = 0; i < predictor.neighborCount; i++) {
if (i == aps.max_num_direct_predictors)
......@@ -627,13 +626,13 @@ AttributeEncoder::computeColorPredictionWeights(
attrPred = pointCloud.getColor(
indexesLOD[predictor.neighbors[i].predictorIndex]);
attrResidualQuant =
computeColorResiduals(attrValue, attrPred, qs, qs2);
attrResidualQuant = computeColorResiduals(attrValue, attrPred, quant);
double idxBits = i + (i == aps.max_num_direct_predictors - 1 ? 1 : 2);
double score = attrResidualQuant[0] + attrResidualQuant[1]
+ attrResidualQuant[2]
+ idxBits * kAttrPredLambdaC * (qs >> kFixedPointAttributeShift);
+ idxBits * kAttrPredLambdaC
* (quant[0].stepSize() >> kFixedPointAttributeShift);
if (score < best_score) {
best_score = score;
......@@ -652,7 +651,7 @@ void
AttributeEncoder::encodeColorsPred(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCPointSet3& pointCloud,
PCCResidualsEncoder& encoder)
{
......@@ -681,21 +680,19 @@ AttributeEncoder::encodeColorsPred(
for (size_t predictorIndex = 0; predictorIndex < pointCount;
++predictorIndex) {
auto& predictor = predictors[predictorIndex];
const int64_t qs = qstep[0];
const int64_t qs2 = qstep[1];
computeColorPredictionWeights(
aps, pointCloud, indexesLOD, predictorIndex, predictor, encoder, context,
qs, qs2);
quant);
const auto pointIndex = indexesLOD[predictorIndex];
const Vec3<uint8_t> color = pointCloud.getColor(pointIndex);
const Vec3<uint8_t> predictedColor =
predictor.predictColor(pointCloud, indexesLOD);
const int64_t quantAttValue = color[0];
const int64_t quantPredAttValue = predictedColor[0];
const int64_t delta = PCCQuantization(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift, qs);
const int64_t reconstructedDelta = divExp2RoundHalfUp(
PCCInverseQuantization(delta, qs), kFixedPointAttributeShift);
const int64_t delta = quant[0].quantize(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift);
const int64_t reconstructedDelta =
divExp2RoundHalfUp(quant[0].scale(delta), kFixedPointAttributeShift);
const int64_t reconstructedQuantAttValue =
quantPredAttValue + reconstructedDelta;
values[0] = uint32_t(IntToUInt(long(delta)));
......@@ -705,10 +702,10 @@ AttributeEncoder::encodeColorsPred(
for (size_t k = 1; k < 3; ++k) {
const int64_t quantAttValue = color[k];
const int64_t quantPredAttValue = predictedColor[k];
const int64_t delta = PCCQuantization(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift, qs2);
const int64_t reconstructedDelta = divExp2RoundHalfUp(
PCCInverseQuantization(delta, qs2), kFixedPointAttributeShift);
const int64_t delta = quant[1].quantize(
(quantAttValue - quantPredAttValue) << kFixedPointAttributeShift);
const int64_t reconstructedDelta =
divExp2RoundHalfUp(quant[1].scale(delta), kFixedPointAttributeShift);
const int64_t reconstructedQuantAttValue =
quantPredAttValue + reconstructedDelta;
values[k] = uint32_t(IntToUInt(long(delta)));
......@@ -760,7 +757,7 @@ void
AttributeEncoder::encodeReflectancesTransformRaht(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCPointSet3& pointCloud,
PCCResidualsEncoder& encoder)
{
......@@ -787,7 +784,7 @@ AttributeEncoder::encodeReflectancesTransformRaht(
// Transform.
regionAdaptiveHierarchicalTransform(
aps.raht_prediction_enabled_flag, qstep, mortonCode, attributes,
aps.raht_prediction_enabled_flag, quant, mortonCode, attributes,
attribCount, voxelCount, coefficients);
// Entropy encode.
......@@ -828,7 +825,7 @@ void
AttributeEncoder::encodeColorsTransformRaht(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCPointSet3& pointCloud,
PCCResidualsEncoder& encoder)
{
......@@ -857,7 +854,7 @@ AttributeEncoder::encodeColorsTransformRaht(
// Transform.
regionAdaptiveHierarchicalTransform(
aps.raht_prediction_enabled_flag, qstep, mortonCode, attributes,
aps.raht_prediction_enabled_flag, quant, mortonCode, attributes,
attribCount, voxelCount, coefficients);
// Entropy encode.
......@@ -902,7 +899,7 @@ void
AttributeEncoder::encodeColorsLift(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCPointSet3& pointCloud,
PCCResidualsEncoder& encoder)
{
......@@ -944,24 +941,22 @@ AttributeEncoder::encodeColorsLift(
// compress
int zero_cnt = 0;
const int64_t qs = qstep[0] << (kFixedPointWeightShift / 2);
const size_t qs2 = qstep[1] << (kFixedPointWeightShift / 2);
for (size_t predictorIndex = 0; predictorIndex < pointCount;
++predictorIndex) {
const int64_t quantWeight = weights[predictorIndex];
auto& color = colors[predictorIndex];
const int64_t delta = PCCQuantization(color[0] * quantWeight, qs);
const int64_t delta = quant[0].quantize(color[0] * quantWeight);
const int64_t detail = IntToUInt(delta);
assert(detail < std::numeric_limits<uint32_t>::max());
const int64_t reconstructedDelta = PCCInverseQuantization(delta, qs);
const int64_t reconstructedDelta = quant[0].scale(delta);
color[0] = reconstructedDelta / quantWeight;
uint32_t values[3];
values[0] = uint32_t(detail);
for (size_t d = 1; d < 3; ++d) {
const int64_t delta = PCCQuantization(color[d] * quantWeight, qs2);
const int64_t delta = quant[1].quantize(color[d] * quantWeight);
const int64_t detail = IntToUInt(delta);
assert(detail < std::numeric_limits<uint32_t>::max());
const int64_t reconstructedDelta = PCCInverseQuantization(delta, qs2);
const int64_t reconstructedDelta = quant[1].scale(delta);
color[d] = reconstructedDelta / quantWeight;
values[d] = uint32_t(detail);
}
......@@ -1001,7 +996,7 @@ void
AttributeEncoder::encodeReflectancesLift(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCPointSet3& pointCloud,
PCCResidualsEncoder& encoder)
{
......@@ -1045,13 +1040,12 @@ AttributeEncoder::encodeReflectancesLift(
int zero_cnt = 0;
for (size_t predictorIndex = 0; predictorIndex < pointCount;
++predictorIndex) {
const int64_t qs = qstep[0] << (kFixedPointWeightShift / 2);
const int64_t quantWeight = weights[predictorIndex];
auto& reflectance = reflectances[predictorIndex];
const int64_t delta = PCCQuantization(reflectance * quantWeight, qs);
const int64_t delta = quant[0].quantize(reflectance * quantWeight);
const int64_t detail = IntToUInt(delta);
assert(detail < std::numeric_limits<uint32_t>::max());
const int64_t reconstructedDelta = PCCInverseQuantization(delta, qs);
const int64_t reconstructedDelta = quant[0].scale(delta);
reflectance = reconstructedDelta / quantWeight;
if (!detail)
++zero_cnt;
......
tmc3/AttributeEncoder.h
View file @ 46efadca
......@@ -67,50 +67,49 @@ protected:
void encodeReflectancesLift(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCPointSet3& pointCloud,
PCCResidualsEncoder& encoder);
void encodeColorsLift(
const AttributeDescription& desc,
const AttributeParameterSet& aps,
const Quantizers& qstep,
const Quantizers& quant,
PCCPointSet3& pointCloud,
PCCResidualsEncoder& encoder);
void encodeReflectancesPred(
const AttributeDescription& desc,