Fem: Calculator filter implementation

Author: marioalexis84

src/Mod/Fem/App/AppFem.cpp

@@ -186,6 +186,7 @@ PyMOD_INIT_FUNC(Fem)
     Fem::FemPostObject                        ::init();
     Fem::FemPostPipeline                      ::init();
     Fem::FemPostFilter                        ::init();
+    Fem::FemPostCalculatorFilter              ::init();
     Fem::FemPostClipFilter                    ::init();
     Fem::FemPostContoursFilter                ::init();
     Fem::FemPostCutFilter                     ::init();

src/Mod/Fem/App/FemPostFilter.cpp

@@ -1158,3 +1158,96 @@ short int FemPostWarpVectorFilter::mustExecute() const
         return App::DocumentObject::mustExecute();
     }
 }
+
+
+// ***************************************************************************
+// calculator filter
+PROPERTY_SOURCE(Fem::FemPostCalculatorFilter, Fem::FemPostFilter)
+
+FemPostCalculatorFilter::FemPostCalculatorFilter()
+    : FemPostFilter()
+{
+    ADD_PROPERTY_TYPE(FieldName,
+                      (""),
+                      "Calculator",
+                      App::Prop_None,
+                      "Name of the calculated field");
+    ADD_PROPERTY_TYPE(Function,
+                      (""),
+                      "Calculator",
+                      App::Prop_None,
+                      "Expression of the unction to evaluate");
+
+    FilterPipeline calculator;
+    m_calculator = vtkSmartPointer<vtkArrayCalculator>::New();
+    calculator.source = m_calculator;
+    calculator.target = m_calculator;
+    addFilterPipeline(calculator, "calculator");
+    setActiveFilterPipeline("calculator");
+}
+
+FemPostCalculatorFilter::~FemPostCalculatorFilter() = default;
+
+DocumentObjectExecReturn* FemPostCalculatorFilter::execute()
+{
+    printf("EXECITER\n");
+    m_calculator->AddCoordinateScalarVariable("X", 0);
+    m_calculator->AddCoordinateScalarVariable("Y", 1);
+    m_calculator->AddCoordinateScalarVariable("Z", 2);
+    //    std::string val;
+    //    if (Vector.getValue() >= 0) {
+    //        val = Vector.getValueAsString();
+    //    }
+    //
+    //    std::vector<std::string> VectorArray;
+    //
+    //    vtkSmartPointer<vtkDataObject> data = getInputData();
+    //    vtkDataSet* dset = vtkDataSet::SafeDownCast(data);
+    //    if (!dset) {
+    //        return StdReturn;
+    //    }
+    //    vtkPointData* pd = dset->GetPointData();
+    //
+    //    // get all vector fields
+    //    for (int i = 0; i < pd->GetNumberOfArrays(); ++i) {
+    //        if (pd->GetArray(i)->GetNumberOfComponents() == 3) {
+    //            VectorArray.emplace_back(pd->GetArrayName(i));
+    //        }
+    //    }
+    //
+    //    App::Enumeration empty;
+    //    Vector.setValue(empty);
+    //    m_vectorFields.setEnums(VectorArray);
+    //    Vector.setValue(m_vectorFields);
+    //
+    //    // search if the current field is in the available ones and set it
+    //    std::vector<std::string>::iterator it = std::find(VectorArray.begin(), VectorArray.end(),
+    //    val); if (!val.empty() && it != VectorArray.end()) {
+    //        Vector.setValue(val.c_str());
+    //    }
+    //
+    // recalculate the filter
+    return FemPostFilter::execute();
+}
+
+void FemPostCalculatorFilter::onChanged(const Property* prop)
+{
+    if (prop == &Function) {
+        m_calculator->SetFunction(Function.getValue());
+    }
+    else if (prop == &FieldName) {
+        m_calculator->SetResultArrayName(FieldName.getValue());
+    }
+    printf("FUNCION: %s\n", m_calculator->GetFunction());
+    Fem::FemPostFilter::onChanged(prop);
+}
+
+short int FemPostCalculatorFilter::mustExecute() const
+{
+    if (Function.isTouched() || FieldName.isTouched()) {
+        return 1;
+    }
+    else {
+        return FemPostFilter::mustExecute();
+    }
+}

src/Mod/Fem/App/FemPostFilter.h

@@ -23,6 +23,7 @@
 #ifndef Fem_FemPostFilter_H
 #define Fem_FemPostFilter_H
 
+#include <vtkArrayCalculator.h>
 #include <vtkContourFilter.h>
 #include <vtkSmoothPolyDataFilter.h>
 #include <vtkCutter.h>
@@ -352,6 +353,34 @@ class FemExport FemPostWarpVectorFilter: public FemPostFilter
     App::Enumeration m_vectorFields;
 };
 
