cellSolver.hpp

#ifndef ABLATELIBRARY_CELLSOLVER_HPP
#define ABLATELIBRARY_CELLSOLVER_HPP
 
#include <petsc.h>
#include <functional>
#include <vector>
#include "solver.hpp"
 
namespace ablate::solver {
 
class CellSolver : public solver::Solver {
   public:
    using AuxFieldUpdateFunction = PetscErrorCode (*)(PetscReal time, PetscInt dim, const PetscFVCellGeom* cellGeom, const PetscInt uOff[], const PetscScalar* u, const PetscInt aOff[],
                                                      PetscScalar* auxField, void* ctx);
 
    using SolutionFieldUpdateFunction = PetscErrorCode (*)(PetscReal time, PetscInt dim, const PetscFVCellGeom* cellGeom, const PetscInt uOff[], PetscScalar* u, void* ctx);
 
   private:
    struct AuxFieldUpdateFunctionDescription {
        AuxFieldUpdateFunction function;
        void* context;
        std::vector<PetscInt> inputFields;
        std::vector<PetscInt> auxFields;
    };
 
    std::vector<AuxFieldUpdateFunctionDescription> auxFieldUpdateFunctionDescriptions;
 
    struct SolutionFieldUpdateFunctionDescription {
        SolutionFieldUpdateFunction function;
        void* context;
        std::vector<PetscInt> inputFieldsOffsets;
    };
 
    std::vector<SolutionFieldUpdateFunctionDescription> solutionFieldUpdateFunctionDescriptions;
 
   protected:
    Vec cellGeomVec = nullptr;
 
    Vec faceGeomVec = nullptr;
 
   public:
    explicit CellSolver(std::string solverId, std::shared_ptr<domain::Region> = {}, std::shared_ptr<parameters::Parameters> options = nullptr);
    ~CellSolver() override;
 
    void RegisterAuxFieldUpdate(AuxFieldUpdateFunction function, void* context, const std::vector<std::string>& auxField, const std::vector<std::string>& inputFields);
 
    void RegisterSolutionFieldUpdate(SolutionFieldUpdateFunction function, void* context, const std::vector<std::string>& inputFields);
 
    void UpdateAuxFields(PetscReal time, Vec locXVec, Vec locAuxField);
 
    void UpdateSolutionFields(PetscReal time, Vec globXVec);
 
    void Setup() override;
 
    void Initialize() override;
};
}  // namespace ablate::solver
 
#endif  // ABLATELIBRARY_CELLSOLVER_HPP