Skip to content
Snippets Groups Projects
Select Git revision
  • main default protected
1 result
Blame History Permalink
main.py 1.22 KiB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57 






from manim import *

from pyPlantUML import *

import argparse

from pyPlantUML.layout.SpringLayout import SpringLayout


class MainScene(MovingCameraScene):
    file: str
    animate: bool

    def construct(self):
        parser = PUMLParser()
        diagram: Diagram = parser.parse_file(self.file)

        layout = SpringLayout(diagram)
        layout.apply()
        layout.scale(7, 5)

        self.camera.background_color = WHITE
        Text.set_default(font_size=16)

        diagram.set_scene(self)

        diagram.animate = self.animate

        diagram.draw()

    def set_file(self, file: str):
        self.file = file

    def set_animate(self, animate: bool):
        self.animate = animate


if __name__ == "__main__":
    config.output_file = "PlantUML"
    config.cairo_path = "media/images/"
    config.ffmpeg_path = "media/videos/"

    argparser = argparse.ArgumentParser(prog="pyPlantUML")

    argparser.add_argument(
        "-f", "--file", required=True, type=str, help="Path to source plantUml file."
    )
    argparser.add_argument("-a", "--animate", action="store_true")

    args = argparser.parse_args()

    scene = MainScene()

    scene.set_file(args.file)
    scene.set_animate(args.animate)

    scene.render()