ppsspp/GPU/Common/TextureScalerCommon.cpp
2023-03-16 10:14:41 +01:00

757 lines
28 KiB
C++

// Copyright (c) 2012- PPSSPP Project.
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, version 2.0 or later versions.
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License 2.0 for more details.
// A copy of the GPL 2.0 should have been included with the program.
// If not, see http://www.gnu.org/licenses/
// Official git repository and contact information can be found at
// https://github.com/hrydgard/ppsspp and http://www.ppsspp.org/.
#include <cstddef>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include "GPU/Common/TextureScalerCommon.h"
#include "Core/Config.h"
#include "Common/Common.h"
#include "Common/Log.h"
#include "Common/CommonFuncs.h"
#include "Common/Thread/ParallelLoop.h"
#include "Core/ThreadPools.h"
#include "Common/CPUDetect.h"
#include "ext/xbrz/xbrz.h"
#if defined(_M_SSE)
#include <emmintrin.h>
#include <smmintrin.h>
#endif
// Report the time and throughput for each larger scaling operation in the log
//#define SCALING_MEASURE_TIME
//#define DEBUG_SCALER_OUTPUT
#ifdef SCALING_MEASURE_TIME
#include "Common/TimeUtil.h"
#endif
/////////////////////////////////////// Helper Functions (mostly math for parallelization)
namespace {
//////////////////////////////////////////////////////////////////// Various image processing
#define R(_col) ((_col>> 0)&0xFF)
#define G(_col) ((_col>> 8)&0xFF)
#define B(_col) ((_col>>16)&0xFF)
#define A(_col) ((_col>>24)&0xFF)
#define DISTANCE(_p1,_p2) ( abs(static_cast<int>(static_cast<int>(R(_p1))-R(_p2))) + abs(static_cast<int>(static_cast<int>(G(_p1))-G(_p2))) \
+ abs(static_cast<int>(static_cast<int>(B(_p1))-B(_p2))) + abs(static_cast<int>(static_cast<int>(A(_p1))-A(_p2))) )
// this is sadly much faster than an inline function with a loop, at least in VC10
#define MIX_PIXELS(_p0, _p1, _factors) \
( (R(_p0)*(_factors)[0] + R(_p1)*(_factors)[1])/255 << 0 ) | \
( (G(_p0)*(_factors)[0] + G(_p1)*(_factors)[1])/255 << 8 ) | \
( (B(_p0)*(_factors)[0] + B(_p1)*(_factors)[1])/255 << 16 ) | \
( (A(_p0)*(_factors)[0] + A(_p1)*(_factors)[1])/255 << 24 )
#define BLOCK_SIZE 32
// 3x3 convolution with Neumann boundary conditions, parallelizable
// quite slow, could be sped up a lot
// especially handling of separable kernels
void convolve3x3(const u32 *data, u32 *out, const int kernel[3][3], int width, int height, int l, int u) {
for (int yb = 0; yb < (u - l) / BLOCK_SIZE + 1; ++yb) {
for (int xb = 0; xb < width / BLOCK_SIZE + 1; ++xb) {
for (int y = l + yb*BLOCK_SIZE; y < l + (yb + 1)*BLOCK_SIZE && y < u; ++y) {
for (int x = xb*BLOCK_SIZE; x < (xb + 1)*BLOCK_SIZE && x < width; ++x) {
int val = 0;
for (int yoff = -1; yoff <= 1; ++yoff) {
int yy = std::max(std::min(y + yoff, height - 1), 0);
for (int xoff = -1; xoff <= 1; ++xoff) {
int xx = std::max(std::min(x + xoff, width - 1), 0);
val += data[yy*width + xx] * kernel[yoff + 1][xoff + 1];
}
}
out[y*width + x] = abs(val);
}
}
}
}
}
// deposterization: smoothes posterized gradients from low-color-depth (e.g. 444, 565, compressed) sources
void deposterizeH(const u32 *data, u32 *out, int w, int l, int u) {
static const int T = 8;
for (int y = l; y < u; ++y) {
for (int x = 0; x < w; ++x) {
int inpos = y*w + x;
u32 center = data[inpos];
if (x == 0 || x == w - 1) {
out[y*w + x] = center;
continue;
}
u32 left = data[inpos - 1];
u32 right = data[inpos + 1];
out[y*w + x] = 0;
for (int c = 0; c < 4; ++c) {
u8 lc = ((left >> c * 8) & 0xFF);
u8 cc = ((center >> c * 8) & 0xFF);
u8 rc = ((right >> c * 8) & 0xFF);
if ((lc != rc) && ((lc == cc && abs((int)((int)rc) - cc) <= T) || (rc == cc && abs((int)((int)lc) - cc) <= T))) {
// blend this component
out[y*w + x] |= ((rc + lc) / 2) << (c * 8);
} else {
// no change for this component
out[y*w + x] |= cc << (c * 8);
}
}
}
}
}
void deposterizeV(const u32 *data, u32 *out, int w, int h, int l, int u) {
static const int T = 8;
for (int xb = 0; xb < w / BLOCK_SIZE + 1; ++xb) {
for (int y = l; y < u; ++y) {
for (int x = xb*BLOCK_SIZE; x < (xb + 1)*BLOCK_SIZE && x < w; ++x) {
u32 center = data[y * w + x];
if (y == 0 || y == h - 1) {
out[y*w + x] = center;
continue;
}
u32 upper = data[(y - 1) * w + x];
u32 lower = data[(y + 1) * w + x];
out[y*w + x] = 0;
for (int c = 0; c < 4; ++c) {
u8 uc = ((upper >> c * 8) & 0xFF);
u8 cc = ((center >> c * 8) & 0xFF);
u8 lc = ((lower >> c * 8) & 0xFF);
if ((uc != lc) && ((uc == cc && abs((int)((int)lc) - cc) <= T) || (lc == cc && abs((int)((int)uc) - cc) <= T))) {
// blend this component
out[y*w + x] |= ((lc + uc) / 2) << (c * 8);
} else {
// no change for this component
out[y*w + x] |= cc << (c * 8);
}
}
}
}
}
}
// generates a distance mask value for each pixel in data
// higher values -> larger distance to the surrounding pixels
void generateDistanceMask(const u32 *data, u32 *out, int width, int height, int l, int u) {
for (int yb = 0; yb < (u - l) / BLOCK_SIZE + 1; ++yb) {
for (int xb = 0; xb < width / BLOCK_SIZE + 1; ++xb) {
for (int y = l + yb*BLOCK_SIZE; y < l + (yb + 1)*BLOCK_SIZE && y < u; ++y) {
for (int x = xb*BLOCK_SIZE; x < (xb + 1)*BLOCK_SIZE && x < width; ++x) {
const u32 center = data[y*width + x];
u32 dist = 0;
for (int yoff = -1; yoff <= 1; ++yoff) {
int yy = y + yoff;
if (yy == height || yy == -1) {
dist += 1200; // assume distance at borders, usually makes for better result
continue;
}
for (int xoff = -1; xoff <= 1; ++xoff) {
if (yoff == 0 && xoff == 0) continue;
int xx = x + xoff;
if (xx == width || xx == -1) {
dist += 400; // assume distance at borders, usually makes for better result
continue;
}
dist += DISTANCE(data[yy*width + xx], center);
}
}
out[y*width + x] = dist;
}
}
}
}
}
// mix two images based on a mask
void mix(u32 *data, const u32 *source, const u32 *mask, u32 maskmax, int width, int l, int u) {
for (int y = l; y < u; ++y) {
for (int x = 0; x < width; ++x) {
int pos = y*width + x;
u8 mixFactors[2] = { 0, static_cast<u8>((std::min(mask[pos], maskmax) * 255) / maskmax) };
mixFactors[0] = 255 - mixFactors[1];
data[pos] = MIX_PIXELS(data[pos], source[pos], mixFactors);
if (A(source[pos]) == 0) data[pos] = data[pos] & 0x00FFFFFF; // xBRZ always does a better job with hard alpha
}
}
}
//////////////////////////////////////////////////////////////////// Bicubic scaling
// Code for the cubic upscaler is pasted below as-is.
// WARNING: different codestyle.
// NOTE: in several places memcpy is used instead of type punning,
// to avoid strict aliasing problems. This may produce suboptimal
// code, especially on MSVC.
// Loads a sample (4 bytes) from image into 'output'.
static void load_sample(ptrdiff_t w, ptrdiff_t h, ptrdiff_t s, const u8 *pixels, int wrap_mode, ptrdiff_t x, ptrdiff_t y, u8 *output) {
// Check if the sample is inside. NOTE: for b>=0
// the expression (UNSIGNED)a<(UNSIGNED)b is
// equivalent to a>=0&&a<b.
static_assert(sizeof(ptrdiff_t) == sizeof(size_t), "Assumes ptrdiff_t same width as size_t");
if((size_t)x >= (size_t)w || (size_t)y >= (size_t)h) {
switch(wrap_mode) {
case 0: // Wrap
if(!((w & (w-1)) | (h & (h-1)))) {
// Both w and h are powers of 2.
x &= w-1;
y &= h-1;
} else {
// For e.g. 1x1 images we might need to wrap several
// times, hence 'while', instead of 'if'. Probably
// still faster, than modulo.
while(x < 0) x += w;
while(y < 0) y += h;
while(x >= w) x -= w;
while(y >= h) y -= h;
}
break;
case 1: // Clamp
if(x < 0) x = 0;
if(y < 0) y = 0;
if(x >= w) x = w-1;
if(y >= h) y = h-1;
break;
case 2: // Zero
memset(output, 0, 4);
break;
}
}
memcpy(output, pixels + s*y + 4*x, 4);
}
#define BLOCK 8
static void init_block(
ptrdiff_t w, ptrdiff_t h,
ptrdiff_t src_stride, const u8 *src_pixels,
int wrap_mode, ptrdiff_t factor, float B, float C,
ptrdiff_t x0, ptrdiff_t y0,
float (*cx)[4], float (*cy)[4],
ptrdiff_t *lx, ptrdiff_t *ly, ptrdiff_t *lx0, ptrdiff_t *ly0, ptrdiff_t *sx, ptrdiff_t *sy,
u8 (*src)[(BLOCK+4)*4]) {
// Precomputed coefficients for pixel weights
// in the Mitchell-Netravali filter:
// output = SUM(wij*pixel[i]*t^j)
// where t is distance from pixel[1] to the
// sampling position.
float w00 = B/6.0f , w01 = -C-0.5f*B, w02 = 2.0f*C+0.5f*B , w03 = -C-B/6.0f ;
float w10 = 1.0f-B/3.0f,/*w11 = 0.0f ,*/w12 = C+2.0f*B-3.0f , w13 = -C-1.5f*B+2.0f;
float w20 = B/6.0f , w21 = C+0.5f*B, w22 = -2.0f*C-2.5f*B+3.0f, w23 = C+1.5f*B-2.0f;
float /*w30 = 0.0f , w31 = 0.0f ,*/w32 = -C , w33 = C+B/6.0f ;
// Express the sampling position as a rational
// number num/den-1 (off by one, so that num is
// always positive, since the C language does
// not do Euclidean division). Sampling points
// for both src and dst are assumed at pixel centers.
ptrdiff_t den = 2*factor;
float inv_den = 1.0f/(float)den;
for(int dir = 0; dir < 2; ++dir) {
ptrdiff_t num = (dir ? 2*y0+1+factor : 2*x0+1+factor);
ptrdiff_t *l = (dir ? ly : lx), *l0 = (dir ? ly0 : lx0), *s = (dir ? sy : sx);
float (*c)[4] = (dir ? cy : cx);
(*l0) = num/den-2;
num = num%den;
for(ptrdiff_t i = 0, j = 0; i < BLOCK; ++i) {
l[i] = j; // i-th dst pixel accesses src pixels (l0+l[i])..(l0+l[i]+3) in {x|y} direction.
float t = (float)num*inv_den; // Fractional part of the sampling position.
// Write out pixel weights.
c[i][0] = ((w03*t+w02)*t +w01 )*t +w00 ;
c[i][1] = ((w13*t+w12)*t/*+w11*/)*t +w10 ;
c[i][2] = ((w23*t+w22)*t +w21 )*t +w20 ;
c[i][3] = ((w33*t+w32)*t/*+w31*/)*t/*+w30*/;
// Increment the sampling position.
if((num += 2) >= den) {num -= den; j += 1;}
}
(*s) = l[BLOCK-1]+4; // Total sampled src pixels in {x|y} direction.
}
// Get a local copy of the source pixels.
if((*lx0) >=0 && (*ly0) >= 0 && *lx0 + (*sx) <= w && *ly0 + (*sy) <= h) {
for(ptrdiff_t iy = 0; iy < (*sy); ++iy)
memcpy(src[iy], src_pixels+src_stride*((*ly0) + iy) + 4*(*lx0), (size_t)(4*(*sx)));
}
else {
for(ptrdiff_t iy = 0; iy < (*sy); ++iy) for(ptrdiff_t ix = 0; ix < (*sx); ++ix)
load_sample(w, h, src_stride, src_pixels, wrap_mode, (*lx0) + ix, (*ly0) + iy, src[iy] + 4*ix);
}
}
static void upscale_block_c(
ptrdiff_t w, ptrdiff_t h,
ptrdiff_t src_stride, const u8 *src_pixels,
int wrap_mode, ptrdiff_t factor, float B, float C,
ptrdiff_t x0, ptrdiff_t y0,
u8 *dst_pixels) {
float cx[BLOCK][4], cy[BLOCK][4];
ptrdiff_t lx[BLOCK], ly[BLOCK], lx0, ly0, sx, sy;
u8 src[BLOCK+4][(BLOCK+4)*4];
float buf[2][BLOCK+4][BLOCK+4][4];
init_block(
w, h, src_stride, src_pixels, wrap_mode, factor, B, C, x0, y0,
cx, cy, lx, ly, &lx0, &ly0, &sx, &sy, src);
// Unpack source pixels.
for(ptrdiff_t iy = 0; iy < sy; ++iy)
for(ptrdiff_t ix = 0; ix < sx; ++ix)
for(ptrdiff_t k = 0; k < 4; ++k)
buf[0][iy][ix][k] = (float)(int)src[iy][4*ix + k];
// Horizontal pass.
for(ptrdiff_t ix = 0; ix < BLOCK; ++ix) {
#define S(i) (buf[0][iy][lx[ix] + i][k])
float C0 = cx[ix][0], C1 = cx[ix][1], C2 = cx[ix][2], C3 = cx[ix][3];
for(ptrdiff_t iy = 0; iy < sy; ++iy)
for(ptrdiff_t k = 0; k < 4; ++k)
buf[1][iy][ix][k] = S(0)*C0 + S(1)*C1 + S(2)*C2 + S(3)*C3;
#undef S
}
// Vertical pass.
for(ptrdiff_t iy = 0; iy < BLOCK; ++iy) {
#define S(i) (buf[1][ly[iy]+i][ix][k])
float C0 = cy[iy][0], C1 = cy[iy][1], C2 = cy[iy][2], C3 = cy[iy][3];
for(ptrdiff_t ix = 0; ix < BLOCK; ++ix)
for(ptrdiff_t k = 0; k < 4; ++k)
buf[0][iy][ix][k] = S(0)*C0 + S(1)*C1 + S(2)*C2 + S(3)*C3;
#undef S
}
// Pack destination pixels.
for(ptrdiff_t iy = 0; iy < BLOCK; ++iy)
for(ptrdiff_t ix = 0; ix < BLOCK; ++ix) {
u8 pixel[4];
for(ptrdiff_t k = 0; k < 4; ++k) {
float C = buf[0][iy][ix][k];
if(!(C>0.0f)) C = 0.0f;
if(C>255.0f) C = 255.0f;
pixel[k] = (u8)(int)(C + 0.5f);
}
memcpy(dst_pixels + 4*(BLOCK*iy + ix), pixel, 4);
}
}
#if defined(_M_SSE)
#if defined(__GNUC__)
#define ALIGNED(n) __attribute__((aligned(n)))
#elif defined(_MSC_VER)
#define ALIGNED(n) __declspec(align(n))
#else
// For our use case, ALIGNED is a hint, not a requirement,
// so it's fine to ignore it.
#define ALIGNED(n)
#endif
static void upscale_block_sse2(
ptrdiff_t w, ptrdiff_t h,
ptrdiff_t src_stride, const u8 *src_pixels,
int wrap_mode, ptrdiff_t factor, float B, float C,
ptrdiff_t x0, ptrdiff_t y0,
u8 *dst_pixels) {
float cx[BLOCK][4], cy[BLOCK][4];
ptrdiff_t lx[BLOCK], ly[BLOCK], lx0, ly0, sx, sy;
ALIGNED(16) u8 src[BLOCK+4][(BLOCK+4)*4];
ALIGNED(16) float buf[2][BLOCK+4][BLOCK+4][4];
init_block(
w, h, src_stride, src_pixels, wrap_mode, factor, B, C, x0, y0,
cx, cy, lx, ly, &lx0, &ly0, &sx, &sy, src);
// Unpack source pixels.
for(ptrdiff_t iy = 0; iy < sy; ++iy)
for(ptrdiff_t ix = 0; ix < sx; ++ix) {
int pixel;
memcpy(&pixel, src[iy] + 4*ix, 4);
__m128i C = _mm_cvtsi32_si128(pixel);
C = _mm_unpacklo_epi8(C, _mm_set1_epi32(0));
C = _mm_unpacklo_epi8(C, _mm_set1_epi32(0));
_mm_storeu_ps(buf[0][iy][ix], _mm_cvtepi32_ps(C));
}
// Horizontal pass.
for(ptrdiff_t ix = 0; ix < BLOCK; ++ix) {
#define S(i) (buf[0][iy][lx[ix] + i])
__m128 C0 = _mm_set1_ps(cx[ix][0]),
C1 = _mm_set1_ps(cx[ix][1]),
C2 = _mm_set1_ps(cx[ix][2]),
C3 = _mm_set1_ps(cx[ix][3]);
for(ptrdiff_t iy = 0; iy < sy; ++iy)
_mm_storeu_ps(buf[1][iy][ix],
_mm_add_ps(_mm_mul_ps(_mm_loadu_ps(S(0)), C0),
_mm_add_ps(_mm_mul_ps(_mm_loadu_ps(S(1)), C1),
_mm_add_ps(_mm_mul_ps(_mm_loadu_ps(S(2)), C2),
_mm_mul_ps(_mm_loadu_ps(S(3)), C3)))));
#undef S
}
// Vertical pass.
for(ptrdiff_t iy = 0; iy < BLOCK; ++iy) {
#define S(i) (buf[1][ly[iy] + i][ix])
__m128 C0 = _mm_set1_ps(cy[iy][0]),
C1 = _mm_set1_ps(cy[iy][1]),
C2 = _mm_set1_ps(cy[iy][2]),
C3 = _mm_set1_ps(cy[iy][3]);
for(ptrdiff_t ix = 0; ix < BLOCK; ++ix)
_mm_storeu_ps(buf[0][iy][ix],
_mm_add_ps(_mm_mul_ps(_mm_loadu_ps(S(0)), C0),
_mm_add_ps(_mm_mul_ps(_mm_loadu_ps(S(1)), C1),
_mm_add_ps(_mm_mul_ps(_mm_loadu_ps(S(2)), C2),
_mm_mul_ps(_mm_loadu_ps(S(3)), C3)))));
#undef S
}
// Pack destination pixels.
for(ptrdiff_t iy = 0; iy < BLOCK; ++iy)
for(ptrdiff_t ix = 0; ix < BLOCK; ++ix) {
__m128 C = _mm_loadu_ps(buf[0][iy][ix]);
C = _mm_min_ps(_mm_max_ps(C, _mm_set1_ps(0.0f)), _mm_set1_ps(255.0f));
C = _mm_add_ps(C, _mm_set1_ps(0.5f));
__m128i R = _mm_cvttps_epi32(C);
R = _mm_packus_epi16(R, R);
R = _mm_packus_epi16(R, R);
int pixel = _mm_cvtsi128_si32(R);
memcpy(dst_pixels + 4*(BLOCK*iy+ix), &pixel, 4);
}
}
#endif // defined(_M_SSE)
static void upscale_cubic(
ptrdiff_t width, ptrdiff_t height, ptrdiff_t src_stride_in_bytes, const void *src_pixels,
ptrdiff_t dst_stride_in_bytes, void *dst_pixels,
ptrdiff_t scale, float B, float C, int wrap_mode,
ptrdiff_t x0, ptrdiff_t y0, ptrdiff_t x1, ptrdiff_t y1) {
u8 pixels[BLOCK*BLOCK*4];
for(ptrdiff_t y = y0; y < y1; y+= BLOCK)
for(ptrdiff_t x = x0; x < x1; x+= BLOCK) {
#if defined(_M_SSE)
upscale_block_sse2(width, height, src_stride_in_bytes, (const u8*)src_pixels, wrap_mode, scale, B, C, x, y, pixels);
#else
upscale_block_c (width, height, src_stride_in_bytes, (const u8*)src_pixels, wrap_mode, scale, B, C, x, y, pixels);
#endif
for(ptrdiff_t iy = 0, ny = (y1-y < BLOCK ? y1-y : BLOCK), nx = (x1-x < BLOCK ? x1-x : BLOCK); iy < ny; ++iy)
memcpy((u8*)dst_pixels + dst_stride_in_bytes*(y+iy) + 4*x, pixels + BLOCK*4*iy, (size_t)(4*nx));
}
}
// End of pasted cubic upscaler.
void scaleBicubicBSpline(int factor, const u32 *data, u32 *out, int w, int h, int l, int u) {
const float B = 1.0f, C = 0.0f;
const int wrap_mode = 1; // Clamp
upscale_cubic(
w, h, w*4, data,
factor*w*4, out,
factor, B, C, wrap_mode,
0, factor*l, factor*w, factor*u);
}
void scaleBicubicMitchell(int factor, const u32 *data, u32 *out, int w, int h, int l, int u) {
const float B = 0.0f, C = 0.5f; // Actually, Catmull-Rom
const int wrap_mode = 1; // Clamp
upscale_cubic(
w, h, w*4, data,
factor*w*4, out,
factor, B, C, wrap_mode,
0, factor*l, factor*w, factor*u);
}
//////////////////////////////////////////////////////////////////// Bilinear scaling
const static u8 BILINEAR_FACTORS[4][3][2] = {
{ { 44, 211 }, { 0, 0 }, { 0, 0 } }, // x2
{ { 64, 191 }, { 0, 255 }, { 0, 0 } }, // x3
{ { 77, 178 }, { 26, 229 }, { 0, 0 } }, // x4
{ { 102, 153 }, { 51, 204 }, { 0, 255 } }, // x5
};
// integral bilinear upscaling by factor f, horizontal part
template<int f>
void bilinearHt(const u32 *data, u32 *out, int w, int l, int u) {
static_assert(f > 1 && f <= 5, "Bilinear scaling only implemented for factors 2 to 5");
int outw = w*f;
for (int y = l; y < u; ++y) {
for (int x = 0; x < w; ++x) {
int inpos = y*w + x;
u32 left = data[inpos - (x == 0 ? 0 : 1)];
u32 center = data[inpos];
u32 right = data[inpos + (x == w - 1 ? 0 : 1)];
int i = 0;
for (; i < f / 2 + f % 2; ++i) { // first half of the new pixels + center, hope the compiler unrolls this
out[y*outw + x*f + i] = MIX_PIXELS(left, center, BILINEAR_FACTORS[f - 2][i]);
}
for (; i < f; ++i) { // second half of the new pixels, hope the compiler unrolls this
out[y*outw + x*f + i] = MIX_PIXELS(right, center, BILINEAR_FACTORS[f - 2][f - 1 - i]);
}
}
}
}
void bilinearH(int factor, const u32 *data, u32 *out, int w, int l, int u) {
switch (factor) {
case 2: bilinearHt<2>(data, out, w, l, u); break;
case 3: bilinearHt<3>(data, out, w, l, u); break;
case 4: bilinearHt<4>(data, out, w, l, u); break;
case 5: bilinearHt<5>(data, out, w, l, u); break;
default: ERROR_LOG(G3D, "Bilinear upsampling only implemented for factors 2 to 5");
}
}
// integral bilinear upscaling by factor f, vertical part
// gl/gu == global lower and upper bound
template<int f>
void bilinearVt(const u32 *data, u32 *out, int w, int gl, int gu, int l, int u) {
static_assert(f>1 && f <= 5, "Bilinear scaling only implemented for 2x, 3x, 4x, and 5x");
int outw = w*f;
for (int xb = 0; xb < outw / BLOCK_SIZE + 1; ++xb) {
for (int y = l; y < u; ++y) {
u32 uy = y - (y == gl ? 0 : 1);
u32 ly = y + (y == gu - 1 ? 0 : 1);
for (int x = xb*BLOCK_SIZE; x < (xb + 1)*BLOCK_SIZE && x < outw; ++x) {
u32 upper = data[uy * outw + x];
u32 center = data[y * outw + x];
u32 lower = data[ly * outw + x];
int i = 0;
for (; i < f / 2 + f % 2; ++i) { // first half of the new pixels + center, hope the compiler unrolls this
out[(y*f + i)*outw + x] = MIX_PIXELS(upper, center, BILINEAR_FACTORS[f - 2][i]);
}
for (; i < f; ++i) { // second half of the new pixels, hope the compiler unrolls this
out[(y*f + i)*outw + x] = MIX_PIXELS(lower, center, BILINEAR_FACTORS[f - 2][f - 1 - i]);
}
}
}
}
}
void bilinearV(int factor, const u32 *data, u32 *out, int w, int gl, int gu, int l, int u) {
switch (factor) {
case 2: bilinearVt<2>(data, out, w, gl, gu, l, u); break;
case 3: bilinearVt<3>(data, out, w, gl, gu, l, u); break;
case 4: bilinearVt<4>(data, out, w, gl, gu, l, u); break;
case 5: bilinearVt<5>(data, out, w, gl, gu, l, u); break;
default: ERROR_LOG(G3D, "Bilinear upsampling only implemented for factors 2 to 5");
}
}
#undef BLOCK_SIZE
#undef MIX_PIXELS
#undef DISTANCE
#undef R
#undef G
#undef B
#undef A
#ifdef DEBUG_SCALER_OUTPUT
// used for debugging texture scaling (writing textures to files)
static int g_imgCount = 0;
void dbgPPM(int w, int h, u8* pixels, const char* prefix = "dbg") { // 3 component RGB
char fn[32];
snprintf(fn, 32, "%s%04d.ppm", prefix, g_imgCount++);
FILE *fp = fopen(fn, "wb");
fprintf(fp, "P6\n%d %d\n255\n", w, h);
for (int j = 0; j < h; ++j) {
for (int i = 0; i < w; ++i) {
static unsigned char color[3];
color[0] = pixels[(j*w + i) * 4 + 0]; /* red */
color[1] = pixels[(j*w + i) * 4 + 1]; /* green */
color[2] = pixels[(j*w + i) * 4 + 2]; /* blue */
fwrite(color, 1, 3, fp);
}
}
fclose(fp);
}
void dbgPGM(int w, int h, u32* pixels, const char* prefix = "dbg") { // 1 component
char fn[32];
snprintf(fn, 32, "%s%04d.pgm", prefix, g_imgCount++);
FILE *fp = fopen(fn, "wb");
fprintf(fp, "P5\n%d %d\n65536\n", w, h);
for (int j = 0; j < h; ++j) {
for (int i = 0; i < w; ++i) {
fwrite((pixels + (j*w + i)), 1, 2, fp);
}
}
fclose(fp);
}
#endif
}
/////////////////////////////////////// Texture Scaler
TextureScalerCommon::TextureScalerCommon() {
// initBicubicWeights() used to be here.
}
TextureScalerCommon::~TextureScalerCommon() {
}
bool TextureScalerCommon::IsEmptyOrFlat(const u32 *data, int pixels) const {
u32 ref = data[0];
// TODO: SIMD-ify this (although, for most textures we'll get out very early)
for (int i = 1; i < pixels; ++i) {
if (data[i] != ref)
return false;
}
return true;
}
void TextureScalerCommon::ScaleAlways(u32 *out, u32 *src, int width, int height, int *scaledWidth, int *scaledHeight, int factor) {
if (IsEmptyOrFlat(src, width * height)) {
// This means it was a flat texture. Vulkan wants the size up front, so we need to make it happen.
u32 pixel = *src;
*scaledWidth = width * factor;
*scaledHeight = height * factor;
size_t pixelCount = *scaledWidth * *scaledHeight;
// ABCD. If A = D, and AB = CD, then they must all be equal (B = C, etc.)
if ((pixel & 0x000000FF) == (pixel >> 24) && (pixel & 0x0000FFFF) == (pixel >> 16)) {
memset(out, pixel & 0xFF, pixelCount * sizeof(u32));
} else {
// Let's hope this is vectorized.
for (int i = 0; i < pixelCount; ++i) {
out[i] = pixel;
}
}
} else {
ScaleInto(out, src, width, height, scaledWidth, scaledHeight, factor);
}
}
bool TextureScalerCommon::ScaleInto(u32 *outputBuf, u32 *src, int width, int height, int *scaledWidth, int *scaledHeight, int factor) {
#ifdef SCALING_MEASURE_TIME
double t_start = time_now_d();
#endif
u32 *inputBuf = src;
// deposterize
if (g_Config.bTexDeposterize) {
bufDeposter.resize(width * height);
DePosterize(inputBuf, bufDeposter.data(), width, height);
inputBuf = bufDeposter.data();
}
// scale
switch (g_Config.iTexScalingType) {
case XBRZ:
ScaleXBRZ(factor, inputBuf, outputBuf, width, height);
break;
case HYBRID:
ScaleHybrid(factor, inputBuf, outputBuf, width, height);
break;
case BICUBIC:
ScaleBicubicMitchell(factor, inputBuf, outputBuf, width, height);
break;
case HYBRID_BICUBIC:
ScaleHybrid(factor, inputBuf, outputBuf, width, height, true);
break;
default:
ERROR_LOG(G3D, "Unknown scaling type: %d", g_Config.iTexScalingType);
}
// update values accordingly
*scaledWidth = width * factor;
*scaledHeight = height * factor;
#ifdef SCALING_MEASURE_TIME
if (*scaledWidth* *scaledHeight > 64 * 64 * factor*factor) {
double t = time_now_d() - t_start;
NOTICE_LOG(G3D, "TextureScaler: processed %9d pixels in %6.5lf seconds. (%9.2lf Mpixels/second)",
*scaledWidth * *scaledHeight, t, (*scaledWidth * *scaledHeight) / (t * 1000 * 1000));
}
#endif
return true;
}
bool TextureScalerCommon::Scale(u32* &data, int width, int height, int *scaledWidth, int *scaledHeight, int factor) {
// prevent processing empty or flat textures (this happens a lot in some games)
// doesn't hurt the standard case, will be very quick for textures with actual texture
if (IsEmptyOrFlat(data, width*height)) {
DEBUG_LOG(G3D, "TextureScaler: early exit -- empty/flat texture");
return false;
}
bufOutput.resize(width * height * (factor * factor)); // used to store the upscaled image
u32 *outputBuf = bufOutput.data();
if (ScaleInto(outputBuf, data, width, height, scaledWidth, scaledHeight, factor)) {
data = outputBuf;
return true;
}
return false;
}
const int MIN_LINES_PER_THREAD = 4;
void TextureScalerCommon::ScaleXBRZ(int factor, u32* source, u32* dest, int width, int height) {
xbrz::ScalerCfg cfg;
ParallelRangeLoop(&g_threadManager, std::bind(&xbrz::scale, factor, source, dest, width, height, xbrz::ColorFormat::ARGB, cfg, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
}
void TextureScalerCommon::ScaleBilinear(int factor, u32* source, u32* dest, int width, int height) {
bufTmp1.resize(width * height * factor);
u32 *tmpBuf = bufTmp1.data();
ParallelRangeLoop(&g_threadManager, std::bind(&bilinearH, factor, source, tmpBuf, width, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
ParallelRangeLoop(&g_threadManager, std::bind(&bilinearV, factor, tmpBuf, dest, width, 0, height, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
}
void TextureScalerCommon::ScaleBicubicBSpline(int factor, u32* source, u32* dest, int width, int height) {
ParallelRangeLoop(&g_threadManager,std::bind(&scaleBicubicBSpline, factor, source, dest, width, height, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
}
void TextureScalerCommon::ScaleBicubicMitchell(int factor, u32* source, u32* dest, int width, int height) {
ParallelRangeLoop(&g_threadManager,std::bind(&scaleBicubicMitchell, factor, source, dest, width, height, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
}
void TextureScalerCommon::ScaleHybrid(int factor, u32* source, u32* dest, int width, int height, bool bicubic) {
// Basic algorithm:
// 1) determine a feature mask C based on a sobel-ish filter + splatting, and upscale that mask bilinearly
// 2) generate 2 scaled images: A - using Bilinear filtering, B - using xBRZ
// 3) output = A*C + B*(1-C)
const static int KERNEL_SPLAT[3][3] = {
{ 1, 1, 1 }, { 1, 1, 1 }, { 1, 1, 1 }
};
bufTmp1.resize(width*height);
bufTmp2.resize(width*height*factor*factor);
bufTmp3.resize(width*height*factor*factor);
ParallelRangeLoop(&g_threadManager,std::bind(&generateDistanceMask, source, bufTmp1.data(), width, height, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
ParallelRangeLoop(&g_threadManager,std::bind(&convolve3x3, bufTmp1.data(), bufTmp2.data(), KERNEL_SPLAT, width, height, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
ScaleBilinear(factor, bufTmp2.data(), bufTmp3.data(), width, height);
// mask C is now in bufTmp3
ScaleXBRZ(factor, source, bufTmp2.data(), width, height);
// xBRZ upscaled source is in bufTmp2
if (bicubic) ScaleBicubicBSpline(factor, source, dest, width, height);
else ScaleBilinear(factor, source, dest, width, height);
// Upscaled source is in dest
// Now we can mix it all together
// The factor 8192 was found through practical testing on a variety of textures
ParallelRangeLoop(&g_threadManager,std::bind(&mix, dest, bufTmp2.data(), bufTmp3.data(), 8192, width*factor, std::placeholders::_1, std::placeholders::_2), 0, height*factor, MIN_LINES_PER_THREAD);
}
void TextureScalerCommon::DePosterize(u32* source, u32* dest, int width, int height) {
bufTmp3.resize(width*height);
ParallelRangeLoop(&g_threadManager,std::bind(&deposterizeH, source, bufTmp3.data(), width, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
ParallelRangeLoop(&g_threadManager,std::bind(&deposterizeV, bufTmp3.data(), dest, width, height, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
ParallelRangeLoop(&g_threadManager,std::bind(&deposterizeH, dest, bufTmp3.data(), width, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
ParallelRangeLoop(&g_threadManager,std::bind(&deposterizeV, bufTmp3.data(), dest, width, height, std::placeholders::_1, std::placeholders::_2), 0, height, MIN_LINES_PER_THREAD);
}