#!/usr/bin/perl

$fn = shift || die "usage: mdp2xgmml.pl filename\n";

open( MDP, "<$fn" ) || die "Couldn't open $fn: $!\n";

print <<EOTEXT;
<?xml version="1.0"?>
<!DOCTYPE graph PUBLIC "-//John Punin//DTD graph description//EN" "http://www.cs.rpi.edu/~puninj/XGMML/xgmml.dtd">
<graph directed="1" id="42" label="Hello, I am a graph">
EOTEXT
;

$nstates = <MDP>;
chomp $nstates;

for ( $i=0; $i<$nstates; $i++ ) {
    print <<EOTEXT;
<node id="$i" label="$i" weight="0" />
EOTEXT
}

for ( $i=0; $i<$nstates; $i++ ) {
    $line = <MDP>;
    chomp $line;

    ($start_state, $nacts) = split( /\s+/, $line );
    $start_state = int( $start_state );

    for ( $a=0; $a<$nacts; $a++ ) {
	$line = <MDP>;

	@parts = split( /\s+/, $line );

	$reward = shift @parts;
	$ndeps = shift @parts;

	for ( $n=0; $n<$ndeps; $n++ ) {
	    $end_state = shift @parts;
	    $prob = shift @parts;

	    $end_state = int( $end_state );

	    print <<EOTEXT;
<edge source="$start_state" target="$end_state" weight="$prob" label="$a" />
EOTEXT

	}
    }
}

close( MDP );

print "</graph>\n";