pyrbd 0.2.5__py3-none-any.whl → 0.3.0__py3-none-any.whl

pyrbd/block.py

@@ -40,7 +40,7 @@ class Block:
             "anchor=west",
             "align=center",
             "fill={fill_color}",
-            "draw=black",
+            "draw=black!70!gray",
             "minimum height=1cm",
             "rounded corners=0.3mm",
             "inner sep=4pt",
@@ -48,6 +48,8 @@ class Block:
         ]
     )
 
+    arrow_options: str = "arrowcolor, thick"
+
     def __init__(
         self,
         text: str,
@@ -90,7 +92,7 @@ class Block:
 
         return "".join(
             [
-                f"\\draw[thick, rectangle connector={connector_position}cm]",
+                f"\\draw[{self.arrow_options}, rectangle connector={connector_position}cm]",
                 f"({self.parent.id}.east) to ({self.id}.west);\n\n",
             ]
         )
@@ -402,7 +404,7 @@ class Group(Block):
         if self.parent is None:
             return ""
 
-        return f"\\draw[thick] ({self.parent.id}.east) to ({self.id}.west);\n"
+        return f"\\draw[{self.arrow_options}] ({self.parent.id}.east) to ({self.id}.west);\n"
 
     @property
     def arrows(self) -> str:
@@ -412,7 +414,7 @@ class Group(Block):
             [
                 " ".join(
                     [
-                        "\\draw[thick, rectangle line]",
+                        f"\\draw[{self.arrow_options}, rectangle line]",
                         f"({block1.id}.east) to ({block2.id}.east);\n",
                     ]
                 )

pyrbd/diagram.py

@@ -1,5 +1,6 @@
 """Module containing Diagram class definition."""
 
+from typing import Optional
 import subprocess
 
 import pymupdf
@@ -18,12 +19,24 @@ class Diagram:
         list of `Block` instances
     hazard : str, optional
         string defining the `hazard` block text
+    colors : dict[str, str], optional
+        dictionary with custom color definitions in HEX format:
+        `{'color name': '6 digit hex code'}`
     """
 
-    def __init__(self, name: str, blocks: list[Block], hazard: str = "") -> None:
+    # Default color definitions
+    colors: dict[str, str] = {"arrowcolor": "4c4d4c", "hazardcolor": "ff6666"}
+
+    def __init__(
+        self,
+        name: str,
+        blocks: list[Block],
+        hazard: str = "",
+        colors: Optional[dict[str, str]] = None,
+    ) -> None:
         self.filename = name
         if hazard:
-            self.head = Block(hazard, "red!60")
+            self.head = Block(hazard, "hazardcolor")
         else:
             self.head = blocks.pop(0)
 
@@ -31,11 +44,14 @@ class Diagram:
         self.blocks = blocks
         self.blocks[0].parent = self.head
 
+        if colors is not None:
+            self.colors = self.colors | colors
+
     def write(self) -> None:
         """Write diagram to .tex file."""
 
         with open(f"{self.filename}.tex", mode="w", encoding="utf-8") as file:
-            file.write(TEX_PREAMBLE)
+            file.write(tex_preamble(self.colors))
             for block in [self.head, *self.blocks]:
                 file.write(block.get_node())
             file.write(TEX_END)
@@ -53,7 +69,19 @@ class Diagram:
         page = pdf_document[0]
 
         # Get and convert page to svg image
-        svg_content = page.get_svg_image()
+        svg_content = page.get_svg_image().splitlines()
+        svg_content.insert(
+            1,
+            "\n".join(
+                [
+                    "<style>",
+                    "   @media (prefers-color-scheme: light) { :root { --color: #000000; } }",
+                    "   @media (prefers-color-scheme: dark) { :root { --color: #DDDDDD; } }",
+                    "</style>",
+                ]
+            ),
+        )
+        svg_content = "\n".join(svg_content).replace(r"#4c4d4c", "var(--color)")
 
         # Save to file
         with open(output_file := f"{self.filename}.svg", "w", encoding="utf-8") as file:
@@ -145,38 +173,54 @@ class Diagram:
         return output_files
 
 
-TEX_PREAMBLE = "\n".join(
-    [
-        r"\documentclass{standalone}",
-        r"\usepackage{tikz}",
-        r"\usetikzlibrary{shapes,arrows,positioning,calc}",
-        r"\pgfdeclarelayer{background}",
-        r"\pgfsetlayers{background, main}",
-        r"\tikzset{",
-        r"connector/.style={",
-        r"-latex,",
-        r"font=\scriptsize},",
-        r"line/.style={",
-        r"font=\scriptsize},",
-        r"rectangle connector/.style={",
-        r"connector,"
-        r"to path={(\tikztostart) -- ++(#1,0pt) \tikztonodes |- (\tikztotarget) },",
-        r"pos=0.5},"
-        r"rectangle connector/.default=0.5cm,",
-        r"rectangle line/.style={",
-        r"line,"
-        r"to path={(\tikztostart) -- ++(#1,0pt) \tikztonodes |- (\tikztotarget) },",
-        r"pos=0.5},"
-        r"rectangle line/.default=0.5cm,",
-        r"straight connector/.style={",
-        r"connector,",
-        r"to path=--(\tikztotarget) \tikztonodes}",
-        r"}",
-        r"\begin{document}",
-        r"\begin{tikzpicture}",
-        "",
-    ]
-)
+def tex_preamble(custom_colors: dict[str, str] | None = None) -> str:
+    """LaTeX file preamble file with definition of custom colors given in dictionary."""
+
+    color_defs = []
+    if custom_colors is not None:
+        color_defs = [
+            f"\\definecolor{{{color_name}}}{{HTML}}{{{hex_code}}}"
+            for (color_name, hex_code) in custom_colors.items()
+        ]
+
+    return "\n".join(
+        [
+            r"\documentclass{standalone}",
+            r"\usepackage[T1]{fontenc}"
+            r"\usepackage{helvet}",
+            r"\renewcommand{\familydefault}{\sfdefault}",
+            r"\usepackage[dvipsnames,svgnames,x11names]{xcolor}",
+            r"\usepackage{tikz}",
+            r"\usetikzlibrary{shapes,arrows,positioning,calc}",
+            r"\pgfdeclarelayer{background}",
+            r"\pgfsetlayers{background, main}",
+            r"\tikzset{",
+            r"connector/.style={",
+            r"-latex,",
+            r"font=\scriptsize},",
+            r"line/.style={",
+            r"font=\scriptsize},",
+            r"rectangle connector/.style={",
+            r"connector,"
+            r"to path={(\tikztostart) -- ++(#1,0pt) \tikztonodes |- (\tikztotarget) },",
+            r"pos=0.5},"
+            r"rectangle connector/.default=0.5cm,",
+            r"rectangle line/.style={",
+            r"line,"
+            r"to path={(\tikztostart) -- ++(#1,0pt) \tikztonodes |- (\tikztotarget) },",
+            r"pos=0.5},"
+            r"rectangle line/.default=0.5cm,",
+            r"straight connector/.style={",
+            r"connector,",
+            r"to path=--(\tikztotarget) \tikztonodes}",
+            r"}",
+            *color_defs,
+            r"\begin{document}",
+            r"\begin{tikzpicture}",
+            "",
+        ]
+    )
+
 
 TEX_END = "\n".join(
     [

{pyrbd-0.2.5.dist-info → pyrbd-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyrbd
-Version: 0.2.5
+Version: 0.3.0
 Summary: Package for creating simple reliability block diagrams (RBDs) using LaTeX and TikZ.
 Project-URL: Repository, https://github.com/hghugdal/pyrbd
 Project-URL: Issues, https://github.com/hghugdal/pyrbd/issues
@@ -14,9 +14,13 @@ Requires-Python: >=3.10
 Requires-Dist: pymupdf>=1.26.3
 Description-Content-Type: text/markdown
 
-# <img alt="pyRBDlogo" src=docs/images/logo.svg width=40 align=top> pyRBD
+# <img alt="pyRBDlogo" src="https://raw.githubusercontent.com/hghugdal/pyrbd/23d7f00ca74e8465df3760821488b4bb78df803c/docs/images/logo.svg" width=40 align=top> pyRBD
 
-<img alt="Python" src="https://img.shields.io/badge/Python->= 3.10-blue?logo=python&link=None"> ![PyPI - Version](https://img.shields.io/pypi/v/pyrbd?link=https%3A%2F%2Fpypi.org%2Fproject%2Fpyrbd) <img alt="Tests" src="https://img.shields.io/badge/Tests-Passing-darkgreen?logo=pytest&link=None"> <img alt="Coverage" src="https://img.shields.io/badge/Coverage-100%25-darkgreen?link=None"> <img alt="Pylint" src="https://img.shields.io/badge/Pylint-10%2F10-darkgreen?link=None">
+[![PyPI - Version](https://img.shields.io/pypi/v/pyrbd)](https://pypi.org/project/pyrbd/)
+<img alt="Python" src="https://img.shields.io/badge/Python->= 3.10-blue?logo=python&link=None">
+<img alt="Tests" src="https://img.shields.io/badge/Tests-Passing-darkgreen?logo=pytest&link=None">
+<img alt="Coverage" src="https://img.shields.io/badge/Coverage-100%25-darkgreen?link=None">
+<img alt="Pylint" src="https://img.shields.io/badge/Pylint-10%2F10-darkgreen?link=None">
 
 A Python package for creating simple reliability block diagrams (RBDs) using `LaTeX` and [`TikZ`](https://en.wikipedia.org/wiki/PGF/TikZ).
 
@@ -25,6 +29,7 @@ A Python package for creating simple reliability block diagrams (RBDs) using `La
 
 ## Simple example diagram
 The blocks of the RBD are defined using `Block`, `Series` and `Group`, and the diagram itself is handled by the `Diagram` class. A simple example is given by the code
+
 ```python linenums="1"
 from pyrbd import Block, Diagram
 
@@ -39,7 +44,8 @@ diagram = Diagram(
 diagram.write()
 diagram.compile()
 ```
+
 producing the following diagram
-<div><img src="docs/examples/simple_RBD.svg" width=500/></div>
+<div><img src="https://raw.githubusercontent.com/hghugdal/pyrbd/23d7f00ca74e8465df3760821488b4bb78df803c/docs/examples/simple_RBD.svg" width=500/></div>
 
 For more examples, visit the [documentation](https://hghugdal.github.io/pyrbd/).

pyrbd-0.3.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+pyrbd/__init__.py,sha256=mpsb022BX9eDGSBhuYIr2gOr4sg_M9sYbPGHLPIdvl0,219
+pyrbd/block.py,sha256=qQJx2Sn_SvAnm3Chz1f078nTzbS61LSFdxaEveu2RcY,12319
+pyrbd/diagram.py,sha256=llZ0ncDmyJRg7zyQfTXAzbEM1W66wyHs9gfiovy-ujI,6762
+pyrbd-0.3.0.dist-info/METADATA,sha256=KEMFXHPv7ykDCYbh2fOOjcKPyqhOchXl0Ck2pc3ODwQ,2369
+pyrbd-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pyrbd-0.3.0.dist-info/licenses/LICENSE,sha256=mxXMgWpFgk71JpEEElFrb9FJxeoaDgvrU8gjUUU9LzA,1069
+pyrbd-0.3.0.dist-info/RECORD,,

pyrbd-0.2.5.dist-info/RECORD

@@ -1,7 +0,0 @@
-pyrbd/__init__.py,sha256=mpsb022BX9eDGSBhuYIr2gOr4sg_M9sYbPGHLPIdvl0,219
-pyrbd/block.py,sha256=ETRZJeEUU4OPk0qGo6zZIsaR1oLkyDPf1kelfuKWNsc,12219
-pyrbd/diagram.py,sha256=NVAsdfmqAu6-6J2mT5Gp2W_8a9WsI2iMbKjcEI6QFlE,5092
-pyrbd-0.2.5.dist-info/METADATA,sha256=WN1raYOu5KJtH5wj-7b1P8wH8-1GGHmnnqvwrt5hmZs,2196
-pyrbd-0.2.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pyrbd-0.2.5.dist-info/licenses/LICENSE,sha256=mxXMgWpFgk71JpEEElFrb9FJxeoaDgvrU8gjUUU9LzA,1069
-pyrbd-0.2.5.dist-info/RECORD,,