AnaTools/plugins/OSUAnalysis.cc

#include "OSUT3Analysis/AnaTools/plugins/OSUAnalysis.h"

OSUAnalysis::OSUAnalysis (const edm::ParameterSet &cfg) :
  // Retrieve parameters from the configuration file.
  jets_ (cfg.getParameter<edm::InputTag> ("jets")),
  muons_ (cfg.getParameter<edm::InputTag> ("muons")),
  electrons_ (cfg.getParameter<edm::InputTag> ("electrons")),
  events_ (cfg.getParameter<edm::InputTag> ("events")),
  taus_ (cfg.getParameter<edm::InputTag> ("taus")),
  mets_ (cfg.getParameter<edm::InputTag> ("mets")),
  tracks_ (cfg.getParameter<edm::InputTag> ("tracks")),
  genjets_ (cfg.getParameter<edm::InputTag> ("genjets")),
  mcparticles_ (cfg.getParameter<edm::InputTag> ("mcparticles")),
  primaryvertexs_ (cfg.getParameter<edm::InputTag> ("primaryvertexs")),
  bxlumis_ (cfg.getParameter<edm::InputTag> ("bxlumis")),
  photons_ (cfg.getParameter<edm::InputTag> ("photons")),
  superclusters_ (cfg.getParameter<edm::InputTag> ("superclusters")),
  triggers_ (cfg.getParameter<edm::InputTag> ("triggers")),


  channels_  (cfg.getParameter<vector<edm::ParameterSet> >("channels")),

  histogramSets_ (cfg.getParameter<vector<edm::ParameterSet> >("histogramSets"))

{


  TH1::SetDefaultSumw2 ();
  TH2::SetDefaultSumw2 ();

  // Construct Cutflow Objects. These store the results of cut decisions and
  // handle filling cut flow histograms.
  masterCutFlow_ = new CutFlow (fs_);
  std::vector<TFileDirectory> directories;

  //always get vertex collection so we can assign the primary vertex in the event
  allNecessaryObjects.push_back("primaryvertexs");


  //parse the histogram definitions
  for(uint currentHistogramSet = 0; currentHistogramSet != histogramSets_.size(); currentHistogramSet++){

    string tempInputCollection = histogramSets_.at(currentHistogramSet).getParameter<string>("inputCollection");
    objectsToPlot.push_back(tempInputCollection);
    allNecessaryObjects.push_back(tempInputCollection);
    vector<edm::ParameterSet> histogramList_  (histogramSets_.at(currentHistogramSet).getParameter<vector<edm::ParameterSet> >("histograms"));
    
    for(uint currentHistogram = 0; currentHistogram != histogramList_.size(); currentHistogram++){

      histogram tempHistogram;
      tempHistogram.inputCollection = tempInputCollection;
      tempHistogram.name = histogramList_.at(currentHistogram).getParameter<string>("name");
      tempHistogram.title = histogramList_.at(currentHistogram).getParameter<string>("title");
      tempHistogram.bins = histogramList_.at(currentHistogram).getParameter<vector<double> >("bins");
      tempHistogram.inputVariables = histogramList_.at(currentHistogram).getParameter<vector<string> >("inputVariables");
      if(histogramList_.at(currentHistogram).exists("function"))
        tempHistogram.function = histogramList_.at(currentHistogram).getParameter<string>("function");
      else
        tempHistogram.function = "";
      
      histograms.push_back(tempHistogram);

    }
  }


  channel tempChannel;
  //loop over all channels (event selections)
  for(uint currentChannel = 0; currentChannel != channels_.size(); currentChannel++){

    //get name of channel
    string channelName  (channels_.at(currentChannel).getParameter<string>("name"));
    tempChannel.name = channelName;
    TString channelLabel = channelName;

    //set triggers for this channel
    vector<string> triggerNames;
    triggerNames.clear();
    tempChannel.triggers.clear();
    if(channels_.at(currentChannel).exists("triggers")){
      triggerNames   = channels_.at(currentChannel).getParameter<vector<string> >("triggers");
      for(uint trigger = 0; trigger!= triggerNames.size(); trigger++)
        tempChannel.triggers.push_back(triggerNames.at(trigger));
      allNecessaryObjects.push_back("triggers");
    }


    //create cutFlow for this channel
    cutFlows_.push_back (new CutFlow (fs_, channelName));

    //book a directory in the output file with the name of the channel
    TFileDirectory subDir = fs_->mkdir( channelName );
    directories.push_back(subDir);

    std::map<std::string, TH1D*> oneDhistoMap;
    oneDHists_.push_back(oneDhistoMap);
    std::map<std::string, TH2D*> twoDhistoMap;
    twoDHists_.push_back(twoDhistoMap);

    //book all histograms included in the configuration
    for(uint currentHistogramIndex = 0; currentHistogramIndex != histograms.size(); currentHistogramIndex++){
      histogram currentHistogram = histograms.at(currentHistogramIndex);
      int numBinsElements = currentHistogram.bins.size();
      int numInputVariables = currentHistogram.inputVariables.size();
      if(numBinsElements != 3 && numBinsElements !=6) {
        std::cout << "Error: Didn't find correct number of bin specifications for histogram named '" << currentHistogram.name << "'\n";
        exit(0);
      }
      else if((numBinsElements == 3 && numInputVariables !=1) || (numBinsElements == 6 && numInputVariables!=2)){
        std::cout << "Error: Didn't find correct number of input variables for histogram named '" << currentHistogram.name << "'\n";
        exit(0);
      }
      else if(numBinsElements == 3)
        oneDHists_.at(currentChannel)[currentHistogram.name] = directories.at(currentChannel).make<TH1D> (TString(currentHistogram.name),channelLabel+" channel: "+currentHistogram.title, currentHistogram.bins.at(0), currentHistogram.bins.at(1), currentHistogram.bins.at(2));
      else if(numBinsElements == 6)
        twoDHists_.at(currentChannel)[currentHistogram.name] = directories.at(currentChannel).make<TH2D> (TString(currentHistogram.name),channelLabel+" channel: "+currentHistogram.title, currentHistogram.bins.at(0), currentHistogram.bins.at(1), currentHistogram.bins.at(2),currentHistogram.bins.at(3),currentHistogram.bins.at(4),currentHistogram.bins.at(5));


    }
    //book a histogram for the number of each object type to be plotted
    for (uint currentObjectIndex = 0; currentObjectIndex != objectsToPlot.size(); currentObjectIndex++){
      string currentObject = objectsToPlot.at(currentObjectIndex);
      int maxNum = 10;
      if(currentObject == "mcparticles") maxNum = 50;
      currentObject[0] = toupper(currentObject[0]);
      string histoName = "num" + currentObject;
      oneDHists_.at(currentChannel)[histoName] = directories.at(currentChannel).make<TH1D> (TString(histoName),channelLabel+" channel: Number of Selected "+currentObject+"; # "+currentObject, maxNum, 0, maxNum);
    }


    //get list of cuts for this channel
    vector<edm::ParameterSet> cuts_  (channels_.at(currentChannel).getParameter<vector<edm::ParameterSet> >("cuts"));

    //loop over and parse all cuts
    for(uint currentCut = 0; currentCut != cuts_.size(); currentCut++){

      cut tempCut;
     //store input collection for cut
      string inputCollection = cuts_.at(currentCut).getParameter<string> ("inputCollection");
      tempCut.inputCollection = inputCollection;
      allNecessaryObjects.push_back(inputCollection);

      //split cut string into parts and store them
      string cutString = cuts_.at(currentCut).getParameter<string> ("cutString");
      std::vector<string> cutStringVector = splitString(cutString);
      if(cutStringVector.size()!=3){
        std::cout << "Error: Didn't find three elements in the following cut string: '" <<cutString << "'\n";
        exit(0);
      }

      tempCut.variable = cutStringVector.at(0);// variable to cut on
      tempCut.comparativeOperator = cutStringVector.at(1);// comparison to make
      tempCut.cutValue = atof(cutStringVector.at(2).c_str());// threshold value to pass cut

      //get number of objects required to pass cut for event to pass
      string numberRequiredString = cuts_.at(currentCut).getParameter<string> ("numberRequired");
      std::vector<string> numberRequiredVector = splitString(numberRequiredString);
      if(numberRequiredVector.size()!=2){
        std::cout << "Error: Didn't find two elements in the following number requirement string: '" << numberRequiredString << "'\n";
        exit(0);
      }

      // determine number required if comparison contains "="
      int numberRequiredInt = atoi(numberRequiredVector.at(1).c_str());
      if(numberRequiredVector.at(0) == ">") numberRequiredInt++;
      else if(numberRequiredVector.at(0) == "<") numberRequiredInt--;

      tempCut.numberRequired = numberRequiredInt;// number of objects required to pass the cut
      tempCut.eventComparativeOperator = numberRequiredVector.at(0);// comparison to make

      if(cuts_.at(currentCut).exists("function")){
        tempCut.function = cuts_.at(currentCut).getParameter<string> ("function");
      }
      else tempCut.function = "";
      string tempCutName;
      if(cuts_.at(currentCut).exists("alias")){
        tempCutName = cuts_.at(currentCut).getParameter<string> ("alias");
      }
      else{
        //construct string for cutflow table
        bool plural = numberRequiredInt != 1;
        string collectionString = plural ? inputCollection : inputCollection.substr(0, inputCollection.size()-1);
        string cutName =  numberRequiredString + " " + collectionString + " with " + cutString;
        tempCutName = cutName;
      }
      tempCut.name = tempCutName;


      tempChannel.cuts.push_back(tempCut);


    }//end loop over cuts


    channels.push_back(tempChannel);
    tempChannel.cuts.clear();

  }//end loop over channels


  //make unique vector of objects we need to cut on (so we make sure to get them from the event)
  sort( allNecessaryObjects.begin(), allNecessaryObjects.end() );
  allNecessaryObjects.erase( unique( allNecessaryObjects.begin(), allNecessaryObjects.end() ), allNecessaryObjects.end() );


}

OSUAnalysis::~OSUAnalysis ()
{
  // Destroying the CutFlow objects causes the cut flow numbers and time
  // information to be printed to standard output.
  for(uint currentChannel = 0; currentChannel != channels_.size(); currentChannel++){
    delete cutFlows_.at(currentChannel);
  }
}

