void data(Images& images) const override {
ColorVal pixel[5];
for (Image& image : images)
for (uint32_t r=0; r<image.rows(); r++) {
for (uint32_t c=0; c<image.cols(); c++) {
for (int p=0; p<ranges->numPlanes(); p++) pixel[p] = image(p,r,c);
image.set(0,r,c, pixel[permutation[0]]);
if (!subtract) { for (int p=1; p<ranges->numPlanes(); p++) image.set(p,r,c, pixel[permutation[p]]); }
else { for (int p=1; p<3 && p<ranges->numPlanes(); p++) image.set(p,r,c, pixel[permutation[p]] - pixel[permutation[0]]);
for (int p=3; p<ranges->numPlanes(); p++) image.set(p,r,c, pixel[permutation[p]]); }
}
}
}
void initPropRanges(Ranges &propRanges, const ColorRanges &ranges, int p) {
propRanges.clear();
int min = ranges.min(p);
int max = ranges.max(p);
int mind = min - max, maxd = max - min;
if (p < 3) { // alpha channel first
for (int pp = 0; pp < p; pp++) {
propRanges.push_back(std::make_pair(ranges.min(pp), ranges.max(pp))); // pixels on previous planes
}
if (ranges.numPlanes()>3) propRanges.push_back(std::make_pair(ranges.min(3), ranges.max(3))); // pixel on alpha plane
}
//if (p<1 || p>2)
propRanges.push_back(std::make_pair(0,2)); // median predictor: which of the three values is the median?
if (p==1 || p==2) propRanges.push_back(std::make_pair(ranges.min(0)-ranges.max(0),ranges.max(0)-ranges.min(0))); // luma prediction miss
propRanges.push_back(std::make_pair(mind,maxd)); // neighbor A - neighbor B (top-bottom or left-right)
propRanges.push_back(std::make_pair(mind,maxd)); // top/left prediction miss (previous pixel)
propRanges.push_back(std::make_pair(mind,maxd)); // left/top prediction miss (other direction)
propRanges.push_back(std::make_pair(mind,maxd)); // bottom/right prediction miss
propRanges.push_back(std::make_pair(min,max)); // guess
// propRanges.push_back(std::make_pair(mind,maxd)); // left - topleft
// propRanges.push_back(std::make_pair(mind,maxd)); // topleft - top
// if (p == 0 || p > 2)
// propRanges.push_back(std::make_pair(mind,maxd)); // top - topright
if (p != 2) {
propRanges.push_back(std::make_pair(mind,maxd)); // toptop - top
propRanges.push_back(std::make_pair(mind,maxd)); // leftleft - left
}
}
void invData(Images& images) const override {
ColorVal pixel[5];
for (Image& image : images) {
for (int p=0; p<ranges->numPlanes(); p++) image.undo_make_constant_plane(p);
for (uint32_t r=0; r<image.rows(); r++) {
for (uint32_t c=0; c<image.cols(); c++) {
for (int p=0; p<ranges->numPlanes(); p++) pixel[p] = image(p,r,c);
for (int p=0; p<ranges->numPlanes(); p++) image.set(permutation[p],r,c, pixel[p]);
image.set(permutation[0],r,c, pixel[0]);
if (!subtract) { for (int p=1; p<ranges->numPlanes(); p++) image.set(permutation[p],r,c, pixel[p]); }
else { for (int p=1; p<3 && p<ranges->numPlanes(); p++) image.set(permutation[p],r,c, CLAMP(pixel[p] + pixel[0], ranges->min(permutation[p]), ranges->max(permutation[p])));
for (int p=3; p<ranges->numPlanes(); p++) image.set(permutation[p],r,c, pixel[p]); }
}
}
}
}
void print() {
printf("Y buckets:\n");
bucket0.print();
printf("\nI buckets:\n");
for (auto b : bucket1) b.print();
printf("\nQ buckets:\n ");
for (auto bs : bucket2) { for (auto b : bs) b.print(); printf("\n ");}
if (ranges->numPlanes() > 3) {
printf("Alpha buckets:\n");
bucket3.print();
}
}
void initPropRanges_scanlines(Ranges &propRanges, const ColorRanges &ranges, int p) {
propRanges.clear();
int min = ranges.min(p);
int max = ranges.max(p);
int mind = min - max, maxd = max - min;
if (p < 3) {
for (int pp = 0; pp < p; pp++) {
propRanges.push_back(std::make_pair(ranges.min(pp), ranges.max(pp))); // pixels on previous planes
}
if (ranges.numPlanes()>3) propRanges.push_back(std::make_pair(ranges.min(3), ranges.max(3))); // pixel on alpha plane
}
propRanges.push_back(std::make_pair(min,max)); // guess (median of 3)
propRanges.push_back(std::make_pair(0,2)); // which predictor was it
propRanges.push_back(std::make_pair(mind,maxd));
propRanges.push_back(std::make_pair(mind,maxd));
propRanges.push_back(std::make_pair(mind,maxd));
propRanges.push_back(std::make_pair(mind,maxd));
propRanges.push_back(std::make_pair(mind,maxd));
}
void initPropRanges(Ranges &propRanges, const ColorRanges &ranges, int p) {
propRanges.clear();
int min = ranges.min(p);
int max = ranges.max(p);
int mind = min - max, maxd = max - min;
if (p < 3) { // alpha channel first
for (int pp = 0; pp < p; pp++) {
propRanges.push_back(std::make_pair(ranges.min(pp), ranges.max(pp))); // pixels on previous planes
}
if (ranges.numPlanes()>3) propRanges.push_back(std::make_pair(ranges.min(3), ranges.max(3))); // pixel on alpha plane
}
propRanges.push_back(std::make_pair(mind,maxd)); // neighbor A - neighbor B (top-bottom or left-right)
propRanges.push_back(std::make_pair(min,max)); // guess (median of 3)
propRanges.push_back(std::make_pair(0,2)); // which predictor was it
propRanges.push_back(std::make_pair(mind,maxd));
propRanges.push_back(std::make_pair(mind,maxd));
propRanges.push_back(std::make_pair(mind,maxd));
if (p < 2 || p >= 3) {
propRanges.push_back(std::make_pair(mind,maxd));
propRanges.push_back(std::make_pair(mind,maxd));
}
}
int numPlanes() const override { return ranges->numPlanes(); }
int numPlanes() const {
return ranges->numPlanes();
}