#!/bin/bash

set -euo pipefail

out=${1:-output}

echo -e "" > $out

echo -e "Below is the feedback for your submission of knight{1,2,3}.scala" >> $out

echo -e "" >> $out

#echo -e "!! Important: !!" >> $out

#echo -e "Because of limitations with our testing infrastructure, we can only" >> $out

#echo -e "let code run for 10 seconds and then have to kill it. This might" >> $out

#echo -e "mean your code is correct, but still marked as Fail. Remember" >> $out

#echo -e "you can test your code on your own machine and benchmark it" >> $out

#echo -e "against the reference implementation." >> $out

#echo -e "" >> $out

# compilation tests

function scala_compile {

  (ulimit -t 30; JAVA_OPTS="-Xmx1g" scala -Xprint:parser "$1" 2> c$out 1> c$out)

}

# functional tests

function scala_assert {

  (ulimit -t 30; JAVA_OPTS="-Xmx1g" scala -i "$1" -- "$2" -e "" 2> /dev/null 1> /dev/null)

}

# purity test

function scala_vars {

   (egrep '\bvar\b|\breturn\b|\.par\.|\.par |ListBuffer|AtomicInteger|mutable|util.control|new Array' c$out 2> /dev/null 1> /dev/null)

}

# compilation test

echo -e "knight1.scala runs?" >> $out

if (scala_compile knight1.scala)

then

    echo -e "  --> passed" >> $out

    tsts1=$(( 0 ))

  else

    echo -e "  --> SCALA DID NOT RUN KNIGHT1.SCALA\n" >> $out

    tsts1=$(( 1 )) 

fi

# knights1: purity test

if [ $tsts1 -eq 0 ]

then  

    echo -e "knight1.scala does not contain vars, returns etc?" >> $out

    if (scala_vars knight1.scala)

    then

	echo -e "  --> FAIL (make triple-sure your program conforms to the required format)" >> $out  

	tsts1=$(( 1 ))

    else

	echo -e "  --> passed" >> $out

	tsts1=$(( 0 )) 

    fi

fi    

### knight1 test

if [ $tsts1 -eq 0 ]

then

    #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out

    echo -e " is_legal(8, Nil, (3, 4)) == true " >> $out

    echo -e " is_legal(8, List((4, 1), (1, 0)), (4, 1)) == false " >> $out

    echo -e " is_legal(2, Nil, (0, 0)) == true" >> $out                          

    if (scala_assert "knight1.scala" "knight_test1.scala")

    then

        echo -e "  --> success" >> $out

    else

        echo -e "  --> \n ONE TEST FAILED\n" >> $out

    fi

fi

### knight2 test

if [ $tsts1 -eq 0 ]

then

  #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out

  echo -e " legal_moves(8, Nil, (2,2)) == List((3,4), (4,3), (4,1), (3,0), (1,0), (0,1), (0,3), (1,4))" >> $out

  echo -e " legal_moves(8, Nil, (7,7)) == List((6,5), (5,6))" >> $out

  echo -e " legal_moves(8, List((4,1), (1,0)), (2,2)) == List((3,4), (4,3), (3,0), (0,1), (0,3), (1,4))" >> $out

  echo -e " legal_moves(8, List((6,6)), (7,7)) == List((6,5), (5,6))" >> $out

  echo -e " legal_moves(1, Nil, (0,0)) == Nil" >> $out

  echo -e " legal_moves(2, Nil, (0,0)) == Nil" >> $out

  echo -e " legal_moves(3, Nil, (0,0)) == List((1,2), (2,1))" >> $out

  if (scala_assert "knight1.scala" "knight_test2.scala")

  then

    echo -e "  --> success" >> $out

  else

    echo -e "  --> \n ONE TEST FAILED\n" >> $out

  fi

fi

### knight3 test

if [ $tsts1 -eq 0 ]

then

  #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out

  echo -e " count_tours from every position on the board" >> $out

  echo -e " dim = 1: 1" >> $out

  echo -e "       2: 0,0,0,0" >>  $out

  echo -e "       3: 0,0,0,0,0,0,0,0,0" >>  $out

  echo -e "       4: 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0" >> $out

  #echo -e "       5: 304,0,56,0,304,0,56,0,56,0,56,0,64,0,56,0,56,0,56,0,304,0,56,0,304" >> $out

  echo -e " enum_tours(5, List((0,0)) ) == 304 and all correct?" >> $out

  echo -e " enum_tours(5, List((0,1)) ) == 0" >> $out

  echo -e " enum_tours(5, List((0,2)) ) == 56 and all correct?" >> $out

  if (scala_assert "knight1.scala" "knight_test3.scala")

  then

    echo -e "  --> success" >> $out

  else

    echo -e "  --> \n ONE TEST FAILED\n" >> $out

  fi

fi

### knight4 test

if [ $tsts1 -eq 0 ]

then

  #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out

  echo -e " Let f = (x:(Int, Int)) => if (x._1 > 3) Some(List(x)) else None " >> $out

  echo -e "   first(List((1,0),(2,0),(3,0),(4,0)), f) == Some(List((4,0)))" >> $out

  echo -e "   first(List((1,0),(2,0),(3,0)), f) == None" >> $out

  if (scala_assert "knight1.scala" "knight_test4.scala") 

  then

    echo -e "  --> success" >> $out

  else

    echo -e "  --> \n ONE TEST FAILED\n" >> $out

  fi

fi

### knight5 test

if [ $tsts1 -eq 0 ]

