- Add code to visualize an instance and matching as a graph (bipartite, rather

[match/match.git] / program / Test.hs

diff --git a/program/Test.hs b/program/Test.hs
index 260ce0b..cef7a53 100644 (file)
--- a/program/Test.hs
+++ b/program/Test.hs
@@ -36,6 +36,10 @@ import Data.Graph.Inductive.Graph
 import Data.Graph.Inductive.Tree
 import ArrayStuff
 
+-- A fixed-seeded random number generator for reproducible experimentation.
+myGen = read "314159265 1" :: StdGen
+
+-- TESTING GRAPH ALGORITHMS
 myGraph = mkGraph [(0, ()), (1, ()), (2, ())]
        [(0, 1, (0, 2)), (0, 2, (1, 3)), (2, 1, (2, -2))] :: Gr () (Int, Int)
 
@@ -46,6 +50,7 @@ flowArray = minCostFlow (0, 2) fst (const 1) snd myGraph (0, 1)
 myNCGraph = mkGraph [(0, ())] [(0, 0, -1)] :: Gr () Int
 bfNCResult = bellmanFord id 0 myNCGraph
 
+-- VISUALIZATION STUFF
 data REdgeF = REdgeF Int Int Int Wt
 instance Show REdgeF where
        show (REdgeF idx cap flow cost) = "#" ++ (show idx) ++ ": "
@@ -54,6 +59,25 @@ flowAnnotate g fa =
        mkGraph (labNodes g) (map (\(n1, n2, REdge i ca co) ->
                (n1, n2, REdgeF i ca (fa ! i) co)) $ labEdges g) :: Gr () REdgeF
 
+showInstanceAsGraph :: Instance -> [(Int, Int)] -> Gr String String
+showInstanceAsGraph (Instance numRvrs numProps rloadA prefA) matchedPairs =
+       let
+               rvrNode i = i
+               propNode j = numRvrs + j
+               numNodes = numRvrs + numProps
+               theNodes = map (\i -> (rvrNode i, "R#" ++ show i ++
+                               " (RLoad " ++ show (rloadA ! i) ++ ")")) [0..numRvrs-1] ++
+                       map (\j -> (propNode j, "P#" ++ show j)) [0..numProps-1]
+               parenthesizeIf False s = s
+               parenthesizeIf True s = "(" ++ s ++ ")"
+               theEdges = do
+                       i <- [0..numRvrs-1]
+                       j <- [0..numProps-1]
+                       return (rvrNode i, propNode j,
+                               parenthesizeIf (elem (i, j) matchedPairs) $ show (prefA ! (i, j)))
+       in mkGraph theNodes theEdges
+
+-- PROPOSAL-MATCHING EXAMPLES
 -- Example from idea book p. 425
 {- 
 (myNumRvrs, myNumProps) = (4, 3)
@@ -80,3 +104,5 @@ ReductionResult rrg rrso rrsi rreib rredi = rdnResult
 rdnFlowArray = minCostFlow rreib reIdx reCap reCost rrg (rrso, rrsi)
 rrg2 = flowAnnotate rrg rdnFlowArray
 myMatching = doMatching myInst
+
+iGraph = showInstanceAsGraph myInst myMatching -- Visualize me!