# IfcSverchok - IFC Sverchok extension

# Copyright (C) 2022 Martina Jakubowska <martina@jakubowska.dk>

#

# This file is part of IfcSverchok.

#

# IfcSverchok 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.

#

# IfcSverchok 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 IfcSverchok. If not, see <http://www.gnu.org/licenses/>.

from typing import Any, Union

import bonsai.tool as tool

import bpy

import ifcopenshell

import ifcopenshell.api.context

import ifcopenshell.util.representation

from ifcopenshell import template

class SvIfcStore:

path = ""

file: Union[ifcopenshell.file, None] = None

schema = None

cache = None

cache_path = None

id_map: dict[str, Any] = {}

"""Mapping `{node_id: Any}`"""

guid_map = {}

deleted_ids = set()

edited_objs = set()

pset_template_path = ""

pset_template_file = None

library_path = ""

library_file = None

element_listeners = set()

undo_redo_stack_objects = set()

undo_redo_stack_object_names = {}

current_transaction = ""

last_transaction = ""

history = []

future = []

schema_identifiers = ["IFC4", "IFC2X3"]

use_bonsai_file = False

@staticmethod

def purge() -> None:

SvIfcStore.path = ""

SvIfcStore.file = None

SvIfcStore.schema = None

SvIfcStore.cache = None

SvIfcStore.cache_path = None

SvIfcStore.id_map = {}

SvIfcStore.guid_map = {}

SvIfcStore.deleted_ids = set()

SvIfcStore.edited_objs = set()

SvIfcStore.pset_template_path = ""

SvIfcStore.pset_template_file = None

SvIfcStore.library_path = ""

SvIfcStore.library_file = None

SvIfcStore.last_transaction = ""

SvIfcStore.history = []

SvIfcStore.future = []

SvIfcStore.schema_identifiers = ["IFC4", "IFC2X3"]

@staticmethod

def create_boilerplate() -> ifcopenshell.file:

file = template.create(

filename="IfcSverchokDemoFile",

organization=None,

creator=None,

project_name="IfcSverchokDemoProject",

)

if bpy.context.scene.unit_settings.system == "IMPERIAL":

# TODO change units to imperial

pass

model = ifcopenshell.util.representation.get_context(file, context="Model")

context = ifcopenshell.api.context.add_context(

file,

context_type="Model",

context_identifier="Body",

target_view="MODEL_VIEW",

parent=model,

)

SvIfcStore.file = file

return SvIfcStore.file

@staticmethod

def get_file() -> ifcopenshell.file:

if SvIfcStore.use_bonsai_file:

return tool.Ifc.get()

if SvIfcStore.file is None:

SvIfcStore.create_boilerplate()

return SvIfcStore.file