Coverage for src/cell_abm_pipeline/tasks/build_svg_image.py: 0%

1import xml.etree.ElementTree as ET 

2 

3from prefect import task 

4 

5PATH_ATTRIBUTES: list[str] = ["fill", "stroke", "stroke-width", "stroke-dasharray"] 

6 

7 

8@task 

9def build_svg_image( 

10 elements: list[dict], width: int, height: int, rotate: float, scale: float 

11) -> str: 

12 root = ET.fromstring("<svg></svg>") 

13 root.set("xmlns", "http://www.w3.org/2000/svg") 

14 root.set("width", str(width)) 

15 root.set("height", str(height)) 

16 

17 for element in elements: 

18 path = ET.fromstring("<path></path>") 

19 

20 cx = width / 2 

21 cy = height / 2 

22 

23 path.set("d", "M" + "L".join([f"{x},{y}" for x, y in element["points"]])) 

24 

25 for attribute in PATH_ATTRIBUTES: 

26 path.set(attribute, str(element[attribute]) if attribute in element else "none") 

27 

28 path.set("transform", f"rotate({rotate},{cx},{cy}) translate({cx},{cy}) scale({scale})") 

29 root.insert(0, path) 

30 

31 return ET.tostring(root, encoding="unicode")