void
OSUAnalysis::analyze (const edm::Event &event, const edm::EventSetup &setup)
{


  // Retrieve necessary collections from the event.
  edm::Handle<BNtriggerCollection> triggers;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "triggers") != allNecessaryObjects.end())
    event.getByLabel (triggers_, triggers);

  edm::Handle<BNjetCollection> jets;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "jets") != allNecessaryObjects.end())
    event.getByLabel (jets_, jets);

  edm::Handle<BNmuonCollection> muons;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "muons") != allNecessaryObjects.end())
    event.getByLabel (muons_, muons);

  edm::Handle<BNelectronCollection> electrons;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "electrons") != allNecessaryObjects.end())
    event.getByLabel (electrons_, electrons);

  edm::Handle<BNeventCollection> events;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "events") != allNecessaryObjects.end())
    event.getByLabel (events_, events);

  edm::Handle<BNtauCollection> taus;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "taus") != allNecessaryObjects.end())
    event.getByLabel (taus_, taus);

  edm::Handle<BNmetCollection> mets;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "mets") != allNecessaryObjects.end())
    event.getByLabel (mets_, mets);

  edm::Handle<BNtrackCollection> tracks;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "tracks") != allNecessaryObjects.end())
    event.getByLabel (tracks_, tracks);

  edm::Handle<BNgenjetCollection> genjets;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "genjets") != allNecessaryObjects.end())
    event.getByLabel (genjets_, genjets);

  edm::Handle<BNmcparticleCollection> mcparticles;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "mcparticles") != allNecessaryObjects.end())
    event.getByLabel (mcparticles_, mcparticles);

  edm::Handle<BNprimaryvertexCollection> primaryvertexs;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "primaryvertexs") != allNecessaryObjects.end())
    event.getByLabel (primaryvertexs_, primaryvertexs);

  edm::Handle<BNbxlumiCollection> bxlumis;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "bxlumis") != allNecessaryObjects.end())
    event.getByLabel (bxlumis_, bxlumis);

  edm::Handle<BNphotonCollection> photons;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "photons") != allNecessaryObjects.end())
    event.getByLabel (photons_, photons);

  edm::Handle<BNsuperclusterCollection> superclusters;
  if (std::find(allNecessaryObjects.begin(), allNecessaryObjects.end(), "superclusters") != allNecessaryObjects.end())
    event.getByLabel (superclusters_, superclusters);


  //loop over all channels

  for(uint currentChannelIndex = 0; currentChannelIndex != channels.size(); currentChannelIndex++){
    channel currentChannel = channels.at(currentChannelIndex);

    flagMap individualFlags;
    flagMap cumulativeFlags;
    counterMap passingCounter;

    bool triggerDecision = true;
    if(currentChannel.triggers.size() != 0){  //triggers specified
      triggerDecision = evaluateTriggers(currentChannel.triggers,triggers.product());
      cutFlows_.at(currentChannelIndex)->at ("trigger") = triggerDecision;
    }

    //loop over all cuts
    for(uint currentCutIndex = 0; currentCutIndex != currentChannel.cuts.size(); currentCutIndex++){
      cut currentCut = currentChannel.cuts.at(currentCutIndex);

      for(uint currentObjectIndex = 0; currentObjectIndex != allNecessaryObjects.size(); currentObjectIndex++){

        string currentObject = allNecessaryObjects.at(currentObjectIndex);
        individualFlags[currentObject].push_back (vector<bool> ());
        cumulativeFlags[currentObject].push_back (vector<bool> ());

        if(currentObject == "jets") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,jets.product(),"jets");
        else if(currentObject == "muons") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,muons.product(),"muons");
        else if(currentObject == "electrons") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,electrons.product(),"electrons");
        else if(currentObject == "events") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,events.product(),"events");
        else if(currentObject == "taus") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,taus.product(),"taus");
        else if(currentObject == "mets") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,mets.product(),"mets");
        else if(currentObject == "tracks") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,tracks.product(),"tracks");
        else if(currentObject == "genjets") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,genjets.product(),"genjets");
        else if(currentObject == "mcparticles") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,mcparticles.product(),"mcparticles");
        else if(currentObject == "primaryvertexs") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,primaryvertexs.product(),"primaryvertexs");
        else if(currentObject == "bxlumis") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,bxlumis.product(),"bxlumis");
        else if(currentObject == "photons") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,photons.product(),"photons");
        else if(currentObject == "superclusters") setObjectFlags(currentCut,currentCutIndex,individualFlags,cumulativeFlags,superclusters.product(),"superclusters");


      }


    }//end loop over all cuts


    //use cumulative flags to apply cuts at event level

    bool eventPassedAllCuts = true;

    //apply trigger (true if none were specified)
    eventPassedAllCuts = eventPassedAllCuts && triggerDecision;


    for(uint currentCutIndex = 0; currentCutIndex != currentChannel.cuts.size(); currentCutIndex++){

      //loop over all objects and count how many passed the cumulative selection up to this point
      cut currentCut = currentChannel.cuts.at(currentCutIndex);
      int numberPassing = 0;

      for (uint object = 0; object != cumulativeFlags[currentCut.inputCollection].at(currentCutIndex).size() ; object++)
          if(cumulativeFlags[currentCut.inputCollection].at(currentCutIndex).at(object)) numberPassing++;


      bool cutDecision = evaluateComparison(numberPassing,currentCut.eventComparativeOperator,currentCut.numberRequired);
      cutFlows_.at(currentChannelIndex)->at (currentCut.name) = cutDecision;

      eventPassedAllCuts = eventPassedAllCuts && cutDecision;

    }

    cutFlows_.at(currentChannelIndex)->fillCutFlow();


    if(!eventPassedAllCuts)continue;


    //set position of primary vertex in event, in order to calculate quantities relative to it
    vector<bool> vertexFlags = cumulativeFlags["primaryvertexs"].back();
    for (uint vertexIndex = 0; vertexIndex != vertexFlags.size(); vertexIndex++){
      if(!vertexFlags.at(vertexIndex)) continue;
      vertexPosition.SetXYZ (primaryvertexs->at(vertexIndex).x,primaryvertexs->at(vertexIndex).y,primaryvertexs->at(vertexIndex).z);
      break;
    }


    //filling histograms
    for (uint histogramIndex = 0; histogramIndex != histograms.size(); histogramIndex++){
      histogram currentHistogram = histograms.at(histogramIndex);

      if(currentHistogram.inputVariables.size() == 1){
        TH1D* histo;
        histo = oneDHists_.at(currentChannelIndex).at(currentHistogram.name);
        if(currentHistogram.inputCollection == "jets") fillHistogram(histo,currentHistogram,jets.product());
        else if(currentHistogram.inputCollection == "muons") fillHistogram(histo,currentHistogram,muons.product());
        else if(currentHistogram.inputCollection == "electrons") fillHistogram(histo,currentHistogram,electrons.product());
        else if(currentHistogram.inputCollection == "events") fillHistogram(histo,currentHistogram,events.product());
        else if(currentHistogram.inputCollection == "taus") fillHistogram(histo,currentHistogram,taus.product());
        else if(currentHistogram.inputCollection == "mets") fillHistogram(histo,currentHistogram,mets.product());
        else if(currentHistogram.inputCollection == "tracks") fillHistogram(histo,currentHistogram,tracks.product());
        else if(currentHistogram.inputCollection == "genjets") fillHistogram(histo,currentHistogram,genjets.product());
        else if(currentHistogram.inputCollection == "mcparticles") fillHistogram(histo,currentHistogram,mcparticles.product());
        else if(currentHistogram.inputCollection == "primaryvertexs") fillHistogram(histo,currentHistogram,primaryvertexs.product());
        else if(currentHistogram.inputCollection == "bxlumis") fillHistogram(histo,currentHistogram,bxlumis.product());
        else if(currentHistogram.inputCollection == "photons") fillHistogram(histo,currentHistogram,photons.product());
        else if(currentHistogram.inputCollection == "superclusters") fillHistogram(histo,currentHistogram,superclusters.product());
      }
      else if(currentHistogram.inputVariables.size() == 2){
        TH2D* histo;
        histo = twoDHists_.at(currentChannelIndex).at(currentHistogram.name);
        if(currentHistogram.inputCollection == "jets") fillHistogram(histo,currentHistogram,jets.product());
        else if(currentHistogram.inputCollection == "muons") fillHistogram(histo,currentHistogram,muons.product());
        else if(currentHistogram.inputCollection == "electrons") fillHistogram(histo,currentHistogram,electrons.product());
        else if(currentHistogram.inputCollection == "events") fillHistogram(histo,currentHistogram,events.product());
        else if(currentHistogram.inputCollection == "taus") fillHistogram(histo,currentHistogram,taus.product());
        else if(currentHistogram.inputCollection == "mets") fillHistogram(histo,currentHistogram,mets.product());
        else if(currentHistogram.inputCollection == "tracks") fillHistogram(histo,currentHistogram,tracks.product());
        else if(currentHistogram.inputCollection == "genjets") fillHistogram(histo,currentHistogram,genjets.product());
        else if(currentHistogram.inputCollection == "mcparticles") fillHistogram(histo,currentHistogram,mcparticles.product());
        else if(currentHistogram.inputCollection == "primaryvertexs") fillHistogram(histo,currentHistogram,primaryvertexs.product());
        else if(currentHistogram.inputCollection == "bxlumis") fillHistogram(histo,currentHistogram,bxlumis.product());
        else if(currentHistogram.inputCollection == "photons") fillHistogram(histo,currentHistogram,photons.product());
        else if(currentHistogram.inputCollection == "superclusters") fillHistogram(histo,currentHistogram,superclusters.product());
      } 
    }


    //fills histograms with the sizes of collections
    for (uint currentObjectIndex = 0; currentObjectIndex != objectsToPlot.size(); currentObjectIndex++){
      string currentObject = objectsToPlot.at(currentObjectIndex);
      string tempCurrentObject = currentObject;
      tempCurrentObject[0] = toupper(tempCurrentObject[0]);
      string histoName = "num" + tempCurrentObject;

      if(currentObject == "jets") oneDHists_.at(currentChannelIndex)[histoName]->Fill(jets->size());
      else if(currentObject == "muons") oneDHists_.at(currentChannelIndex)[histoName]->Fill(muons->size());
      else if(currentObject == "electrons") oneDHists_.at(currentChannelIndex)[histoName]->Fill(electrons->size());
      else if(currentObject == "events") oneDHists_.at(currentChannelIndex)[histoName]->Fill(events->size());
      else if(currentObject == "taus") oneDHists_.at(currentChannelIndex)[histoName]->Fill(taus->size());
      else if(currentObject == "mets") oneDHists_.at(currentChannelIndex)[histoName]->Fill(mets->size());
      else if(currentObject == "tracks") oneDHists_.at(currentChannelIndex)[histoName]->Fill(tracks->size());
      else if(currentObject == "genjets") oneDHists_.at(currentChannelIndex)[histoName]->Fill(genjets->size());
      else if(currentObject == "mcparticles") oneDHists_.at(currentChannelIndex)[histoName]->Fill(mcparticles->size());
      else if(currentObject == "primaryvertexs") oneDHists_.at(currentChannelIndex)[histoName]->Fill(primaryvertexs->size());
      else if(currentObject == "bxlumis") oneDHists_.at(currentChannelIndex)[histoName]->Fill(bxlumis->size());
      else if(currentObject == "photons") oneDHists_.at(currentChannelIndex)[histoName]->Fill(photons->size());
      else if(currentObject == "superclusters") oneDHists_.at(currentChannelIndex)[histoName]->Fill(superclusters->size());

    }
  } //end loop over channel

  masterCutFlow_->fillCutFlow();


}


