Commit | Line | Data |
---|---|---|
56b565b1 MM |
1 | import PMInstance |
2 | import PMDefaults | |
3 | import ProposalMatcher | |
4 | import System.IO | |
5 | import Data.Array.IArray | |
578d7d98 | 6 | import Data.Array.Unboxed |
56b565b1 MM |
7 | import ArrayStuff |
8 | import Text.CSV | |
9 | ||
bbabbd01 MM |
10 | -- Command-line interface with simple tab-separated input/output formats. |
11 | -- ./match <example.in | |
12 | ||
56b565b1 MM |
13 | -- pretty silly but it does the job |
14 | swapTabCommaIn s = map (\c -> if c == '\t' then ',' else if c == ',' then '\t' else c) s | |
15 | removeQuotes s = filter (\c -> not (c == '"')) s | |
bbabbd01 MM |
16 | |
17 | -- Some versions of Text.CSV do not use or accept a trailing newline; compensate for that. | |
18 | removeTrailingNewline s = if not (null s) && last s == '\n' then init s else s | |
19 | addTrailingNewline s = if not (null s) && last s /= '\n' then s ++ ['\n'] else s | |
20 | ||
21 | parseTSV fname str = case parseCSV fname (swapTabCommaIn $ removeTrailingNewline str) of | |
56b565b1 MM |
22 | Left pe -> Left pe |
23 | Right ll -> Right $ map (map swapTabCommaIn) ll | |
bbabbd01 | 24 | printTSV ll = addTrailingNewline $ removeQuotes $ swapTabCommaIn $ printCSV $ map (map swapTabCommaIn) ll |
56b565b1 MM |
25 | |
26 | main = do | |
27 | incsv <- hGetContents stdin | |
28 | -- handle errors another day, or let the platform do it | |
29 | let Right inll = parseTSV "standard input" incsv | |
578d7d98 | 30 | let loadList = tail (head inll) |
56b565b1 MM |
31 | let numRvrs = length loadList |
32 | let loadA = listArray (0, numRvrs-1) (map read loadList) | |
33 | let numProps = length (tail inll) `div` 2 | |
34 | -- explicit type on the next line appears to be necessary | |
35 | let pxarr = listOfListsToArray2D (tail inll) :: Array (Int,Int) String | |
36 | -- careful, we end up transposing the array in here | |
578d7d98 MM |
37 | let prefFixA = funcArray ((0,0), (numRvrs-1,numProps-1)) (\(i,j) -> |
38 | let | |
39 | cell = pxarr ! (2*j, i+1) | |
40 | (fix, pstr) = if last cell == '*' then (True, init cell) else (False, cell) | |
41 | pref = read pstr | |
42 | in (pref, fix)) :: Array (Int,Int) (Wt,Bool) | |
43 | let prefA = amap2 fst prefFixA; fixA = amap2 snd prefFixA | |
44 | let expA = funcArray ((0,0), (numRvrs-1,numProps-1)) (\(i,j) -> read $ pxarr ! (2*j+1, i+1)) | |
45 | let pnrA = funcArray (0, numProps-1) (\j -> read $ pxarr ! (2*j, 0)) | |
46 | let theInst = PMInstance numRvrs numProps loadA prefA expA fixA pnrA | |
56b565b1 MM |
47 | let PMatching theMatching = doMatching pmDefaults theInst |
48 | hPutStr stdout $ printTSV $ map (\(i, j) -> map show [i, j]) theMatching |