diff options
Diffstat (limited to 'Spear/Physics/Rigid.hs')
| -rw-r--r-- | Spear/Physics/Rigid.hs | 52 |
1 files changed, 23 insertions, 29 deletions
diff --git a/Spear/Physics/Rigid.hs b/Spear/Physics/Rigid.hs index 6d3c4d7..396cae4 100644 --- a/Spear/Physics/Rigid.hs +++ b/Spear/Physics/Rigid.hs | |||
| @@ -10,9 +10,9 @@ module Spear.Physics.Rigid | |||
| 10 | where | 10 | where |
| 11 | 11 | ||
| 12 | 12 | ||
| 13 | import qualified Spear.Math.Matrix4 as M4 | 13 | import qualified Spear.Math.Matrix3 as M3 |
| 14 | import Spear.Math.Spatial | 14 | import Spear.Math.Spatial2 |
| 15 | import Spear.Math.Vector3 as V3 | 15 | import Spear.Math.Vector2 |
| 16 | import Spear.Physics.Types | 16 | import Spear.Physics.Types |
| 17 | 17 | ||
| 18 | import Data.List (foldl') | 18 | import Data.List (foldl') |
| @@ -20,55 +20,49 @@ import Control.Monad.State | |||
| 20 | 20 | ||
| 21 | 21 | ||
| 22 | data RigidBody = RigidBody | 22 | data RigidBody = RigidBody |
| 23 | { mass :: !Float | 23 | { mass :: {-# UNPACK #-} !Float |
| 24 | , position :: !Vector3 | 24 | , position :: {-# UNPACK #-} !Vector2 |
| 25 | , velocity :: !Vector3 | 25 | , velocity :: {-# UNPACK #-} !Vector2 |
| 26 | , acceleration :: !Vector3 | 26 | , acceleration :: {-# UNPACK #-} !Vector2 |
| 27 | } | 27 | } |
| 28 | 28 | ||
| 29 | 29 | ||
| 30 | instance Spatial RigidBody where | 30 | instance Spatial2 RigidBody where |
| 31 | 31 | ||
| 32 | move v body = body { position = v + position body } | 32 | move v body = body { position = v + position body } |
| 33 | 33 | ||
| 34 | moveFwd speed body = body { position = position body + scale (-speed) unitZ } | 34 | moveFwd speed body = body { position = position body + scale speed unity } |
| 35 | 35 | ||
| 36 | moveBack speed body = body { position = position body + scale speed unitZ } | 36 | moveBack speed body = body { position = position body + scale (-speed) unity } |
| 37 | 37 | ||
| 38 | strafeLeft speed body = body { position = position body + scale (-speed) unitX } | 38 | strafeLeft speed body = body { position = position body + scale (-speed) unitx } |
| 39 | 39 | ||
| 40 | strafeRight speed body = body { position = position body + scale speed unitX } | 40 | strafeRight speed body = body { position = position body + scale speed unitx } |
| 41 | 41 | ||
| 42 | pitch angle = id | 42 | rotate angle = id |
| 43 | |||
| 44 | yaw angle = id | ||
| 45 | |||
| 46 | roll angle = id | ||
| 47 | 43 | ||
| 48 | pos = position | 44 | pos = position |
| 49 | 45 | ||
| 50 | fwd _ = unitZ | 46 | fwd _ = unity |
| 51 | |||
| 52 | up _ = unitY | ||
| 53 | 47 | ||
| 54 | right _ = unitX | 48 | right _ = unitx |
| 55 | 49 | ||
| 56 | transform body = M4.transform unitX unitY unitZ $ position body | 50 | transform body = M3.transform unitx unity $ position body |
| 57 | 51 | ||
| 58 | setTransform transf body = body { position = M4.position transf } | 52 | setTransform transf body = body { position = M3.position transf } |
| 59 | 53 | ||
| 60 | setPos p body = body { position = p } | 54 | setPos p body = body { position = p } |
| 61 | 55 | ||
| 62 | 56 | ||
| 63 | -- | Build a 'RigidBody'. | 57 | -- | Build a 'RigidBody'. |
| 64 | rigidBody :: Mass -> Position -> RigidBody | 58 | rigidBody :: Mass -> Position -> RigidBody |
| 65 | rigidBody m x = RigidBody m x V3.zero V3.zero | 59 | rigidBody m x = RigidBody m x zero zero |
| 66 | 60 | ||
| 67 | 61 | ||
| 68 | -- | Update the given 'RigidBody'. | 62 | -- | Update the given 'RigidBody'. |
| 69 | update :: [Force] -> Dt -> RigidBody -> RigidBody | 63 | update :: [Force] -> Dt -> RigidBody -> RigidBody |
| 70 | update forces dt body = | 64 | update forces dt body = |
| 71 | let netforce = foldl' (+) V3.zero forces | 65 | let netforce = foldl' (+) zero forces |
| 72 | m = mass body | 66 | m = mass body |
| 73 | r1 = position body | 67 | r1 = position body |
| 74 | v1 = velocity body | 68 | v1 = velocity body |
| @@ -92,8 +86,8 @@ setAcceleration a body = body { acceleration = a } | |||
| 92 | 86 | ||
| 93 | 87 | ||
| 94 | -- test | 88 | -- test |
| 95 | gravity = vec3 0 (-10) 0 | 89 | gravity = vec2 0 (-10) |
| 96 | b0 = rigidBody 50 $ vec3 0 1000 0 | 90 | b0 = rigidBody 50 $ vec2 0 1000 |
| 97 | 91 | ||
| 98 | 92 | ||
| 99 | debug :: IO () | 93 | debug :: IO () |
| @@ -110,7 +104,7 @@ debug' = do | |||
| 110 | step $ update [gravity*50] 1 | 104 | step $ update [gravity*50] 1 |
| 111 | step $ update [gravity*50] 1 | 105 | step $ update [gravity*50] 1 |
| 112 | lift . putStrLn $ "Jumping" | 106 | lift . putStrLn $ "Jumping" |
| 113 | step $ update [gravity*50, vec3 0 9000 0] 1 | 107 | step $ update [gravity*50, vec2 0 9000] 1 |
| 114 | lift . putStrLn $ "Falling..." | 108 | lift . putStrLn $ "Falling..." |
| 115 | step $ update [gravity*50] 1 | 109 | step $ update [gravity*50] 1 |
| 116 | step $ update [gravity*50] 1 | 110 | step $ update [gravity*50] 1 |
| @@ -131,4 +125,4 @@ show' body = | |||
| 131 | ", acceleration " ++ (showVec $ acceleration body) | 125 | ", acceleration " ++ (showVec $ acceleration body) |
| 132 | 126 | ||
| 133 | 127 | ||
| 134 | showVec v = (show $ x v) ++ ", " ++ (show $ y v) ++ ", " ++ (show $ z v) | 128 | showVec v = (show $ x v) ++ ", " ++ (show $ y v) |
