hashirama/reanimate
Watch
1
0
Fork
You've already forked reanimate
 0
mirror of https://github.com/reanimate/reanimate.git synced 2026-09-15 18:12:20 +00:00
Code Issues Projects Releases Wiki Activity

Clean up fourier_draw sligthly.

Browse source
This commit is contained in:
David Himmelstrup 2019-07-07 11:33:14 +08:00
commit 02f422b019
1 changed files with 4 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

23
examples/fourier_draw.hs
View file

@ -1,10 +1,9 @@
#!/usr/bin/env stack #!/usr/bin/env stack
-- stack --resolver lts-12.26 runghc --package reanimate -- stack --resolver lts-13.14 runghc --package reanimate
{-# LANGUAGE OverloadedStrings #-} {-# LANGUAGE OverloadedStrings #-}
module Main (main) where module Main (main) where
import Control.Lens import Control.Lens
import Control.Monad
import Data.Complex import Data.Complex
import qualified Data.Text as T import qualified Data.Text as T
@ -14,11 +13,6 @@ import Reanimate.Driver (reanimate)
import Reanimate.LaTeX import Reanimate.LaTeX
import Reanimate.Monad import Reanimate.Monad
import Reanimate.Svg import Reanimate.Svg
import Reanimate.Combinators
waveMultiplier :: Int
-- waveMultiplier = 1 -- Sawtooth wave
waveMultiplier = 2 -- Square wave
main :: IO () main :: IO ()
main = reanimate $ pauseAtEnd 2 $ main = reanimate $ pauseAtEnd 2 $
@ -37,6 +31,7 @@ main = reanimate $ pauseAtEnd 2 $
fourierAnimation 25 40 `before` fourierAnimation 25 40 `before`
fourierAnimation 40 40 fourierAnimation 40 40
sWidth :: Double
sWidth = 0.5 sWidth = 0.5
applyMorph :: Int -> [Complex Double] -> Complex Double -> [Complex Double] applyMorph :: Int -> [Complex Double] -> Complex Double -> [Complex Double]
@ -69,7 +64,7 @@ piPoints = lineToPoints 500 $
findCoefficient :: Int -> Complex Double findCoefficient :: Int -> Complex Double
findCoefficient n = findCoefficient n =
sum [ toComplex point * exp (negate (fromIntegral n) * 2 *pi * i*t) * deltaT sum [ toComplex point * exp (negate (fromIntegral n) * 2 *pi * i*t) * deltaT
| (idx, point) <- zip [0..] piPoints, let t = fromIntegral idx/nPoints ] | (idx, point) <- zip [0::Int ..] piPoints, let t = fromIntegral idx/nPoints ]
where where
i = 0 :+ 1 i = 0 :+ 1
toComplex (V2 x y) = x :+ y toComplex (V2 x y) = x :+ y
@ -116,6 +111,7 @@ drawNCircles circles = do
& circleRadius .~ Num radius & circleRadius .~ Num radius
mapF (translate x y) $ worker rest mapF (translate x y) $ worker rest
mkCirclePath :: [Complex Double] -> [(Double, Double)]
mkCirclePath circles = mkCirclePath circles =
[ (x, y) [ (x, y)
| idx <- [0 .. granularity] | idx <- [0 .. granularity]
@ -124,14 +120,3 @@ mkCirclePath circles =
] ]
where where
granularity = 500 granularity = 500
fourierYValue n phi =
imagPart (sum [ nthCircle i phi | i <- [0..n-1]])
nthCircle :: Int -> Double -> Complex Double
nthCircle n phi = x :+ y
where
n' = fromIntegral (n*waveMultiplier+1)
x = cos (n'*phi) * radius
y = sin (n'*phi) * radius
radius = 40 * (2 / (n'*pi))