icplot 0.0.1__tar.gz → 0.0.2__tar.gz

{icplot-0.0.1/src/icplot.egg-info → icplot-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: icplot
-Version: 0.0.1
+Version: 0.0.2
 Summary: Utilities for generating plots and graphics for technical reports.
 Author-email: "James Grogan, Irish Centre for High End Computing" <james.grogan@ichec.ie>
 Project-URL: Repository, https://git.ichec.ie/performance/toolshed/icplot
@@ -14,6 +14,7 @@ Classifier: Topic :: System :: Distributed Computing
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: wand
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-cov; extra == "test"

{icplot-0.0.1 → icplot-0.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "icplot"
-version = "0.0.1"
+version = "0.0.2"
 authors = [
   { name="James Grogan, Irish Centre for High End Computing", email="james.grogan@ichec.ie" },
 ]
@@ -16,6 +16,8 @@ classifiers = [
 ]
 keywords = ["Publishing", "Technical Reports", "Graphics"]
 
+dependencies = ["wand"]
+
 [project.urls]
 Repository = "https://git.ichec.ie/performance/toolshed/icplot"
 Homepage = "https://git.ichec.ie/performance/toolshed/icplot"

icplot-0.0.2/src/icplot/gantt.py

@@ -0,0 +1,64 @@
+from pathlib import Path
+from .cairo_interface import CairoInterface
+from .geometry import Scene, Color, Rectangle
+
+
+class GanttChart:
+
+    def __init__(self):
+        self.milestones = []
+        self.start_date = None
+        self.end_date = None
+        self.milestone_bar_max_height = 0.1
+        self.height = 100
+        self.width = 500
+        self.milestone_color = Color(0.7, 0, 0)
+
+    def _add_milestone(self, scene, milestone, chart_range, yloc):
+        chart_delta = chart_range[1] - chart_range[0]
+        start_delta = milestone.start_date - chart_range[0]
+        milestone_delta = milestone.due_date - milestone.start_date
+
+        start_frac = float(start_delta / chart_delta)
+        milestone_frac = float(milestone_delta / chart_delta)
+
+        w = milestone_frac * self.width
+        h = self.milestone_bar_max_height * self.height
+        x = start_frac * self.width
+        y = yloc
+
+        rect = Rectangle(w, h)
+        rect.location = (x, y)
+        rect.fill = self.milestone_color
+        scene.items.append(rect)
+
+    def _add_milestones(self, scene):
+
+        if not self.milestones:
+            return
+
+        milestones = self.milestones
+        milestones.sort(key=lambda x: x.start_date, reverse=True)
+
+        start_date = self.start_date
+        end_date = self.end_date
+        if not start_date:
+            start_date = milestones[0].start_date
+
+        if not end_date:
+            end_date = max(m.due_date for m in milestones)
+
+        chart_range = (start_date, end_date)
+        yloc = 0
+        bar_height = self.milestone_bar_max_height * self.height
+        for milestone in milestones:
+            self._add_milestone(scene, milestone, chart_range, yloc)
+            yloc += bar_height
+
+    def plot(self, path: Path):
+        scene = Scene()
+
+        self._add_milestones(scene)
+
+        renderer = CairoInterface()
+        renderer.draw_svg(scene, path)

icplot-0.0.2/src/icplot/image_utils.py

@@ -0,0 +1,16 @@
+import logging
+from pathlib import Path
+
+from wand.image import Image
+from wand.color import Color
+
+logger = logging.getLogger(__name__)
+
+
+def pdf_to_png(source_path: Path):
+    png_filename = source_path.parent / f"{source_path.stem}.png"
+    with Image(filename=source_path, resolution=300) as img:
+        img.format = "png"
+        img.background_color = Color("white")
+        img.alpha_channel = "remove"
+        img.save(filename=png_filename)

icplot-0.0.2/src/icplot/tex_interface.py

@@ -0,0 +1,95 @@
+import argparse
+import subprocess
+import shutil
+import os
+from pathlib import Path
+import logging
+
+from .image_utils import pdf_to_png
+
+logger = logging.getLogger(__name__)
+
+
+class TexInterface:
+
+    def __init__(self, build_dir: Path, output_dir: Path):
+
+        if not build_dir.is_absolute():
+            build_dir = Path(os.getcwd()) / build_dir
+
+        if not output_dir:
+            output_dir = build_dir
+
+        if not output_dir.is_absolute():
+            output_dir = Path(os.getcwd()) / output_dir
+        self.build_dir = build_dir
+        self.output_dir = output_dir
+        self.build_engine = "pdflatex"
+
+    def build_pdf(self, source_path: Path, build_dir: Path):
+        cmd = f"{self.build_engine} {source_path}"
+        subprocess.run(cmd, shell=True, check=True, cwd=build_dir)
+
+    def build_single(self, source_path: Path):
+        logging.info("Building source: %s", source_path)
+
+        # Make a working dir
+        work_dir = self.build_dir / source_path.stem
+        os.makedirs(work_dir, exist_ok=True)
+
+        tex_path = work_dir / source_path.name
+        shutil.copy(source_path, tex_path)
+
+        self.build_pdf(tex_path, work_dir)
+
+        pdf_path = tex_path.parent / f"{tex_path.stem}.pdf"
+        pdf_to_png(pdf_path)
+
+        # If output dir is different to build dir copy final content there
+        if self.output_dir != self.build_dir:
+            png_path = pdf_path.parent / f"{tex_path.stem}.png"
+            shutil.copy(pdf_path, self.output_dir)
+            shutil.copy(png_path, self.output_dir)
+
+    def build(self, search_path: Path):
+
+        if not search_path.is_absolute():
+            search_path = Path(os.getcwd()) / search_path
+
+        os.makedirs(self.output_dir, exist_ok=True)
+
+        if search_path.is_dir():
+            for tex_file in search_path.glob("*.tex"):
+                self.build_single(tex_file)
+        else:
+            self.build_single(search_path)
+
+        logger.info("Finished building sources")
+
+
+if __name__ == "__main__":
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "--source",
+        type=Path,
+        help="Path to the tikz source",
+    )
+    parser.add_argument(
+        "--build_dir",
+        type=Path,
+        default=Path(os.getcwd()) / "_build/tikz",
+        help="Path for build output",
+    )
+    parser.add_argument(
+        "--output_dir",
+        type=Path,
+        default="",
+        help="Optional dir to collect build output",
+    )
+
+    logging.basicConfig(encoding="utf-8", level=logging.INFO)
+    args = parser.parse_args()
+
+    tex = TexInterface(args.build_dir, args.output_dir)
+    tex.build(args.source)

{icplot-0.0.1 → icplot-0.0.2/src/icplot.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: icplot
-Version: 0.0.1
+Version: 0.0.2
 Summary: Utilities for generating plots and graphics for technical reports.
 Author-email: "James Grogan, Irish Centre for High End Computing" <james.grogan@ichec.ie>
 Project-URL: Repository, https://git.ichec.ie/performance/toolshed/icplot
@@ -14,6 +14,7 @@ Classifier: Topic :: System :: Distributed Computing
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: wand
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-cov; extra == "test"

{icplot-0.0.1 → icplot-0.0.2}/src/icplot.egg-info/SOURCES.txt

@@ -6,10 +6,14 @@ src/icplot/__init__.py
 src/icplot/cairo_interface.py
 src/icplot/gantt.py
 src/icplot/geometry.py
+src/icplot/image_utils.py
 src/icplot/project_elements.py
+src/icplot/tex_interface.py
 src/icplot.egg-info/PKG-INFO
 src/icplot.egg-info/SOURCES.txt
 src/icplot.egg-info/dependency_links.txt
 src/icplot.egg-info/requires.txt
 src/icplot.egg-info/top_level.txt
-test/test_cairo_interface.py
+test/test_cairo_interface.py
+test/test_gantt_chart.py
+test/test_tex_interface.py

{icplot-0.0.1 → icplot-0.0.2}/src/icplot.egg-info/requires.txt

@@ -1,3 +1,4 @@
+wand
 
 [cairo]
 pycairo

icplot-0.0.2/test/test_gantt_chart.py

@@ -0,0 +1,28 @@
+import os
+import datetime
+from pathlib import Path
+from icplot.gantt import GanttChart
+from icplot.project_elements import Milestone
+
+
+def test_gantt_chart():
+
+    milestone0 = Milestone()
+    milestone0.title = "My Milestone 0"
+    milestone0.description = "Description of Milestone 0."
+    milestone0.start_date = datetime.date(2024, 6, 30)
+    milestone0.due_date = datetime.date(2024, 7, 12)
+
+    milestone1 = Milestone()
+    milestone1.title = "My Milestone 1"
+    milestone1.description = "Description of Milestone 1."
+    milestone1.start_date = datetime.date(2024, 7, 1)
+    milestone1.due_date = datetime.date(2024, 7, 15)
+    
+    gantt = GanttChart()
+    gantt.milestones = [milestone0,
+                        milestone1]
+
+    output_path = Path(os.getcwd()) / "gantt.svg"
+
+    gantt.plot(output_path)

icplot-0.0.2/test/test_tex_interface.py

@@ -0,0 +1,17 @@
+import os
+import shutil
+from pathlib import Path
+from icplot.tex_interface import TexInterface
+
+def test_tex_interface():
+
+    build_dir = Path(os.getcwd()) / "tmp_build"
+    tex = TexInterface(build_dir, build_dir)
+
+    data_dir = Path(__file__).parent / "data"
+    tex_file = data_dir / "test.tex"
+
+    tex = TexInterface(build_dir, build_dir)
+    tex.build(tex_file)
+
+    shutil.rmtree(build_dir)

icplot-0.0.1/src/icplot/gantt.py

@@ -1,7 +0,0 @@
-class GanttChart:
-
-    def __init__(self):
-        self.milestones = []
-
-    def plot(self):
-        pass