プログラミング コンテストで Haskell に粒子ベースの流体シミュレーションを実装する際に少し問題があります。現在、各シミュレーション ステップで変更される粒子の配列があります。各粒子は、位置と速度 (私自身の Vec3D モジュール) の 2 つのベクトルのタプルです。ある時点で、粒子から位置を抽出する必要があります (解凍リストのようなもの)。これを次のようにしようとしました。
let xs = runSTArray $ do
xs' <- newArray (min, max) (0.0,0.0,0.0) :: ST s (STArray s Int Vec3D)
forM_ [min..max] $ \j -> do
(x, v) <- readArray ps' j
writeArray xs' j x
return xs'
let displacements = doubleDensityRelaxation xs restDen k kNear t h
whereps'
とdoubleDensityRelaxation
はタイプです
type Vec3D = (Double, Double, Double)
ps' :: ST s (STArray s Int (Vec3D, Vec3D))
doubleDensityRelaxation :: Array Int Vec3D -> Double -> Double -> Double -> Double -> Double -> Array Int Vec3D
xs
タイプのはずですxs :: Array Int Vec3D
。しかし、私は得る
Simulator.hs:76:35:
No instance for (MArray (STArray s) (Vec3D, Vec3D) (ST s1))
arising from a use of `readArray'
Possible fix:
add an instance declaration for
(MArray (STArray s) (Vec3D, Vec3D) (ST s1))
In a stmt of a 'do' block: (x, v) <- readArray ps' j
In the expression:
do { (x, v) <- readArray ps' j;
writeArray xs' j x }
In the second argument of `($)', namely
`\ j
-> do { (x, v) <- readArray ps' j;
writeArray xs' j x }'
readArray
配列全体を返すことになっていないことを考えると、私が本当に理解していないコンパイラから。たった一つの(Vec3D, Vec3D)
要素。
修正として、直接 doubleDensityRelaxation
取ることはできますか?ST s (STArray s Int Vec3D)
let xs = runSTArray $ do
そのようなタイプを変更して、取得した部分を 削除すると
Couldn't match expected type `ST s0 (STArray s0 Int Vec3D)'
with actual type `STArray s Int Vec3D'
(ST s xs')
しかし、それを入力として与えると、データコンストラクターがスコープ内にないというxs'
不平を言うだけではありません。ST
私の輸入品は現在
import Data.List
import Data.Array
import Data.Array.ST
import Control.Monad
import Control.Monad.ST
import Vec3D
完全な機能:
step :: Array Int (Vec3D, Vec3D) -> Vec3D -> Double -> Double -> Double -> Double -> Double -> Array Int (Vec3D, Vec3D)
step ps g restDen k kNear t h = runSTArray $ do
ps' <- thaw ps :: ST s (STArray s Int Particle)
--GRAVITY
forM_ [min..max] $ \i -> do
(x, v) <- readArray ps' i
writeArray ps' i (x, addGravity v g t)
--TODO - VISCOSITY
--MOVE
xsOld <- newArray (min, max) (0.0,0.0,0.0) :: ST s(STArray s Int Vec3D)
forM_ [min..max] $ \i -> do
(x, v) <- readArray ps' i
writeArray xsOld i x
writeArray ps' i (x `add` (v `mulSc` t), v)
--TODO - SPRINGS
--DOUBLE DENSITY RELAXATION
xs' <- newArray (min, max) (0.0,0.0,0.0) :: ST s (STArray s Int Vec3D)
forM_ [min..max] $ \j -> do
(x, v) <- readArray ps' j
writeArray xs' j x
let displacements = doubleDensityRelaxation (freeze xs') restDen k kNear t h
ps <- newArray (min, max) ((0.0,0.0,0.0), (0.0,0.0,0.0)) :: ST s (STArray s Int (Vec3D, Vec3D))
--TODO incomplete
return ps
where
addGravity v g t = v `add` (g `mulSc` t)
(min, max) = bounds ps