Mustergenerator/mustergenerator.py

import drawsvg as draw
import os
import math


class Circle:
    def __init__(self, name: str, filepath: str, length=472, height=472, radius=10, distance=60, offset=50):
        assert length >= 0, "length can not be negative"
        assert height >= 0, "height can not be negative"
        assert radius >= 0, "radius can not be negative"
        assert distance >= 2*radius, "distance must be grater then two times the radius"
        assert offset >= 0, "offset can only be 0 - 100 %"
        assert offset <= 100, "offset can only be 0 - 100 %"

        self.__name = name
        self.__filepath = filepath
        self.__length = length
        self.__height = height
        self.__radius = radius
        self.__distance = distance
        self.__offset_percentage = offset
        self.__offset = (distance / 100) * offset

    @property
    def length(self):
        return self.__length

    @length.setter
    def length(self, val):
        assert val >= 0, "length can not be negative"
        self.__length = val

    @property
    def height(self):
        return self.__height

    @height.setter
    def height(self, val):
        assert val >= 0, "height can not be negative"
        self.__height = val

    @property
    def radius(self):
        return self.__radius

    @radius.setter
    def radius(self, val):
        assert val >= 0, "radius can not be negative"
        self.__radius = val

    @property
    def distance(self):
        return self.__distance

    @distance.setter
    def distance(self, val):
        assert val >= 2 * radius, "distance must be grater then two times the radius"
        self.__distance = val

    @property
    def offset_percentage(self):
        return self.__offset_percentage

    @property
    def offset(self):
        return self.__offset

    @offset.setter
    def offset(self, val):
        assert val >= 0, "offset can only be 0 - 100 %"
        assert val <= 100, "offset can only be 0 - 100 %"
        self.__offset = (self.__distance / 100) * val

    def circle_generation(self, filetype='svg'):
        assert filetype == 'svg' or 'png', "Dateityp mus svg oder png sein"
        x = self.__radius
        y = self.__radius
        i = 1

        canvas = draw.Drawing(self.__length, self.__height, origin=(0, 0))

        while y + self.__radius <= self.__height:
            while x + self.__radius <= self.__length:
                canvas.append(draw.Circle(x, y, self.__radius, fill='none', stroke_width=1, stroke='black'))
                x = x + self.__distance

            if i % 2 == 0:
                x = self.__radius
            else:
                x = self.__radius + self.__offset

            y = y + self.__distance
            i = i + 1

        if filetype == 'svg':
            canvas.save_svg(f'{self.__filepath}/{self.__name}.svg')
        if filetype == 'png':
            canvas.save_png(f'{self.__filepath}/{self.__name}.png')


class Hexagon:
    def __init__(self, name: str, filepath: str, length=472, height=472, key_width=50, bridge_width=10):
        assert length >= 0, "length can not be negative"
        assert height >= 0, "height can not be negative"
        assert key_width >= 0, "key width can not be negative"
        assert bridge_width >= 0, "bridge width can not be negative"

        self.__name = name
        self.__filepath = filepath
        self.__length = length
        self.__height = height
        self.__key_width = key_width
        self.__bridge_width = bridge_width

    @property
    def length(self):
        return self.__length

    @length.setter
    def length(self, val):
        assert val >= 0, "length can not be negative"
        self.__length = val

    @property
    def height(self):
        return self.__height

    @height.setter
    def height(self, val):
        assert val >= 0, "height can not be negative"
        self.__height = val

    @property
    def key_width(self):
        return self.__key_width

    @key_width.setter
    def key_width(self, val):
        assert val >= 0, "length can not be negative"
        self.__key_width = val

    @property
    def bridge_width(self):
        return self.__bridge_width

    @bridge_width.setter
    def bridge_width(self, val):
        assert val >= 0, "bridge width can not be negative"
        self.__bridge_width = val

    def hexagon_generation(self, filetype='svg'):
        assert filetype == 'svg' or 'png', "Dateityp mus svg oder png sein"

        i = 1
        x = 0
        y = 0
        side_length = self.__key_width / math.sqrt(3)
        sin_expression = math.sin(math.radians(30)) * side_length
        distance = self.__key_width + self.__bridge_width
        distance_y = (math.sqrt(3) / 2) * distance

        canvas = draw.Drawing(self.__length, self.__height, origin=(0, 0))

        while y + (2 * side_length) < self.__height:
            while x + self.__key_width < self.__length:
                p1_x = x + self.__key_width / 2
                p1_y = y
                p2_x = x
                p2_y = y + sin_expression
                p3_x = x
                p3_y = y + sin_expression + side_length
                p4_x = x + self.__key_width / 2
                p4_y = y + 2 * side_length
                p5_x = x + self.__key_width
                p5_y = y + sin_expression + side_length
                p6_x = x + self.__key_width
                p6_y = y + sin_expression

                hexagon = draw.Lines(p1_x, p1_y, p2_x, p2_y, p3_x, p3_y, p4_x, p4_y, p5_x, p5_y, p6_x, p6_y,
                                     fill='none', stroke='black', close='true')
                canvas.append(hexagon)

                x = x + distance

            if i % 2 == 0:
                x = 0
            else:
                x = distance / 2

            y = y + distance_y
            i = i + 1

        if filetype == 'svg':
            canvas.save_svg(f'{self.__filepath}/{self.__name}.svg')
        if filetype == 'png':
            canvas.save_png(f'{self.__filepath}/{self.__name}.png')


