]> gitweb.michael.orlitzky.com - numerical-analysis.git/blob - fem1.hs
59d48bb7ad0b454a6699715f34149a3f9dc7e714
[numerical-analysis.git] / fem1.hs
1 module Main
2 where
3
4 import FEM.R1
5 import Linear.Matrix hiding ((*))
6 import Naturals
7 import Piecewise
8
9 big_A' :: Double -> Double
10 big_A' = const 1
11
12 c' :: Double -> Double
13 c' x = sin x
14
15 f' :: Double -> Double
16 f' x = x*(sin x)
17
18 bdy' :: BoundaryConditions Double
19 bdy' = Left (Dirichlet (0,1))
20
21
22 pde' :: PDE Double
23 pde' = PDE big_A' c' f' bdy'
24
25 mesh' :: Col4 (Double,Double)
26 mesh' = fromList [[(0.0,1/3)],
27 [(1/3,2/3)],
28 [(2/3,4/5)],
29 [(4/5,1.0)]]
30
31 pvec' :: Col4 Int
32 pvec' = fromList [[2],[3],[2],[1]]
33
34 params' :: Params N4 N3 N7 Double
35 params' = Params mesh' pvec'
36
37
38 soln :: Piecewise Double
39 soln = solution pde' params'