bool
OSUAnalysis::evaluateComparison (double testValue, string comparison, double cutValue){


  if(comparison == ">")       return testValue >  cutValue;
  else if(comparison == ">=") return testValue >= cutValue;
  else if(comparison == "<")  return testValue <  cutValue;
  else if(comparison == "<=") return testValue <= cutValue;
  else if(comparison == "==") return testValue == cutValue;
  else if(comparison == "!=") return testValue != cutValue;
  else {std::cout << "WARNING: invalid comparison operator '" << comparison << "'\n"; return false;}

}

bool
OSUAnalysis::evaluateTriggers (vector<string> triggersToTest, const BNtriggerCollection* triggerCollection){

  bool triggerDecision = false;

  for (BNtriggerCollection::const_iterator trigger = triggerCollection->begin (); trigger != triggerCollection->end (); trigger++)
    {
      if(trigger->pass != 1) continue;
      for(uint triggerName = 0; triggerName != triggersToTest.size(); triggerName++)
        {
          if(trigger->name.find(triggersToTest.at(triggerName))!=std::string::npos){
            triggerDecision = true;
          }
        }
    }
  return triggerDecision;

}

std::vector<std::string>
OSUAnalysis::splitString (string inputString){

  std::stringstream stringStream(inputString);
  std::istream_iterator<std::string> begin(stringStream);
  std::istream_iterator<std::string> end;
  std::vector<std::string> stringVector(begin, end);
  return stringVector;

}

double
OSUAnalysis::valueLookup (const BNjet* object, string variable, string function){

  double value = 0.0;
  if(variable == "energy") value = object->energy;
  else if(variable == "et") value = object->et;
  else if(variable == "pt") value = object->pt;
  else if(variable == "px") value = object->px;
  else if(variable == "py") value = object->py;
  else if(variable == "pz") value = object->pz;
  else if(variable == "phi") value = object->phi;
  else if(variable == "eta") value = object->eta;
  else if(variable == "theta") value = object->theta;
  else if(variable == "Upt") value = object->Upt;
  else if(variable == "Uenergy") value = object->Uenergy;
  else if(variable == "L2pt") value = object->L2pt;
  else if(variable == "L2L3pt") value = object->L2L3pt;
  else if(variable == "L2L3respt") value = object->L2L3respt;
  else if(variable == "respt") value = object->respt;
  else if(variable == "EMfrac") value = object->EMfrac;
  else if(variable == "Hadfrac") value = object->Hadfrac;
  else if(variable == "charge") value = object->charge;
  else if(variable == "mass") value = object->mass;
  else if(variable == "area") value = object->area;
  else if(variable == "fHPD") value = object->fHPD;
  else if(variable == "approximatefHPD") value = object->approximatefHPD;
  else if(variable == "genPartonET") value = object->genPartonET;
  else if(variable == "genPartonPT") value = object->genPartonPT;
  else if(variable == "genPartonEta") value = object->genPartonEta;
  else if(variable == "genPartonPhi") value = object->genPartonPhi;
  else if(variable == "genJetET") value = object->genJetET;
  else if(variable == "genJetPT") value = object->genJetPT;
  else if(variable == "genJetEta") value = object->genJetEta;
  else if(variable == "genJetPhi") value = object->genJetPhi;
  else if(variable == "btagTChighPur") value = object->btagTChighPur;
  else if(variable == "btagTChighEff") value = object->btagTChighEff;
  else if(variable == "btagJetProb") value = object->btagJetProb;
  else if(variable == "btagJetBProb") value = object->btagJetBProb;
  else if(variable == "btagSoftEle") value = object->btagSoftEle;
  else if(variable == "btagSoftMuon") value = object->btagSoftMuon;
  else if(variable == "btagSoftMuonNoIP") value = object->btagSoftMuonNoIP;
  else if(variable == "btagSecVertex") value = object->btagSecVertex;
  else if(variable == "btagSecVertexHighEff") value = object->btagSecVertexHighEff;
  else if(variable == "btagSecVertexHighPur") value = object->btagSecVertexHighPur;
  else if(variable == "btagCombinedSecVertex") value = object->btagCombinedSecVertex;
  else if(variable == "btagCombinedSecVertexMVA") value = object->btagCombinedSecVertexMVA;
  else if(variable == "btagSoftMuonByPt") value = object->btagSoftMuonByPt;
  else if(variable == "btagSoftMuonByIP3") value = object->btagSoftMuonByIP3;
  else if(variable == "btagSoftElectronByPt") value = object->btagSoftElectronByPt;
  else if(variable == "btagSoftElectronByIP3") value = object->btagSoftElectronByIP3;
  else if(variable == "n90Hits") value = object->n90Hits;
  else if(variable == "hitsInN90") value = object->hitsInN90;
  else if(variable == "chargedHadronEnergyFraction") value = object->chargedHadronEnergyFraction;
  else if(variable == "neutralHadronEnergyFraction") value = object->neutralHadronEnergyFraction;
  else if(variable == "chargedEmEnergyFraction") value = object->chargedEmEnergyFraction;
  else if(variable == "neutralEmEnergyFraction") value = object->neutralEmEnergyFraction;
  else if(variable == "fLong") value = object->fLong;
  else if(variable == "fShort") value = object->fShort;
  else if(variable == "etaetaMoment") value = object->etaetaMoment;
  else if(variable == "phiphiMoment") value = object->phiphiMoment;
  else if(variable == "JESunc") value = object->JESunc;
  else if(variable == "JECuncUp") value = object->JECuncUp;
  else if(variable == "JECuncDown") value = object->JECuncDown;
  else if(variable == "puJetMVA_full") value = object->puJetMVA_full;
  else if(variable == "puJetMVA_simple") value = object->puJetMVA_simple;
  else if(variable == "puJetMVA_cutbased") value = object->puJetMVA_cutbased;
  else if(variable == "dZ") value = object->dZ;
  else if(variable == "dR2Mean") value = object->dR2Mean;
  else if(variable == "dRMean") value = object->dRMean;
  else if(variable == "frac01") value = object->frac01;
  else if(variable == "frac02") value = object->frac02;
  else if(variable == "frac03") value = object->frac03;
  else if(variable == "frac04") value = object->frac04;
  else if(variable == "frac05") value = object->frac05;
  else if(variable == "frac06") value = object->frac06;
  else if(variable == "frac07") value = object->frac07;
  else if(variable == "beta") value = object->beta;
  else if(variable == "betaStar") value = object->betaStar;
  else if(variable == "betaClassic") value = object->betaClassic;
  else if(variable == "betaStarClassic") value = object->betaStarClassic;
  else if(variable == "ptD") value = object->ptD;
  else if(variable == "nvtx") value = object->nvtx;
  else if(variable == "d0") value = object->d0;
  else if(variable == "leadCandPt") value = object->leadCandPt;
  else if(variable == "leadCandVx") value = object->leadCandVx;
  else if(variable == "leadCandVy") value = object->leadCandVy;
  else if(variable == "leadCandVz") value = object->leadCandVz;
  else if(variable == "leadCandDistFromPV") value = object->leadCandDistFromPV;
  else if(variable == "flavour") value = object->flavour;
  else if(variable == "Nconst") value = object->Nconst;
  else if(variable == "jetIDMinimal") value = object->jetIDMinimal;
  else if(variable == "jetIDLooseAOD") value = object->jetIDLooseAOD;
  else if(variable == "jetIDLoose") value = object->jetIDLoose;
  else if(variable == "jetIDTight") value = object->jetIDTight;
  else if(variable == "genPartonId") value = object->genPartonId;
  else if(variable == "genPartonMotherId") value = object->genPartonMotherId;
  else if(variable == "genPartonMother0Id") value = object->genPartonMother0Id;
  else if(variable == "genPartonMother1Id") value = object->genPartonMother1Id;
  else if(variable == "genPartonGrandMotherId") value = object->genPartonGrandMotherId;
  else if(variable == "genPartonGrandMother00Id") value = object->genPartonGrandMother00Id;
  else if(variable == "genPartonGrandMother01Id") value = object->genPartonGrandMother01Id;
  else if(variable == "genPartonGrandMother10Id") value = object->genPartonGrandMother10Id;
  else if(variable == "genPartonGrandMother11Id") value = object->genPartonGrandMother11Id;
  else if(variable == "chargedMultiplicity") value = object->chargedMultiplicity;
  else if(variable == "neutralMultiplicity") value = object->neutralMultiplicity;
  else if(variable == "nconstituents") value = object->nconstituents;
  else if(variable == "nHit") value = object->nHit;
  else if(variable == "puJetId_full") value = object->puJetId_full;
  else if(variable == "puJetId_simple") value = object->puJetId_simple;
  else if(variable == "puJetId_cutbased") value = object->puJetId_cutbased;
  else if(variable == "puJetId_tight_full") value = object->puJetId_tight_full;
  else if(variable == "puJetId_tight_simple") value = object->puJetId_tight_simple;
  else if(variable == "puJetId_tight_cutbased") value = object->puJetId_tight_cutbased;
  else if(variable == "puJetId_medium_full") value = object->puJetId_medium_full;
  else if(variable == "puJetId_medium_simple") value = object->puJetId_medium_simple;
  else if(variable == "puJetId_medium_cutbased") value = object->puJetId_medium_cutbased;
  else if(variable == "puJetId_loose_full") value = object->puJetId_loose_full;
  else if(variable == "puJetId_loose_simple") value = object->puJetId_loose_simple;
  else if(variable == "puJetId_loose_cutbased") value = object->puJetId_loose_cutbased;


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}


