probes.hpp

#ifndef ABLATELIBRARY_PROBES_HPP
#define ABLATELIBRARY_PROBES_HPP
 
#include <utility>
#include "io/interval/interval.hpp"
#include "monitor.hpp"
#include "probes/probe.hpp"
#include "probes/probeInitializer.hpp"
 
namespace ablate::monitors {
 
class Probes : public Monitor {
   public:
    class ProbeRecorder {
       private:
        const int bufferSize;
 
        std::filesystem::path outputPath;
 
        PetscReal lastOutputTime = PETSC_MIN_REAL;
 
        int activeIndex = -1;
 
        std::vector<std::vector<double>> buffer;
 
        std::vector<double> timeHistory;
 
       public:
        ProbeRecorder(int bufferSize, const std::vector<std::string>& variables, const std::filesystem::path& outputPath);
 
        ~ProbeRecorder();
 
        void AdvanceTime(double time);
 
        void SetValue(std::size_t index, double value);
 
        void WriteBuffer();
    };
 
    const std::shared_ptr<ablate::monitors::probes::ProbeInitializer> initializer;
 
    const std::vector<std::string> variableNames;
 
    const std::shared_ptr<io::interval::Interval> interval;
 
    const int bufferSize;
 
    std::vector<probes::Probe> localProbes;
 
    std::vector<domain::Field> fields;
 
    std::vector<int> fieldOffset;
 
    std::vector<DMInterpolationInfo> interpolants;
 
    std::vector<ProbeRecorder> recorders;
 
    static PetscErrorCode UpdateProbes(TS ts, PetscInt step, PetscReal crtime, Vec u, void* ctx);
 
   public:
    Probes(const std::shared_ptr<ablate::monitors::probes::ProbeInitializer>&, std::vector<std::string> variableNames, const std::shared_ptr<io::interval::Interval>& interval = {},
           const int bufferSize = 0);
 
    ~Probes() override;
 
    void Register(std::shared_ptr<solver::Solver> solver) override;
 
    PetscMonitorFunction GetPetscFunction() override { return UpdateProbes; }
};
 
}  // namespace ablate::monitors
 
#endif  // ABLATELIBRARY_PROBES_HPP