psnr.h 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  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;
  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* org, const uint8* rec, int size);
  28. #endif
  29. // PSNR formula: psnr = 10 * log10 (Peak Signal^2 * size / sse)
  30. // Returns 128.0 (kMaxPSNR) if sse is 0 (perfect match).
  31. double ComputePSNR(double sse, double size);
  32. #ifdef __cplusplus
  33. } // extern "C"
  34. #endif
  35. #endif // UTIL_PSNR_H_ // NOLINT