schrod_eig.hpp

Go to the documentation of this file.

/**
 * @file schrod_eig.hpp
 * @brief Calculate the square of wave function
 * @date 2025-11-14
 *
 * @copyright Copyright (c) 2025
 *
 */
 
#include <MoFEM.hpp>
 
using namespace MoFEM;
 
namespace schrod_eig {
 
//! [Calculate square of wave function]
struct OpSquare : public ForcesAndSourcesCore::UserDataOperator {
  OpSquare(boost::shared_ptr<VectorDouble> squareField,
           boost::shared_ptr<VectorDouble> field)
      : ForcesAndSourcesCore::UserDataOperator(NOSPACE, OPLAST),
        squareField(squareField), field(field) {}
 
  MoFEMErrorCode doWork(int side, EntityType type,
                        DataForcesAndSourcesCore::EntData &data) {
    MoFEMFunctionBegin;
 
    const size_t nb_gauss_pts = getGaussPts().size2();
    squareField->resize(nb_gauss_pts, 0);
 
    auto t_squareField = getFTensor0FromVec(*squareField);
 
    auto t_field = getFTensor0FromVec(*field);
 
    for (int gg = 0; gg != nb_gauss_pts; gg++) {
      t_squareField = t_field * t_field;
      ++t_field;
      ++t_squareField;
    }
 
    MoFEMFunctionReturn(0);
  }
 
private:
  boost::shared_ptr<VectorDouble> squareField;
  boost::shared_ptr<VectorDouble> field;
};
//! [Calculate square of wave function]
 
} // namespace schrod_eig