TagProbeFitter.h

Go to the documentation of this file.

#ifndef TagProbeFitter_h
#define TagProbeFitter_h

#include "TFile.h"
#include "TChain.h"
#include "TGraphAsymmErrors.h"
#include "RooWorkspace.h"
#include "RooFitResult.h"
#include "RooDataSet.h"

class TagProbeFitter {
  public:
  TagProbeFitter(const std::vector<std::string>& inputFileNames, std::string inputDirectoryName, std::string inputTreeName, std::string outputFileName, int numCPU = 1, bool saveWorkspace_ = false, bool floatShapeParameters = true, const std::vector<std::string>& fixVars_ = std::vector<std::string>() );

  ~TagProbeFitter();

  bool addVariable(std::string variableName, std::string title, double low, double hi, std::string units);

  bool addCategory(std::string categoryName, std::string title, std::string expression);

  bool addExpression(std::string expressionName, std::string title, std::string expression, const std::vector<std::string>& arguments);

  bool addThresholdCategory(std::string categoryName, std::string title, std::string varName, double cutValue);

  void addPdf(std::string pdfName, std::vector<std::string>& pdfCommands);

  void addFixedVariavles(const std::vector<std::string>&);

  std::string calculateEfficiency(std::string dirName, std::string efficiencyCategory, std::string efficiencyState, std::vector<std::string>& unbinnedVariables, std::map<std::string, std::vector<double> >& binnedReals, std::map<std::string, std::vector<std::string> >& binnedCategories, std::vector<std::string>& binToPDFmap) {
    std::vector<std::string> efficiencyCategories(1,efficiencyCategory);
    std::vector<std::string> efficiencyStates(1,efficiencyState);
    return calculateEfficiency(dirName, efficiencyCategories, efficiencyStates, unbinnedVariables, binnedReals, binnedCategories, binToPDFmap);
  }

  std::string calculateEfficiency(std::string dirName,const std::vector<std::string>& efficiencyCategories, const std::vector<std::string>& efficiencyStates, std::vector<std::string>& unbinnedVariables, std::map<std::string, std::vector<double> >& binnedReals, std::map<std::string, std::vector<std::string> >& binnedCategories, std::vector<std::string>& binToPDFmap);

  void setBinnedFit(bool binned, int bins=0) { binnedFit = binned; massBins = bins; }

  void setBinsForMassPlots(int bins) ;

  void setSaveDistributionsPlot(bool saveDistributionsPlot_) { doSaveDistributionsPlot = saveDistributionsPlot_; }

  void setWeightVar(const std::string &weight);

  void setQuiet(bool quiet_=true);

  protected:
  TChain* inputTree;

  TFile* outputFile;

  TDirectory* outputDirectory;

  int numCPU;

  bool doSaveDistributionsPlot;

  bool saveWorkspace;

  int binnedFit;

  int massBins;

  std::map<std::string, std::vector<std::string> > pdfs;

  RooArgSet variables;

  std::string weightVar;

  //RooArgSet expressionVars;
  std::vector<std::pair<std::pair<std::string,std::string>, std::pair<std::string, std::vector<std::string> > > > expressionVars;

  // Threshold categories have to be created at the last minute
  // so we store just the info about them
  std::vector<std::pair<std::pair<std::string,std::string>, std::pair<std::string, double> > > thresholdCategories;

  std::vector<std::string> fixVars;
  std::vector<double> fixVarValues;

  bool floatShapeParameters;

  RooWorkspace parameterParser;

  bool quiet;

  void varFixer(RooWorkspace* w, bool fix);
  void varSaver(RooWorkspace* w);
  void varRestorer(RooWorkspace* w);

  void doFitEfficiency(RooWorkspace* w, std::string pdfName, RooRealVar& efficiency);

  void doSBSEfficiency(RooWorkspace* w, RooRealVar& efficiency);

  void doCntEfficiency(RooWorkspace* w, RooRealVar& efficiency);

  void createPdf(RooWorkspace* w, std::vector<std::string>& pdfCommands);

  void setInitialValues(RooWorkspace* w);

  void saveFitPlot(RooWorkspace* w);

  void saveDistributionsPlot(RooWorkspace* w);

  void saveEfficiencyPlots(RooDataSet& eff, const TString& effName, RooArgSet& binnedVariables, RooArgSet& mappedCategories);
  
  void makeEfficiencyPlot1D(RooDataSet& eff, RooRealVar& v, const TString& plotName, const TString& plotTitle, const TString& effName, const char *catName = 0, int catIndex = -1);
   
  void makeEfficiencyPlot2D(RooDataSet& eff, RooRealVar& v1, RooRealVar& v2, const TString& plotName, const TString& plotTitle, const TString& effName, const char *catName = 0, int catIndex = -1);
  
};

#endif //TagProbeFitter_h