WalkCorrector.hpp

Go to the documentation of this file.

#ifndef WALKCORRECTOR_H
#define WALKCORRECTOR_H

#include <vector>
#include <map>
#include <string>

#include "Identifier.hpp"

enum WalkModel {
    walk_none,
    walk_A,
    walk_B1,
    walk_B2
};

struct CorrectionParams {
    WalkModel model; 
    double min; 
    double max;
    std::vector<double> parameters;
};

class WalkCorrector {
public:
    WalkCorrector() {};

    void AddChannel(const Identifier& chanID,
                    const std::string model,
                    double min, double max,
                    const std::vector<double>& par);

    double GetCorrection(Identifier& chanID, double raw) const;

private:
    std::map<Identifier, std::vector<CorrectionParams> > channels_;

    double Model_None() const;

    double Model_A(const std::vector<double>& par, double raw) const;

    double Model_B1(const std::vector<double>& par, double raw) const;

    double Model_B2(const std::vector<double>& par, double raw) const;
};
#endif