coupledParticleSolver.hpp

#ifndef ABLATELIBRARY_COUPLEDPARTICLESOLVER_HPP
#define ABLATELIBRARY_COUPLEDPARTICLESOLVER_HPP
 
#include "particleSolver.hpp"
#include "processes/coupledProcess.hpp"
#include "solver/rhsFunction.hpp"
 
namespace ablate::particles {
 
class CoupledParticleSolver : public ParticleSolver, public ablate::solver::RHSFunction {
    // A string to hold the previous source term name name
    inline static const char PreviousPackedSolution[] = "PreviousPackedSolution";
 
   public:
    CoupledParticleSolver(std::string solverId, std::shared_ptr<domain::Region>, std::shared_ptr<parameters::Parameters> options, std::vector<FieldDescription> fields,
                          std::vector<std::shared_ptr<processes::Process>> processes, std::shared_ptr<initializers::Initializer> initializer,
                          std::vector<std::shared_ptr<mathFunctions::FieldFunction>> fieldInitialization, std::vector<std::shared_ptr<mathFunctions::FieldFunction>> exactSolutions = {},
                          const std::vector<std::string>& coupledFields = {});
 
    CoupledParticleSolver(std::string solverId, std::shared_ptr<domain::Region>, std::shared_ptr<parameters::Parameters> options, const std::vector<std::shared_ptr<FieldDescription>>& fields,
                          std::vector<std::shared_ptr<processes::Process>> processes, std::shared_ptr<initializers::Initializer> initializer,
                          std::vector<std::shared_ptr<mathFunctions::FieldFunction>> fieldInitialization, std::vector<std::shared_ptr<mathFunctions::FieldFunction>> exactSolutions = {},
                          const std::vector<std::string>& = {});
 
    ~CoupledParticleSolver() override;
 
    void Setup() override;
 
    void Initialize() override;
 
    PetscErrorCode PreRHSFunction(TS ts, PetscReal time, bool initialStage, Vec locX) override;
 
    PetscErrorCode ComputeRHSFunction(PetscReal time, Vec locX, Vec locF) override;
 
   protected:
    void MacroStepParticles(TS macroTS, bool swarmMigrate) override;
 
   private:
    std::vector<std::shared_ptr<processes::CoupledProcess>> coupledProcesses;
 
    std::vector<std::string> coupledFieldsNames;
 
    std::vector<std::string> coupledParticleFieldsNames;
 
    std::vector<ablate::domain::Field> coupledFields;
 
    Vec localEulerianSourceVec{};
 
    Vec localEulerianVolumeFactor{};
 
    void ComputeEulerianSource(PetscReal startTime, PetscReal endTime);
};
 
}  // namespace ablate::particles
#endif  // ABLATELIBRARY_COUPLEDPARTICLESOLVER_HPP