initial commit

This commit is contained in:
Barrett Ruth 2025-09-11 23:52:32 -05:00
commit dcb7debff6
29 changed files with 1276 additions and 0 deletions

View file

@ -0,0 +1,29 @@
#!/bin/sh
. ./scripts/utils.sh
SRC="$1"
BASE=$(basename "$SRC" .cc)
INPUT="${BASE}.in"
OUTPUT="${BASE}.out"
DBG_BIN="${BASE}.debug"
test -d build || mkdir -p build
test -d io || mkdir -p io
test -f "$INPUT" && test ! -f "io/$INPUT" && mv "$INPUT" "io/"
test -f "$OUTPUT" && test ! -f "io/$OUTPUT" && mv "$OUTPUT" "io/"
test -f "io/$INPUT" || touch "io/$INPUT"
test -f "io/$OUTPUT" || touch "io/$OUTPUT"
INPUT="io/$INPUT"
OUTPUT="io/$OUTPUT"
DBG_BIN="build/$DBG_BIN"
compile_source "$SRC" "$DBG_BIN" "$OUTPUT" @debug_flags.txt
CODE=$?
test $CODE -gt 0 && exit $CODE
execute_binary "$DBG_BIN" "$INPUT" "$OUTPUT" true
exit $?

29
templates/scripts/run.sh Normal file
View file

@ -0,0 +1,29 @@
#!/bin/sh
. ./scripts/utils.sh
SRC="$1"
BASE=$(basename "$SRC" .cc)
INPUT="${BASE}.in"
OUTPUT="${BASE}.out"
RUN_BIN="${BASE}.run"
test -d build || mkdir -p build
test -d io || mkdir -p io
test -f "$INPUT" && test ! -f "io/$INPUT" && mv "$INPUT" "io/"
test -f "$OUTPUT" && test ! -f "io/$OUTPUT" && mv "$OUTPUT" "io/"
test -f "io/$INPUT" || touch "io/$INPUT"
test -f "io/$OUTPUT" || touch "io/$OUTPUT"
INPUT="io/$INPUT"
OUTPUT="io/$OUTPUT"
RUN_BIN="build/$RUN_BIN"
compile_source "$SRC" "$RUN_BIN" "$OUTPUT" ""
CODE=$?
test $CODE -gt 0 && exit $CODE
execute_binary "$RUN_BIN" "$INPUT" "$OUTPUT"
exit $?

85
templates/scripts/scrape.sh Executable file
View file

@ -0,0 +1,85 @@
#!/bin/sh
CONTEST="$1"
PROBLEM="$2"
PROBLEM_LETTER="$3"
if [ -z "$CONTEST" ] || [ -z "$PROBLEM" ]; then
echo "Usage: make scrape <contest> <problem_id> [problem_letter]"
echo "Available contests: cses, atcoder, codeforces"
echo "Examples:"
echo " make scrape cses 1068"
echo " make scrape atcoder abc042 a"
echo " make scrape codeforces 1234 A"
exit
fi
test -d io && true || mkdir -p io
TMPFILE=$(mktemp)
ORIGDIR=$(pwd)
case "$CONTEST" in
cses)
cd "$(dirname "$0")/../.." && uv run scrapers/cses.py "$PROBLEM" > "$TMPFILE"
if [ $? -eq 0 ]; then
cd "$ORIGDIR"
awk '/^---INPUT---$/ {getline; while ($0 != "---OUTPUT---") {print; getline}} END {}' "$TMPFILE" > "io/$PROBLEM.in"
awk '/^---OUTPUT---$/ {getline; while ($0 != "---END---") {print; getline}} END {}' "$TMPFILE" > "io/$PROBLEM.expected"
echo "Scraped problem $PROBLEM to io/$PROBLEM.in and io/$PROBLEM.expected"
else
echo "Failed to scrape problem $PROBLEM"
cat "$TMPFILE"
rm "$TMPFILE"
exit
fi
;;
atcoder)
if [ -z "$PROBLEM_LETTER" ]; then
echo "AtCoder requires problem letter (e.g., make scrape atcoder abc042 a)"
rm "$TMPFILE"
exit
fi
FULL_PROBLEM_ID="${PROBLEM}${PROBLEM_LETTER}"
cd "$(dirname "$0")/../.." && uv run scrapers/atcoder.py "$PROBLEM" "$PROBLEM_LETTER" > "$TMPFILE"
if [ $? -eq 0 ]; then
cd "$ORIGDIR"
awk '/^---INPUT---$/ {getline; while ($0 != "---OUTPUT---") {print; getline}} END {}' "$TMPFILE" > "io/$FULL_PROBLEM_ID.in"
awk '/^---OUTPUT---$/ {getline; while ($0 != "---END---") {print; getline}} END {}' "$TMPFILE" > "io/$FULL_PROBLEM_ID.expected"
echo "Scraped problem $FULL_PROBLEM_ID to io/$FULL_PROBLEM_ID.in and io/$FULL_PROBLEM_ID.expected"
else
echo "Failed to scrape problem $FULL_PROBLEM_ID"
cat "$TMPFILE"
rm "$TMPFILE"
exit
fi
;;
codeforces)
if [ -z "$PROBLEM_LETTER" ]; then
echo "Codeforces requires problem letter (e.g., make scrape codeforces 1234 A)"
rm "$TMPFILE"
exit
fi
FULL_PROBLEM_ID="${PROBLEM}${PROBLEM_LETTER}"
cd "$(dirname "$0")/../.." && uv run scrapers/codeforces.py "$PROBLEM" "$PROBLEM_LETTER" > "$TMPFILE"
if [ $? -eq 0 ]; then
cd "$ORIGDIR"
awk '/^---INPUT---$/ {getline; while ($0 != "---OUTPUT---") {print; getline}} END {}' "$TMPFILE" > "io/$FULL_PROBLEM_ID.in"
awk '/^---OUTPUT---$/ {getline; while ($0 != "---END---") {print; getline}} END {}' "$TMPFILE" > "io/$FULL_PROBLEM_ID.expected"
echo "Scraped problem $FULL_PROBLEM_ID to io/$FULL_PROBLEM_ID.in and io/$FULL_PROBLEM_ID.expected"
else
echo "Failed to scrape problem $FULL_PROBLEM_ID"
echo "You can manually add test cases to io/$FULL_PROBLEM_ID.in and io/$FULL_PROBLEM_ID.expected"
cat "$TMPFILE"
rm "$TMPFILE"
exit
fi
;;
*)
echo "Unknown contest type: $CONTEST"
echo "Available contests: cses, atcoder, codeforces"
rm "$TMPFILE"
exit
;;
esac
rm "$TMPFILE"

