New features from Tuesday call:

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

diff --git a/program/Main.hs b/program/Main.hs
index e3dbfff..0bbfdda 100644 (file)
--- a/program/Main.hs
+++ b/program/Main.hs
@@ -3,6 +3,7 @@ import PMDefaults
 import ProposalMatcher
 import System.IO
 import Data.Array.IArray
+import Data.Array.Unboxed
 import ArrayStuff
 import Text.CSV
 
@@ -18,15 +19,22 @@ main = do
        incsv <- hGetContents stdin
        -- handle errors another day, or let the platform do it
        let Right inll = parseTSV "standard input" incsv
-       let loadList = head inll
+       let loadList = tail (head inll)
        let numRvrs = length loadList
        let loadA = listArray (0, numRvrs-1) (map read loadList)
        let numProps = length (tail inll) `div` 2
        -- explicit type on the next line appears to be necessary
        let pxarr = listOfListsToArray2D (tail inll) :: Array (Int,Int) String
        -- careful, we end up transposing the array in here
-       let prefA = funcArray ((0,0), (numRvrs-1,numProps-1)) (\(i,j) -> read $ pxarr ! (2*j, i))
-       let expA = funcArray ((0,0), (numRvrs-1,numProps-1)) (\(i,j) -> read $ pxarr ! (2*j+1, i))
-       let theInst = PMInstance numRvrs numProps loadA prefA expA
+       let prefFixA = funcArray ((0,0), (numRvrs-1,numProps-1)) (\(i,j) ->
+               let
+                       cell = pxarr ! (2*j, i+1)
+                       (fix, pstr) = if last cell == '*' then (True, init cell) else (False, cell)
+                       pref = read pstr
+               in (pref, fix)) :: Array (Int,Int) (Wt,Bool)
+       let prefA = amap2 fst prefFixA; fixA = amap2 snd prefFixA
+       let expA = funcArray ((0,0), (numRvrs-1,numProps-1)) (\(i,j) -> read $ pxarr ! (2*j+1, i+1))
+       let pnrA = funcArray (0, numProps-1) (\j -> read $ pxarr ! (2*j, 0))
+       let theInst = PMInstance numRvrs numProps loadA prefA expA fixA pnrA
        let PMatching theMatching = doMatching pmDefaults theInst
        hPutStr stdout $ printTSV $ map (\(i, j) -> map show [i, j]) theMatching