fix partiality warning

diagrams · Dec 18, 2024 · ff3068d · ff3068d
1 parent f17501b
commit ff3068d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/Data/Colour/Palette/RandomColor.hs b/src/Data/Colour/Palette/RandomColor.hs
@@ -41,6 +41,7 @@ import           Data.Colour.Palette.Types
 import           Data.Colour.RGBSpace.HSV
 import           Data.Colour.SRGB            (RGB (..), sRGB)
 import           Data.List                   (find)
+import qualified Data.List.NonEmpty as NE
 import           Data.Maybe                  (fromMaybe)
 
 getColorDefinition :: Hue -> ColorDefinition
@@ -106,9 +107,9 @@ saturationRange :: Hue -> (Int, Int)
 saturationRange hue = result
   where
     lbs = lowerBounds $ getColorDefinition hue
-    result = case lbs of
-      [] -> error "Can\'t obtain saturationRange from an empty lowerBounds"
-      _  -> (fst . head $ lbs, fst . last $ lbs)
+    result = case NE.nonEmpty lbs of
+      Nothing -> error "Can\'t obtain saturationRange from an empty lowerBounds"
+      Just lbsNE -> (fst . NE.head $ lbsNE, fst . NE.last $ lbsNE)
 
 randomSaturation :: MonadRandom m => Hue -> Luminosity -> m Int
 randomSaturation HueMonochrome _   = return 0