class Triangle:
    def __init__(self, name: str, filepath: str, length=472, height=472, side_length=50, bridge_width=10):
        assert length >= 0, "length can not be negative"
        assert height >= 0, "height can not be negative"
        assert side_length >= 0, "side length can not be negative"
        assert bridge_width >= 0, "bridge width can not be negative"

        self.__name = name
        self.__filepath = filepath
        self.__length = length
        self.__height = height
        self.__side_length = side_length
        self.__bridge_width = bridge_width

    @property
    def length(self):
        return self.__length

    @length.setter
    def length(self, val):
        assert val >= 0, "length can not be negative"
        self.__length = val

    @property
    def height(self):
        return self.__height

    @height.setter
    def height(self, val):
        assert val >= 0, "height can not be negative"
        self.__height = val

    @property
    def side_length(self):
        return self.__side_length

    @side_length.setter
    def side_length(self, val):
        assert val >= 0, "length can not be negative"
        self.__side_length = val

    @property
    def bridge_width(self):
        return self.__bridge_width

    @bridge_width.setter
    def bridge_width(self, val):
        assert val >= 0, "bridge width can not be negative"
        self.__bridge_width = val

    def triangle_generation(self, filetype='svg'):
        assert filetype == 'svg' or 'png', "Dateityp mus svg oder png sein"
        x = 0
        y = 0
        i = 1
        distance = self.__bridge_width / math.sin(math.radians(60))

        canvas = draw.Drawing(self.__length, self.__height, origin=(0, 0))

        while y + math.sqrt(3) / 2 * self.__side_length <= self.__height:
            while x + 1.5 * self.__side_length + distance <= self.__length:
                triangle1_p1_x = x
                triangle1_p1_y = y
                triangle1_p2_x = x + self.__side_length
                triangle1_p2_y = y
                triangle1_p3_x = x + math.sin(math.radians(30)) * self.__side_length
                triangle1_p3_y = y + math.cos(math.radians(30)) * self.__side_length

                triangle = draw.Lines(triangle1_p1_x, triangle1_p1_y, triangle1_p2_x, triangle1_p2_y, triangle1_p3_x,
                                      triangle1_p3_y, close='true', stroke='black', fill='none')
                canvas.append(triangle)

                x = x + self.__side_length + distance

                triangle2_p1_x = x
                triangle2_p1_y = y
                triangle2_p2_x = x - self.__side_length / 2
                triangle2_p2_y = y + math.sqrt(3) / 2 * self.__side_length
                triangle2_p3_x = x + self.__side_length / 2
                triangle2_p3_y = triangle2_p2_y

                triangle2 = draw.Lines(triangle2_p1_x, triangle2_p1_y, triangle2_p2_x, triangle2_p2_y, triangle2_p3_x,
                                       triangle2_p3_y, close='true', stroke='black', fill='none')
                canvas.append(triangle2)

                x = x + distance
            if i % 2 == 0:
                x = 0
            else:
                x = self.__side_length / 2 + distance
            y = y + self.__bridge_width + math.sqrt(3) / 2 * self.__side_length
            i = i + 1

        if filetype == 'svg':
            canvas.save_svg(f'{self.__filepath}/{self.__name}.svg')
        if filetype == 'png':
            canvas.save_png(f'{self.__filepath}/{self.__name}.png')


