New paper-side gadget for PC + ERC

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

import PMInstance
import PMDefaults
import ProposalMatcher
import System.IO
import Data.Array.IArray
import ArrayStuff
import Text.CSV

-- pretty silly but it does the job
swapTabCommaIn s = map (\c -> if c == '\t' then ',' else if c == ',' then '\t' else c) s
removeQuotes s = filter (\c -> not (c == '"')) s
parseTSV fname str = case parseCSV fname (swapTabCommaIn str) of
	Left pe -> Left pe
	Right ll -> Right $ map (map swapTabCommaIn) ll
printTSV ll = removeQuotes $ swapTabCommaIn $ printCSV $ map (map swapTabCommaIn) ll

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 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 PMatching theMatching = doMatching pmDefaults theInst
	hPutStr stdout $ printTSV $ map (\(i, j) -> map show [i, j]) theMatching