2
0

psnr.h 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. /*
  2. * Copyright 2013 The LibYuv Project Authors. All rights reserved.
  3. *
  4. * Use of this source code is governed by a BSD-style license
  5. * that can be found in the LICENSE file in the root of the source
  6. * tree. An additional intellectual property rights grant can be found
  7. * in the file PATENTS. All contributing project authors may
  8. * be found in the AUTHORS file in the root of the source tree.
  9. */
  10. // Get PSNR for video sequence. Assuming RAW 4:2:0 Y:Cb:Cr format
  11. #ifndef UTIL_PSNR_H_ // NOLINT
  12. #define UTIL_PSNR_H_
  13. #include <math.h> // For log10()
  14. #ifdef __cplusplus
  15. extern "C" {
  16. #endif
  17. #if !defined(INT_TYPES_DEFINED) && !defined(UINT8_TYPE_DEFINED)
  18. typedef unsigned char uint8_t;
  19. #define UINT8_TYPE_DEFINED
  20. #endif
  21. static const double kMaxPSNR = 128.0;
  22. // libyuv provides this function when linking library for jpeg support.
  23. // TODO(fbarchard): make psnr lib compatible subset of libyuv.
  24. #if !defined(HAVE_JPEG)
  25. // Computer Sum of Squared Error (SSE).
  26. // Pass this to ComputePSNR for final result.
  27. double ComputeSumSquareError(const uint8_t* src_a,
  28. const uint8_t* src_b,
  29. int count);
  30. #endif
  31. // PSNR formula: psnr = 10 * log10 (Peak Signal^2 * size / sse)
  32. // Returns 128.0 (kMaxPSNR) if sse is 0 (perfect match).
  33. double ComputePSNR(double sse, double size);
  34. #ifdef __cplusplus
  35. } // extern "C"
  36. #endif
  37. #endif // UTIL_PSNR_H_ // NOLINT