if __name__ == '__main__':
    name = input("Dateinamen:")

    path = input("Dateipfad (0 für aktuelles Verzeichnis):")
    if path == "0":
        path = os.getcwd()

    shape = int(input("Form (1: Kreis, 2: Hexagon, 3: Dreieck):"))

    filetype = 'svg'

    if shape == 1:
        circle = Circle(name, path)
        print(f"\nStandardwerte\n1. Höhe: {circle.height} px\n2. Länge: {circle.length} px\n3. Radius: {circle.radius} px\n4. Abstand: {circle.distance} px\n5. Offset: {circle.offset_percentage} %\n6. Dateityp: svg\n")
        selection_flag = True
        while selection_flag:
            try:
                parameter_selection = int(input("Parameter ändern (0: keinen Parameter ändern, 1: Höhe, 2: Länge, 3: Radius, 4: Abstand, 5: Offste, 6: Dateityp):"))
                if parameter_selection == 0:
                    selection_flag = False
                elif parameter_selection == 1:
                    height = int(input("Höhe in Pixeln eingeben:"))
                    circle.height = height
                elif parameter_selection == 2:
                    length = int(input("Länge in Pixeln:"))
                    circle.length = length
                elif parameter_selection == 3:
                    radius = int(input("Radius in Pixeln:"))
                    circle.radius = radius
                elif parameter_selection == 4:
                    distance = int(input("Abstand in Pixeln:"))
                    circle.distance = distance
                elif parameter_selection == 5:
                    offset = int(input("Offset in %:"))
                    circle.offset = offset
                elif parameter_selection == 6:
                    filetype_selection = int(input(f"1: {name}.svg, 2: {name}.png:"))
                    if filetype_selection == 1:
                        filetype = "svg"
                    elif filetype_selection == 2:
                        filetype = "png"
                    else:
                        raise ValueError
                else:
                    print("ungültiger Wert")
            except ValueError:
                print("ungültiger Wert")
            except AssertionError as error:
                print(error)

        circle.circle_generation(filetype)

    elif shape == 2:
        hexagon = Hexagon(name, path)
        print(
            f"\nStandardwerte\n1. Höhe: {hexagon.height} px\n2. Länge: {hexagon.length} px\n3. Schlüsselweite: {hexagon.key_width} px\n4. Stegbreite: {hexagon.bridge_width} px\n5. Dateityp: svg\n")
        selection_flag = True
        while selection_flag:
            try:
                parameter_selection = int(input("Parameter ändern (0: keinen Parameter ändern, 1: Höhe, 2: Länge, 3: Schlüsselweite, 4: Stegbreite, 5: Dateityp):"))
                if parameter_selection == 0:
                    selection_flag = False
                elif parameter_selection == 1:
                    height = int(input("Höhe in Pixeln eingeben:"))
                    hexagon.height = height
                elif parameter_selection == 2:
                    length = int(input("Länge in Pixeln:"))
                    hexagon.length = length
                elif parameter_selection == 3:
                    key_width = int(input("Schlüsselweite in Pixeln:"))
                    hexagon.key_width = key_width
                elif parameter_selection == 4:
                    bridge_width = int(input("Stegbreite in Pixeln:"))
                    hexagon.bridge_width = bridge_width
                elif parameter_selection == 5:
                    filetype_selection = int(input(f"1: {name}.svg, 2: {name}.png:"))
                    if filetype_selection == 1:
                        filetype = "svg"
                    elif filetype_selection == 2:
                        filetype = "png"
                    else:
                        raise ValueError
                else:
                    print("ungültiger Wert")
            except ValueError:
                print("ungültiger Wert")
            except AssertionError as error:
                print(error)

        hexagon.hexagon_generation(filetype)
    elif shape == 3:
        triangle = Triangle(name, path)
        print(
            f"\nStandardwerte\n1. Höhe: {triangle.height} px\n2. Länge: {triangle.length} px\n3. Seitenlänge: {triangle.side_length} px\n4. Stegbreite: {triangle.bridge_width} px\n5. Dateityp: svg\n")
        selection_flag = True
        while selection_flag:
            try:
                parameter_selection = int(input(
                    "Parameter ändern (0: keinen Parameter ändern, 1: Höhe, 2: Länge, 3: Seitenlänge, 4: Stegbreite, 5: Dateityp):"))
                if parameter_selection == 0:
                    selection_flag = False
                elif parameter_selection == 1:
                    height = int(input("Höhe in Pixeln eingeben:"))
                    triangle.height = height
                elif parameter_selection == 2:
                    length = int(input("Länge in Pixeln:"))
                    triangle.length = length
                elif parameter_selection == 3:
                    side_length = int(input("Seitenlänge in Pixeln:"))
                    triangle.side_length = side_length
                elif parameter_selection == 4:
                    bridge_width = int(input("Stegbreite in Pixeln:"))
                    triangle.bridge_width = bridge_width
                elif parameter_selection == 5:
                    filetype_selection = int(input(f"1: {name}.svg, 2: {name}.png:"))
                    if filetype_selection == 1:
                        filetype = "svg"
                    elif filetype_selection == 2:
                        filetype = "png"
                    else:
                        raise ValueError
                else:
                    print("ungültiger Wert")
            except ValueError:
                print("ungültiger Wert")
            except AssertionError as error:
                print(error)

        triangle.triangle_generation(filetype)
    else:
        print("ungültiger Wert")