From c4fb6ead71da7eadc96b65caab2865cc13690c85 Mon Sep 17 00:00:00 2001
From: Francesco Bariatti <francesco.bariatti@insa-rennes.fr>
Date: Tue, 8 Mar 2016 13:48:59 +0100
Subject: [PATCH] drawState main loop executed only when executed as a program
---
tools/drawState.py | 39 ++++++++++++++++++++-------------------
1 file changed, 20 insertions(+), 19 deletions(-)
diff --git a/tools/drawState.py b/tools/drawState.py
index 2f1c457..ee01d4a 100755
--- a/tools/drawState.py
+++ b/tools/drawState.py
@@ -34,24 +34,25 @@ def drawPenguin(penguin):
}
print("Pos: {pos}, A:{A}, B:{B}, C:{C}, D:{D}, E:{E}, F:{F}, Tot: {tot}".format(**args))
-try:
- while True:
- read = []
- print("Enter state:")
+if __name__ == "__main__":
+ try:
while True:
- line = input()
- if line == "":
- break
- read.append(line)
+ read = []
+ print("Enter state:")
+ while True:
+ line = input()
+ if line == "":
+ break
+ read.append(line)
+
+ state = json.loads(''.join(read))
+ drawBitboard(state["bitboards"]["onefish"],state["bitboards"]["twofish"],state["bitboards"]["threefish"])
+ print("Red penguins")
+ for i in range(4):
+ drawPenguin(state["penguins"]["red"][i])
+ print("Blue penguins")
+ for i in range(4):
+ drawPenguin(state["penguins"]["blue"][i])
- state = json.loads(''.join(read))
- drawBitboard(state["bitboards"]["onefish"],state["bitboards"]["twofish"],state["bitboards"]["threefish"])
- print("Red penguins")
- for i in range(4):
- drawPenguin(state["penguins"]["red"][i])
- print("Blue penguins")
- for i in range(4):
- drawPenguin(state["penguins"]["blue"][i])
-
-except KeyboardInterrupt:
- print("\nMay the fish be with you.")
+ except KeyboardInterrupt:
+ print("\nMay the fish be with you.")
--
GitLab