MA0301/exam_template/python/Truthtable.py

from sys import argv
from pathlib import Path

from common import printerr, replaceContent

try:
  from tabulate import tabulate
except:
  printerr('Couldn\'t find tabulate. Do you have it installed?')


def parseExpressionList(inputData):
  return [e.strip() for e in inputData.split(',')]


class OverloadedBool:
  """Overloads the bool in order to make Implies and Iff functions"""

  def __init__(self, val):
    self.val = val
  
  def __bool__(self):
    val = self.val
    while(type(val) is OverloadedBool):
      val = val.val
    return val

  def __str__(self):
    return str(self.val)

  def __add__(self, bool2):
    """ Implies """
    return OverloadedBool(not self.val or bool2)

  def __sub__(self, bool2):
    """ Iff """
    return OverloadedBool((not self.val or bool2) and (not bool2 or self.val))

def flattenImpliesIff(exps):
  return [exp.replace('implies', '+').replace('iff', '-').replace('E', '') for exp in exps]

def generateTruthTable(exps):

  exps = flattenImpliesIff(exps)

  boolvars = [e for e in exps if len(e) == 1]

  truthtable = []

  for x in reversed(range(2 ** len(boolvars))):
    for i,digit in enumerate('{0:b}'.format(x).zfill(len(boolvars))):
      exec(f'{boolvars[i]} = OverloadedBool({digit == "1"})', globals())
    
    def calculateGridLine():
      line = []
      for exp in exps:
        exec(f'line.append(({exp}))')
      return line
    
    truthtable.append(calculateGridLine())
  
  return truthtable
    
  
def latexify(exps, truthtable):

  latex_expressions = \
    [ (
        'E' in exp,
        exp
          .replace('not', '\\neg')
          .replace('and', '\\wedge')
          .replace('or', '\\vee')
          .replace('implies', '\\Rightarrow')
          .replace('iff', '\\Leftrightarrow')
          .replace('E', ''
          .strip())
      )
        for exp in exps]

  return \
  """
  \\begin{{truthtable}}
    {{{}}}
    {{{}}}
{}
  \\end{{truthtable}}
  """.format(
    '|'.join('e' if e else 'c' for e,_ in latex_expressions),
    ' & '.join(f'${exp}$' for _,exp in latex_expressions),
    '\n'.join('    ' + ' & '.join('\\T' if b else '\\F' for b in line) + ' \\\\' for line in truthtable)
  )

def printTruthtable(exps, truthtable):
  stringTable = [ ['\033[32mT\033[0m ' if b else '\033[31mF\033[0m' for b in line] for line in truthtable]
  try:
    print(tabulate(stringTable, headers=exps))
  except:
    pass

def processFileContent(raw):
  exps = parseExpressionList(raw)
  truthtable = generateTruthTable(exps)

  printTruthtable(exps, generateTruthTable(exps))

  content = latexify(exps, truthtable)
  return replaceContent(content, 'Truthtable')

if __name__ == '__main__':
  pass