View file

@ -0,0 +1,73 @@
#!/bin/sh
execute_binary() {
binary="$1"
input="$2"
output="$3"
is_debug="$4"
start=$(date '+%s.%N')
if [ -n "$is_debug" ]; then
asan="$(ldconfig -p | grep libasan.so | head -n1 | awk '{print $4}')"
LD_PRELOAD="$asan" timeout 2s ./"$binary" <"$input" >"$output" 2>&1
else
timeout 2s ./"$binary" <"$input" >"$output" 2>&1
fi
CODE=$?
end=$(date '+%s.%N')
truncate -s "$(head -n 1000 "$output" | wc -c)" "$output"
if [ $CODE -ge 124 ]; then
MSG=''
case $CODE in
124) MSG='TIMEOUT' ;;
128) MSG='SIGILL' ;;
130) MSG='SIGABRT' ;;
131) MSG='SIGBUS' ;;
136) MSG='SIGFPE' ;;
135) MSG='SIGSEGV' ;;
137) MSG='SIGPIPE' ;;
139) MSG='SIGTERM' ;;
esac
[ $CODE -ne 124 ] && sed -i '$d' "$output"
test -n "$MSG" && printf '\n[code]: %s (%s)' "$CODE" "$MSG" >>"$output"
else
printf '\n[code]: %s' "$CODE" >>"$output"
fi
printf '\n[time]: %s ms' "$(awk "BEGIN {print ($end - $start) * 1000}")" >>$output
test -n "$is_debug" && is_debug_string=true || is_debug_string=false
printf '\n[debug]: %s' "$is_debug_string" >>$output
expected_file="${output%.out}.expected"
if [ -f "$expected_file" ] && [ $CODE -eq 0 ]; then
awk '/^\[[^]]*\]:/ {exit} {print}' "$output" > /tmp/program_output
if cmp -s /tmp/program_output "$expected_file"; then
printf '\n[matches]: true' >>"$output"
else
printf '\n[matches]: false' >>"$output"
fi
rm -f /tmp/program_output
fi
return $CODE
}
compile_source() {
src="$1"
bin="$2"
output="$3"
flags="$4"
test -f "$bin" && rm "$bin" || true
g++ @compile_flags.txt $flags "$src" -o "$bin" 2>"$output"
CODE=$?
if [ $CODE -gt 0 ]; then
printf '\n[code]: %s' "$CODE" >>"$output"
return $CODE
else
echo '' >"$output"
return 0
fi
}