





"""simple docstring"""


import math
import os

import cva
import numpy as np

lowerCamelCase__      =      2_5_5.0




def     lowercase__			(    lowercase_						,lowercase_	)    ->						float:




      """simple docstring"""

      _UpperCamelCase					:		Any  =				np.mean((original - contrast) ** 2	)
      if mse == 0:
            return 100

      return 20 * math.logaa(PIXEL_MAX / math.sqrt(__A	)	)




def     lowercase__			(    )    ->						None:




      """simple docstring"""

      _UpperCamelCase					:		int  =				os.path.dirname(os.path.realpath(__A	)	)
      # Loading images (original image and compressed image)
      _UpperCamelCase					:		Tuple  =				cva.imread(os.path.join(__A						,"image_data/original_image.png"	)	)
      _UpperCamelCase					:		Dict  =				cva.imread(os.path.join(__A						,"image_data/compressed_image.png"	)						,1	)

      _UpperCamelCase					:		Optional[int]  =				cva.imread(os.path.join(__A						,"image_data/PSNR-example-base.png"	)	)
      _UpperCamelCase					:		Optional[int]  =				cva.imread(
          os.path.join(__A						,"image_data/PSNR-example-comp-10.jpg"	)						,1	)

      # Value expected: 29.73dB
      print("-- First Test --"	)
      print(F'''PSNR value is {peak_signal_to_noise_ratio(__A						,__A	)} dB'''	)

      # # Value expected: 31.53dB (Wikipedia Example)
      print("\n-- Second Test --"	)
      print(F'''PSNR value is {peak_signal_to_noise_ratio(__A						,__A	)} dB'''	)


if __name__ == "__main__":
  main()



import math
import os

import cva
import numpy as np

__UpperCAmelCase      = 255.0





def 	snake_case_      (__A     :						float					,   __A     :						float     )			->	float:
   __lowerCAmelCase  :			Any					    =						np.mean((original - contrast) ** 2     )
   if mse == 0:
      return 1_0_0

   return 2_0 * math.logaa(PIXEL_MAX / math.sqrt(__A     )     )





def 	snake_case_      ()			->	None:
   __lowerCAmelCase  :			int					    =						os.path.dirname(os.path.realpath(__A     )     )
   # Loading images (original image and compressed image)
   __lowerCAmelCase  :			Tuple					    =						cva.imread(os.path.join(__A					,   """image_data/original_image.png"""     )     )
   __lowerCAmelCase  :			Dict					    =						cva.imread(os.path.join(__A					,   """image_data/compressed_image.png"""     )					,   1     )

   __lowerCAmelCase  :			Optional[int]					    =						cva.imread(os.path.join(__A					,   """image_data/PSNR-example-base.png"""     )     )
   __lowerCAmelCase  :			Optional[int]					    =						cva.imread(
       os.path.join(__A					,   """image_data/PSNR-example-comp-10.jpg"""     )					,   1     )

   # Value expected: 29.73dB
   print("""-- First Test --"""     )
   print(f'''PSNR value is {peak_signal_to_noise_ratio(__A					,   __A     )} dB'''     )

   # # Value expected: 31.53dB (Wikipedia Example)
   print("""\n-- Second Test --"""     )
   print(f'''PSNR value is {peak_signal_to_noise_ratio(__A					,   __A     )} dB'''     )


if __name__ == "__main__":
   main()


