mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-19 19:54:07 +00:00
Andrej730
117 lines
4.2 KiB
Python
117 lines
4.2 KiB
Python
# the SVG_FILE should be the file that contains all the icons (each icon should be in a unique layer group, and colored for dark mode)
|
|
# the EXPORT_FOLDER is where the png files get exported to
|
|
# the script will export dark mode version and modify colors for a light mode verion
|
|
# requires inkscape to run
|
|
# to run on mac: python3 generate_png_icons.py
|
|
|
|
import os
|
|
import subprocess
|
|
import xml.etree.ElementTree as ET
|
|
import re
|
|
|
|
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
SVG_FILE = os.path.join(SCRIPT_DIR, "bonsai_icons.svg")
|
|
EXPORT_FOLDER = os.path.join(SCRIPT_DIR, "..", "bonsai", "bim", "data", "icons")
|
|
|
|
overwrite_all = False
|
|
skip_all = False
|
|
|
|
|
|
def dark_to_light(svg_content):
|
|
svg_content = re.sub(r"#ffffff", "#434343", svg_content, flags=re.IGNORECASE) # white to drak grey
|
|
|
|
svg_content = re.sub(r"#00d27b", "#009d5c", svg_content, flags=re.IGNORECASE) # make green darker
|
|
svg_content = re.sub(r"#009a5f", "#00663f", svg_content, flags=re.IGNORECASE) # make dark green darker
|
|
svg_content = re.sub(r"#75ffaf", "#3dff8f", svg_content, flags=re.IGNORECASE) # make light green darker
|
|
|
|
svg_content = re.sub(r"#cce3ff", "#3391ff", svg_content, flags=re.IGNORECASE) # make blue darker
|
|
svg_content = re.sub(r"#4c9fff", "#005fcc", svg_content, flags=re.IGNORECASE) # make dark blue darker
|
|
svg_content = re.sub(r"#c9e0ff", "#80b8ff", svg_content, flags=re.IGNORECASE) # make light blue darker
|
|
|
|
return svg_content
|
|
|
|
|
|
def prompt_overwrite(file_path, png_name):
|
|
global overwrite_all, skip_all
|
|
|
|
if skip_all and os.path.exists(file_path):
|
|
return "n"
|
|
|
|
if overwrite_all:
|
|
return "y"
|
|
|
|
if os.path.exists(file_path):
|
|
while True:
|
|
response = (
|
|
input(
|
|
f"File '{png_name}' already exists.\nOverwrite? (Enter = yes, s = skip, y = yes all, n = no all): "
|
|
)
|
|
.strip()
|
|
.lower()
|
|
)
|
|
if response in {"", "y", "n", "s"}:
|
|
if response == "y":
|
|
overwrite_all = True
|
|
elif response == "n":
|
|
skip_all = True
|
|
return response
|
|
else:
|
|
print("Invalid option. Please enter Enter, y, n, or s.")
|
|
return ""
|
|
|
|
|
|
def export_svg_group(group_content, group_label, mode, output_folder):
|
|
temp_svg_file = os.path.join(output_folder, f"{group_label}_{mode}.svg")
|
|
svg_content = f"""<svg xmlns="http://www.w3.org/2000/svg" xmlns:svg="http://www.w3.org/2000/svg"
|
|
width="31.999998" height="31.999998" viewBox="0 0 8.4666661 8.4666661">
|
|
{group_content}
|
|
</svg>"""
|
|
|
|
with open(temp_svg_file, "w") as file:
|
|
file.write(svg_content)
|
|
|
|
png_name = f"{mode}_{group_label}.png"
|
|
output_png_file = os.path.join(output_folder, png_name)
|
|
|
|
user_choice = prompt_overwrite(output_png_file, png_name)
|
|
|
|
if user_choice in {"y", ""}:
|
|
subprocess.run(["inkscape", temp_svg_file, "--export-type=png", "--export-filename=" + output_png_file])
|
|
print(f"\033[32mExported PNG file: {output_png_file}\033[0m")
|
|
|
|
elif user_choice == "n":
|
|
print(f"Skipping {png_name}")
|
|
|
|
elif user_choice == "s":
|
|
print(f"Skipped exporting {output_png_file}.")
|
|
|
|
os.remove(temp_svg_file)
|
|
|
|
|
|
def export_svg_groups_to_png(svg_file, output_folder):
|
|
tree = ET.parse(svg_file)
|
|
root = tree.getroot()
|
|
ns = {"svg": "http://www.w3.org/2000/svg"}
|
|
|
|
for group in root.findall(".//svg:g", ns):
|
|
group_label = group.get("{http://www.inkscape.org/namespaces/inkscape}label")
|
|
if not group_label:
|
|
continue
|
|
|
|
style = group.get("style", "")
|
|
if "display:none" in style:
|
|
style = style.replace("display:none", "display:inline")
|
|
elif "display" not in style:
|
|
style += ";display:inline"
|
|
group.set("style", style)
|
|
|
|
group_content = ET.tostring(group, encoding="unicode", method="xml")
|
|
export_svg_group(group_content, group_label, "dm", output_folder)
|
|
light_mode_svg = dark_to_light(group_content)
|
|
export_svg_group(light_mode_svg, group_label, "lm", output_folder)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
os.makedirs(EXPORT_FOLDER, exist_ok=True)
|
|
export_svg_groups_to_png(SVG_FILE, EXPORT_FOLDER)
|