X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FMisc.hs;h=d3c716283b624449276ae25e1fd271187cf43efb;hb=8964e1ca87745763daa42f7aa838b80c7dbc0192;hp=b2ebeef53c42e470af7e12c58e5e55fdeac74a10;hpb=c864b66c83f2be395fa590321ca313e227d79fab;p=spline3.git diff --git a/src/Misc.hs b/src/Misc.hs index b2ebeef..d3c7162 100644 --- a/src/Misc.hs +++ b/src/Misc.hs @@ -7,6 +7,15 @@ where -- | The standard factorial function. See -- for -- possible improvements. +-- +-- Examples: +-- +-- >>> factorial 0 +-- 1 +-- +-- >>> factorial 4 +-- 24 +-- factorial :: Int -> Int factorial n | n <= 1 = 1 @@ -16,10 +25,30 @@ factorial n -- | Takes a three-dimensional list, and flattens it into a -- one-dimensional one. +-- +-- Examples: +-- +-- >>> flatten [ [[1,2], [3,4]], [[5,6], [7,8]] ] +-- [1,2,3,4,5,6,7,8] +-- flatten :: [[[a]]] -> [a] flatten xs = concat $ concat xs +-- | Switch the x and z dimensions of a three-dimensional list. +transpose_xz :: [[[a]]] -> [[[a]]] +transpose_xz [] = [] +transpose_xz [[]] = [[]] +transpose_xz [[[]]] = [[[]]] +transpose_xz m = + [[[ m !! x !! y !! z | x <- [0..xsize]] + | y <- [0..ysize]] + | z <- [0..zsize]] + where + zsize = (length m) - 1 + ysize = (length (m !! 0))- 1 + xsize = (length ((m !! 0) !! 0)) - 1 + -- | Takes a list, and returns True if its elements are pairwise -- equal. Returns False otherwise. all_equal :: (Eq a) => [a] -> Bool