Utilities.h File Reference

#include <vector>
#include <string>
#include <sstream>
#include <algorithm>
#include <utility>
#include "TMath.h"

Go to the source code of this file.

Functions
void	mergeNtuples (const char *ntuplename, const char *filestomerge, const char *outputfile)
template<typename T >
std::string	numToStr (T num)
float	processPrediction (float BDTPt, int Quality, float PrelimFit)
template<class bidiiter >
bidiiter	shuffle (bidiiter begin, bidiiter end, size_t num_random)
void	sortNtupleByEvent (const char *ntuplename, const char *filenametosort, const char *outputfile)

Variables
const Double_t	ptscale [31]
const std::vector< Double_t >	ptScale
const Double_t	twoJets_scale [16]
const std::vector< Double_t >	twoJetsScale

Function Documentation

void mergeNtuples	(	const char *	ntuplename,
		const char *	filestomerge,
		const char *	outputfile
	)

Definition at line 85 of file Utilities.cc.

Referenced by numToStr().

{
  TChain chain(ntuplename);
  chain.Add(filestomerge);
  chain.Merge(outputfile);
}

template<typename T >

std::string numToStr

(

T

num

)

Definition at line 43 of file Utilities.h.

References mergeNtuples(), pileupDistInMC::num, estimatePileup_makeJSON::outputfile, processPrediction(), alignCSCRings::s, sortNtupleByEvent(), and AlCaHLTBitMon_QueryRunRegistry::string.

{
  // Convert a number to a string.
  std::stringstream ss;
  ss << num;
  std::string s = ss.str();
  return s;
};

float processPrediction	(	float	BDTPt,
		int	Quality,
		float	PrelimFit
	)

Definition at line 47 of file Utilities.cc.

References RPCpg::pts, and ptscale.

Referenced by numToStr().

{
// Discretize and scale the BDTPt prediction 

  // Fix terrible predictions
  if(BDTPt < 0) BDTPt = PrelimFit;
  if(BDTPt > 250) BDTPt = PrelimFit;
  
  float BDTPt1 = BDTPt;
  float scaleF = 1.0;
  
  // Scale based upon quality
  if (Quality == 3) scaleF = 1.15;
  if (Quality == 2) scaleF = 1.3;
  if (Quality == 1) scaleF = 1.7;
  
  BDTPt1 = scaleF*BDTPt1;
  
  // Discretize based upon ptscale
  for (int pts=0; pts<31; pts++)
    {
      if (ptscale[pts]<=BDTPt1 && ptscale[pts+1]>BDTPt1)
    {
      BDTPt1 = ptscale[pts];
      break;
    }
    }
  
  if (BDTPt1 > 140) BDTPt1 = 140;
  if (BDTPt1 < 0) BDTPt1 = 0;
  
  return BDTPt1;
}

template<class bidiiter >

bidiiter shuffle	(	bidiiter	begin,
		bidiiter	end,
		size_t	num_random
	)

Definition at line 26 of file Utilities.h.

References begin, SoftLeptonByDistance_cfi::distance, alignCSCRings::r, rand(), and std::swap().

Referenced by L1TForest::prepareRandomSubsample().

{
  // We will end up with the same elements in the collection except that
  // the first num_random elements will be randomized.
  
  size_t left = std::distance(begin, end);
  while (num_random--) {
    bidiiter r = begin;
    std::advance(r, rand()%left);
    std::swap(*begin, *r);
    ++begin;
    --left;
  }
  return begin;
}

void sortNtupleByEvent	(	const char *	ntuplename,
		const char *	filenametosort,
		const char *	outputfile
	)

Definition at line 96 of file Utilities.cc.

References f, connectstrParser::f2, i, and diffTreeTool::index.

Referenced by numToStr().

{
  //TFile f("../../all_test_redux_post.root");
  TFile f(filenametosort);
  TNtuple *tree = (TNtuple*)f.Get(ntuplename);
  Int_t nentries = (Int_t)tree->GetEntries();
  //Drawing variable pz with no graphics option.
  //variable pz stored in array fV1 (see TTree::Draw)
  tree->Draw("Event","","goff");
  Int_t *index = new Int_t[nentries];
  //sort array containing pz in decreasing order
  //The array index contains the entry numbers in decreasing order
  TMath::Sort(nentries,tree->GetV1(),index);
  
  //open new file to store the sorted Tree
  //TFile f2("../../test_events_sorted.root","recreate");
  TFile f2(outputfile,"recreate");
  
  //Create an empty clone of the original tree
  TTree *tsorted = (TTree*)tree->CloneTree(0);
  for (Int_t i=0;i<nentries;i++) {
    tree->GetEntry(index[i]);
    tsorted->Fill();
  }
  tsorted->Write();
  delete [] index;
}

Variable Documentation

const Double_t ptscale[31]

Definition at line 26 of file Utilities.cc.

Referenced by L1TGMT::bookHistograms(), and processPrediction().

const std::vector<Double_t> ptScale

Definition at line 32 of file Utilities.cc.

Referenced by CSCTFPtLUT::CSCTFPtLUT(), and CSCTFTrackProducer::produce().

const Double_t twoJets_scale[16]

Definition at line 35 of file Utilities.cc.

const std::vector<Double_t> twoJetsScale

Definition at line 40 of file Utilities.cc.