double
OSUAnalysis::valueLookup (const BNmuon* object, string variable, string function){

  double value = 0.0;
  if(variable == "energy") value = object->energy;
  else if(variable == "et") value = object->et;
  else if(variable == "pt") value = object->pt;
  else if(variable == "px") value = object->px;
  else if(variable == "py") value = object->py;
  else if(variable == "pz") value = object->pz;
  else if(variable == "phi") value = object->phi;
  else if(variable == "eta") value = object->eta;
  else if(variable == "theta") value = object->theta;
  else if(variable == "trackIso") value = object->trackIso;
  else if(variable == "ecalIso") value = object->ecalIso;
  else if(variable == "hcalIso") value = object->hcalIso;
  else if(variable == "caloIso") value = object->caloIso;
  else if(variable == "trackIsoDR03") value = object->trackIsoDR03;
  else if(variable == "ecalIsoDR03") value = object->ecalIsoDR03;
  else if(variable == "hcalIsoDR03") value = object->hcalIsoDR03;
  else if(variable == "caloIsoDR03") value = object->caloIsoDR03;
  else if(variable == "trackVetoIsoDR03") value = object->trackVetoIsoDR03;
  else if(variable == "ecalVetoIsoDR03") value = object->ecalVetoIsoDR03;
  else if(variable == "hcalVetoIsoDR03") value = object->hcalVetoIsoDR03;
  else if(variable == "caloVetoIsoDR03") value = object->caloVetoIsoDR03;
  else if(variable == "trackIsoDR05") value = object->trackIsoDR05;
  else if(variable == "ecalIsoDR05") value = object->ecalIsoDR05;
  else if(variable == "hcalIsoDR05") value = object->hcalIsoDR05;
  else if(variable == "caloIsoDR05") value = object->caloIsoDR05;
  else if(variable == "trackVetoIsoDR05") value = object->trackVetoIsoDR05;
  else if(variable == "ecalVetoIsoDR05") value = object->ecalVetoIsoDR05;
  else if(variable == "hcalVetoIsoDR05") value = object->hcalVetoIsoDR05;
  else if(variable == "caloVetoIsoDR05") value = object->caloVetoIsoDR05;
  else if(variable == "hcalE") value = object->hcalE;
  else if(variable == "ecalE") value = object->ecalE;
  else if(variable == "genET") value = object->genET;
  else if(variable == "genPT") value = object->genPT;
  else if(variable == "genPhi") value = object->genPhi;
  else if(variable == "genEta") value = object->genEta;
  else if(variable == "genMotherET") value = object->genMotherET;
  else if(variable == "genMotherPT") value = object->genMotherPT;
  else if(variable == "genMotherPhi") value = object->genMotherPhi;
  else if(variable == "genMotherEta") value = object->genMotherEta;
  else if(variable == "vx") value = object->vx;
  else if(variable == "vy") value = object->vy;
  else if(variable == "vz") value = object->vz;
  else if(variable == "tkNormChi2") value = object->tkNormChi2;
  else if(variable == "tkPT") value = object->tkPT;
  else if(variable == "tkEta") value = object->tkEta;
  else if(variable == "tkPhi") value = object->tkPhi;
  else if(variable == "tkDZ") value = object->tkDZ;
  else if(variable == "tkD0") value = object->tkD0;
  else if(variable == "tkD0bs") value = object->tkD0bs;
  else if(variable == "tkD0err") value = object->tkD0err;
  else if(variable == "samNormChi2") value = object->samNormChi2;
  else if(variable == "samPT") value = object->samPT;
  else if(variable == "samEta") value = object->samEta;
  else if(variable == "samPhi") value = object->samPhi;
  else if(variable == "samDZ") value = object->samDZ;
  else if(variable == "samD0") value = object->samD0;
  else if(variable == "samD0bs") value = object->samD0bs;
  else if(variable == "samD0err") value = object->samD0err;
  else if(variable == "comNormChi2") value = object->comNormChi2;
  else if(variable == "comPT") value = object->comPT;
  else if(variable == "comEta") value = object->comEta;
  else if(variable == "comPhi") value = object->comPhi;
  else if(variable == "comDZ") value = object->comDZ;
  else if(variable == "comD0") value = object->comD0;
  else if(variable == "comD0bs") value = object->comD0bs;
  else if(variable == "comD0err") value = object->comD0err;
  else if(variable == "isolationR03emVetoEt") value = object->isolationR03emVetoEt;
  else if(variable == "isolationR03hadVetoEt") value = object->isolationR03hadVetoEt;
  else if(variable == "normalizedChi2") value = object->normalizedChi2;
  else if(variable == "dVzPVz") value = object->dVzPVz;
  else if(variable == "dB") value = object->dB;
  else if(variable == "ptErr") value = object->ptErr;
  else if(variable == "innerTrackNormChi2") value = object->innerTrackNormChi2;
  else if(variable == "correctedD0") value = object->correctedD0;
  else if(variable == "correctedD0Vertex") value = object->correctedD0Vertex;
  else if(variable == "correctedDZ") value = object->correctedDZ;
  else if(variable == "particleIso") value = object->particleIso;
  else if(variable == "chargedHadronIso") value = object->chargedHadronIso;
  else if(variable == "neutralHadronIso") value = object->neutralHadronIso;
  else if(variable == "photonIso") value = object->photonIso;
  else if(variable == "puChargedHadronIso") value = object->puChargedHadronIso;
  else if(variable == "chargedHadronIsoDR03") value = object->chargedHadronIsoDR03;
  else if(variable == "neutralHadronIsoDR03") value = object->neutralHadronIsoDR03;
  else if(variable == "photonIsoDR03") value = object->photonIsoDR03;
  else if(variable == "puChargedHadronIsoDR03") value = object->puChargedHadronIsoDR03;
  else if(variable == "chargedHadronIsoDR04") value = object->chargedHadronIsoDR04;
  else if(variable == "neutralHadronIsoDR04") value = object->neutralHadronIsoDR04;
  else if(variable == "photonIsoDR04") value = object->photonIsoDR04;
  else if(variable == "puChargedHadronIsoDR04") value = object->puChargedHadronIsoDR04;
  else if(variable == "rhoPrime") value = object->rhoPrime;
  else if(variable == "AEffDr03") value = object->AEffDr03;
  else if(variable == "AEffDr04") value = object->AEffDr04;
  else if(variable == "pfIsoR03SumChargedHadronPt") value = object->pfIsoR03SumChargedHadronPt;
  else if(variable == "pfIsoR03SumNeutralHadronEt") value = object->pfIsoR03SumNeutralHadronEt;
  else if(variable == "pfIsoR03SumPhotonEt") value = object->pfIsoR03SumPhotonEt;
  else if(variable == "pfIsoR03SumPUPt") value = object->pfIsoR03SumPUPt;
  else if(variable == "pfIsoR04SumChargedHadronPt") value = object->pfIsoR04SumChargedHadronPt;
  else if(variable == "pfIsoR04SumNeutralHadronEt") value = object->pfIsoR04SumNeutralHadronEt;
  else if(variable == "pfIsoR04SumPhotonEt") value = object->pfIsoR04SumPhotonEt;
  else if(variable == "pfIsoR04SumPUPt") value = object->pfIsoR04SumPUPt;
  else if(variable == "IP") value = object->IP;
  else if(variable == "IPError") value = object->IPError;
  else if(variable == "timeAtIpInOut") value = object->timeAtIpInOut;
  else if(variable == "timeAtIpInOutErr") value = object->timeAtIpInOutErr;
  else if(variable == "timeAtIpOutIn") value = object->timeAtIpOutIn;
  else if(variable == "timeAtIpOutInErr") value = object->timeAtIpOutInErr;
  else if(variable == "ecal_time") value = object->ecal_time;
  else if(variable == "hcal_time") value = object->hcal_time;
  else if(variable == "ecal_timeError") value = object->ecal_timeError;
  else if(variable == "hcal_timeError") value = object->hcal_timeError;
  else if(variable == "energy_ecal") value = object->energy_ecal;
  else if(variable == "energy_hcal") value = object->energy_hcal;
  else if(variable == "e3x3_ecal") value = object->e3x3_ecal;
  else if(variable == "e3x3_hcal") value = object->e3x3_hcal;
  else if(variable == "energyMax_ecal") value = object->energyMax_ecal;
  else if(variable == "energyMax_hcal") value = object->energyMax_hcal;
  else if(variable == "charge") value = object->charge;
  else if(variable == "IDGMPTight") value = object->IDGMPTight;
  else if(variable == "tkNumValidHits") value = object->tkNumValidHits;
  else if(variable == "tkCharge") value = object->tkCharge;
  else if(variable == "samNumValidHits") value = object->samNumValidHits;
  else if(variable == "samCharge") value = object->samCharge;
  else if(variable == "comNumValidHits") value = object->comNumValidHits;
  else if(variable == "comCharge") value = object->comCharge;
  else if(variable == "genId") value = object->genId;
  else if(variable == "genCharge") value = object->genCharge;
  else if(variable == "genNumberOfMothers") value = object->genNumberOfMothers;
  else if(variable == "genMotherId") value = object->genMotherId;
  else if(variable == "genMotherCharge") value = object->genMotherCharge;
  else if(variable == "genMother0Id") value = object->genMother0Id;
  else if(variable == "genMother1Id") value = object->genMother1Id;
  else if(variable == "genGrandMother00Id") value = object->genGrandMother00Id;
  else if(variable == "genGrandMother01Id") value = object->genGrandMother01Id;
  else if(variable == "genGrandMother10Id") value = object->genGrandMother10Id;
  else if(variable == "genGrandMother11Id") value = object->genGrandMother11Id;
  else if(variable == "isPFMuon") value = object->isPFMuon;
  else if(variable == "isGoodMuon_1StationTight") value = object->isGoodMuon_1StationTight;
  else if(variable == "isGlobalMuon") value = object->isGlobalMuon;
  else if(variable == "isTrackerMuon") value = object->isTrackerMuon;
  else if(variable == "isStandAloneMuon") value = object->isStandAloneMuon;
  else if(variable == "isGlobalMuonPromptTight") value = object->isGlobalMuonPromptTight;
  else if(variable == "numberOfValidMuonHits") value = object->numberOfValidMuonHits;
  else if(variable == "numberOfValidTrackerHits") value = object->numberOfValidTrackerHits;
  else if(variable == "numberOfLayersWithMeasurement") value = object->numberOfLayersWithMeasurement;
  else if(variable == "pixelLayersWithMeasurement") value = object->pixelLayersWithMeasurement;
  else if(variable == "numberOfMatches") value = object->numberOfMatches;
  else if(variable == "numberOfValidTrackerHitsInnerTrack") value = object->numberOfValidTrackerHitsInnerTrack;
  else if(variable == "numberOfValidPixelHits") value = object->numberOfValidPixelHits;
  else if(variable == "numberOfMatchedStations") value = object->numberOfMatchedStations;
  else if(variable == "time_ndof") value = object->time_ndof;

  //user-defined variables
  else if(variable == "d0Sig") value =  object->correctedD0Vertex / object->tkD0err;
  else if(variable == "detIso") value = (object->trackIso + object->caloIso) / object->pt;
  else if(variable == "relPFdBetaIso") value = (object->pfIsoR04SumChargedHadronPt + max(0.0, object->pfIsoR04SumNeutralHadronEt + object->pfIsoR04SumPhotonEt - 0.5*object->pfIsoR04SumPUPt)) / object->pt; 
  else if(variable == "relPFrhoIso") value = ( object->chargedHadronIso + max(0.0, object->neutralHadronIso + object->photonIso - object->AEffDr03*object->rhoPrime) ) / object->pt;
  else if(variable == "tightID") {
    value = object->isGlobalMuon > 0            \
      && object->isPFMuon > 0                   \
      && object->normalizedChi2 < 10            \
      && object->numberOfValidMuonHits > 0      \
      && object->numberOfMatchedStations > 1    \
      && fabs(object->correctedD0Vertex) < 0.2  \
      && fabs(object->correctedDZ) < 0.5        \
      && object->numberOfValidPixelHits > 0     \
      && object->numberOfLayersWithMeasurement > 5;
    

  }
  else if(variable == "tightIDdisplaced"){
    value = object->isGlobalMuon > 0            \
      && object->isPFMuon > 0                   \
      && object->normalizedChi2 < 10            \
      && object->numberOfValidMuonHits > 0      \
      && object->numberOfMatchedStations > 1    \
      && object->numberOfValidPixelHits > 0     \
      && object->numberOfLayersWithMeasurement > 5;
  }

  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}


