This repository has been archived on 2024-05-19. You can view files and clone it, but cannot push or open issues or pull requests.
l-systems/converter.py

139 lines
4.2 KiB
Python

from argparse import ArgumentParser
from csv import reader as csv_reader
from pprint import pformat
from xml.dom.minidom import parseString as xml_parser
from xml.etree.ElementTree import Element as XMLElement
from xml.etree.ElementTree import SubElement as XMLSubElement
from xml.etree.ElementTree import tostring as element_stringifier
class System:
"""Represents an L-System"""
def __init__(
self,
name: str,
base: list[str],
axiom: str,
substitutions: list[tuple[str, str]],
interpretations: list[tuple[str, str]],
):
# Space trimming for class members
self.name = name.strip()
"""System name"""
self.base = map(str.strip, base)
"""Set used"""
self.axiom = axiom.strip()
"""Axiom used at the start of iterations"""
self.substitutions = [(m.strip(), s.strip()) for m, s in substitutions]
"""
Substitution for each member of the base,
represented as a couple (member, substitution)
"""
self.interpretations = [(m.strip(), i.strip()) for m, i in interpretations]
"""
Interpretation for each member of the base,
represented as a couple (member, interpretation)
"""
def __repr__(self):
return pformat(object=self.__dict__, compact=True, width=120, sort_dicts=False)
def to_xml(self) -> XMLElement:
"""Convert the current system into an XML element"""
system = XMLElement("lsystem")
base = XMLSubElement(system, "name")
base.text = self.name
base = XMLSubElement(system, "base")
base.text = "".join(self.base)
axiom = XMLSubElement(system, "axiom")
axiom.text = self.axiom
substitutions = XMLSubElement(system, "substitutions")
for member, substitution in self.substitutions:
sub_element = XMLSubElement(substitutions, "substitution")
sub_element.set("member", member)
sub_element.text = substitution
interpretations = XMLSubElement(system, "interpretations")
for member, interpretation in self.interpretations:
inter_element = XMLSubElement(interpretations, "interpretation")
inter_element.set("member", member)
# If the interpretation is "do nothing", then do nothing
if len(interpretation) == 0:
interpretation = "TURN 0"
inter_element.text = interpretation
return system
def data_reader(path: str, delimiter: str = ","):
"""Read a CSV file and returns a list of L-System"""
res: list[System] = []
with open(path) as csv_file:
data = csv_reader(csv_file, delimiter=delimiter)
for system in data:
if len(system) == 0:
continue
name = system[0]
base = list(system[1])
axiom = system[2]
substitutions = [(v, system[3 + i]) for i, v in enumerate(base)]
interpretations = [
(v, system[3 + i + len(substitutions)]) for i, v in enumerate(base)
]
res.append(System(name, base, axiom, substitutions, interpretations))
return res
def lsystems_xml(systems: list[System]) -> XMLElement:
"""Convert list of L-system structure into XML"""
root = XMLElement("lsystems")
for system in systems:
root.append(system.to_xml())
return root
if __name__ == "__main__":
parser = ArgumentParser(description="Generate XML representation of L-systems")
parser.add_argument(
"file", metavar="FILE", help="CSV file containing L-system data"
)
parser.add_argument(
"-o", "--output", metavar="OUTPUT_FILE", help="Specify output file for XML"
)
args = parser.parse_args()
# Read data
lsystems = data_reader(args.file)
# Generate XML
xml = lsystems_xml(lsystems)
# Add XML model declaration
xml_string = (
'<?xml-model href="l-systems.xsd"?>\n' + element_stringifier(xml).decode()
)
# Output XML
dom = xml_parser(xml_string)
pretty_xml = dom.toprettyxml(indent=" ", encoding="UTF-8").decode()
if args.output:
with open(args.output, "w") as output_file:
output_file.write(pretty_xml)
else:
print(pretty_xml)