Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-20 15:08:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #721 from s-leger/patch-3

Browse Source 
Update ui.py
This commit is contained in:
Dion Moult
2019-11-15 15:03:59 +11:00
committed by GitHub
parent e1bf82b88e 0532048f37
commit 836e350a15
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcblenderexport/io_export_ifc/ui.py
+9 -8
View File
@@ -29,6 +29,7 @@ types_enum = []
psetnames_enum = [] psetnames_enum = []
psetfiles_enum = [] psetfiles_enum = []
classification_enum = [] classification_enum = []
reference_enum = []
attributes_enum = [] attributes_enum = []
documents_enum = [] documents_enum = []
@@ -36,7 +37,7 @@ documents_enum = []
# enums should be cached especially when file io is involved # enums should be cached especially when file io is involved
# we may clear on need to regen # we may clear on need to regen
def getIfcPredefinedTypes(self): def getIfcPredefinedTypes(self, context):
global types_enum global types_enum
types_enum.clear() types_enum.clear()
for name, data in ifc_schema.elements.items(): for name, data in ifc_schema.elements.items():
@@ -48,20 +49,20 @@ def getIfcPredefinedTypes(self):
types_enum.extend([(e, e, '') for e in attribute['enum_values']]) types_enum.extend([(e, e, '') for e in attribute['enum_values']])
return types_enum return types_enum
def getIfcClasses(self): def getIfcClasses(self, context):
global classes_enum global classes_enum
classes_enum.clear() classes_enum.clear()
classes_enum.extend([(e, e, '') for e in ifc_schema.elements]) classes_enum.extend([(e, e, '') for e in ifc_schema.elements])
return classes_enum return classes_enum
def getPsetNames(self): def getPsetNames(self, context):
global psetnames_enum global psetnames_enum
psetnames_enum.clear() psetnames_enum.clear()
files = os.listdir(bpy.context.scene.BIMProperties.data_dir + 'pset' + os.path.sep) files = os.listdir(bpy.context.scene.BIMProperties.data_dir + 'pset' + os.path.sep)
psetnames_enum.extend([(f, f, '') for f in files]) psetnames_enum.extend([(f, f, '') for f in files])
return psetnames_enum return psetnames_enum
def getPsetFiles(self): def getPsetFiles(self, context):
global psetfiles_enum global psetfiles_enum
psetfiles_enum.clear() psetfiles_enum.clear()
if not bpy.context.scene.BIMProperties.pset_name: if not bpy.context.scene.BIMProperties.pset_name:
@@ -74,7 +75,7 @@ def getPsetFiles(self):
psetfiles_enum.extend([(f.replace('.csv', ''), f.replace('.csv', ''), '') for f in files]) psetfiles_enum.extend([(f.replace('.csv', ''), f.replace('.csv', ''), '') for f in files])
return psetfiles_enum return psetfiles_enum
def getClassifications(self): def getClassifications(self, context):
global classification_enum global classification_enum
classification_enum.clear() classification_enum.clear()
with open(bpy.context.scene.BIMProperties.data_dir + 'class/classifications.csv', 'r') as f: with open(bpy.context.scene.BIMProperties.data_dir + 'class/classifications.csv', 'r') as f:
@@ -84,7 +85,7 @@ def getClassifications(self):
classification_enum.extend([(str(i), d[index], '') for i, d in enumerate(data)]) classification_enum.extend([(str(i), d[index], '') for i, d in enumerate(data)])
return classification_enum return classification_enum
def getReferences(self): def getReferences(self, context):
global reference_enum global reference_enum
reference_enum.clear() reference_enum.clear()
if not bpy.context.scene.BIMProperties.classification: if not bpy.context.scene.BIMProperties.classification:
@@ -96,7 +97,7 @@ def getReferences(self):
d[2] == bpy.context.scene.BIMProperties.classification]) d[2] == bpy.context.scene.BIMProperties.classification])
return reference_enum return reference_enum
def getApplicableAttributes(self): def getApplicableAttributes(self, context):
global attributes_enum global attributes_enum
attributes_enum.clear() attributes_enum.clear()
if '/' in bpy.context.active_object.name \ if '/' in bpy.context.active_object.name \
@@ -106,7 +107,7 @@ def getApplicableAttributes(self):
if bpy.context.active_object.BIMObjectProperties.attributes.find(a['name']) == -1]) if bpy.context.active_object.BIMObjectProperties.attributes.find(a['name']) == -1])
return attributes_enum return attributes_enum
def getApplicableDocuments(self): def getApplicableDocuments(self, context):
global documents_enum global documents_enum
documents_enum.clear() documents_enum.clear()
doc_path = bpy.context.scene.BIMProperties.data_dir + 'doc' + os.path.sep doc_path = bpy.context.scene.BIMProperties.data_dir + 'doc' + os.path.sep