double
OSUAnalysis::valueLookup (const BNelectron* object, string variable, string function){

  double value = 0.0;
  if(variable == "energy") value = object->energy;
  else if(variable == "et") value = object->et;
  else if(variable == "gsfEt") value = object->gsfEt;
  else if(variable == "pt") value = object->pt;
  else if(variable == "px") value = object->px;
  else if(variable == "py") value = object->py;
  else if(variable == "pz") value = object->pz;
  else if(variable == "phi") value = object->phi;
  else if(variable == "eta") value = object->eta;
  else if(variable == "theta") value = object->theta;
  else if(variable == "pIn") value = object->pIn;
  else if(variable == "pOut") value = object->pOut;
  else if(variable == "EscOverPin") value = object->EscOverPin;
  else if(variable == "EseedOverPout") value = object->EseedOverPout;
  else if(variable == "hadOverEm") value = object->hadOverEm;
  else if(variable == "trackIso") value = object->trackIso;
  else if(variable == "ecalIso") value = object->ecalIso;
  else if(variable == "hcalIso") value = object->hcalIso;
  else if(variable == "caloIso") value = object->caloIso;
  else if(variable == "trackIsoDR03") value = object->trackIsoDR03;
  else if(variable == "ecalIsoDR03") value = object->ecalIsoDR03;
  else if(variable == "hcalIsoDR03") value = object->hcalIsoDR03;
  else if(variable == "hcalIsoDR03depth1") value = object->hcalIsoDR03depth1;
  else if(variable == "hcalIsoDR03depth2") value = object->hcalIsoDR03depth2;
  else if(variable == "caloIsoDR03") value = object->caloIsoDR03;
  else if(variable == "trackIsoDR04") value = object->trackIsoDR04;
  else if(variable == "ecalIsoDR04") value = object->ecalIsoDR04;
  else if(variable == "hcalIsoDR04") value = object->hcalIsoDR04;
  else if(variable == "hcalIsoDR04depth1") value = object->hcalIsoDR04depth1;
  else if(variable == "hcalIsoDR04depth2") value = object->hcalIsoDR04depth2;
  else if(variable == "caloIsoDR04") value = object->caloIsoDR04;
  else if(variable == "fbrem") value = object->fbrem;
  else if(variable == "absInvEMinusInvPin") value = object->absInvEMinusInvPin;
  else if(variable == "delPhiIn") value = object->delPhiIn;
  else if(variable == "delEtaIn") value = object->delEtaIn;
  else if(variable == "genET") value = object->genET;
  else if(variable == "genPT") value = object->genPT;
  else if(variable == "genPhi") value = object->genPhi;
  else if(variable == "genEta") value = object->genEta;
  else if(variable == "genMotherET") value = object->genMotherET;
  else if(variable == "genMotherPT") value = object->genMotherPT;
  else if(variable == "genMotherPhi") value = object->genMotherPhi;
  else if(variable == "genMotherEta") value = object->genMotherEta;
  else if(variable == "vx") value = object->vx;
  else if(variable == "vy") value = object->vy;
  else if(variable == "vz") value = object->vz;
  else if(variable == "scEnergy") value = object->scEnergy;
  else if(variable == "scRawEnergy") value = object->scRawEnergy;
  else if(variable == "scSigmaEtaEta") value = object->scSigmaEtaEta;
  else if(variable == "scSigmaIEtaIEta") value = object->scSigmaIEtaIEta;
  else if(variable == "scE1x5") value = object->scE1x5;
  else if(variable == "scE2x5Max") value = object->scE2x5Max;
  else if(variable == "scE5x5") value = object->scE5x5;
  else if(variable == "scEt") value = object->scEt;
  else if(variable == "scEta") value = object->scEta;
  else if(variable == "scPhi") value = object->scPhi;
  else if(variable == "scZ") value = object->scZ;
  else if(variable == "tkNormChi2") value = object->tkNormChi2;
  else if(variable == "tkPT") value = object->tkPT;
  else if(variable == "tkEta") value = object->tkEta;
  else if(variable == "tkPhi") value = object->tkPhi;
  else if(variable == "tkDZ") value = object->tkDZ;
  else if(variable == "tkD0") value = object->tkD0;
  else if(variable == "tkD0bs") value = object->tkD0bs;
  else if(variable == "tkD0err") value = object->tkD0err;
  else if(variable == "mva") value = object->mva;
  else if(variable == "mvaTrigV0") value = object->mvaTrigV0;
  else if(variable == "mvaNonTrigV0") value = object->mvaNonTrigV0;
  else if(variable == "dist") value = object->dist;
  else if(variable == "dcot") value = object->dcot;
  else if(variable == "convradius") value = object->convradius;
  else if(variable == "convPointX") value = object->convPointX;
  else if(variable == "convPointY") value = object->convPointY;
  else if(variable == "convPointZ") value = object->convPointZ;
  else if(variable == "eMax") value = object->eMax;
  else if(variable == "eLeft") value = object->eLeft;
  else if(variable == "eRight") value = object->eRight;
  else if(variable == "eTop") value = object->eTop;
  else if(variable == "eBottom") value = object->eBottom;
  else if(variable == "e3x3") value = object->e3x3;
  else if(variable == "swissCross") value = object->swissCross;
  else if(variable == "seedEnergy") value = object->seedEnergy;
  else if(variable == "seedTime") value = object->seedTime;
  else if(variable == "swissCrossNoI85") value = object->swissCrossNoI85;
  else if(variable == "swissCrossI85") value = object->swissCrossI85;
  else if(variable == "E2overE9NoI85") value = object->E2overE9NoI85;
  else if(variable == "E2overE9I85") value = object->E2overE9I85;
  else if(variable == "correctedD0") value = object->correctedD0;
  else if(variable == "correctedD0Vertex") value = object->correctedD0Vertex;
  else if(variable == "correctedDZ") value = object->correctedDZ;
  else if(variable == "particleIso") value = object->particleIso;
  else if(variable == "chargedHadronIso") value = object->chargedHadronIso;
  else if(variable == "neutralHadronIso") value = object->neutralHadronIso;
  else if(variable == "photonIso") value = object->photonIso;
  else if(variable == "puChargedHadronIso") value = object->puChargedHadronIso;
  else if(variable == "chargedHadronIsoDR03") value = object->chargedHadronIsoDR03;
  else if(variable == "neutralHadronIsoDR03") value = object->neutralHadronIsoDR03;
  else if(variable == "photonIsoDR03") value = object->photonIsoDR03;
  else if(variable == "puChargedHadronIsoDR03") value = object->puChargedHadronIsoDR03;
  else if(variable == "chargedHadronIsoDR04") value = object->chargedHadronIsoDR04;
  else if(variable == "neutralHadronIsoDR04") value = object->neutralHadronIsoDR04;
  else if(variable == "photonIsoDR04") value = object->photonIsoDR04;
  else if(variable == "puChargedHadronIsoDR04") value = object->puChargedHadronIsoDR04;
  else if(variable == "rhoPrime") value = object->rhoPrime;
  else if(variable == "AEffDr03") value = object->AEffDr03;
  else if(variable == "AEffDr04") value = object->AEffDr04;
  else if(variable == "IP") value = object->IP;
  else if(variable == "IPError") value = object->IPError;
  else if(variable == "charge") value = object->charge;
  else if(variable == "classification") value = object->classification;
  else if(variable == "genId") value = object->genId;
  else if(variable == "genCharge") value = object->genCharge;
  else if(variable == "genNumberOfMothers") value = object->genNumberOfMothers;
  else if(variable == "genMotherId") value = object->genMotherId;
  else if(variable == "genMotherCharge") value = object->genMotherCharge;
  else if(variable == "genMother0Id") value = object->genMother0Id;
  else if(variable == "genMother1Id") value = object->genMother1Id;
  else if(variable == "genGrandMother00Id") value = object->genGrandMother00Id;
  else if(variable == "genGrandMother01Id") value = object->genGrandMother01Id;
  else if(variable == "genGrandMother10Id") value = object->genGrandMother10Id;
  else if(variable == "genGrandMother11Id") value = object->genGrandMother11Id;
  else if(variable == "numClusters") value = object->numClusters;
  else if(variable == "tkNumValidHits") value = object->tkNumValidHits;
  else if(variable == "tkCharge") value = object->tkCharge;
  else if(variable == "gsfCharge") value = object->gsfCharge;
  else if(variable == "isEB") value = object->isEB;
  else if(variable == "isEE") value = object->isEE;
  else if(variable == "isGap") value = object->isGap;
  else if(variable == "isEBEEGap") value = object->isEBEEGap;
  else if(variable == "isEBGap") value = object->isEBGap;
  else if(variable == "isEEGap") value = object->isEEGap;
  else if(variable == "isEcalDriven") value = object->isEcalDriven;
  else if(variable == "isTrackerDriven") value = object->isTrackerDriven;
  else if(variable == "numberOfLostHits") value = object->numberOfLostHits;
  else if(variable == "numberOfExpectedInnerHits") value = object->numberOfExpectedInnerHits;
  else if(variable == "numberOfValidPixelHits") value = object->numberOfValidPixelHits;
  else if(variable == "numberOfValidPixelBarrelHits") value = object->numberOfValidPixelBarrelHits;
  else if(variable == "numberOfValidPixelEndcapHits") value = object->numberOfValidPixelEndcapHits;
  else if(variable == "isHEEP") value = object->isHEEP;
  else if(variable == "isHEEPnoEt") value = object->isHEEPnoEt;
  else if(variable == "seedRecoFlag") value = object->seedRecoFlag;
  else if(variable == "eidRobustHighEnergy") value = object->eidRobustHighEnergy;
  else if(variable == "eidRobustLoose") value = object->eidRobustLoose;
  else if(variable == "eidRobustTight") value = object->eidRobustTight;
  else if(variable == "eidLoose") value = object->eidLoose;
  else if(variable == "eidTight") value = object->eidTight;
  else if(variable == "eidVeryLooseMC") value = object->eidVeryLooseMC;
  else if(variable == "eidLooseMC") value = object->eidLooseMC;
  else if(variable == "eidMediumMC") value = object->eidMediumMC;
  else if(variable == "eidTightMC") value = object->eidTightMC;
  else if(variable == "eidSuperTightMC") value = object->eidSuperTightMC;
  else if(variable == "eidHyperTight1MC") value = object->eidHyperTight1MC;
  else if(variable == "eidHyperTight2MC") value = object->eidHyperTight2MC;
  else if(variable == "eidHyperTight3MC") value = object->eidHyperTight3MC;
  else if(variable == "eidHyperTight4MC") value = object->eidHyperTight4MC;
  else if(variable == "passConvVeto") value = object->passConvVeto;

  //user-defined variables
  else if(variable == "d0Sig") value =  object->correctedD0Vertex / object->tkD0err;
  else if(variable == "detIso") value = (object->trackIso + object->caloIso) / object->pt;
  else if(variable == "relPFrhoIso") value = ( object->chargedHadronIsoDR03 + max(0.0, object->neutralHadronIsoDR03 + object->photonIsoDR03 - object->AEffDr03*object->rhoPrime) ) / object->pt;


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}


