Import: move STEP reader & writer to own classes

2023-09-30 18:02:29 +02:00
parent 5dc9c484b9
commit f40dfe7193
7 changed files with 253 additions and 126 deletions
--- a/src/Mod/Import/App/WriterStep.cpp
+++ b/src/Mod/Import/App/WriterStep.cpp
@@ -0,0 +1,72 @@
+// SPDX-License-Identifier: LGPL-2.1-or-later
+
+/***************************************************************************
+ *   Copyright (c) 2023 Werner Mayer <wmayer[at]users.sourceforge.net>     *
+ *                                                                         *
+ *   This file is part of FreeCAD.                                         *
+ *                                                                         *
+ *   FreeCAD is free software: you can redistribute it and/or modify it    *
+ *   under the terms of the GNU Lesser General Public License as           *
+ *   published by the Free Software Foundation, either version 2.1 of the  *
+ *   License, or (at your option) any later version.                       *
+ *                                                                         *
+ *   FreeCAD 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      *
+ *   Lesser General Public License for more details.                       *
+ *                                                                         *
+ *   You should have received a copy of the GNU Lesser General Public      *
+ *   License along with FreeCAD. If not, see                               *
+ *   <https://www.gnu.org/licenses/>.                                      *
+ *                                                                         *
+ **************************************************************************/
+
+
+#include "PreCompiled.h"
+#ifndef _PreComp_
+#include <APIHeaderSection_MakeHeader.hxx>
+#include <STEPCAFControl_Writer.hxx>
+#endif
+
+#include "WriterStep.h"
+#include <Base/Exception.h>
+#include <App/Application.h>
+#include <Mod/Part/App/Interface.h>
+
+using namespace Import;
+
+WriterStep::WriterStep(const Base::FileInfo& file)  // NOLINT
+    : file {file}
+{}
+
+void WriterStep::write(Handle(TDocStd_Document) hDoc) const  // NOLINT
+{
+    STEPCAFControl_Writer writer;
+    Part::Interface::writeStepAssembly(Part::Interface::Assembly::On);
+    writer.Transfer(hDoc, STEPControl_AsIs);
+
+    APIHeaderSection_MakeHeader makeHeader(writer.ChangeWriter().Model());
+    Base::Reference<ParameterGrp> hGrp = App::GetApplication()
+                                             .GetUserParameter()
+                                             .GetGroup("BaseApp")
+                                             ->GetGroup("Preferences")
+                                             ->GetGroup("Mod/Part")
+                                             ->GetGroup("STEP");
+
+    // Don't set name because STEP doesn't support UTF-8
+    // https://forum.freecad.org/viewtopic.php?f=8&t=52967
+    makeHeader.SetAuthorValue(
+        1,
+        new TCollection_HAsciiString(hGrp->GetASCII("Author", "Author").c_str()));
+    makeHeader.SetOrganizationValue(
+        1,
+        new TCollection_HAsciiString(hGrp->GetASCII("Company").c_str()));
+    makeHeader.SetOriginatingSystem(
+        new TCollection_HAsciiString(App::Application::getExecutableName().c_str()));
+    makeHeader.SetDescriptionValue(1, new TCollection_HAsciiString("FreeCAD Model"));
+    IFSelect_ReturnStatus ret = writer.Write(name8bit.c_str());
+    if (ret == IFSelect_RetError || ret == IFSelect_RetFail || ret == IFSelect_RetStop) {
+        std::string utf8Name = file.filePath();
+        throw Base::FileException("Cannot open file '%s'", utf8Name.c_str());
+    }
+}