Implement dumping filters to Python

[modules/shaper.git] / src / FiltersAPI / FiltersAPI_Filter.cpp

diff --git a/src/FiltersAPI/FiltersAPI_Filter.cpp b/src/FiltersAPI/FiltersAPI_Filter.cpp
new file mode 100644 (file)
index 0000000..668a89f
--- /dev/null
+++ b/src/FiltersAPI/FiltersAPI_Filter.cpp
@@ -0,0 +1,81 @@
+// Copyright (C) 2014-2019  CEA/DEN, EDF R&D
+//
+// This library 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.
+//
+// This library 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 this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+
+#include "FiltersAPI_Filter.h"
+
+#include <ModelAPI_Attribute.h>
+#include <ModelAPI_AttributeBoolean.h>
+#include <ModelAPI_AttributeSelectionList.h>
+
+#include <ModelHighAPI_Dumper.h>
+
+FiltersAPI_Filter::FiltersAPI_Filter(
+    const std::string& theName,
+    const bool theRevertFilter,
+    const std::list<ModelHighAPI_Selection>& theArguments)
+  : myName(theName), myReversed(theRevertFilter), myFilterArguments(theArguments)
+{
+}
+
+FiltersAPI_Filter::FiltersAPI_Filter(const std::string& theName,
+                                     const std::list<AttributePtr>& theArguments)
+  : myName(theName)
+{
+  for (std::list<AttributePtr>::const_iterator anArgIt = theArguments.begin();
+       anArgIt != theArguments.end(); ++anArgIt) {
+    AttributeBooleanPtr aBoolAttr =
+      std::dynamic_pointer_cast<ModelAPI_AttributeBoolean>(*anArgIt);
+    if (aBoolAttr)
+      myReversed = aBoolAttr->value();
+    else {
+      AttributeSelectionListPtr aSelList =
+          std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(*anArgIt);
+      if (aSelList) {
+        int aSize = aSelList->size();
+        for (int i = 0; i < aSize; ++i) {
+          AttributeSelectionPtr aSelection = aSelList->value(i);
+          myFilterArguments.push_back(
+              ModelHighAPI_Selection(aSelection->context(), aSelection->value()));
+        }
+      }
+    }
+  }
+}
+
+FiltersAPI_Filter::~FiltersAPI_Filter()
+{
+}
+
+void FiltersAPI_Filter::dump(ModelHighAPI_Dumper& theDumper) const
+{
+  theDumper << "model.addFilter(name = \"" << myName << "\"";
+  if (myReversed)
+    theDumper << ", exclude = " << myReversed;
+  if (!myFilterArguments.empty())
+    theDumper << ", args = []";
+  theDumper << ")";
+}
+
+// ================================================================================================
+FilterAPIPtr addFilter(const std::string& name,
+                       const bool exclude,
+                       const std::list<ModelHighAPI_Selection>& args)
+{
+  return FilterAPIPtr(new FiltersAPI_Filter(name, exclude, args));
+}