guess output visualizer for an interactive problem

Author: thorehusfeldt

test/problems/guess/output_visualizer_disabled/guess-visualizer.py

@@ -0,0 +1,32 @@
+import sys
+from pathlib import Path
+
+
+with open(sys.argv[1]) as in_file, open(sys.argv[3] / Path("judgemessage.txt"), "r") as msg_file:
+    mode = in_file.read().split()[0]
+    assert mode in ("random", "fixed", "adaptive"), mode
+    judgemessages = iter(msg_file)
+
+    print(r"""\documentclass[varwidth]{standalone}
+\usepackage{tikz}
+\usetikzlibrary{patterns}
+\tikzset{every node/.style={font=\sffamily}}
+\begin{document}
+\begin{tikzpicture}
+    """)
+    if not mode == "adaptive":
+        secret = int(next(judgemessages).split()[-1])
+        print(rf"\node at ({secret/100},-1.5) {{ {secret} ({mode}) }};")
+    else:
+        next(judgemessages)
+        print(rf"\node at (5,-.5) {{ adaptive }};")
+    for line in judgemessages:
+        rnd, guess = int(line.split()[1]), int(line.split()[3])
+        y = -1 - rnd
+        print(rf"\draw [very thick, blue!20] (0, {y}) -- (10, {y});")
+        print(rf"\node at ({guess/100}, {y})[anchor=north]", r"{$\uparrow$};")
+        print(rf"\node at ({guess/100}, {y-.5})[anchor=north] {{ { guess } }};")
+    if not mode == "adaptive":
+        print(rf"\draw [red] ({secret/100}, {-rnd-1}) -- ({secret/100}, 0);")
+
+    print(r"\end{tikzpicture}\end{document}")

test/problems/guess/output_visualizer_disabled/run

@@ -0,0 +1,41 @@
+#!/bin/bash
+
+# Set script directory
+SCRIPT_DIR="$(dirname "$0")"
+
+# Check if visualize.py exists
+if [[ ! -f "$SCRIPT_DIR/guess-visualizer.py" ]]; then
+    echo "Error: guess-visualizer.py not found in $SCRIPT_DIR" >&2
+    exit 1
+fi
+
+tmptexdir=$(mktemp -d)  # Create a unique temporary directory
+OUTPUT_FILE="$tmptexdir/judgeimage.tex"
+
+# Run visualize.py
+python3 "$SCRIPT_DIR/guess-visualizer.py" $1 $2 $3 >  "$OUTPUT_FILE"
+if [[ $? -ne 0 ]]; then
+    echo "Error: guess-visualizer.py failed" >&2
+    exit 1
+fi
+
+# Check if judgeimage.tex exists
+if [[ ! -f "$OUTPUT_FILE" ]]; then
+    echo "Error: texfile not found in $SCRIPT_DIR" >&2
+    exit 1
+fi
+
+# Run pdflatex
+(
+    cd "$tmptexdir" && pdflatex judgeimage.tex
+)
+if [[ $? -ne 0 ]]; then
+    echo "Error: pdflatex failed" >&2
+    exit 1
+fi
+
+mv "$tmptexdir/judgeimage.pdf" $3
+rm -r "$tmptexdir"
+
+echo "Script completed successfully."
+exit 0