3 importJava "gnu.trove.map.hash.TIntFloatHashMap" where
6 @JavaName adjustOrPutValue
7 adjustLMap_ :: LMap -> Integer -> Float -> Float -> <Proc> ()
10 adjustLMap :: LMap -> Integer -> Float -> <Proc> ()
11 adjustLMap m k v = adjustLMap_ m k v v
13 data LPTerm = LPTerm (LMap -> Float -> <Proc> ())
15 instance Additive LPTerm where
17 zero = LPTerm (\_ _ -> ())
19 LPTerm a + LPTerm b = LPTerm (\m s -> do a m s ; b m s)
20 sum ts = LPTerm (\m s -> for ts (\(LPTerm t) -> t m s))
22 instance Ring LPTerm where
24 neg (LPTerm a) = LPTerm (\m s -> a m (-s))
26 LPTerm a - LPTerm b = LPTerm (\m s -> do a m s ; b m (-s))
28 fromInteger c = LPTerm (\m s -> adjustLMap m (-1) (fromInteger c*s))
30 one = LPTerm (\m s -> adjustLMap m (-1) s)
31 _ * _ = fail "Multiplication is not supported."
33 data LPProblem = LPProblem (Ref Integer)
35 newProblem :: () -> <Proc> LPProblem
36 newProblem _ = LPProblem (ref 0)
38 newVar :: LPProblem -> <Proc> LPTerm
39 newVar (LPProblem varCounter) = do
40 curId = getRef varCounter
41 varCounter := curId + 1
42 LPTerm (\m s -> adjustLMap m curId s)
47 (**) :: Float -> LPTerm -> LPTerm
48 s0 ** LPTerm t = LPTerm (\m s -> t m (s0*s))
51 data LPTerm = LPTerm Double (Map.T String Double)
53 instance Additive LPTerm where
54 zero = LPTerm 0 Map.empty
55 LPTerm c1 m1 + LPTerm c2 m2 = LPTerm (c1+c2) (Map.merge (+) m1 m2)
57 instance Ring LPTerm where
58 one = LPTerm 1 Map.empty
59 neg (LPTerm c m) = LPTerm (-c) (map neg m)
60 LPTerm c1 m1 - LPTerm c2 m2 = LPTerm (c1-c2) (Map.merge (-) m1 m2)
62 LPTerm c1 [] * LPTerm c2 m2 = LPTerm (c1*c2) (Map.merge (\x -> c1*x) m2)
63 LPTerm c1 m1 * LPTerm c2 [] = LPTerm (c1*c2) (Map.merge (\x -> c2*x) m1)
64 _ * _ = fail "Invalid expression: not linear."
66 fromInteger i = LPTerm (fromInteger i) Map.empty
68 data LPConstraint = LPConstraint String LPTerm
70 (>==) :: LPTerm -> LPTerm -> String -> [LPConstraint]
71 (a >== b) name = [LPConstraint name (a-b)]
73 (<==) :: LPTerm -> LPTerm -> String -> [LPConstraint]
74 (a <== b) name = [LPConstraint name (b-a)]
79 problem = newProblem ()