SNOPO/SNOPO

258 lines
12 KiB
Python
Executable File

#!/usr/bin/env python3
"""
SNOPO
Create diagram of SatNOGS Optical Process Overview.
Copyright (C) 2022, Jeff Moe
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
import graphviz
good_color = "green"
ok_color = "yellow"
bad_color = "red"
vid_shape = "rect"
cam_shape = "egg"
py_shape = "circle"
header_font = "helvetica bold"
header_font_size = "20"
header_font = "helvetica bold"
header_font_size = "20"
sub_font = "helvetica italic"
sub_font_size = "14"
start_font = "helvetica"
start_font_size = "18"
node_font = "ariel"
node_font_size = "14"
code_font = "courier"
code_font_size = "14"
g = graphviz.Digraph(
name="SatNOGS Optical",
comment="Process Graph",
filename="SNOPO",
directory="tmp",
engine="dot",
renderer="cairo",
formatter="cairo",
graph_attr="",
node_attr="",
edge_attr="",
)
g.attr(rankdir="LR")
g.attr(
bgcolor="seashell2:seashell",
label="SatNOGS Optical Process Overview",
fontcolor="black",
fontname=header_font,
fontsize=header_font_size
)
with g.subgraph(name="cluster_hardware") as a:
a.attr(style="filled", color="lightgrey")
a.attr(label="Hardware")
with a.subgraph(name="cluster_video") as b:
b.attr(style="filled", color="cornsilk", label="Motion Video Cameras", fontname=sub_font, fontsize=sub_font_size)
b.node_attr.update(style="filled", color="darkkhaki", shape="rect", fontname=node_font, fontsize=node_font_size)
b.node("n_uvc", label="UVC/V4L2")
b.node("n_cv", label="OpenCV")
b.node("n_pi", label="Raspberry Pi")
b.node("n_tis", label="The Imaging Source")
b.node("n_asi", label="ZWO ASI")
b.edge("n_uvc", "n_video", color=good_color)
b.edge("n_cv", "n_video", color=good_color)
b.edge("n_pi", "n_video", color=good_color)
b.edge("n_tis", "n_video", color=good_color)
b.edge("n_asi", "n_video", color=good_color)
with a.subgraph(name="cluster_camera") as c:
c.attr(
style="filled",
color="lightskyblue",
label="Still Cameras",
fontcolor="black",
fontname=sub_font,
fontsize=sub_font_size,
)
c.node_attr.update(
style="filled",
color="midnightblue",
shape="rect",
fontcolor="white",
fontname=node_font, fontsize=node_font_size
)
c.node("n_gphoto", label="gphoto2")
c.node("n_canon", label="Canon DSLR")
c.node("n_nikon", label="Nikon DSLR")
c.node("n_telescope", label="Telescope")
c.node("n_indi", label="INDI")
c.edge("n_gphoto", "n_camera", color=good_color)
c.edge("n_canon", "n_camera", color=good_color)
c.edge("n_nikon", "n_camera", color=good_color)
c.edge("n_telescope", "n_camera", color=bad_color)
c.edge("n_indi", "n_camera", color=bad_color)
with a.subgraph(name="cluster_allsky") as d:
d.attr(
style="filled", color="plum2", label="Allsky Cameras", fontcolor="black", fontname=sub_font, fontsize=sub_font_size
)
d.node_attr.update(
style="filled", color="seashell2", shape="rect", fontcolor="black", fontname=node_font, fontsize=node_font_size
)
d.node("n_asi_allsky", label="ZWO ASI Allsky")
d.node("n_oculus", label="Starlight Xpress Oculus")
d.node("n_indi_allsky", label="INDI Allsky")
d.edge("n_asi_allsky", "n_allsky", color=good_color)
d.edge("n_oculus", "n_allsky", color=bad_color)
d.edge("n_indi_allsky", "n_allsky", color=bad_color)
with g.subgraph(name="cluster_repository") as a:
a.attr(style="filled", color="palegreen4", label="Repository")
a.attr(label="Software\nRepository")
with a.subgraph(name="cluster_camera_software") as b:
b.attr(color="lightblue")
b.node_attr.update(style="filled", color="lightgoldenrodyellow", shape="rect", fontname=node_font, fontsize=node_font_size)
b.node("n_video", label="Video")
b.node("n_camera", label="Camera")
b.node("n_allsky", label="Allsky")
b.node("n_stvid", label="stvid")
b.node("n_stphot", label="stphot")
b.node("n_asm", label="asm")
b.edge("n_video", "n_stvid", color=good_color)
b.edge("n_camera", "n_stphot", color=good_color)
b.edge("n_allsky", "n_asm", color=good_color)
b.attr(label="Development\nPython", fontname=sub_font, fontsize=sub_font_size)
with a.subgraph(name="cluster_c_software") as c:
c.attr(color="palevioletred4")
c.node_attr.update(style="filled", color="lightsteelblue1", shape="rect", fontname=node_font, fontsize=node_font_size)
c.node_attr["style"] = "filled"
c.node("n_sattools", label="sattools")
c.edge("n_video", "n_sattools", color=good_color)
c.edge("n_camera", "n_sattools", color=good_color)
c.attr(label="Legacy\nC", fontname=sub_font, fontsize=sub_font_size)
with g.subgraph(name="cluster_acquire") as a:
a.attr(style="filled", color="pink", label="Acquire")
a.attr(label="Acquire Sample")
with a.subgraph(name="cluster_camera_acquire") as b:
b.attr(color="lightblue")
b.node_attr.update(style="filled", color="wheat3", shape="rect", fontname=node_font, fontsize=node_font_size)
b.node(
"n_sattools_cacquire", label="sattools\nget.sh\ngphoto2\njpg2fits"
)
b.node("n_stphot_acquire", label="stphot\nacquire.py")
b.node("n_asm_acquire", label="asm\ncapture")
b.edge("n_sattools", "n_sattools_cacquire", color=good_color)
b.edge("n_stphot", "n_stphot_acquire", color=good_color)
b.edge("n_asm", "n_asm_acquire", color=good_color)
b.attr(label="Camera", fontname=sub_font, fontsize=sub_font_size)
with a.subgraph(name="cluster_video_acquire") as c:
c.attr(color="olivedrab3")
c.node_attr.update(style="filled", color="plum", shape="rect", fontname=node_font, fontsize=node_font_size)
c.node("n_sattools_vacquire", label="sattools\nUNKNOWN")
c.node("n_stvid_acquire", label="stvid\nacquire.py")
c.edge("n_sattools", "n_sattools_vacquire", color=ok_color)
c.edge("n_stvid", "n_stvid_acquire", color=good_color)
c.attr(label="Video", fontname=sub_font, fontsize=sub_font_size)
with g.subgraph(name="cluster_platesolver") as a:
a.attr(style="filled", color="tan", label="Plate Solver")
a.attr(label="Plate Solver")
with a.subgraph(name="cluster_camera_plate_solver") as b:
b.attr(color="lightblue")
b.node_attr.update(style="filled", color="silver", shape="rect", fontname=node_font, fontsize=node_font_size)
b.node("n_sattools_cplate", label="sattools\nfind.sh\nsource-extractor")
b.node("n_stphot_plate", label="stphot\nUse sattools")
b.node("n_asm_plate", label="asm\nNONE")
b.edge("n_sattools_cacquire", "n_sattools_cplate", color=good_color)
b.edge("n_stphot_acquire", "n_sattools_cplate", color=good_color)
b.edge("n_stphot_acquire", "n_stphot_plate", color=bad_color)
b.edge("n_asm_acquire", "n_asm_plate", color=bad_color)
b.attr(label="Camera", fontname=sub_font, fontsize=sub_font_size)
with a.subgraph(name="cluster_video_platesolver") as c:
c.attr(color="goldenrod3")
c.node_attr.update(style="filled", color="paleturquoise3", shape="rect", fontname=node_font, fontsize=node_font_size)
c.node("n_sattools_vplate", label="sattools\nUNKNOWN")
c.node("n_stvid_plate", label="stvid\nprocess.py")
c.edge("n_sattools_vacquire", "n_sattools_vplate", color=ok_color)
c.edge("n_stvid_acquire", "n_stvid_plate", color=good_color)
c.attr(label="Video", fontname=sub_font, fontsize=sub_font_size)
with g.subgraph(name="cluster_satdetect") as a:
a.attr(style="filled", color="lightblue", label="satdetect")
a.attr(label="Detect Satellites")
with a.subgraph(name="cluster_camera_satdetect") as b:
b.attr(color="khaki2")
b.node_attr.update(style="filled", color="olivedrab3", shape="rect", fontname=node_font, fontsize=node_font_size)
b.node("n_sattools_cdetect", label="sattools\nid.sh\naddwcs\nsatid")
b.node("n_stphot_detect", label="stphot\nNONE")
b.node("n_asm_detect", label="asm\nNONE")
b.edge("n_sattools_cplate", "n_sattools_cdetect", color=good_color)
b.edge("n_stphot_plate", "n_stphot_detect", color=bad_color)
b.edge("n_asm_plate", "n_asm_detect", color=bad_color)
b.attr(label="Camera", fontname=sub_font, fontsize=sub_font_size)
with a.subgraph(name="cluster_video_satdetect") as c:
c.attr(color="aquamarine4")
c.node_attr.update(style="filled", color="ivory2", shape="rect", fontname=node_font, fontsize=node_font_size)
c.node("n_sattools_vdetect", label="sattools\nUNKNOWN")
c.node("n_stvid_detect", label="stvid\nprocess.py")
c.edge("n_sattools_vplate", "n_sattools_vdetect", color=ok_color)
c.edge("n_stvid_plate", "n_stvid_detect", color=good_color)
c.attr(label="Video", fontname=sub_font, fontsize=sub_font_size)
with g.subgraph(name="cluster_satid") as a:
a.attr(style="filled", color="darkseagreen4", label="satid")
a.attr(label="Identify Satellites")
with a.subgraph(name="cluster_camera_satid") as b:
b.attr(color="yellow3")
b.node_attr.update(style="filled", color="darkseagreen3", shape="rect", fontname=node_font, fontsize=node_font_size)
b.node("n_sattools_cid", label="sattools\nid.sh\naddwcs\nsatid")
b.node("n_stphot_id", label="stphot\nNONE")
b.node("n_asm_id", label="asm\nNONE")
b.edge("n_sattools_cdetect", "n_sattools_cid", color=good_color)
b.edge("n_stphot_detect", "n_stphot_id", color=bad_color)
b.edge("n_asm_detect", "n_asm_id", color=bad_color)
b.attr(label="Camera", fontname=sub_font, fontsize=sub_font_size)
with a.subgraph(name="cluster_video_satid") as c:
c.attr(color="thistle")
c.node_attr.update(style="filled", color="steelblue", shape="rect", fontname=node_font, fontsize=node_font_size)
c.node("n_sattools_vid", label="sattools\nUNKNOWN")
c.node("n_stvid_id", label="stvid\nprocess.py")
c.edge("n_sattools_vdetect", "n_sattools_vid", color=ok_color)
c.edge("n_stvid_detect", "n_stvid_id", color=good_color)
c.attr(label="Video", fontname=sub_font, fontsize=sub_font_size)
g.attr("node", shape="doublecircle", style="filled", color="orange")
g.node("n_start", label="Start\nHere", fontname=start_font, fontsize=start_font_size)
g.edge("n_start", "n_indi", color=good_color)
g.edge("n_start", "n_gphoto", color=good_color)
g.edge("n_start", "n_canon", color=good_color)
g.edge("n_start", "n_nikon", color=good_color)
g.edge("n_start", "n_telescope", color=good_color)
g.edge("n_start", "n_asi", color=good_color)
g.edge("n_start", "n_pi", color=good_color)
g.edge("n_start", "n_tis", color=good_color)
g.edge("n_start", "n_cv", color=good_color)
g.edge("n_start", "n_uvc", color=good_color)
g.edge("n_start", "n_indi_allsky", color=good_color)
g.edge("n_start", "n_asi_allsky", color=good_color)
g.edge("n_start", "n_oculus", color=good_color)
g.attr("node", shape="rectangle", style="filled", color="grey")
g.render(view=False, format="png")