From 5849798cf0e699688b8a72dcd638d00d155804af Mon Sep 17 00:00:00 2001
From: Dave Parker <dave.x.parker@gmail.com>
Date: Tue, 13 Jan 2015 18:32:59 +0000
Subject: [PATCH] Allow -importinitdist (and anything else that uses
 StateValues.readFromFile) to read vector files with lines of the form i:x,
 not just x.

git-svn-id: https://www.prismmodelchecker.org/svn/prism/prism/trunk@9521 bbc10eb1-c90d-0410-af57-cb519fbb1720
---
 prism/src/explicit/StateValues.java   | 19 ++++++++++++++-----
 prism/src/prism/StateValuesDV.java    | 23 +++++++++++++++++------
 prism/src/prism/StateValuesMTBDD.java | 19 +++++++++++++++----
 3 files changed, 46 insertions(+), 15 deletions(-)

diff --git a/prism/src/explicit/StateValues.java b/prism/src/explicit/StateValues.java
index 9dbd578a..4cb60de5 100644
--- a/prism/src/explicit/StateValues.java
+++ b/prism/src/explicit/StateValues.java
@@ -1226,16 +1226,25 @@ public class StateValues implements StateVector
 		BufferedReader in;
 		String s;
 		int lineNum = 0, count = 0;
+		boolean hasIndices = false;
 
 		try {
-			// open file for reading
+			// Open file for reading
 			in = new BufferedReader(new FileReader(file));
-			// read remaining lines
+			// Read remaining lines
 			s = in.readLine();
 			lineNum++;
 			while (s != null) {
 				s = s.trim();
 				if (!("".equals(s))) {
+					// If entry is of form "i=x", use i as index not count
+					// (otherwise, assume line i contains value for state index i)
+					if (s.contains("=")) {
+						hasIndices = true;
+						String ss[] = s.split("=");
+						count = Integer.parseInt(ss[0]);
+						s = ss[1];
+					}
 					if (count + 1 > size) {
 						in.close();
 						throw new PrismException("Too many values in file \"" + file + "\" (more than " + size + ")");
@@ -1255,10 +1264,10 @@ public class StateValues implements StateVector
 				s = in.readLine();
 				lineNum++;
 			}
-			// close file
+			// Close file
 			in.close();
-			// check size
-			if (count < size)
+			// Check size
+			if (!hasIndices && count < size)
 				throw new PrismException("Too few values in file \"" + file + "\" (" + count + ", not " + size + ")");
 		} catch (IOException e) {
 			throw new PrismException("File I/O error reading from \"" + file + "\"");
diff --git a/prism/src/prism/StateValuesDV.java b/prism/src/prism/StateValuesDV.java
index 93a0ce1f..69563918 100644
--- a/prism/src/prism/StateValuesDV.java
+++ b/prism/src/prism/StateValuesDV.java
@@ -134,19 +134,30 @@ public class StateValuesDV implements StateValues
 		String s;
 		int lineNum = 0, count = 0;
 		double d;
+		boolean hasIndices = false;
 		int size = values.getSize();
 
 		try {
-			// open file for reading
+			// Open file for reading
 			in = new BufferedReader(new FileReader(file));
-			// read remaining lines
+			// Read remaining lines
 			s = in.readLine();
 			lineNum++;
 			while (s != null) {
 				s = s.trim();
 				if (!("".equals(s))) {
-					if (count + 1 > size)
+					// If entry is of form "i=x", use i as index not count
+					// (otherwise, assume line i contains value for state index i)
+					if (s.contains("=")) {
+						hasIndices = true;
+						String ss[] = s.split("=");
+						count = Integer.parseInt(ss[0]);
+						s = ss[1];
+					}
+					if (count + 1 > size) {
+						in.close();
 						throw new PrismException("Too many values in file \"" + file + "\" (more than " + size + ")");
+					}
 					d = Double.parseDouble(s);
 					setElement(count, d);
 					count++;
@@ -154,10 +165,10 @@ public class StateValuesDV implements StateValues
 				s = in.readLine();
 				lineNum++;
 			}
-			// close file
+			// Close file
 			in.close();
-			// check size
-			if (count < size)
+			// Check size
+			if (!hasIndices && count < size)
 				throw new PrismException("Too few values in file \"" + file + "\" (" + count + ", not " + size + ")");
 		} catch (IOException e) {
 			throw new PrismException("File I/O error reading from \"" + file + "\"");
diff --git a/prism/src/prism/StateValuesMTBDD.java b/prism/src/prism/StateValuesMTBDD.java
index eeb72d11..44761232 100644
--- a/prism/src/prism/StateValuesMTBDD.java
+++ b/prism/src/prism/StateValuesMTBDD.java
@@ -144,6 +144,7 @@ public class StateValuesMTBDD implements StateValues
 		String s;
 		int lineNum = 0, count = 0;
 		double d;
+		boolean hasIndices = false;
 		long size = model.getNumStates();
 		
 		try {
@@ -154,18 +155,28 @@ public class StateValuesMTBDD implements StateValues
 			while (s != null) {
 				s = s.trim();
 				if (!("".equals(s))) {
-					if (count + 1 > size)
+					// If entry is of form "i=x", use i as index not count
+					// (otherwise, assume line i contains value for state index i)
+					if (s.contains("=")) {
+						hasIndices = true;
+						String ss[] = s.split("=");
+						count = Integer.parseInt(ss[0]);
+						s = ss[1];
+					}
+					if (count + 1 > size) {
+						in.close();
 						throw new PrismException("Too many values in file \"" + file + "\" (more than " + size + ")");
+					}
 					d = Double.parseDouble(s);
 					setElement(count, d);
 					count++;
 				}
 				s = in.readLine(); lineNum++;
 			}
-			// close file
+			// Close file
 			in.close();
-			// check size
-			if (count < size)
+			// Check size
+			if (!hasIndices && count < size)
 				throw new PrismException("Too few values in file \"" + file + "\" (" + count + ", not " + size + ")");
 		}
 		catch (IOException e) {