#!/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"

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="DRAFT SatNOGS Optical Process Overview DRAFT",
    fontcolor="black",
)


with g.subgraph(name="cluster_hardware") as a:
    a.attr(style="filled", color="lightgrey")
    a.node_attr.update(style="filled", color="pink", shape="rectangle")
    a.attr(label="Hardware\nChoose One")
    with a.subgraph(name="cluster_video") as b:
        b.attr(style="filled", color="cornsilk", label="Motion Video Cameras")
        b.node_attr.update(style="filled", color="darkkhaki", shape="tripleoctagon")
        b.node("n_uvc", label="UVC")
        b.node("n_cv", label="OpenCV")
        b.node("n_pi", label="Raspberry Pi")
        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_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",
        )
        c.node_attr.update(
            style="filled",
            color="midnightblue",
            shape="doubleoctagon",
            fontcolor="white",
        )
        c.node("n_gphoto", label="gphoto2")
        c.node("n_canon", label="Canon DSLR")
        c.node("n_nikon", label="Nikon DSLR")
        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_indi", "n_camera", color=good_color)
    with a.subgraph(name="cluster_allsky") as d:
        d.attr(
            style="filled", color="slateblue", label="Allsky Cameras", fontcolor="white"
        )
        d.node_attr.update(
            style="filled", color="peru", shape="octagon", fontcolor="black"
        )
        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=good_color)
        d.edge("n_indi_allsky", "n_allsky", color=good_color)

with g.subgraph(name="cluster_repository") as a:
    a.attr(style="filled", color="palegreen4", label="Repository")
    a.attr(label="Software\nRepository\nChoose One")
    with a.subgraph(name="cluster_camera_software") as b:
        b.attr(color="lightblue")
        b.node_attr["style"] = "filled"
        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")
    with a.subgraph(name="cluster_c_software") as c:
        c.attr(color="tan")
        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")

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["style"] = "filled"
        b.node(
            "n_sattools_cacquire", label="sattools/scripts/get.sh\ngphoto2\njpg2fits"
        )
        b.node("n_stphot_acquire", label="stphot/acquire.py")
        b.node("n_asm_acquire", label="asm/capture")
        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\nSample")
    with a.subgraph(name="cluster_video_acquire") as c:
        c.attr(color="tan")
        c.node_attr["style"] = "filled"
        c.node("n_sattools_vacquire", label="sattools/UNKNOWN-vid-get.sh")
        c.node("n_stvid_acquire", label="stvid/acquire.py")
        c.edge("n_sattools", "n_sattools_vacquire", color=good_color)
        c.edge("n_stvid", "n_stvid_acquire", color=good_color)
        c.attr(label="Video\nSample")


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["style"] = "filled"
        b.node("n_sattools_cplate", label="sattools/scripts/find.sh\nsource-extractor")
        b.node("n_stphot_plate", label="stphot/UNKNOWN-find.py")
        b.node("n_asm_plate", label="asm/UNKNOWN-find")
        b.edge("n_sattools_cacquire", "n_sattools_cplate", color=good_color)
        b.edge("n_stphot_acquire", "n_stphot_plate", color=good_color)
        b.edge("n_asm_acquire", "n_asm_plate", color=good_color)
        b.attr(label="Camera\nPlate Solver")
    with a.subgraph(name="cluster_video_platesolver") as c:
        c.attr(color="pink")
        c.node_attr["style"] = "filled"
        c.node("n_sattools_vplate", label="sattools/UNKNOWN-vid-find.sh")
        c.node("n_stvid_plate", label="stvid/process.py")
        c.edge("n_sattools_vacquire", "n_sattools_vplate", color=good_color)
        c.edge("n_stvid_acquire", "n_stvid_plate", color=good_color)
        c.attr(label="Video\nPlate Solver")

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="tan")
        b.node_attr["style"] = "filled"
        b.node("n_sattools_cdetect", label="sattools/scripts/id.sh\naddwcs\nsatid")
        b.node("n_stphot_detect", label="stphot/UNKNOWN-detect.py")
        b.node("n_asm_detect", label="asm/UNKNOWN-detect")
        b.edge("n_sattools_cplate", "n_sattools_cdetect", color=good_color)
        b.edge("n_stphot_plate", "n_stphot_detect", color=good_color)
        b.edge("n_asm_plate", "n_asm_detect", color=good_color)
        b.attr(label="Camera\nDetect Satellites")
    with a.subgraph(name="cluster_video_satdetect") as c:
        c.attr(color="tan")
        c.node_attr["style"] = "filled"
        c.node("n_sattools_vdetect", label="sattools/UNKNOWN-vid-detect.sh")
        c.node("n_stvid_detect", label="stvid/process.py")
        c.edge("n_sattools_vplate", "n_sattools_vdetect", color=good_color)
        c.edge("n_stvid_plate", "n_stvid_detect", color=good_color)
        c.attr(label="Video\nDetect Satellites")

with g.subgraph(name="cluster_satid") as a:
    a.attr(style="filled", color="lightblue", label="satid")
    a.attr(label="Identify Satellites")
    with a.subgraph(name="cluster_camera_satid") as b:
        b.attr(color="tan")
        b.node_attr["style"] = "filled"
        b.node("n_sattools_cid", label="sattools/scripts/id.sh\naddwcs\nsatid")
        b.node("n_stphot_id", label="stphot/UNKNOWN-id.py")
        b.node("n_asm_id", label="asm/UNKNOWN-id")
        b.edge("n_sattools_cdetect", "n_sattools_cid", color=good_color)
        b.edge("n_stphot_detect", "n_stphot_id", color=good_color)
        b.edge("n_asm_detect", "n_asm_id", color=good_color)
        b.attr(label="Camera\nID Satellite")
    with a.subgraph(name="cluster_video_satid") as c:
        c.attr(color="tan")
        c.node_attr["style"] = "filled"
        c.node("n_sattools_vid", label="sattools/UNKNOWN-vid-id.sh")
        c.node("n_stvid_id", label="stvid/process.py")
        c.edge("n_sattools_vdetect", "n_sattools_vid", color=good_color)
        c.edge("n_stvid_detect", "n_stvid_id", color=good_color)
        c.attr(label="Video\nID Satellite")

g.attr("node", shape="doublecircle", style="filled", color="orange")
g.node("n_start", label="Start\nHere")
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_asi", color=good_color)
g.edge("n_start", "n_pi", 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")