99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
"""NassiShneidermann.py: Classes for the Java Instruction found by the Interpreter"""
|
|
|
|
__author__ = "Weckyy702"
|
|
|
|
|
|
|
|
from typing import Dict, List, Optional
|
|
import logging
|
|
from enum import IntEnum
|
|
import os.path
|
|
import secrets
|
|
|
|
from interpreter.Tokenizer import Tokenizer
|
|
from interpreter.Lexer import Lexer
|
|
from interpreter.function_scope import Function_scope
|
|
from draw.code_to_image_wrapper import NSD_writer
|
|
import draw.code_to_image as cti
|
|
|
|
class Overwrite_behaviour(IntEnum):
|
|
SKIP = 0
|
|
OVERWRITE = 1
|
|
RANDOM_NAME = 2
|
|
|
|
OB = Overwrite_behaviour
|
|
|
|
class NassiShneidermanDiagram:
|
|
|
|
def __init__(self, do_debug: bool):
|
|
self.function_scopes: List[Function_scope] = []
|
|
logLevel = logging.INFO
|
|
if do_debug:
|
|
logLevel = logging.DEBUG
|
|
|
|
logging.basicConfig(force=True, level=logLevel)
|
|
|
|
@staticmethod
|
|
def set_font(font_filepath: str):
|
|
cti.set_font(font_filepath)
|
|
|
|
@staticmethod
|
|
def _save_scope(scope: Function_scope, output_path: str):
|
|
width = scope.get_width()
|
|
height = scope.get_height()
|
|
with NSD_writer(output_path, width, height):
|
|
y = 0
|
|
for instruction in scope:
|
|
print(instruction.instruction_text)
|
|
y = instruction.convert_to_image(0, y, width)
|
|
|
|
@staticmethod
|
|
def check_conflicts(filepath:str, behavoiur: Overwrite_behaviour):
|
|
if os.path.exists(filepath + ".png"):
|
|
if behavoiur == OB.SKIP:
|
|
return None
|
|
elif behavoiur == OB.OVERWRITE:
|
|
return filepath
|
|
else:
|
|
while os.path.exists(filepath+".png"):
|
|
filepath += str(secrets.token_hex(1))
|
|
return filepath
|
|
return filepath
|
|
|
|
def convert_to_image(self, output_path: str, on_conflict: Overwrite_behaviour=OB.SKIP):
|
|
i = 0
|
|
for scope in self.function_scopes:
|
|
|
|
filepath = f"{output_path}/{scope.name}"
|
|
filepath = self.check_conflicts(filepath, on_conflict)
|
|
|
|
if filepath is not None:
|
|
logging.info(f"Saving NSD to {filepath}.png...")
|
|
|
|
try:
|
|
self._save_scope(scope, filepath)
|
|
except Exception as e:
|
|
logging.error(f"Failed to save image {filepath} with error '{e}'")
|
|
raise e
|
|
except:
|
|
logging.error(f"Failed to save image {filepath}. Unknown error")
|
|
raise
|
|
|
|
|
|
yield i+1
|
|
i+=1
|
|
|
|
def load_from_file(self, filepath:str, itp_custom_tags: Optional[Dict[str, List[str]]]):
|
|
|
|
tokenizer = Tokenizer(filepath)
|
|
|
|
tokens = tokenizer.get_tokens()
|
|
|
|
lexer = Lexer(tokens)
|
|
|
|
self.function_scopes = lexer.get_instructions()[:-1]
|
|
|
|
if not self.function_scopes:
|
|
return True
|
|
else:
|
|
return False |