forked from arunatma/LearnYouAHaskell
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Shapes.hs
36 lines (30 loc) · 1.39 KB
/
Shapes.hs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
-- Part of 08_custom_types.hs (but independent on its own)
-- Learn You a Haskell For Great Good
-- Chapter 8: Making Our Own Types and Typeclasses
-- http://learnyouahaskell.com/making-our-own-types-and-typeclasses
-- (..) after data type inside module command signifies all "value constructors"
-- are also exported with the said data type (here Shape and Point)
module Shapes(
Point(..), -- exporting a data type
Shape(..), -- exporting a data type
surface, -- exporting a function
nudge,
baseCircle,
baseRect
) where
data Point = Point Float Float deriving (Show)
data Shape = Circle Point Float | Rectangle Point Point deriving (Show)
-- Function to calculate the surface area
surface :: Shape -> Float
surface (Circle _ r) = pi * r ^ 2
surface (Rectangle (Point x1 y1) (Point x2 y2)) = (abs $ x2 - x1) * (abs $ y2 - y1)
-- Function to nudge a shape! be it rectangle or circle, move by defined amount.
nudge :: Shape -> Float -> Float -> Shape
nudge (Circle (Point x y) r) a b = Circle (Point (x + a) (y + b)) r
nudge (Rectangle (Point x1 y1) (Point x2 y2)) a b =
Rectangle (Point (x1 + a) (y1 + b)) (Point (x2 + a) (y2 + b))
-- Definitions of base circle and base rectangle
baseCircle :: Float -> Shape
baseCircle r = Circle (Point 0 0) r
baseRect :: Float -> Float -> Shape
baseRect width height = Rectangle (Point 0 0) (Point width height)