blob: f74bdb83c9c8347e7fb9c782150dc222b9624e91 [file] [log] [blame]
#ifndef LIBGAV1_SRC_UTILS_SEGMENTATION_H_
#define LIBGAV1_SRC_UTILS_SEGMENTATION_H_
#include <cstdint>
#include "src/utils/constants.h"
namespace libgav1 {
// The corresponding segment feature constants in the AV1 spec are named
// SEG_LVL_xxx.
enum SegmentFeature : uint8_t {
kSegmentFeatureQuantizer,
kSegmentFeatureLoopFilterYVertical,
kSegmentFeatureLoopFilterYHorizontal,
kSegmentFeatureLoopFilterU,
kSegmentFeatureLoopFilterV,
kSegmentFeatureReferenceFrame,
kSegmentFeatureSkip,
kSegmentFeatureGlobalMv,
kSegmentFeatureMax
};
extern const int8_t kSegmentationFeatureBits[kSegmentFeatureMax];
extern const int kSegmentationFeatureMaxValues[kSegmentFeatureMax];
struct Segmentation {
// 5.11.14.
// Returns true if the feature is enabled in the segment.
bool FeatureActive(int segment_id, SegmentFeature feature) const {
return enabled && segment_id < kMaxSegments &&
feature_enabled[segment_id][feature];
}
// Returns true if the feature is signed.
static bool FeatureSigned(SegmentFeature feature) {
// Only the first five segment features are signed, so this comparison
// suffices.
return feature <= kSegmentFeatureLoopFilterV;
}
bool enabled;
bool update_map;
bool update_data;
bool temporal_update;
// True if the segment id will be read before the skip syntax element. False
// if the skip syntax element will be read first.
bool segment_id_pre_skip;
// The highest numbered segment id that has some enabled feature. Used as
// the upper bound for decoding segment ids.
int8_t last_active_segment_id;
bool feature_enabled[kMaxSegments][kSegmentFeatureMax];
int16_t feature_data[kMaxSegments][kSegmentFeatureMax];
bool lossless[kMaxSegments];
// Cached values of get_qindex(1, segmentId), to be consumed by
// Tile::ReadTransformType(). The values are in the range [0, 255].
uint8_t qindex[kMaxSegments];
};
} // namespace libgav1
#endif // LIBGAV1_SRC_UTILS_SEGMENTATION_H_