Pastable version of Code from a novice Haskell user

<pre class='prettyprint' language='haskell'>module Main where

import Text.XML.HaXml
import Text.XML.HaXml.Parse
import Text.XML.HaXml.Posn
import Text.XML.HaXml.Util
import System.Environment (getArgs)
import System.IO (IOMode(..), hClose, hGetContents, hPutStr, openFile)
import Data.List

main = do 
  args &lt;- getArgs
  case args of
    [collada, ma] -&gt; process collada ma
    _ -&gt; argError
    
process :: String -&gt; String -&gt; IO ()  
process collada ma = do
  colladah &lt;- openFile collada  ReadMode
  mah &lt;- openFile ma ReadMode
  colladaStr &lt;- hGetContents colladah
  maStr &lt;- hGetContents mah
  
  -- Parse the COLLADA file and get the FOV animation data.
  let (Document _ _ root _) = xmlParse collada colladaStr
  let rootElem = CElem root noPos
  let animations = tag &quot;COLLADA&quot; /&gt; tag &quot;library_animations&quot; /&gt; tag &quot;animation&quot; $ rootElem
  let fovAnimation = (filter isFovAnimation animations) !! 0
  let sources = keep /&gt; tag &quot;source&quot; /&gt; tag &quot;float_array&quot; $ fovAnimation
  let times = map ((*24.0) . (read :: String -&gt; Float)) $ words $ tagTextContent (sources !! 0)
  let values = map (yfovToFocalLength . (read :: String -&gt; Float)) $ words $ tagTextContent (sources !! 1)
  let (CElem (Elem _ attributes _) _) = sources !! 0
  let (_, AttValue (Left countVal:_)) = attributes !! 0
  let count = read countVal :: Int
--putStrLn $ show count
--putStrLn $ show values
  
  -- Parse the ma file and get the name of the camera node.
  let createLine = (filter isCustomCameraCreate $ lines maStr) !! 0
  let !cameraName = takeWhile isNotQuote $ drop 1 $ dropWhile isNotQuote createLine
--putStrLn cameraName
  hClose mah
  
  -- Append the ma file with the FOV animation.
  mah2 &lt;- openFile ma AppendMode
  let createNode = &quot;createNode animCurveTU -n \&quot;&quot; ++ cameraName ++ &quot;_focalLength\&quot;;\n\tsetAttr \&quot;.tan\&quot; 10;\n\tsetAttr \&quot;.wgt\&quot; no;\n\tsetAttr -s &quot; ++ (show count) ++ &quot; \&quot;.ktv[0:&quot; ++ (show (count - 1)) ++ &quot;]\&quot; &quot;
  hPutStr mah2 createNode
  hPutStr mah2 (concat $ map pairToString $ zip times values) 
  hPutStr mah2 &quot;;\n&quot;
  let connectAttr = &quot;connectAttr \&quot;&quot; ++ cameraName ++ &quot;_focalLength.o\&quot; \&quot;&quot; ++ cameraName ++ &quot;.fl\&quot;;&quot;
  hPutStr mah2 connectAttr
  
  hClose colladah
  hClose mah2

isFovAnimation animation = 
  let (CElem (Elem _ attributes _) _) = animation
  in let (_, AttValue (Left idStr:_)) = attributes !! 0
     in isSuffixOf &quot;FOV&quot; idStr

yfovToFocalLength yfov = yAperture / (tan (yfov * 0.5 * pi / 180.0) * 2.0)
                            where yAperture = 25.4 * 0.581
     
isCustomCameraCreate line = &quot;createNode camera -n&quot; `isPrefixOf` line

isNotQuote char = char /= '&quot;'

pairToString pair = &quot; &quot; ++ (show . fst $ pair) ++ &quot; &quot; ++ (show . snd $ pair)

argError = putStrLn
  &quot;Usage: add_fov_to_ma.exe COLLADA_filename ma_filename\n&quot;</pre>
<a href="http://www.refactormycode.com/codes/1167-code-from-a-novice-haskell-user" style="color:#fff" title="As seen on RefactorMyCode.com"><img alt="Small_logo" src="http://www.refactormycode.com/images/small_logo.gif" style="border:0" /></a>

Refactor :my => 'code'

Recent

Popular

Pastable version of Code from a novice Haskell user