rhsAccessor.hpp

#ifndef ABLATELIBRARY_RHSACCESSOR_HPP
#define ABLATELIBRARY_RHSACCESSOR_HPP
 
#include <petsc.h>
#include <map>
#include "accessor.hpp"
#include "particles/field.hpp"
#include "utilities/petscUtilities.hpp"
 
namespace ablate::particles::accessors {
class RhsAccessor : public Accessor<PetscReal> {
   private:
    const std::map<std::string, Field>& fieldsMap;
 
    Vec rhsVec;
 
    PetscScalar* rhsValues{};
 
   public:
    RhsAccessor(bool cachePointData, const std::map<std::string, Field>& fieldsMap, Vec rhsVec) : Accessor(cachePointData), fieldsMap(fieldsMap), rhsVec(rhsVec) {
        // extract the array from the vector
        VecGetArray(rhsVec, &rhsValues) >> utilities::PetscUtilities::checkError;
    }
 
    ~RhsAccessor() override { VecRestoreArray(rhsVec, &rhsValues) >> utilities::PetscUtilities::checkError; }
 
    PointData CreateData(const std::string& fieldName) override {
        try {
            const auto& field = fieldsMap.at(fieldName);
            if (field.location == domain::FieldLocation::SOL) {
                return {rhsValues, field};
            } else {
                throw std::invalid_argument("The field " + std::string(fieldName) + " is not a solution variable");
            }
        } catch (std::exception& exception) {
            throw std::invalid_argument("Unable to locate particle field " + fieldName);
        }
    }
 
    RhsAccessor(const RhsAccessor&) = delete;
};
}  // namespace ablate::particles::accessors
#endif  // ABLATELIBRARY_RHSACCESSOR_HPP