DataTweaker.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef LArSamples_DataTweaker_H
#define LArSamples_DataTweaker_H
 
#include <map>
#include "LArSamplesMon/Chi2Calc.h"
#include "CxxUtils/checker_macros.h"
 
namespace LArSamples {
  
  class Data;
  class EventData;
  class ScaledErrorData;
  
  class ATLAS_NOT_THREAD_SAFE DataTweaker  {
  
   public:
   
    DataTweaker() : 
      m_nSamplesTruncation(0), m_removeNoiseCorrelations(false),
      m_refit(false), m_adjust(false), m_fitParams(DefaultChi2) { }
    
    bool set(const TString& tweaks);
    bool findOption(TString& tweaks, const TString& option) const;
 
    Data* tweak(const Data& data, int evtIndex = -1) const;
    EventData* tweak(const EventData& eventData, int runIndex = -1) const;
 
    void setNSamplesTrunctation(unsigned int n) { m_nSamplesTruncation = n; }
    void addSample(unsigned int n) { m_samples[n] = true; }
    void setRemoveNoiseCorrelations(bool remove = true) { m_removeNoiseCorrelations = remove; }
    void setRefit(bool refit = true) { m_refit = refit; }
    void setFitParams(Chi2Params params) { m_fitParams = params; }
    void setAdjust(bool adjust = true) { m_adjust = adjust; }
    void setRemoveRoIs(bool removeRoIs = true) { m_removeRoIs = removeRoIs; }
 
  private:
   
    unsigned int m_nSamplesTruncation;
    bool m_removeNoiseCorrelations, m_refit, m_adjust, m_removeRoIs;
    std::map<unsigned int, bool> m_samples; //automatically ordered... 
    Chi2Params m_fitParams;
  };
}
#endif