{-# OPTIONS_GHC -fno-warn-orphans #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
module Graphics.Image.ColorSpaceSpec (spec) where

import Test.Hspec
import Test.QuickCheck
  
import Graphics.Image as I
import Graphics.Image.Interface as II
import Graphics.Image.ColorSpace


instance Arbitrary RGB where
  arbitrary = do
    NonNegative c <- arbitrary
    return $ toEnum (c `mod` 3)

instance Arbitrary e => Arbitrary (Pixel RGB e) where
  arbitrary = PixelRGB <$> arbitrary <*> arbitrary <*> arbitrary


instance (SequentialArray VU RGB e, Arbitrary e) => Arbitrary (Image VU RGB e) where
  arbitrary = do
    (Positive m, Positive n) <- arbitrary
    II.mapM (const arbitrary) $ I.makeImage (m, n) (const $ PixelGray (0 :: Double))
  

prop_ToFromElt :: (ColorSpace cs, Arbitrary cs, Arbitrary (Pixel cs e), Eq (Pixel cs e)) =>
                  Pixel cs e -> Bool
prop_ToFromElt px = px == fromElt (toElt px)


spec :: Spec
spec = describe "ColorSpace" $ do
  it "RGBElt" $ property (prop_ToFromElt :: Pixel RGB Int -> Bool)