User can now explicitly choose the quantity type via the edit qto API

This commit is contained in:
Dion Moult
2022-09-12 18:05:13 +10:00
parent 5b5540494d
commit 5ee4e2704d
2 changed files with 176 additions and 8 deletions
@@ -17,6 +17,7 @@
# along with IfcOpenShell. If not, see <http://www.gnu.org/licenses/>.
import ifcopenshell
import ifcopenshell.util.pset
class Usecase:
@@ -54,6 +55,7 @@ class Usecase:
if value is None:
self.file.remove(prop)
elif prop.is_a("IfcPhysicalSimpleQuantity"):
value = value.wrappedValue if isinstance(value, ifcopenshell.entity_instance) else value
prop[3] = float(value)
del self.settings["properties"][name]
@@ -62,7 +64,8 @@ class Usecase:
for name, value in self.settings["properties"].items():
if value is None:
continue
property_type = self.get_canonical_property_type(name)
property_type = self.get_canonical_property_type(name, value)
value = value.wrappedValue if isinstance(value, ifcopenshell.entity_instance) else value
properties.append(
self.file.create_entity(
"IfcQuantity{}".format(property_type),
@@ -76,13 +79,20 @@ class Usecase:
props.extend(new_properties)
self.settings["qto"].Quantities = props
def get_canonical_property_type(self, name):
if not self.qto_template:
return "Length"
for prop_template in self.qto_template.HasPropertyTemplates:
if prop_template.Name != name:
continue
return prop_template.TemplateType[2:].lower().capitalize()
def get_canonical_property_type(self, name, value):
if isinstance(value, ifcopenshell.entity_instance):
result = value.is_a().replace("Ifc", "").replace("Measure", "")
# Sigh, IFC inconsistencies
if result == "Numeric":
result = "Number"
elif result == "Mass":
result = "Weight"
return result
if self.qto_template:
for prop_template in self.qto_template.HasPropertyTemplates:
if prop_template.Name != name:
continue
return prop_template.TemplateType[2:].lower().capitalize()
return "Length"
def get_primary_measure_type(self, name, previous_value=None):