Unique filter on IfcEntityList

aothms · aothms · commit e60202b738af · 2017-06-20T18:54:55.000+02:00
diff --git a/src/ifcparse/IfcEntityList.h b/src/ifcparse/IfcEntityList.h
@@ -50,6 +50,7 @@ class IFC_PARSE_API IfcEntityList {
 	}
 	void remove(IfcUtil::IfcBaseClass*);
 	IfcEntityList::ptr filtered(const std::set<IfcSchema::Type::Enum>& entities);
+	IfcEntityList::ptr unique();
 };
 
 template <class T>
diff --git a/src/ifcparse/IfcUtil.cpp b/src/ifcparse/IfcUtil.cpp
@@ -62,6 +62,7 @@ void IfcEntityList::remove(IfcUtil::IfcBaseClass* instance) {
 		ls.erase(it);
 	}
 }
+
 IfcEntityList::ptr IfcEntityList::filtered(const std::set<IfcSchema::Type::Enum>& entities) {
 	IfcEntityList::ptr return_value(new IfcEntityList);
 	for (it it = begin(); it != end(); ++it) {
@@ -79,6 +80,18 @@ IfcEntityList::ptr IfcEntityList::filtered(const std::set<IfcSchema::Type::Enum>
 	return return_value;
 }
 
+IfcEntityList::ptr IfcEntityList::unique() {
+	std::set<IfcUtil::IfcBaseClass*> encountered;
+	IfcEntityList::ptr return_value(new IfcEntityList);
+	for (it it = begin(); it != end(); ++it) {
+		if (encountered.find(*it) == encountered.end()) {
+			return_value->push(*it);
+			encountered.insert(*it);
+		}
+	}
+	return return_value;
+}
+
 
 unsigned int IfcUtil::IfcBaseType::getArgumentCount() const { return 1; }
 Argument* IfcUtil::IfcBaseType::getArgument(unsigned int i) const { return entity->getArgument(i); }

Original file line number	Diff line number	Diff line change
`@@ -50,6 +50,7 @@ class IFC_PARSE_API IfcEntityList {`
`50`	`50`	`}`
`51`	`51`	`void remove(IfcUtil::IfcBaseClass*);`
`52`	`52`	`IfcEntityList::ptr filtered(const std::set<IfcSchema::Type::Enum>& entities);`
	`53`	`+ IfcEntityList::ptr unique();`
`53`	`54`	`};`
`54`	`55`
`55`	`56`	`template <class T>`