123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423 |
- #include "libyuv/compare.h"
- #include <float.h>
- #include <math.h>
- #ifdef _OPENMP
- #include <omp.h>
- #endif
- #include "libyuv/basic_types.h"
- #include "libyuv/compare_row.h"
- #include "libyuv/cpu_id.h"
- #include "libyuv/row.h"
- #include "libyuv/video_common.h"
- #ifdef __cplusplus
- namespace libyuv {
- extern "C" {
- #endif
- LIBYUV_API
- uint32 HashDjb2(const uint8* src, uint64 count, uint32 seed) {
- const int kBlockSize = 1 << 15;
- int remainder;
- uint32 (*HashDjb2_SSE)(const uint8* src, int count, uint32 seed) = HashDjb2_C;
- #if defined(HAS_HASHDJB2_SSE41)
- if (TestCpuFlag(kCpuHasSSE41)) {
- HashDjb2_SSE = HashDjb2_SSE41;
- }
- #endif
- #if defined(HAS_HASHDJB2_AVX2)
- if (TestCpuFlag(kCpuHasAVX2)) {
- HashDjb2_SSE = HashDjb2_AVX2;
- }
- #endif
- while (count >= (uint64)(kBlockSize)) {
- seed = HashDjb2_SSE(src, kBlockSize, seed);
- src += kBlockSize;
- count -= kBlockSize;
- }
- remainder = (int)count & ~15;
- if (remainder) {
- seed = HashDjb2_SSE(src, remainder, seed);
- src += remainder;
- count -= remainder;
- }
- remainder = (int)count & 15;
- if (remainder) {
- seed = HashDjb2_C(src, remainder, seed);
- }
- return seed;
- }
- static uint32 ARGBDetectRow_C(const uint8* argb, int width) {
- int x;
- for (x = 0; x < width - 1; x += 2) {
- if (argb[0] != 255) {
- return FOURCC_BGRA;
- }
- if (argb[3] != 255) {
- return FOURCC_ARGB;
- }
- if (argb[4] != 255) {
- return FOURCC_BGRA;
- }
- if (argb[7] != 255) {
- return FOURCC_ARGB;
- }
- argb += 8;
- }
- if (width & 1) {
- if (argb[0] != 255) {
- return FOURCC_BGRA;
- }
- if (argb[3] != 255) {
- return FOURCC_ARGB;
- }
- }
- return 0;
- }
- LIBYUV_API
- uint32 ARGBDetect(const uint8* argb, int stride_argb, int width, int height) {
- uint32 fourcc = 0;
- int h;
-
- if (stride_argb == width * 4) {
- width *= height;
- height = 1;
- stride_argb = 0;
- }
- for (h = 0; h < height && fourcc == 0; ++h) {
- fourcc = ARGBDetectRow_C(argb, width);
- argb += stride_argb;
- }
- return fourcc;
- }
- LIBYUV_API
- uint64 ComputeHammingDistance(const uint8* src_a,
- const uint8* src_b,
- int count) {
- const int kBlockSize = 1 << 15;
- const int kSimdSize = 64;
-
- int remainder = count & (kBlockSize - 1) & ~(kSimdSize - 1);
- uint64 diff = 0;
- int i;
- uint32 (*HammingDistance)(const uint8* src_a, const uint8* src_b, int count) =
- HammingDistance_C;
- #if defined(HAS_HAMMINGDISTANCE_NEON)
- if (TestCpuFlag(kCpuHasNEON)) {
- HammingDistance = HammingDistance_NEON;
- }
- #endif
- #if defined(HAS_HAMMINGDISTANCE_SSSE3)
- if (TestCpuFlag(kCpuHasSSSE3)) {
- HammingDistance = HammingDistance_SSSE3;
- }
- #endif
- #if defined(HAS_HAMMINGDISTANCE_SSE42)
- if (TestCpuFlag(kCpuHasSSE42)) {
- HammingDistance = HammingDistance_SSE42;
- }
- #endif
- #if defined(HAS_HAMMINGDISTANCE_AVX2)
- if (TestCpuFlag(kCpuHasAVX2)) {
- HammingDistance = HammingDistance_AVX2;
- }
- #endif
- #if defined(HAS_HAMMINGDISTANCE_MSA)
- if (TestCpuFlag(kCpuHasMSA)) {
- HammingDistance = HammingDistance_MSA;
- }
- #endif
- #ifdef _OPENMP
- #pragma omp parallel for reduction(+ : diff)
- #endif
- for (i = 0; i < (count - (kBlockSize - 1)); i += kBlockSize) {
- diff += HammingDistance(src_a + i, src_b + i, kBlockSize);
- }
- src_a += count & ~(kBlockSize - 1);
- src_b += count & ~(kBlockSize - 1);
- if (remainder) {
- diff += HammingDistance(src_a, src_b, remainder);
- src_a += remainder;
- src_b += remainder;
- }
- remainder = count & (kSimdSize - 1);
- if (remainder) {
- diff += HammingDistance_C(src_a, src_b, remainder);
- }
- return diff;
- }
- LIBYUV_API
- uint64 ComputeSumSquareError(const uint8* src_a,
- const uint8* src_b,
- int count) {
-
-
-
- const int kBlockSize = 65536;
- int remainder = count & (kBlockSize - 1) & ~31;
- uint64 sse = 0;
- int i;
- uint32 (*SumSquareError)(const uint8* src_a, const uint8* src_b, int count) =
- SumSquareError_C;
- #if defined(HAS_SUMSQUAREERROR_NEON)
- if (TestCpuFlag(kCpuHasNEON)) {
- SumSquareError = SumSquareError_NEON;
- }
- #endif
- #if defined(HAS_SUMSQUAREERROR_SSE2)
- if (TestCpuFlag(kCpuHasSSE2)) {
-
- SumSquareError = SumSquareError_SSE2;
- }
- #endif
- #if defined(HAS_SUMSQUAREERROR_AVX2)
- if (TestCpuFlag(kCpuHasAVX2)) {
-
- SumSquareError = SumSquareError_AVX2;
- }
- #endif
- #if defined(HAS_SUMSQUAREERROR_MSA)
- if (TestCpuFlag(kCpuHasMSA)) {
- SumSquareError = SumSquareError_MSA;
- }
- #endif
- #ifdef _OPENMP
- #pragma omp parallel for reduction(+ : sse)
- #endif
- for (i = 0; i < (count - (kBlockSize - 1)); i += kBlockSize) {
- sse += SumSquareError(src_a + i, src_b + i, kBlockSize);
- }
- src_a += count & ~(kBlockSize - 1);
- src_b += count & ~(kBlockSize - 1);
- if (remainder) {
- sse += SumSquareError(src_a, src_b, remainder);
- src_a += remainder;
- src_b += remainder;
- }
- remainder = count & 31;
- if (remainder) {
- sse += SumSquareError_C(src_a, src_b, remainder);
- }
- return sse;
- }
- LIBYUV_API
- uint64 ComputeSumSquareErrorPlane(const uint8* src_a,
- int stride_a,
- const uint8* src_b,
- int stride_b,
- int width,
- int height) {
- uint64 sse = 0;
- int h;
-
- if (stride_a == width && stride_b == width) {
- width *= height;
- height = 1;
- stride_a = stride_b = 0;
- }
- for (h = 0; h < height; ++h) {
- sse += ComputeSumSquareError(src_a, src_b, width);
- src_a += stride_a;
- src_b += stride_b;
- }
- return sse;
- }
- LIBYUV_API
- double SumSquareErrorToPsnr(uint64 sse, uint64 count) {
- double psnr;
- if (sse > 0) {
- double mse = (double)count / (double)sse;
- psnr = 10.0 * log10(255.0 * 255.0 * mse);
- } else {
- psnr = kMaxPsnr;
- }
- if (psnr > kMaxPsnr)
- psnr = kMaxPsnr;
- return psnr;
- }
- LIBYUV_API
- double CalcFramePsnr(const uint8* src_a,
- int stride_a,
- const uint8* src_b,
- int stride_b,
- int width,
- int height) {
- const uint64 samples = width * height;
- const uint64 sse = ComputeSumSquareErrorPlane(src_a, stride_a, src_b,
- stride_b, width, height);
- return SumSquareErrorToPsnr(sse, samples);
- }
- LIBYUV_API
- double I420Psnr(const uint8* src_y_a,
- int stride_y_a,
- const uint8* src_u_a,
- int stride_u_a,
- const uint8* src_v_a,
- int stride_v_a,
- const uint8* src_y_b,
- int stride_y_b,
- const uint8* src_u_b,
- int stride_u_b,
- const uint8* src_v_b,
- int stride_v_b,
- int width,
- int height) {
- const uint64 sse_y = ComputeSumSquareErrorPlane(src_y_a, stride_y_a, src_y_b,
- stride_y_b, width, height);
- const int width_uv = (width + 1) >> 1;
- const int height_uv = (height + 1) >> 1;
- const uint64 sse_u = ComputeSumSquareErrorPlane(
- src_u_a, stride_u_a, src_u_b, stride_u_b, width_uv, height_uv);
- const uint64 sse_v = ComputeSumSquareErrorPlane(
- src_v_a, stride_v_a, src_v_b, stride_v_b, width_uv, height_uv);
- const uint64 samples = width * height + 2 * (width_uv * height_uv);
- const uint64 sse = sse_y + sse_u + sse_v;
- return SumSquareErrorToPsnr(sse, samples);
- }
- static const int64 cc1 = 26634;
- static const int64 cc2 = 239708;
- static double Ssim8x8_C(const uint8* src_a,
- int stride_a,
- const uint8* src_b,
- int stride_b) {
- int64 sum_a = 0;
- int64 sum_b = 0;
- int64 sum_sq_a = 0;
- int64 sum_sq_b = 0;
- int64 sum_axb = 0;
- int i;
- for (i = 0; i < 8; ++i) {
- int j;
- for (j = 0; j < 8; ++j) {
- sum_a += src_a[j];
- sum_b += src_b[j];
- sum_sq_a += src_a[j] * src_a[j];
- sum_sq_b += src_b[j] * src_b[j];
- sum_axb += src_a[j] * src_b[j];
- }
- src_a += stride_a;
- src_b += stride_b;
- }
- {
- const int64 count = 64;
-
- const int64 c1 = (cc1 * count * count) >> 12;
- const int64 c2 = (cc2 * count * count) >> 12;
- const int64 sum_a_x_sum_b = sum_a * sum_b;
- const int64 ssim_n = (2 * sum_a_x_sum_b + c1) *
- (2 * count * sum_axb - 2 * sum_a_x_sum_b + c2);
- const int64 sum_a_sq = sum_a * sum_a;
- const int64 sum_b_sq = sum_b * sum_b;
- const int64 ssim_d =
- (sum_a_sq + sum_b_sq + c1) *
- (count * sum_sq_a - sum_a_sq + count * sum_sq_b - sum_b_sq + c2);
- if (ssim_d == 0.0) {
- return DBL_MAX;
- }
- return ssim_n * 1.0 / ssim_d;
- }
- }
- LIBYUV_API
- double CalcFrameSsim(const uint8* src_a,
- int stride_a,
- const uint8* src_b,
- int stride_b,
- int width,
- int height) {
- int samples = 0;
- double ssim_total = 0;
- double (*Ssim8x8)(const uint8* src_a, int stride_a, const uint8* src_b,
- int stride_b) = Ssim8x8_C;
-
- int i;
- for (i = 0; i < height - 8; i += 4) {
- int j;
- for (j = 0; j < width - 8; j += 4) {
- ssim_total += Ssim8x8(src_a + j, stride_a, src_b + j, stride_b);
- samples++;
- }
- src_a += stride_a * 4;
- src_b += stride_b * 4;
- }
- ssim_total /= samples;
- return ssim_total;
- }
- LIBYUV_API
- double I420Ssim(const uint8* src_y_a,
- int stride_y_a,
- const uint8* src_u_a,
- int stride_u_a,
- const uint8* src_v_a,
- int stride_v_a,
- const uint8* src_y_b,
- int stride_y_b,
- const uint8* src_u_b,
- int stride_u_b,
- const uint8* src_v_b,
- int stride_v_b,
- int width,
- int height) {
- const double ssim_y =
- CalcFrameSsim(src_y_a, stride_y_a, src_y_b, stride_y_b, width, height);
- const int width_uv = (width + 1) >> 1;
- const int height_uv = (height + 1) >> 1;
- const double ssim_u = CalcFrameSsim(src_u_a, stride_u_a, src_u_b, stride_u_b,
- width_uv, height_uv);
- const double ssim_v = CalcFrameSsim(src_v_a, stride_v_a, src_v_b, stride_v_b,
- width_uv, height_uv);
- return ssim_y * 0.8 + 0.1 * (ssim_u + ssim_v);
- }
- #ifdef __cplusplus
- }
- }
- #endif
|