Add printDTMC

1 files changed, 31 insertions, 1 deletions

diff --git a/Assignment2/src/DTMC.icl b/Assignment2/src/DTMC.icl
index 9f1e938..9d1ec00 100644
--- a/Assignment2/src/DTMC.icl
+++ b/Assignment2/src/DTMC.icl
@@ -164,6 +164,34 @@ where
 	| isEmpty current      = groupLines head [line] lines
 	| otherwise            = [reverse current:groupLines head [line] lines]
 
+printDTMC :: !*DTMC -> *(!String, !*DTMC)
+printDTMC dtmc=:{nr_states,states}
+# (state_strings,states) = printStates 0 states
+# s = join "\n"
+	[ "@type: DTMC"
+	, "@parameters"
+	, "@reward_models"
+	, "@nr_states"
+	, toString nr_states
+	, "@model"
+	: state_strings
+	]
+= (s, {dtmc & states=states})
+where
+	printStates :: !Int !*{Maybe State} -> *(![String], !*{Maybe State})
+	printStates i ss
+	| i == nr_states = ([], ss)
+	# (s,ss) = ss![i]
+	| isNothing s = printStates (i+1) ss
+	# s = fromJust s
+	# repr =
+		[ "state " <+ s.state_id
+		, "\taction 0"
+		: ["\t\t" <+ to <+ " : " <+ prob \\ (to,prob) <- 'M'.toList s.transitions]
+		]
+	# (reprs,ss) = printStates (i+1) ss
+	= (repr ++ reprs,ss)
+
 addVars :: *DTMC -> *DTMC
 addVars dtmc = {DTMC | nr_states = dtmc.nr_states, states = addVarsS 0 dtmc.nr_states dtmc.states}
 where
@@ -186,4 +214,6 @@ where
 
 Start w
 # (dtmc,w) = parseDTMC "die.drn" w
-= [(s.state_id, 'M'.toList s.transitions, '\n') \\ Just s <-: (stateElimination dtmc).states]
+# dtmc = addVars dtmc
+# dtmc = stateElimination dtmc
+= printDTMC dtmc