then

  #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out

  echo -e " is first_tour(6, List((0, 0))) ok? " >> $out

  echo -e " is first_tour(4, List((0, 0))) == None " >> $out

  if (scala_assert "knight1.scala" "knight_test5.scala") 

  then

    echo -e "  --> success" >> $out

  else

    echo -e "  --> \n ONE TEST FAILED\n" >> $out

  fi

fi

echo -e "" >> $out

echo -e "" >> $out

# compilation test

echo -e "knight2.scala runs?" >> $out

if (scala_compile knight2.scala)

then

    echo -e "  --> passed" >> $out

    tsts2=$(( 0 ))

else

    echo -e "  --> SCALA DID NOT RUN KNIGHT2.SCALA\n" >> $out  

    tsts2=$(( 1 )) 

fi

# knights2: purity test

#

if [ $tsts2 -eq 0 ]

then

    echo -e "knight2.scala does not contain vars, returns, Arrays, ListBuffers etc?" >> $out

    if (scala_vars knight2.scala)

    then

	echo -e "  --> Fail (make triple-sure your program conforms to the required format)" >> $out    

	tsts2=$(( 1 ))

    else

	echo -e "  --> passed" >> $out

	tsts2=$(( 0 )) 

    fi

fi

# ordered move test

if [ $tsts2 -eq 0 ]

then

  #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out  

  echo -e " ordered_moves(8, List((3,4), (3,2)), (1,3)) == List((0,1), (0,5), (2,1), (2,5))" >> $out

  echo -e " ordered_moves(8, List((4,0)), (0,0)) == List((2,1), (1,2))" >> $out

  echo -e " ordered_moves(8, List((0,4)), (0,0)) == List((1,2), (2,1))" >> $out

  if (scala_assert "knight2.scala" "knight_test6.scala")

  then

      echo -e "  --> success" >> $out

  else

      echo -e "  --> \n ONE TEST FAILED\n" >> $out  

  fi

fi

# first-closed-tour test

if [ $tsts2 -eq 0 ]

then

  #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out  

  echo -e " first_closed_tour_heuristics(6, List((3,3))) found and correct?" >> $out

  if (scala_assert "knight2.scala" "knight_test7.scala")

  then

      echo -e "  --> success" >> $out

  else

      echo -e "  --> \n ONE TEST FAILED\n" >> $out  

  fi

fi

if [ $tsts2 -eq 0 ]

then

  #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out   

  echo -e " first_tour_heuristics(8, List((0,0))) found and correct?" >> $out

  echo -e " first_tour_heuristics(30, List((0,0))) found and correct?" >> $out

  if (scala_assert "knight2.scala" "knight_test8.scala")

  then

      echo -e "  --> success" >> $out

  else

      echo -e "  --> \n ONE TEST FAILED\n" >> $out 

  fi

fi

echo -e "" >> $out

echo -e "" >> $out

# compilation test

echo "knight3.scala runs?" >> $out

if (scala_compile knight3.scala)

then

    echo "  --> passed" >> $out

    tsts3=$(( 0 ))

else

    echo -e "  --> SCALA DID NOT RUN KNIGHT3.SCALA\n" >> $out  

    tsts3=$(( 1 )) 

fi

# knights3: purity test

#

if  [ $tsts3 -eq 0 ]

then 

    echo -e "knight3.scala does not contain vars, returns, Arrays, ListBuffers etc?" >> $out

    if (scala_vars knight3.scala)

    then

	echo "  --> Fail (make triple-sure your program conforms to the required format)" >> $out

	tsts3=$(( 1 ))

    else

	echo "  --> passed" >> $out

	tsts3=$(( 0 )) 

    fi

fi

if [ $tsts3 -eq 0 ]

then

  #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out  

  echo -e " tour_on_mega_board(70, List((0,0))) found and correct?" >> $out

  START=$(date +%s)

  if (scala_assert "knight3.scala" "knight_test9.scala")

  then

      END=$(date +%s)

      DIFF=$(( $END - $START ))

      echo " This test ran for $DIFF seconds." >> $out

      echo -e "  --> success" >> $out

  else

      END=$(date +%s)

      DIFF=$(( $END - $START ))

      echo " This test ran for $DIFF seconds." >> $out

      echo -e "  --> \n ONE TEST FAILED\n" >> $out 

  fi

fi

echo -e "" >> $out

echo -e "" >> $out

# compilation test

echo "knight4.scala runs?" >> $out

if (scala_compile knight4.scala)

then

    echo "  --> passed" >> $out

    tsts4=$(( 0 ))

else

    echo -e "  --> SCALA DID NOT RUN KNIGHT4.SCALA\n" >> $out  

    tsts4=$(( 1 )) 

fi

# knights4: purity test

#

if  [ $tsts4 -eq 0 ]

then 

    echo -e "knight4.scala does not contain vars, returns, Arrays, ListBuffers etc?" >> $out

    if (scala_vars knight4.scala)

    then

	echo "  --> Fail (make triple-sure your program conforms to the required format)" >> $out

	tsts4=$(( 1 ))

    else

	echo "  --> passed" >> $out

	tsts4=$(( 0 )) 

    fi

fi

if [ $tsts4 -eq 0 ]

then

  #echo -e "For testing needs to take 10 seconds or less to execute: " >> $out  

  echo -e " one_tour_pred(8, List((0, 0)), 40, x => x._1 < 5) is ok?" >> $out

  if (scala_assert "knight4.scala" "knight_test10.scala")

  then

      echo -e "  --> success" >> $out

  else

      echo -e "  --> \n ONE TEST FAILED\n" >> $out 

  fi

fi