FitManager.h

Go to the documentation of this file.

/*
    This software is Copyright by the Board of Trustees of Michigan
    State University (c) Copyright 2017.
 
    You may use this software under the terms of the GNU public license
    (GPL).  The terms of this license are described at:
 
     http://www.gnu.org/licenses/gpl.txt
 
     Authors:
             Aaron Chester
         FRIB
         Michigan State University
         East Lansing, MI 48824-1321
*/
 
#ifndef FITMANAGER_H
#define FITMANAGER_H
 
#include <vector>
#include <cstdint>
#include <string>
 
class QWidget;
 
namespace ddastoys {    
    struct HitExtension;
    class DDASFitHit;    
    class Configuration;
}
 
// Overkill probably but:
//   1) I don't want to remember flag values
//   2) Adding more fitting methods by name may be nice
//   3) I felt like making an enum
 
enum fitMethod {
    ANALYTIC,   
    TEMPLATE,   
    ML_INFERENCE 
};
 
class FitManager
{
public:
    FitManager();
    ~FitManager();
 
    void configure(std::string method);
    void readConfigFile();
    void readTemplateFile();
    std::vector<double> getSinglePulseFit(
    const ddastoys::HitExtension& ext, unsigned low, unsigned high
    );
    std::vector<double> getDoublePulseFit(
    const ddastoys::HitExtension& ext, unsigned low, unsigned high
    );
    unsigned getLowFitLimit(const ddastoys::DDASFitHit& hit);
    unsigned getHighFitLimit(const ddastoys::DDASFitHit& hit);
 
    void setMethod(fitMethod m) { m_method = m; };
    enum fitMethod getMethod() { return m_method; };
  
private:
    double singlePulse(
    double A1, double k1, double k2, double x1, double C, double x
    );
    double doublePulse(
    double A1, double k1, double k2, double x1,
    double A2, double k3, double k4, double x2,
    double C, double x
    );
    bool checkParamValue(double param);
    void issueWarning(std::string msg);
  
private:
    ddastoys::Configuration* m_pConfig; //<! Configuration file manager.
    fitMethod m_method;       
    bool m_config;            
    bool m_templateConfig;    
};
 
#endif