double
OSUAnalysis::valueLookup (const BNevent* object, string variable, string function){

  double value = 0.0;

  if(variable == "weight") value = object->weight;
  else if(variable == "pthat") value = object->pthat;
  else if(variable == "qScale") value = object->qScale;
  else if(variable == "alphaQCD") value = object->alphaQCD;
  else if(variable == "alphaQED") value = object->alphaQED;
  else if(variable == "scalePDF") value = object->scalePDF;
  else if(variable == "x1") value = object->x1;
  else if(variable == "x2") value = object->x2;
  else if(variable == "xPDF1") value = object->xPDF1;
  else if(variable == "xPDF2") value = object->xPDF2;
  else if(variable == "BSx") value = object->BSx;
  else if(variable == "BSy") value = object->BSy;
  else if(variable == "BSz") value = object->BSz;
  else if(variable == "PVx") value = object->PVx;
  else if(variable == "PVy") value = object->PVy;
  else if(variable == "PVz") value = object->PVz;
  else if(variable == "bField") value = object->bField;
  else if(variable == "instLumi") value = object->instLumi;
  else if(variable == "bxLumi") value = object->bxLumi;
  else if(variable == "FilterOutScrapingFraction") value = object->FilterOutScrapingFraction;
  else if(variable == "sumNVtx") value = object->sumNVtx;
  else if(variable == "sumTrueNVtx") value = object->sumTrueNVtx;
  else if(variable == "nm1_true") value = object->nm1_true;
  else if(variable == "n0_true") value = object->n0_true;
  else if(variable == "np1_true") value = object->np1_true;
  else if(variable == "numTruePV") value = object->numTruePV;
  else if(variable == "Q2ScaleUpWgt") value = object->Q2ScaleUpWgt;
  else if(variable == "Q2ScaleDownWgt") value = object->Q2ScaleDownWgt;
  else if(variable == "rho_kt6PFJets") value = object->rho_kt6PFJets;
  else if(variable == "rho_kt6PFJetsCentralChargedPileUp") value = object->rho_kt6PFJetsCentralChargedPileUp;
  else if(variable == "rho_kt6PFJetsCentralNeutral") value = object->rho_kt6PFJetsCentralNeutral;
  else if(variable == "rho_kt6PFJetsCentralNeutralTight") value = object->rho_kt6PFJetsCentralNeutralTight;
  else if(variable == "run") value = object->run;
  else if(variable == "lumi") value = object->lumi;
  else if(variable == "sample") value = object->sample;
  else if(variable == "numPV") value = object->numPV;
  else if(variable == "W0decay") value = object->W0decay;
  else if(variable == "W1decay") value = object->W1decay;
  else if(variable == "Z0decay") value = object->Z0decay;
  else if(variable == "Z1decay") value = object->Z1decay;
  else if(variable == "H0decay") value = object->H0decay;
  else if(variable == "H1decay") value = object->H1decay;
  else if(variable == "hcalnoiseLoose") value = object->hcalnoiseLoose;
  else if(variable == "hcalnoiseTight") value = object->hcalnoiseTight;
  else if(variable == "GoodVertex") value = object->GoodVertex;
  else if(variable == "FilterOutScraping") value = object->FilterOutScraping;
  else if(variable == "HBHENoiseFilter") value = object->HBHENoiseFilter;
  else if(variable == "CSCLooseHaloId") value = object->CSCLooseHaloId;
  else if(variable == "CSCTightHaloId") value = object->CSCTightHaloId;
  else if(variable == "EcalLooseHaloId") value = object->EcalLooseHaloId;
  else if(variable == "EcalTightHaloId") value = object->EcalTightHaloId;
  else if(variable == "HcalLooseHaloId") value = object->HcalLooseHaloId;
  else if(variable == "HcalTightHaloId") value = object->HcalTightHaloId;
  else if(variable == "GlobalLooseHaloId") value = object->GlobalLooseHaloId;
  else if(variable == "GlobalTightHaloId") value = object->GlobalTightHaloId;
  else if(variable == "LooseId") value = object->LooseId;
  else if(variable == "TightId") value = object->TightId;
  else if(variable == "numGenPV") value = object->numGenPV;
  else if(variable == "nm1") value = object->nm1;
  else if(variable == "n0") value = object->n0;
  else if(variable == "np1") value = object->np1;
  else if(variable == "id1") value = object->id1;
  else if(variable == "id2") value = object->id2;
  else if(variable == "evt") value = object->evt;

  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}

double
OSUAnalysis::valueLookup (const BNtau* object, string variable, string function){

  double value = 0.0;

  if(variable == "px") value = object->px;
  else if(variable == "py") value = object->py;
  else if(variable == "pz") value = object->pz;
  else if(variable == "energy") value = object->energy;
  else if(variable == "et") value = object->et;
  else if(variable == "pt") value = object->pt;
  else if(variable == "eta") value = object->eta;
  else if(variable == "phi") value = object->phi;
  else if(variable == "emFraction") value = object->emFraction;
  else if(variable == "leadingTrackPt") value = object->leadingTrackPt;
  else if(variable == "leadingTrackIpVtdxy") value = object->leadingTrackIpVtdxy;
  else if(variable == "leadingTrackIpVtdz") value = object->leadingTrackIpVtdz;
  else if(variable == "leadingTrackIpVtdxyError") value = object->leadingTrackIpVtdxyError;
  else if(variable == "leadingTrackIpVtdzError") value = object->leadingTrackIpVtdzError;
  else if(variable == "leadingTrackVx") value = object->leadingTrackVx;
  else if(variable == "leadingTrackVy") value = object->leadingTrackVy;
  else if(variable == "leadingTrackVz") value = object->leadingTrackVz;
  else if(variable == "leadingTrackValidHits") value = object->leadingTrackValidHits;
  else if(variable == "leadingTrackNormChiSqrd") value = object->leadingTrackNormChiSqrd;
  else if(variable == "numProngs") value = object->numProngs;
  else if(variable == "numSignalGammas") value = object->numSignalGammas;
  else if(variable == "numSignalNeutrals") value = object->numSignalNeutrals;
  else if(variable == "numSignalPiZeros") value = object->numSignalPiZeros;
  else if(variable == "decayMode") value = object->decayMode;
  else if(variable == "charge") value = object->charge;
  else if(variable == "inTheCracks") value = object->inTheCracks;
  else if(variable == "HPSagainstElectronLoose") value = object->HPSagainstElectronLoose;
  else if(variable == "HPSagainstElectronMVA") value = object->HPSagainstElectronMVA;
  else if(variable == "HPSagainstElectronMedium") value = object->HPSagainstElectronMedium;
  else if(variable == "HPSagainstElectronTight") value = object->HPSagainstElectronTight;
  else if(variable == "HPSagainstMuonLoose") value = object->HPSagainstMuonLoose;
  else if(variable == "HPSagainstMuonMedium") value = object->HPSagainstMuonMedium;
  else if(variable == "HPSagainstMuonTight") value = object->HPSagainstMuonTight;
  else if(variable == "HPSbyLooseCombinedIsolationDeltaBetaCorr") value = object->HPSbyLooseCombinedIsolationDeltaBetaCorr;
  else if(variable == "HPSbyMediumCombinedIsolationDeltaBetaCorr") value = object->HPSbyMediumCombinedIsolationDeltaBetaCorr;
  else if(variable == "HPSbyTightCombinedIsolationDeltaBetaCorr") value = object->HPSbyTightCombinedIsolationDeltaBetaCorr;
  else if(variable == "HPSbyVLooseCombinedIsolationDeltaBetaCorr") value = object->HPSbyVLooseCombinedIsolationDeltaBetaCorr;
  else if(variable == "HPSdecayModeFinding") value = object->HPSdecayModeFinding;
  else if(variable == "leadingTrackValid") value = object->leadingTrackValid;


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}