+// ***************************************************************************
+// calculator filter
+class FemExport FemPostCalculatorFilter: public FemPostFilter
+{
+
+    PROPERTY_HEADER_WITH_OVERRIDE(Fem::FemPostCalculatorFilter);
+
+public:
+    FemPostCalculatorFilter();
+    ~FemPostCalculatorFilter() override;
+
+    App::PropertyString FieldName;
+    App::PropertyString Function;
+
+    const char* getViewProviderName() const override
+    {
+        return "FemGui::ViewProviderFemPostCalculator";
+    }
+    short int mustExecute() const override;
+
+protected:
+    App::DocumentObjectExecReturn* execute() override;
+    void onChanged(const App::Property* prop) override;
+
+private:
+    vtkSmartPointer<vtkArrayCalculator> m_calculator;
+};
+
 }  // namespace Fem
 
 

src/Mod/Fem/App/PreCompiled.h

@@ -154,6 +154,7 @@
 
 // VTK
 #include <vtkAppendFilter.h>
+#include <vtkArrayCalculator.h>
 #include <vtkCellArray.h>
 #include <vtkCompositeDataSet.h>
 #include <vtkDataArray.h>

src/Mod/Fem/Gui/AppFemGui.cpp

@@ -160,6 +160,7 @@ PyMOD_INIT_FUNC(FemGui)
 #ifdef FC_USE_VTK
     FemGui::ViewProviderFemPostObject                           ::init();
     FemGui::ViewProviderFemPostPipeline                         ::init();
+    FemGui::ViewProviderFemPostCalculator                       ::init();
     FemGui::ViewProviderFemPostClip                             ::init();
     FemGui::ViewProviderFemPostContours                         ::init();
     FemGui::ViewProviderFemPostCut                              ::init();

src/Mod/Fem/Gui/CMakeLists.txt

@@ -94,6 +94,7 @@ if(BUILD_FEM_VTK)
         CylinderWidget.ui
         PlaneWidget.ui
         SphereWidget.ui
+        TaskPostCalculator.ui
         TaskPostClip.ui
         TaskPostContours.ui
         TaskPostCut.ui
@@ -286,6 +287,7 @@ if(BUILD_FEM_VTK)
         SphereWidget.ui
         TaskPostBoxes.h
         TaskPostBoxes.cpp
+        TaskPostCalculator.ui
         TaskPostClip.ui
         TaskPostContours.ui
         TaskPostCut.ui

src/Mod/Fem/Gui/Command.cpp

@@ -2443,6 +2443,42 @@ bool CmdFemPostContoursFilter::isActive()
 }
 
 
+//================================================================================================
+DEF_STD_CMD_A(CmdFemPostCalculatorFilter)
+
+CmdFemPostCalculatorFilter::CmdFemPostCalculatorFilter()
+    : Command("FEM_PostFilterCalculator")
+{
+    sAppModule = "Fem";
+    sGroup = QT_TR_NOOP("Fem");
+    sMenuText = QT_TR_NOOP("Calculator filter");
+    sToolTipText = QT_TR_NOOP("Create new fields from current data");
+    sWhatsThis = "FEM_PostFilterCalculator";
+    sStatusTip = sToolTipText;
+    sPixmap = "FEM_PostFilterCalculator";
+}
+
+void CmdFemPostCalculatorFilter::activated(int)
+{
+    setupFilter(this, "Calculator");
+}
+
+bool CmdFemPostCalculatorFilter::isActive()
+{
+    // only allow one object
+    auto selection = getSelection().getSelection();
+    if (selection.size() > 1) {
+        return false;
+    }
+    for (auto obj : selection) {
+        if (obj.pObject->isDerivedFrom<Fem::FemPostObject>()) {
+            return true;
+        }
+    }
+    return false;
+}
+
+
 //================================================================================================
 DEF_STD_CMD_ACL(CmdFemPostFunctions)
 
@@ -2858,6 +2894,7 @@ void CreateFemCommands()
     // vtk post processing
 #ifdef FC_USE_VTK
     rcCmdMgr.addCommand(new CmdFemPostApllyChanges);
+    rcCmdMgr.addCommand(new CmdFemPostCalculatorFilter);
     rcCmdMgr.addCommand(new CmdFemPostClipFilter);
     rcCmdMgr.addCommand(new CmdFemPostContoursFilter);
     rcCmdMgr.addCommand(new CmdFemPostCutFilter);

src/Mod/Fem/Gui/PreCompiled.h

@@ -81,6 +81,8 @@
 #include <QSlider>
 #include <QStackedWidget>
 #include <QString>
+#include <QStringList>
+#include <QStringListModel>
 #include <QTextCharFormat>
 #include <QTextStream>
 #include <QThread>

src/Mod/Fem/Gui/Resources/Fem.qrc

@@ -72,6 +72,7 @@
         <file>icons/FEM_MaterialSolid.svg</file>
 
         <!-- gui command icons: post processing -->
+        <file>icons/FEM_PostFilterCalculator.svg</file>
         <file>icons/FEM_PostFilterClipRegion.svg</file>
         <file>icons/FEM_PostFilterClipScalar.svg</file>
         <file>icons/FEM_PostFilterContours.svg</file>