Line data    Source code

       1             : // This file is part of the Mamico project. For conditions of distribution
       2             : // and use, please see the copyright notice in Mamico's main folder
       3             : 
       4             : #pragma once
       5             : #include "coupling/filtering/interfaces/FilterInterface.h"
       6             : #include <algorithm>
       7             : #include <vector>
       8             : 
       9             : namespace coupling {
      10             : namespace filtering {
      11             : template <unsigned int dim> class Constant;
      12             : }
      13             : } // namespace coupling
      14             : 
      15             : /**
      16             :  * Filter applying a constant floating point value to all cells for all filtered
      17             :  * values. Optionally, one can specify to apply this filter only to certain
      18             :  * directions of multidimensional cell properties.
      19             :  *
      20             :  * @author Felix Maurer
      21             :  */
      22             : 
      23             : template <unsigned int dim> class coupling::filtering::Constant : public coupling::filtering::FilterInterface<dim> {
      24             : public:
      25           0 :   Constant(const std::vector<coupling::datastructures::CouplingCell<dim>*>& inputCellVector,
      26             :            const std::vector<coupling::datastructures::CouplingCell<dim>*>& outputCellVector, const std::array<bool, 7> filteredValues,
      27             :            const tarch::la::Vector<dim, bool> filteredDims, const double constant)
      28           0 :       : coupling::filtering::FilterInterface<dim>(inputCellVector, outputCellVector, filteredValues, "Constant"), _constant(constant),
      29           0 :         _filteredDims(filteredDims) {}
      30             : 
      31           0 :   void operator()() {
      32           0 :     tarch::la::Vector<dim, double> vec_buf;
      33           0 :     for (unsigned int i = 0; i < this->_inputCells.size(); ++i) {
      34             :       // apply to scalars
      35           0 :       for (auto scalarProperty : FilterInterface<dim>::_scalarAccessFunctionPairs) {
      36           0 :         (FilterInterface<dim>::_outputCells[i]->*scalarProperty.set)(_constant);
      37             :       }
      38             : 
      39             :       // apply to vectors
      40           0 :       for (auto vectorProperty : FilterInterface<dim>::_vectorAccessFunctionPairs) {
      41             :         // TODO: perhaps check if _filteredDims == true,..,true before this for
      42             :         // performance reasons?
      43           0 :         vec_buf = (FilterInterface<dim>::_inputCells[i]->*vectorProperty.get)();
      44             : 
      45           0 :         for (unsigned int d = 0; d < dim; d++) {
      46           0 :           if (_filteredDims[d])
      47           0 :             vec_buf[d] = _constant;
      48             :         }
      49             : 
      50           0 :         (FilterInterface<dim>::_outputCells[i]->*vectorProperty.set)(vec_buf);
      51             :       }
      52             :     }
      53           0 :   }
      54             : 
      55             : private:
      56             :   const double _constant;
      57             :   const tarch::la::Vector<dim, bool> _filteredDims;
      58             : };