Schema dependent serializers

2026-08-13 02:47:48 +00:00 · 2018-03-04 10:43:40 +01:00
parent 0a809ba127
commit 6685ea6d7a
15 changed files with 545 additions and 0 deletions
@@ -0,0 +1,60 @@
+/********************************************************************************
+ *                                                                              *
+ * This file is part of IfcOpenShell.                                           *
+ *                                                                              *
+ * IfcOpenShell is free software: you can redistribute it and/or modify         *
+ * it under the terms of the Lesser GNU General Public License as published by  *
+ * the Free Software Foundation, either version 3.0 of the License, or          *
+ * (at your option) any later version.                                          *
+ *                                                                              *
+ * IfcOpenShell 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                 *
+ * Lesser GNU General Public License for more details.                          *
+ *                                                                              *
+ * You should have received a copy of the Lesser GNU General Public License     *
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.         *
+ *                                                                              *
+ ********************************************************************************/
+
+#ifndef STEPSERIALIZER_H
+#define STEPSERIALIZER_H
+
+#include <STEPControl_Writer.hxx>
+#include <Interface_Static.hxx>
+
+#include "../ifcgeom_schema_agnostic/IfcGeomIterator.h"
+
+#include "../ifcconvert/OpenCascadeBasedSerializer.h"
+
+class StepSerializer : public OpenCascadeBasedSerializer
+{
+private:
+	STEPControl_Writer writer;	
+public:
+	explicit StepSerializer(const std::string& out_filename, const SerializerSettings& settings)
+		: OpenCascadeBasedSerializer(out_filename, settings)
+	{}
+	virtual ~StepSerializer() {}
+	void writeShape(const TopoDS_Shape& shape) {
+		std::stringstream ss;
+		std::streambuf *sb = std::cout.rdbuf(ss.rdbuf());
+		writer.Transfer(shape, STEPControl_AsIs);
+		std::cout.rdbuf(sb);
+	}
+	void finalize() {
+		std::stringstream ss;
+		std::streambuf *sb = std::cout.rdbuf(ss.rdbuf());
+		writer.Write(out_filename.c_str());
+		std::cout.rdbuf(sb);
+	}
+	void setUnitNameAndMagnitude(const std::string& /*name*/, float magnitude) {
+		const char* symbol = getSymbolForUnitMagnitude(magnitude);
+		if (symbol) {
+			Interface_Static::SetCVal("xstep.cascade.unit", symbol);
+			Interface_Static::SetCVal("write.step.unit", symbol);
+		}
+	}
+};
+
+#endif