IfcOpenShell/src/blenderbim/blenderbim/bim/module/tester/operator.py

# BlenderBIM Add-on - OpenBIM Blender Add-on
# Copyright (C) 2020, 2021 Dion Moult <dion@thinkmoult.com>
#
# This file is part of BlenderBIM Add-on.
#
# BlenderBIM Add-on 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.
#
# BlenderBIM Add-on 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 BlenderBIM Add-on.  If not, see <http://www.gnu.org/licenses/>.

import os
import bpy
import time
import tempfile
import webbrowser
import json
import ifctester
import ifctester.ids
import ifctester.reporter
import ifcopenshell
import blenderbim.tool as tool


class ExecuteIfcTester(bpy.types.Operator):
    bl_idname = "bim.execute_ifc_tester"
    bl_label = "Execute IfcTester"

    @classmethod
    def poll(cls, context):
        props = context.scene.IfcTesterProperties
        return (props.ifc_file or props.should_load_from_memory) and props.specs

    def execute(self, context):
        props = context.scene.IfcTesterProperties

        with tempfile.TemporaryDirectory() as dirpath:
            start = time.time()
            output = os.path.join(dirpath, "{}.html".format(props.specs))

            if props.should_load_from_memory and tool.Ifc.get():
                ifc = tool.Ifc.get()
            else:
                ifc = ifcopenshell.open(props.ifc_file)

            specs = ifctester.ids.open(props.specs)
            print("Finished loading:", time.time() - start)
            start = time.time()
            specs.validate(ifc)
            print(specs)
            
            print("Finished validating:", time.time() - start)
            start = time.time()

            engine = ifctester.reporter.Html(specs)
            engine.report()            
            engine.to_file(output)
            webbrowser.open("file://" + output)

            report = None
            report = ifctester.reporter.Json(specs).report()['specifications']
            if report:
                props.has_report = True
                props.report = json.dumps(report)
            props.specifications.clear()
            c=0
            for spec in report:
                new_spec = props.specifications.add()
                new_spec.name = spec['name']
                new_spec.status = spec['status']

        return {"FINISHED"}


class SelectSpecs(bpy.types.Operator):
    bl_idname = "bim.select_specs"
    bl_label = "Select IDS"
    bl_options = {"REGISTER", "UNDO"}
    filename_ext = ".ids"
    filter_glob: bpy.props.StringProperty(default="*.ids;*.xml", options={"HIDDEN"})
    filepath: bpy.props.StringProperty(subtype="FILE_PATH")

    def execute(self, context):
        context.scene.IfcTesterProperties.specs = self.filepath
        return {"FINISHED"}

    def invoke(self, context, event):
        context.window_manager.fileselect_add(self)
        return {"RUNNING_MODAL"}


class SelectIfcTesterIfcFile(bpy.types.Operator):
    bl_idname = "bim.select_ifctester_ifc_file"
    bl_label = "Select IfcTester IFC File"
    bl_options = {"REGISTER", "UNDO"}
    filename_ext = ".ifc"
    filter_glob: bpy.props.StringProperty(default="*.ifc;*.ifczip;*.ifcxml", options={"HIDDEN"})
    filepath: bpy.props.StringProperty(subtype="FILE_PATH")

    def execute(self, context):
        context.scene.IfcTesterProperties.ifc_file = self.filepath
        return {"FINISHED"}

    def invoke(self, context, event):
        context.window_manager.fileselect_add(self)
        return {"RUNNING_MODAL"}


class SelectRequirement(bpy.types.Operator):
    bl_idname = "bim.select_requirement"
    bl_label = "Select Specification"
    bl_options = {"REGISTER", "UNDO"}
    spec_index: bpy.props.IntProperty()
    req_index: bpy.props.IntProperty() 

    def execute(self, context):
        props = context.scene.IfcTesterProperties
        props.has_entities = True
        report = json.loads(props.report)
        props.failed_entities.clear()
        failed_entities = report[self.spec_index]['requirements'] [self.req_index]['failed_entities']
        print(failed_entities)
        for e in failed_entities:
            new_entity = props.failed_entities.add()            
            new_entity.element = e['element']
            new_entity.reason = e['reason']
        return {"FINISHED"}

class SelectEntity(bpy.types.Operator):
    bl_idname = "bim.select_entity"
    bl_label = "Select Entity"
    bl_options = {"REGISTER", "UNDO"}
    ifc_id: bpy.props.IntProperty()

    def execute(self, context):
        bpy.ops.object.select_all(action='DESELECT')
        for obj in context.scene.objects:
            if obj.BIMObjectProperties.ifc_definition_id == self.ifc_id:
                obj.select_set(True)
                bpy.context.view_layer.objects.active = obj
        return {"FINISHED"}