#!/bin/zsh

# to make the script fail safely

set -euo pipefail

out=${1:-output}

echo -e "" > $out

echo -e `date` >> $out

echo -e  "" >> $out

echo -e "Below is the feedback and provisional marks for your submission" >> $out

echo -e "for the Main Part 3 (Scala). Please note all marks are provisional until" >> $out

echo -e "ratified by the assessment board -- this is not an official" >> $out

echo -e "results transcript." >> $out

echo -e "" >> $out

# marks for CW8

marks=$(( 0.0 ))

# 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 -nc -i "$1" -- "$2" -e "" 2> /dev/null 1> /dev/null)

}

function scala_assert_thirty {

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

}

# purity test

function scala_vars {

   (sed 's/immutable/ok/g' c$out > cb$out;

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

}

echo -e "" >> $out

echo -e  "Below is the feedback for your submission re.scala" >> $out

echo -e  "" >> $out

### compilation test

echo -e "re.scala runs?" | tee -a $out

if (scala_compile re.scala)

then

    echo -e "  --> success" | tee -a $out

    tsts=$(( 0 ))

else

    echo -e "  --> SCALA DID NOT RUN re.scala" | tee -a $out

    tsts=$(( 1 )) 

fi

# var, return, ListBuffer test

#

if [ $tsts -eq 0 ]

then 

    echo -e "re.scala does not contain VARS, RETURNS etc?" | tee -a $out

    if (scala_vars re.scala)

    then

	echo -e  "  --> TEST FAILED\n" | tee -a $out

	tsts=$(( 1 ))  

    else

	echo -e "  --> success" | tee -a $out

	tsts=$(( 0 )) 

    fi

else

    tsts=$(( 1 ))  

fi

### make sure datatypes are not tampered with

if [ $tsts -eq 0 ]

then

    echo -e "re.scala has case classes for ALTs and SEQs?" | tee -a $out

    if (scala_assert "re.scala" "re_test0.scala")

    then

	echo -e "  --> passed" | tee -a $out

	tsts=$(( 0 ))

    else	

	echo -e "  --> FAIL" | tee -a $out

	tsts=$(( 1 ))

    fi

else

    tsts=$(( 1 ))   

fi

### re1 nullable test

if [ $tsts -eq 0 ]

then

  echo -e " nullable(ZERO) == false" | tee -a $out

  echo -e " nullable(ONE) == true" | tee -a $out

  echo -e " nullable(CHAR('a')) == false" | tee -a $out

  echo -e " nullable(ZERO | ONE) == true" | tee -a $out

  echo -e " nullable(ZERO | CHAR('a')) == false" | tee -a $out

  echo -e " nullable(ONE ~  ONE) == true" | tee -a $out

  echo -e " nullable(ONE ~ CHAR('a')) == false" | tee -a $out

  echo -e " nullable(STAR(ZERO)) == true" | tee -a $out

  echo -e " nullable(ALTs(List(ONE, CHAR('a'), ZERO))) == true" | tee -a $out

  echo -e " nullable(SEQs(List(ONE, ALTs(List(ONE, CHAR('a'), ZERO)), STAR(ZERO)))) == true" | tee -a $out

  if (scala_assert "re.scala" "re_test1.scala")

  then

      echo -e "  --> success (+ 0.5 Marks)\n" | tee -a $out

      marks=$(( marks + 0.5 ))

  else

      echo -e "  --> ONE OF THE TESTS FAILED\n" | tee -a $out

  fi

fi

### re2 der test

if [ $tsts -eq 0 ]

then

  echo -e " der('a', ZERO | ONE) == (ZERO | ZERO)" | tee -a $out

  echo -e " der('a', (CHAR('a') | ONE) ~ CHAR('a')) == ALT((ONE | ZERO) ~ CHAR('a'), SEQs(List(ONE)))" | tee -a $out

  echo -e " der('a', (CHAR('a') | CHAR('a')) ~ CHAR('a')) == (ONE | ONE) ~ CHAR('a')" | tee -a $out

  echo -e " der('a', STAR(CHAR('a'))) == (ONE ~ STAR(CHAR('a')))" | tee -a $out

  echo -e " der('b', STAR(CHAR('a'))) == (ZERO ~ STAR(CHAR('a')))" | tee -a $out

  echo -e "" | tee -a $out

  echo -e " val r0 = \"a\" ~ \"b\" ~ \"c\"" | tee -a $out

  echo -e " der('a', r0) == (ONE ~ \"b\") ~ \"c\"" | tee -a $out

  echo -e " der('b', r0) == (ZERO ~ \"b\") ~ \"c\"" | tee -a $out

  echo -e " der('c', r0) == (ZERO ~ \"b\") ~ \"c\"" | tee -a $out

  echo -e "" | tee -a $out

  echo -e " val r1 = (ONE ~ \"b\") ~ \"c\"" | tee -a $out

  echo -e " der('a', r1) == ((ZERO ~ \"b\") | SEQs(List(ZERO))) ~ \"c\"" | tee -a $out

  echo -e " der('b', r1) == ((ZERO ~ \"b\") | SEQs(List(ONE))) ~ \"c\"" | tee -a $out

  echo -e " der('c', r1) == ((ZERO ~ \"b\") | SEQs(List(ZERO))) ~ \"c\"" | tee -a $out

  echo -e "" | tee -a $out

  echo -e " val r2 = ((ZERO ~ \"b\") | ONE) ~ \"c\"" | tee -a $out

  echo -e " der('a', r2) == ((((ZERO ~ \"b\") | ZERO) ~ \"c\") | SEQs(List(ZERO)))" | tee -a $out

  echo -e " der('b', r2) == ((((ZERO ~ \"b\") | ZERO) ~ \"c\") | SEQs(List(ZERO)))" | tee -a $out

  echo -e " der('c', r2) == ((((ZERO ~ \"b\") | ZERO) ~ \"c\") | SEQs(List(ONE)))" | tee -a $out

  if (scala_assert "re.scala" "re_test2.scala")

  then

      echo -e "  --> success (+ 1 Mark)\n" | tee -a $out

      marks=$(( marks + 1.0 ))

  else

      echo -e "  --> ONE OF THE TESTS FAILED\n" | tee -a $out

  fi

fi

### re3 denest test

if [ $tsts -eq 0 ]

then

    echo -e " denest(List(ONE, ZERO, ALTs(List(ONE, CHAR('a'))))) == List(ONE, ONE, CHAR('a'))" | tee -a $out

    echo -e " denest(List(ONE ~ ONE, ZERO, ZERO | ONE)) == List(ONE ~ ONE, ZERO, ONE)" | tee -a $out

  if (scala_assert "re.scala" "re_test4.scala")

  then

      echo -e "  --> success (+ 1 Mark)\n" | tee -a $out

      marks=$(( marks + 1.0 ))

  else

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

  fi

fi

### re3a flts test

if [ $tsts -eq 0 ]

then

  echo -e " flts(Nil) == Nil" | tee -a $out

  echo -e " flts(ZERO::ZERO::Nil) == List(ZERO)" | tee -a $out

  echo -e " flts(ZERO::ONE::ZERO::ONE::Nil) == List(ZERO)" | tee -a $out

  echo -e " flts(ONE::ALTs(List(ONE))::ONE::Nil) == List(ALTs(List(ONE)))" | tee -a $out

  echo -e " flts(ONE::ALTs(List(ONE))::ONE::ALTs(List(ONE))::Nil) == List(ALTs(List(ONE)), ALTs(List(ONE)))" | tee -a $out

  echo -e " flts(List(CHAR('a'), ONE, ONE, CHAR('b')), Nil) == List(CHAR('a'), CHAR('b'))" | tee -a $out

  echo -e " flts(List(ONE ~ CHAR('a'), CHAR('b') ~ ONE), Nil) == List(ONE, CHAR('a'), CHAR('b'), ONE)" | tee -a $out

  if (scala_assert "re.scala" "re_test3a.scala")

  then

      echo -e "  --> success (+ 1 Mark)\n" | tee -a $out

      marks=$(( marks + 1.0 ))

  else

      echo -e "  --> ONE OF THE TESTS FAILED\n" | tee -a $out

  fi

fi

### re-smart test

if [ $tsts -eq 0 ]

then

    echo -e " SEQs_smart(Nil) == ONE" | tee -a $out

    echo -e " SEQs_smart(List(ZERO)) == ZERO" | tee -a $out

    echo -e " SEQs_smart(List(CHAR('a'))) == CHAR('a')" | tee -a $out

    echo -e " SEQs_smart(List(ONE ~ ONE)) == ONE ~ ONE" | tee -a $out

    echo -e " SEQs_smart(List(ONE, ONE)) == SEQs(List(ONE, ONE))" | tee -a $out

    echo -e " ALTs_smart(Nil) == ZERO" | tee -a $out

    echo -e " ALTs_smart(List(ONE ~ ONE)) == ONE ~ ONE" | tee -a $out

    echo -e " ALTs_smart(List(ZERO, ZERO)) == ALTs(List(ZERO, ZERO))" | tee -a $out

  if (scala_assert "re.scala" "re_test6.scala")

  then

    echo -e "  --> success (+ 0.5 Marks)\n" | tee -a $out

    marks=$(( marks + 0.5 ))

  else

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

  fi

fi

### re3 test

if [ $tsts -eq 0 ]

then

  echo -e " simp(ZERO | ONE) == ONE" | tee -a $out

  echo -e " simp(STAR(ZERO | ONE)) == STAR(ZERO | ONE)" | tee -a $out

  echo -e " simp(ONE ~ (ONE ~ (ONE ~ CHAR('a')))) == CHAR('a')" | tee -a $out

  echo -e " simp(((ONE ~ ONE) ~ ONE) ~ CHAR('a')) == CHAR('a')" | tee -a $out

  echo -e " simp(((ONE | ONE) ~ ONE) ~ CHAR('a')) == CHAR('a')" | tee -a $out

  echo -e " simp(ONE ~ (ONE ~ (ONE ~ ZERO))) == ZERO" | tee -a $out

  echo -e " simp(ALT(ONE ~ (ONE ~ (ONE ~ ZERO)), CHAR('a'))) == CHAR('a')" | tee -a $out

  echo -e " simp(CHAR('a') | CHAR('a')) == CHAR('a')" | tee -a $out

  echo -e " simp(CHAR('a') ~ CHAR('a')) == CHAR('a') ~ CHAR('a')" | tee -a $out

  echo -e " simp(ONE | CHAR('a')) == (ONE | CHAR('a'))" | tee -a $out

  echo -e " simp(ALT((CHAR('a') | ZERO) ~ ONE," | tee -a $out

  echo -e "          ((ONE | CHAR('b')) | CHAR('c')) ~ (CHAR('d') ~ ZERO))) == CHAR('a')" | tee -a $out

  echo -e " simp((ZERO | ((ZERO | ZERO) | (ZERO | ZERO))) ~ ((ONE | ZERO) | ONE ) ~ (CHAR('a'))) == ZERO" | tee -a $out

  echo -e " simp(ALT(ONE | ONE, ONE | ONE)) == ONE" | tee -a $out

  echo -e " simp(ALT(ZERO | CHAR('a'), CHAR('a') | ZERO)) == CHAR('a')" | tee -a $out

  echo -e " simp(ALT(ONE | CHAR('a'), CHAR('a') | ONE)) == ALT(ONE, CHAR('a'))" | tee -a $out

  echo -e " simp(ALTs(Nil)) == ZERO" | tee -a $out

  echo -e " simp(SEQs(List(CHAR('a')))) == CHAR('a')" | tee -a $out

  if (scala_assert "re.scala" "re_test3.scala")

  then

      echo -e "  --> success (+ 1 Mark)\n" | tee -a $out

      marks=$(( marks + 1.0 ))

  else

      echo -e "  --> ONE OF THE TESTS FAILED\n" | tee -a $out

  fi

fi

### re4 test

if [ $tsts -eq 0 ]

then

  echo -e " val EVIL = SEQ(STAR(STAR(CHAR('a'))), CHAR('b'))" | tee -a $out

  echo -e " ders((\"a\" * 5).toList,EVIL) == SEQs(List(STAR(CHAR('a')), STAR(STAR(CHAR('a'))), CHAR('b')))" | tee -a $out

  echo -e " ders(List('b'),EVIL) == ONE" | tee -a $out

  echo -e " ders(List('b','b'),EVIL) == ZERO" | tee -a $out

  echo -e " matcher(EVIL, \"a\" * 5 ++ \"b\") == true" | tee -a $out

  echo -e " matcher(EVIL, \"a\" * 50 ++ \"b\") == true" | tee -a $out

  echo -e " matcher(EVIL, \"a\" * 50) == false" | tee -a $out

  echo -e " matcher(EVIL, \"b\") == true" | tee -a $out

  echo -e " matcher(EVIL, \"bb\") == false" | tee -a $out

  echo -e " matcher(\"abc\", \"abc\") == true" | tee -a $out

  echo -e " matcher(\"abc\", \"ab\") == false" | tee -a $out

  echo -e " matcher((\"ab\" | \"a\") ~ (ONE | \"bc\"), \"abc\") == true" | tee -a $out

  echo -e " matcher(ONE, \"\") == true" | tee -a $out

  echo -e " matcher(ZERO, \"\") == false" | tee -a $out

  echo -e " matcher(ONE | CHAR('a'), \"\") == true" | tee -a $out

  echo -e " matcher(ONE | CHAR('a'), \"a\") == true" | tee -a $out

  if (scala_assert "re.scala" "re_test7.scala")

  then

      echo -e "  --> success (+ 0.5 Marks)\n" | tee -a $out

      marks=$(( marks + 0.5 ))

  else

      echo -e "  --> ONE OF THE TESTS FAILED\n" | tee -a $out

  fi

fi

### re5 test

if [ $tsts -eq 0 ]

then

  echo -e " val EVIL = SEQ(STAR(STAR(CHAR('a'))), CHAR('b'))" | tee -a $out

  echo -e " size(der('a', der('a', EVIL))) == 36" | tee -a $out

  echo -e " size(der('a', der('a', der('a', EVIL)))) == 83" | tee -a $out

  echo -e " size(ders(\"aaaaaa\".toList, EVIL)) == 7" | tee -a $out

  echo -e " size(ders((\"a\" * 50).toList, EVIL)) == 7" | tee -a $out

  if (scala_assert "re.scala" "re_test8.scala")

  then

      echo -e "  --> success (+ 0.5 Marks)\n" | tee -a $out

      marks=$(( marks + 0.5 ))

  else

      echo -e "  --> ONE OF THE TESTS FAILED\n" | tee -a $out

  fi

fi

### re6 'power' test 1

if [ $tsts -eq 0 ]

then

  echo -e " simp(Iterator.iterate(ONE:Rexp)(r => SEQ(r, ONE | ONE)).drop(50).next) == ONE" | tee -a $out

  echo -e "    ...the Iterator produces the rexp" | tee -a $out

  echo -e "" | tee -a $out

  echo -e "      SEQ(SEQ(SEQ(..., ONE | ONE) , ONE | ONE), ONE | ONE)" | tee -a $out

  echo -e "" | tee -a $out

  echo -e "    where SEQ is nested 50 times." | tee -a $out  

  echo -e "" | tee -a $out

  echo -e " simp(Iterator.iterate(ONE:Rexp)(r => ALT(r, r)).drop(20).next) == ONE" | tee -a $out

  echo -e "    ... the Iterator produces a rexp of size 2097151" | tee -a $out

  echo -e "" | tee -a $out

  echo -e " val EVIL = SEQ(STAR(STAR(CHAR('a'))), CHAR('b'))" | tee -a $out

  echo -e " matcher(EVIL, \"a\" * 1000000) == false" | tee -a $out

  echo -e " matcher(EVIL, \"a\" * 1000000 ++ "b") == true" | tee -a $out

  START=$(date +%s)

  if (scala_assert_thirty "re.scala" "re_test9.scala")

  then

      END=$(date +%s)

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

      echo "  This test ran for $DIFF seconds" | tee -a $out

      echo -e "  --> success (+ 1.0 Mark)\n" | tee -a $out

      marks=$(( marks + 1.0 ))

  else

      END=$(date +%s)

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

      echo "  This test ran for $DIFF seconds" | tee -a $out

      echo -e "  --> ONE OF THE TESTS FAILED\n" | tee -a $out

  fi

fi

## final marks

echo -e "Overall mark for Main Part 3 (Scala)" | tee -a $out

printf " %0.1f\n" $marks | tee -a $out