double
OSUAnalysis::valueLookup (const BNmet* object, string variable, string function){

  double value = 0.0;

  if(variable == "et") value = object->et;
  else if(variable == "pt") value = object->pt;
  else if(variable == "px") value = object->px;
  else if(variable == "py") value = object->py;
  else if(variable == "phi") value = object->phi;
  else if(variable == "Upt") value = object->Upt;
  else if(variable == "Uphi") value = object->Uphi;
  else if(variable == "NeutralEMFraction") value = object->NeutralEMFraction;
  else if(variable == "NeutralHadEtFraction") value = object->NeutralHadEtFraction;
  else if(variable == "ChargedEMEtFraction") value = object->ChargedEMEtFraction;
  else if(variable == "ChargedHadEtFraction") value = object->ChargedHadEtFraction;
  else if(variable == "MuonEtFraction") value = object->MuonEtFraction;
  else if(variable == "Type6EtFraction") value = object->Type6EtFraction;
  else if(variable == "Type7EtFraction") value = object->Type7EtFraction;
  else if(variable == "genPT") value = object->genPT;
  else if(variable == "genPhi") value = object->genPhi;
  else if(variable == "muonCorEx") value = object->muonCorEx;
  else if(variable == "muonCorEy") value = object->muonCorEy;
  else if(variable == "jet20CorEx") value = object->jet20CorEx;
  else if(variable == "jet20CorEy") value = object->jet20CorEy;
  else if(variable == "jet1CorEx") value = object->jet1CorEx;
  else if(variable == "jet1CorEy") value = object->jet1CorEy;
  else if(variable == "sumET") value = object->sumET;
  else if(variable == "corSumET") value = object->corSumET;
  else if(variable == "mEtSig") value = object->mEtSig;
  else if(variable == "metSignificance") value = object->metSignificance;
  else if(variable == "significance") value = object->significance;
  else if(variable == "sigmaX2") value = object->sigmaX2;
  else if(variable == "sigmaY2") value = object->sigmaY2;
  else if(variable == "sigmaXY") value = object->sigmaXY;
  else if(variable == "sigmaYX") value = object->sigmaYX;
  else if(variable == "maxEtInEmTowers") value = object->maxEtInEmTowers;
  else if(variable == "emEtFraction") value = object->emEtFraction;
  else if(variable == "emEtInEB") value = object->emEtInEB;
  else if(variable == "emEtInEE") value = object->emEtInEE;
  else if(variable == "emEtInHF") value = object->emEtInHF;
  else if(variable == "maxEtInHadTowers") value = object->maxEtInHadTowers;
  else if(variable == "hadEtFraction") value = object->hadEtFraction;
  else if(variable == "hadEtInHB") value = object->hadEtInHB;
  else if(variable == "hadEtInHE") value = object->hadEtInHE;
  else if(variable == "hadEtInHF") value = object->hadEtInHF;
  else if(variable == "hadEtInHO") value = object->hadEtInHO;
  else if(variable == "UDeltaPx") value = object->UDeltaPx;
  else if(variable == "UDeltaPy") value = object->UDeltaPy;
  else if(variable == "UDeltaP") value = object->UDeltaP;
  else if(variable == "Uscale") value = object->Uscale;
  else if(variable == "type2corPx") value = object->type2corPx;
  else if(variable == "type2corPy") value = object->type2corPy;
  else if(variable == "T2pt") value = object->T2pt;
  else if(variable == "T2px") value = object->T2px;
  else if(variable == "T2py") value = object->T2py;
  else if(variable == "T2phi") value = object->T2phi;
  else if(variable == "T2sumET") value = object->T2sumET;
  else if(variable == "pfT1jet1pt") value = object->pfT1jet1pt;
  else if(variable == "pfT1jet1phi") value = object->pfT1jet1phi;
  else if(variable == "pfT1jet6pt") value = object->pfT1jet6pt;
  else if(variable == "pfT1jet6phi") value = object->pfT1jet6phi;
  else if(variable == "pfT1jet10pt") value = object->pfT1jet10pt;
  else if(variable == "pfT1jet10phi") value = object->pfT1jet10phi;

  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}

double
OSUAnalysis::valueLookup (const BNtrack* object, string variable, string function){

  double value = 0.0;

  if(variable == "pt") value = object->pt;
  else if(variable == "px") value = object->px;
  else if(variable == "py") value = object->py;
  else if(variable == "pz") value = object->pz;
  else if(variable == "phi") value = object->phi;
  else if(variable == "eta") value = object->eta;
  else if(variable == "theta") value = object->theta;
  else if(variable == "normChi2") value = object->normChi2;
  else if(variable == "dZ") value = object->dZ;
  else if(variable == "d0") value = object->d0;
  else if(variable == "d0err") value = object->d0err;
  else if(variable == "vx") value = object->vx;
  else if(variable == "vy") value = object->vy;
  else if(variable == "vz") value = object->vz;
  else if(variable == "charge") value = object->charge;
  else if(variable == "numValidHits") value = object->numValidHits;
  else if(variable == "isHighPurity") value = object->isHighPurity;


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}

double
OSUAnalysis::valueLookup (const BNgenjet* object, string variable, string function){

  double value = 0.0;

  if(variable == "pt") value = object->pt;
  else if(variable == "eta") value = object->eta;
  else if(variable == "phi") value = object->phi;
  else if(variable == "px") value = object->px;
  else if(variable == "py") value = object->py;
  else if(variable == "pz") value = object->pz;
  else if(variable == "et") value = object->et;
  else if(variable == "energy") value = object->energy;
  else if(variable == "mass") value = object->mass;
  else if(variable == "emEnergy") value = object->emEnergy;
  else if(variable == "hadEnergy") value = object->hadEnergy;
  else if(variable == "invisibleEnergy") value = object->invisibleEnergy;
  else if(variable == "auxiliaryEnergy") value = object->auxiliaryEnergy;
  else if(variable == "charge") value = object->charge;


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}

double
OSUAnalysis::valueLookup (const BNmcparticle* object, string variable, string function){

  double value = 0.0;

  if(variable == "energy") value = object->energy;
  else if(variable == "et") value = object->et;
  else if(variable == "pt") value = object->pt;
  else if(variable == "px") value = object->px;
  else if(variable == "py") value = object->py;
  else if(variable == "pz") value = object->pz;
  else if(variable == "phi") value = object->phi;
  else if(variable == "eta") value = object->eta;
  else if(variable == "theta") value = object->theta;
  else if(variable == "mass") value = object->mass;
  else if(variable == "vx") value = object->vx;
  else if(variable == "vy") value = object->vy;
  else if(variable == "vz") value = object->vz;
  else if(variable == "motherET") value = object->motherET;
  else if(variable == "motherPT") value = object->motherPT;
  else if(variable == "motherPhi") value = object->motherPhi;
  else if(variable == "motherEta") value = object->motherEta;
  else if(variable == "mother0ET") value = object->mother0ET;
  else if(variable == "mother0PT") value = object->mother0PT;
  else if(variable == "mother0Phi") value = object->mother0Phi;
  else if(variable == "mother0Eta") value = object->mother0Eta;
  else if(variable == "mother1ET") value = object->mother1ET;
  else if(variable == "mother1PT") value = object->mother1PT;
  else if(variable == "mother1Phi") value = object->mother1Phi;
  else if(variable == "mother1Eta") value = object->mother1Eta;
  else if(variable == "daughter0ET") value = object->daughter0ET;
  else if(variable == "daughter0PT") value = object->daughter0PT;
  else if(variable == "daughter0Phi") value = object->daughter0Phi;
  else if(variable == "daughter0Eta") value = object->daughter0Eta;
  else if(variable == "daughter1ET") value = object->daughter1ET;
  else if(variable == "daughter1PT") value = object->daughter1PT;
  else if(variable == "daughter1Phi") value = object->daughter1Phi;
  else if(variable == "daughter1Eta") value = object->daughter1Eta;
  else if(variable == "grandMotherET") value = object->grandMotherET;
  else if(variable == "grandMotherPT") value = object->grandMotherPT;
  else if(variable == "grandMotherPhi") value = object->grandMotherPhi;
  else if(variable == "grandMotherEta") value = object->grandMotherEta;
  else if(variable == "grandMother00ET") value = object->grandMother00ET;
  else if(variable == "grandMother00PT") value = object->grandMother00PT;
  else if(variable == "grandMother00Phi") value = object->grandMother00Phi;
  else if(variable == "grandMother00Eta") value = object->grandMother00Eta;
  else if(variable == "grandMother01ET") value = object->grandMother01ET;
  else if(variable == "grandMother01PT") value = object->grandMother01PT;
  else if(variable == "grandMother01Phi") value = object->grandMother01Phi;
  else if(variable == "grandMother01Eta") value = object->grandMother01Eta;
  else if(variable == "grandMother10ET") value = object->grandMother10ET;
  else if(variable == "grandMother10PT") value = object->grandMother10PT;
  else if(variable == "grandMother10Phi") value = object->grandMother10Phi;
  else if(variable == "grandMother10Eta") value = object->grandMother10Eta;
  else if(variable == "grandMother11ET") value = object->grandMother11ET;
  else if(variable == "grandMother11PT") value = object->grandMother11PT;
  else if(variable == "grandMother11Phi") value = object->grandMother11Phi;
  else if(variable == "grandMother11Eta") value = object->grandMother11Eta;
  else if(variable == "charge") value = object->charge;
  else if(variable == "id") value = object->id;
  else if(variable == "status") value = object->status;
  else if(variable == "motherId") value = object->motherId;
  else if(variable == "motherCharge") value = object->motherCharge;
  else if(variable == "mother0Id") value = object->mother0Id;
  else if(variable == "mother0Status") value = object->mother0Status;
  else if(variable == "mother0Charge") value = object->mother0Charge;
  else if(variable == "mother1Id") value = object->mother1Id;
  else if(variable == "mother1Status") value = object->mother1Status;
  else if(variable == "mother1Charge") value = object->mother1Charge;
  else if(variable == "daughter0Id") value = object->daughter0Id;
  else if(variable == "daughter0Status") value = object->daughter0Status;
  else if(variable == "daughter0Charge") value = object->daughter0Charge;
  else if(variable == "daughter1Id") value = object->daughter1Id;
  else if(variable == "daughter1Status") value = object->daughter1Status;
  else if(variable == "daughter1Charge") value = object->daughter1Charge;
  else if(variable == "grandMotherId") value = object->grandMotherId;
  else if(variable == "grandMotherCharge") value = object->grandMotherCharge;
  else if(variable == "grandMother00Id") value = object->grandMother00Id;
  else if(variable == "grandMother00Status") value = object->grandMother00Status;
  else if(variable == "grandMother00Charge") value = object->grandMother00Charge;
  else if(variable == "grandMother01Id") value = object->grandMother01Id;
  else if(variable == "grandMother01Status") value = object->grandMother01Status;
  else if(variable == "grandMother01Charge") value = object->grandMother01Charge;
  else if(variable == "grandMother10Id") value = object->grandMother10Id;
  else if(variable == "grandMother10Status") value = object->grandMother10Status;
  else if(variable == "grandMother10Charge") value = object->grandMother10Charge;
  else if(variable == "grandMother11Id") value = object->grandMother11Id;
  else if(variable == "grandMother11Status") value = object->grandMother11Status;
  else if(variable == "grandMother11Charge") value = object->grandMother11Charge;

  //user-defined variables
  else if (variable == "d0"){
    double vx = object->vx - vertexPosition.X (),
      vy = object->vy - vertexPosition.Y (),
      px = object->px,
      py = object->py,
      pt = object->pt;
    value = (-vx * py + vy * px) / pt;
  }
  else if (variable == "dz"){
    double vx = object->vx - vertexPosition.X (),
      vy = object->vy - vertexPosition.Y (),
      vz = object->vz - vertexPosition.Z (),
      px = object->px,
      py = object->py,
      pz = object->pz,
      pt = object->pt;
    value = vz - (vx * px + vy * py)/pt * (pz/pt);
  }


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}

