111 lines
2.6 KiB
Python
111 lines
2.6 KiB
Python
|
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
|