MHO_FringePass.hh

Go to the documentation of this file.

#ifndef MHO_FringePass_HH__
#define MHO_FringePass_HH__
 
#include <functional>
#include <memory>
#include <string>
#include <vector>
 
#include "MHO_Profiler.hh"
 
#include "MHO_FringeControlInitialization.hh"
#include "MHO_FringeData.hh"
#include "MHO_FringeDataInitializer.hh"
#include "MHO_FringeFitter.hh"
#include "MHO_FringeFitterFactory.hh"
#include "MHO_FringePlotVisitor.hh"
#include "MHO_ParameterStore.hh"
 
namespace hops
{
 
class MHO_FringePass
{
    public:
        using ControlEvaluatorFn = std::function< bool(MHO_ParameterStore*, const mho_json&, mho_json&) >;
 
        MHO_FringePass();
        virtual ~MHO_FringePass() = default;
 
        // -----------------------------------------------------------------------
        // Input setters
        // Call these before Initialize().
        // -----------------------------------------------------------------------
 
        void CopyCommandLineParams(const MHO_ParameterStore& cmdline_params);
 
        void SetScanDirectory(const std::string& dir);
        void SetBaseline(const std::string& baseline);
        void SetPolProduct(const std::string& polprod);
        void SetFrequencyGroup(const std::string& fgroup);
        void SetScanName(const std::string& scan);
        void SetRootFile(const std::string& root_file);
        void SetBuildTimestamp(const std::string& ts);
 
        void SetControlFile(const std::string& path);
 
        void SetPythonControlEvaluator(ControlEvaluatorFn fn);
 
        // -----------------------------------------------------------------------
        // Lifecycle
        // -----------------------------------------------------------------------
 
        bool Initialize();
 
        bool Configure();
 
        bool Run(const std::vector< MHO_FringeFitterVisitor* >& plugin_visitors = {},
                 const std::vector< MHO_FringeFitterVisitor* >& output_visitors = {},
                 const std::vector< MHO_FringePlotVisitor* >& plot_visitors = {});
 
        // -----------------------------------------------------------------------
        // Status and results
        // -----------------------------------------------------------------------
 
        bool IsSkipped();
 
        void FlushProfileEvents();
 
        MHO_FringeData* GetFringeData() { return &fData; }
 
    private:
        MHO_FringeData fData;
        std::unique_ptr< MHO_FringeFitterFactory > fFactory; // owns the fitter
 
        std::string fControlFileOverride; // empty -> use /cmdline/control_file
        ControlEvaluatorFn fPythonEvaluator;
};
 
} // namespace hops
 
#endif