57 lines
1.3 KiB
Haskell
57 lines
1.3 KiB
Haskell
module Main where
|
|
|
|
import Lib
|
|
|
|
import Linear.Affine
|
|
import Linear.V3
|
|
|
|
import Graphics.Gloss hiding (Point)
|
|
|
|
csvFromPoint :: Point V3 Double -> String
|
|
csvFromPoint (P (V3 x y z)) =
|
|
show x ++ "," ++ show y ++ "," ++ show z
|
|
|
|
csvFromBodies :: [Body] -> [String]
|
|
csvFromBodies [] = []
|
|
csvFromBodies (x:xs) =
|
|
(bodyName x ++ ","
|
|
++ (show $ bodyMass x) ++ ","
|
|
++ (csvFromPoint $ bodyPosition x) ++ "\n")
|
|
:(csvFromBodies xs)
|
|
|
|
steps :: Int -> Double -> [Body] -> IO ()
|
|
steps 0 _ _ = return ()
|
|
steps n dt bodies = do
|
|
putStr . concat $ map ((++) (show n ++ ",")) $ csvFromBodies bodies
|
|
steps (n-1) dt (updateAll dt bodies)
|
|
|
|
width, height, offset :: Int
|
|
width = 1000
|
|
height = 750
|
|
offset = 100
|
|
|
|
window :: Display
|
|
window = InWindow "Orbit" (width, height) (offset, offset)
|
|
|
|
displayBody :: Body -> Picture
|
|
displayBody b = translate (realToFrac x/1e9) (realToFrac y/1e9) $ circle (realToFrac (bodyRadius b)/1e8)
|
|
where P (V3 x y _) = bodyPosition b
|
|
|
|
displayBodies :: [Body] -> Picture
|
|
displayBodies = (color white) . Pictures . (map displayBody)
|
|
|
|
drawing :: Picture
|
|
drawing = color white $ circle 80
|
|
|
|
main :: IO ()
|
|
main = simulate
|
|
window
|
|
black
|
|
25
|
|
[sun, earth, moon, mercury, venus, mars]
|
|
displayBodies
|
|
(\_ dt bs -> updateAll (realToFrac dt*1e6) bs)
|
|
|
|
--main :: IO ()
|
|
--main = steps 1000000 10 [sun, earth, moon, mercury, venus, mars]
|
|
|