double
OSUAnalysis::valueLookup (const BNprimaryvertex* object, string variable, string function){

  double value = 0.0;

  if(variable == "x") value = object->x;
  else if(variable == "xError") value = object->xError;
  else if(variable == "y") value = object->y;
  else if(variable == "yError") value = object->yError;
  else if(variable == "z") value = object->z;
  else if(variable == "zError") value = object->zError;
  else if(variable == "rho") value = object->rho;
  else if(variable == "normalizedChi2") value = object->normalizedChi2;
  else if(variable == "ndof") value = object->ndof;
  else if(variable == "isFake") value = object->isFake;
  else if(variable == "isValid") value = object->isValid;
  else if(variable == "tracksSize") value = object->tracksSize;
  else if(variable == "isGood") value = object->isGood;


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}

double
OSUAnalysis::valueLookup (const BNbxlumi* object, string variable, string function){

  double value = 0.0;

  if(variable == "bx_B1_now") value = object->bx_B1_now;
  else if(variable == "bx_B2_now") value = object->bx_B2_now;
  else if(variable == "bx_LUMI_now") value = object->bx_LUMI_now;


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}

double
OSUAnalysis::valueLookup (const BNphoton* object, string variable, string function){

  double value = 0.0;

  if(variable == "energy") value = object->energy;
  else if(variable == "et") value = object->et;
  else if(variable == "pt") value = object->pt;
  else if(variable == "px") value = object->px;
  else if(variable == "py") value = object->py;
  else if(variable == "pz") value = object->pz;
  else if(variable == "phi") value = object->phi;
  else if(variable == "eta") value = object->eta;
  else if(variable == "theta") value = object->theta;
  else if(variable == "trackIso") value = object->trackIso;
  else if(variable == "ecalIso") value = object->ecalIso;
  else if(variable == "hcalIso") value = object->hcalIso;
  else if(variable == "caloIso") value = object->caloIso;
  else if(variable == "trackIsoHollowConeDR03") value = object->trackIsoHollowConeDR03;
  else if(variable == "trackIsoSolidConeDR03") value = object->trackIsoSolidConeDR03;
  else if(variable == "ecalIsoDR03") value = object->ecalIsoDR03;
  else if(variable == "hcalIsoDR03") value = object->hcalIsoDR03;
  else if(variable == "caloIsoDR03") value = object->caloIsoDR03;
  else if(variable == "trackIsoHollowConeDR04") value = object->trackIsoHollowConeDR04;
  else if(variable == "trackIsoSolidConeDR04") value = object->trackIsoSolidConeDR04;
  else if(variable == "ecalIsoDR04") value = object->ecalIsoDR04;
  else if(variable == "hcalIsoDR04") value = object->hcalIsoDR04;
  else if(variable == "caloIsoDR04") value = object->caloIsoDR04;
  else if(variable == "hadOverEm") value = object->hadOverEm;
  else if(variable == "sigmaEtaEta") value = object->sigmaEtaEta;
  else if(variable == "sigmaIetaIeta") value = object->sigmaIetaIeta;
  else if(variable == "r9") value = object->r9;
  else if(variable == "scEnergy") value = object->scEnergy;
  else if(variable == "scRawEnergy") value = object->scRawEnergy;
  else if(variable == "scSeedEnergy") value = object->scSeedEnergy;
  else if(variable == "scEta") value = object->scEta;
  else if(variable == "scPhi") value = object->scPhi;
  else if(variable == "scZ") value = object->scZ;
  else if(variable == "genET") value = object->genET;
  else if(variable == "genPT") value = object->genPT;
  else if(variable == "genPhi") value = object->genPhi;
  else if(variable == "genEta") value = object->genEta;
  else if(variable == "genMotherET") value = object->genMotherET;
  else if(variable == "genMotherPT") value = object->genMotherPT;
  else if(variable == "genMotherPhi") value = object->genMotherPhi;
  else if(variable == "genMotherEta") value = object->genMotherEta;
  else if(variable == "eMax") value = object->eMax;
  else if(variable == "eLeft") value = object->eLeft;
  else if(variable == "eRight") value = object->eRight;
  else if(variable == "eTop") value = object->eTop;
  else if(variable == "eBottom") value = object->eBottom;
  else if(variable == "e3x3") value = object->e3x3;
  else if(variable == "swissCross") value = object->swissCross;
  else if(variable == "seedEnergy") value = object->seedEnergy;
  else if(variable == "seedTime") value = object->seedTime;
  else if(variable == "swissCrossNoI85") value = object->swissCrossNoI85;
  else if(variable == "swissCrossI85") value = object->swissCrossI85;
  else if(variable == "E2overE9NoI85") value = object->E2overE9NoI85;
  else if(variable == "E2overE9I85") value = object->E2overE9I85;
  else if(variable == "IDTight") value = object->IDTight;
  else if(variable == "IDLoose") value = object->IDLoose;
  else if(variable == "IDLooseEM") value = object->IDLooseEM;
  else if(variable == "genId") value = object->genId;
  else if(variable == "genCharge") value = object->genCharge;
  else if(variable == "genMotherId") value = object->genMotherId;
  else if(variable == "genMotherCharge") value = object->genMotherCharge;
  else if(variable == "isEB") value = object->isEB;
  else if(variable == "isEE") value = object->isEE;
  else if(variable == "isGap") value = object->isGap;
  else if(variable == "isEBEEGap") value = object->isEBEEGap;
  else if(variable == "isEBGap") value = object->isEBGap;
  else if(variable == "isEEGap") value = object->isEEGap;
  else if(variable == "hasPixelSeed") value = object->hasPixelSeed;
  else if(variable == "seedRecoFlag") value = object->seedRecoFlag;


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}

double
OSUAnalysis::valueLookup (const BNsupercluster* object, string variable, string function){

  double value = 0.0;

  if(variable == "energy") value = object->energy;
  else if(variable == "et") value = object->et;
  else if(variable == "ex") value = object->ex;
  else if(variable == "ey") value = object->ey;
  else if(variable == "ez") value = object->ez;
  else if(variable == "phi") value = object->phi;
  else if(variable == "eta") value = object->eta;
  else if(variable == "theta") value = object->theta;


  else{std::cout << "WARNING: invalid variable '" << variable << "'\n"; value = -999;}

  value = applyFunction(function, value);

  return value;
}


double
OSUAnalysis::applyFunction(string function, double value){

  if(function == "abs") value = fabs(value);


  return value;

}


template <class InputCollection>
void OSUAnalysis::setObjectFlags(cut &currentCut, uint currentCutIndex, flagMap &individualFlags, flagMap &cumulativeFlags, InputCollection inputCollection, string inputType){


  for (uint object = 0; object != inputCollection->size(); object++){

    bool decision = true;//object passes if this cut doesn't cut on that type of object

    if(currentCut.inputCollection == inputType){

      double value = valueLookup(&inputCollection->at(object), currentCut.variable, currentCut.function);

      decision = evaluateComparison(value,currentCut.comparativeOperator,currentCut.cutValue);
    }
    individualFlags[inputType].at(currentCutIndex).push_back(decision);


    //set flags for objects that pass each cut AND all the previous cuts
    bool previousCumulativeFlag = true;
    for(uint previousCutIndex = 0; previousCutIndex != currentCutIndex; previousCutIndex++){
      if(previousCumulativeFlag && individualFlags[inputType].at(previousCutIndex).at(object)) previousCumulativeFlag = true;
      else{ previousCumulativeFlag = false; break;}
    }
    cumulativeFlags[inputType].at(currentCutIndex).push_back(previousCumulativeFlag && decision);

  }

}


template <class InputCollection>
void OSUAnalysis::fillHistogram(TH1* histo, histogram parameters, InputCollection inputCollection){


  for (uint object = 0; object != inputCollection->size(); object++){
    if(parameters.inputVariables.size() == 1){
      double value = valueLookup(&inputCollection->at(object), parameters.inputVariables.at(0), parameters.function);
      histo->Fill(value);
    }
    else if(parameters.inputVariables.size() == 2){
      double valueX = valueLookup(&inputCollection->at(object), parameters.inputVariables.at(0), parameters.function);
      double valueY = valueLookup(&inputCollection->at(object), parameters.inputVariables.at(1), parameters.function);
      histo->Fill(valueX,valueY);
    }
  }

}


DEFINE_FWK_MODULE(OSUAnalysis);


Revision:	1.11
Committed:	Thu Feb 7 15:51:44 2013 UTC (12 years, 2 months ago) by lantonel
Content type:	text/plain
Branch:	MAIN
Changes since 1.10:	+18 -10 lines
Log Message:	corrected muon tightID to match twiki