CovarianceParameterization Class Reference

#include <CovarianceParameterization.h>

Classes
struct	CompressionSchema

Public Member Functions
	CovarianceParameterization ()
bool	isValid () const
void	load (int version)
int	loadedVersion () const
float	meanValue (int i, int j, int sign, float pt, float eta, int nHits, int pixelHits, float cii=1., float cjj=1.) const
float	pack (float value, int schema, int i, int j, float pt, float eta, int nHits, int pixelHits, float cii=1., float cjj=1.) const
float	unpack (uint16_t packed, int schema, int i, int j, float pt, float eta, int nHits, int pixelHits, float cii=1., float cjj=1.) const

Static Public Member Functions
static int	index (int i, int j)

Private Member Functions
void	addTheHistogram (std::vector< TH3D * > *HistoVector, std::string StringToAddInTheName, int i, int j, TFile &fileToRead)
void	readFile (TFile &)

Private Attributes
std::vector< TH3D * >	cov_elements_noPixelHit
std::vector< TH3D * >	cov_elements_pixelHit
TFile *	fileToRead_
int	loadedVersion_
std::unordered_map< uint16_t, CompressionSchema >	schemas

Detailed Description

Definition at line 24 of file CovarianceParameterization.h.

Constructor & Destructor Documentation

CovarianceParameterization::CovarianceParameterization ( )

inline

Definition at line 38 of file CovarianceParameterization.h.

: loadedVersion_(-1) {}

Member Function Documentation

void CovarianceParameterization::addTheHistogram ( std::vector< TH3D * > *  HistoVector, std::string  StringToAddInTheName, int  i, int  j, TFile &  fileToRead  )

private

Definition at line 141 of file CovarianceParameterization.cc.

References mps_fire::i, dqmiolumiharvest::j, and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by readFile().

                                                                                                     {
  std::string List_covName[5] = {"qoverp", "lambda", "phi", "dxy", "dsz"};

  std::string histoNameString = "covariance_" + List_covName[i] + "_" + List_covName[j] + StringToAddInTheName +
                                "parametrization";  // + "_entries";
  TH3D *matrixElememtHistogramm = (TH3D *)fileToRead.Get(histoNameString.c_str());
  HistoVector->push_back(matrixElememtHistogramm);
}

static int CovarianceParameterization::index ( int  i, int  j  )

inlinestatic

Definition at line 26 of file CovarianceParameterization.h.

Referenced by BeautifulSoup.PageElement::_invert(), and CovarianceParameterization::CompressionSchema::operator()().

                                 {
    if (i >= j)
      return j + i * (i + 1) / 2;
    else
      return i + j * (j + 1) / 2;
  }

bool CovarianceParameterization::isValid ( void  ) const

inline

Definition at line 39 of file CovarianceParameterization.h.

References loadedVersion_.

Referenced by ntupleDataFormat._Object::_checkIsValid(), core.AutoHandle.AutoHandle::ReallyLoad(), and pat::PackedCandidate::unpackCovariance().

{ return loadedVersion_ != -1; }

void CovarianceParameterization::load

(

int

version

)

Definition at line 81 of file CovarianceParameterization.cc.

References haddnano::cl, fileToRead_, printsummarytable::folder, edm::FileInPath::fullPath(), mps_fire::i, dqmiolumiharvest::j, isotrackApplyRegressor::k, submitPVResolutionJobs::key, loadedVersion_, GetRecoTauVFromDQM_MC_cff::next, AlCaHLTBitMon_ParallelJobs::p, readFile(), schemas, AlCaHLTBitMon_QueryRunRegistry::string, and relval_steps::version.

                                                 {
  edm::FileInPath fip(
      fmt::sprintf("DataFormats/PatCandidates/data/CovarianceParameterization_version%d.root", version));
  fileToRead_ = TFile::Open(fip.fullPath().c_str());
  TFile &fileToRead = *fileToRead_;
  //Read files from here fip.fullPath().c_str();
  if (fileToRead.IsOpen()) {
    readFile(fileToRead);

    TIter next(((TDirectoryFile *)fileToRead.Get("schemas"))->GetListOfKeys());
    TKey *key;
    while ((key = (TKey *)next())) {
      TClass *cl = gROOT->GetClass(key->GetClassName());
      if (!cl->InheritsFrom("TDirectoryFile"))
        continue;
      std::string schemaNumber = key->ReadObj()->GetName();
      uint16_t schemaN = std::stoi(schemaNumber);
      //for (int folderNumber = 0; folderNumber < 6 ; folderNumber++) {
      CompressionSchema schema;
      for (int i = 0; i < 5; i++) {
        for (int j = i; j < 5; j++) {  //FILLING ONLY THE SCHEMA OF SOME ELEMENTS
          std::string folder = "schemas/" + schemaNumber + "/" + char(48 + i) + char(48 + j);
          std::string methodString = folder + "/method";
          std::string targetString = folder + "/target";
          std::string bitString = folder + "/bit";
          std::vector<float> vParams;
          TVector *p = (TVector *)fileToRead.Get((folder + "/param").c_str());
          vParams.reserve(p->GetNoElements());
          for (int k = 0; k < p->GetNoElements(); k++) {
            vParams.push_back((*p)[k]);
          }

          schema(i, j) = CompressionElement(
              (CompressionElement::Method)((TParameter<int> *)fileToRead.Get(methodString.c_str()))->GetVal(),
              (CompressionElement::Target)((TParameter<int> *)fileToRead.Get(targetString.c_str()))->GetVal(),
              (int)((TParameter<int> *)fileToRead.Get(bitString.c_str()))->GetVal(),
              vParams);
        }
      }
      schemas[schemaN] = schema;
    }

    loadedVersion_ = version;
  } else {
    loadedVersion_ = -1;
  }
}

int CovarianceParameterization::loadedVersion ( ) const

inline

Definition at line 40 of file CovarianceParameterization.h.

References loadedVersion_.

{ return loadedVersion_; }

float CovarianceParameterization::meanValue	(	int	i,
		int	j,
		int	sign,
		float	pt,
		float	eta,
		int	nHits,
		int	pixelHits,
		float	cii = 1.,
		float	cjj = 1.
	)		const

Definition at line 151 of file CovarianceParameterization.cc.

References funct::abs(), cov_elements_noPixelHit, cov_elements_pixelHit, etaBin(), mps_fire::i, dqmiolumiharvest::j, and nHits.

Referenced by pack(), and unpack().

                                                                                                       {
  int hitNumberToUse = nHits;
  if (hitNumberToUse < 2)
    hitNumberToUse = 2;
  if (hitNumberToUse > 32)
    hitNumberToUse = 32;
  int ptBin = cov_elements_pixelHit[0]->GetXaxis()->FindBin(pt);
  int etaBin = cov_elements_pixelHit[0]->GetYaxis()->FindBin(std::abs(eta));
  int hitBin = cov_elements_pixelHit[0]->GetZaxis()->FindBin(hitNumberToUse);
  int min_idx = i;
  int max_idx = j;

  if (i > j) {
    min_idx = j;
    max_idx = i;
  }

  int indexOfTheHitogramInTheList = ((9 - min_idx) * min_idx) / 2 + max_idx;

  double meanValue = 0.;
  if (pixelHits > 0) {
    meanValue = sign * cov_elements_pixelHit[indexOfTheHitogramInTheList]->GetBinContent(ptBin, etaBin, hitBin);
  } else {
    meanValue = sign * cov_elements_noPixelHit[indexOfTheHitogramInTheList]->GetBinContent(ptBin, etaBin, hitBin);
  }
  return meanValue;
}

float CovarianceParameterization::pack	(	float	value,
		int	schema,
		int	i,
		int	j,
		float	pt,
		float	eta,
		int	nHits,
		int	pixelHits,
		float	cii = 1.,
		float	cjj = 1.
	)		const

Definition at line 180 of file CovarianceParameterization.cc.

References meanValue(), DiDispStaMuonMonitor_cfi::pt, schemas, edm::second(), and std::swap().

Referenced by pat::PackedCandidate::packCovarianceElement().

                                                                                                                      {
  if (i > j)
    std::swap(i, j);
  float ref = meanValue(i, j, 1., pt, eta, nHits, pixelHits, cii, cjj);
  if (ref == 0) {
    schema = 0;
  }
  if (schema == 0 && i == j && (i == 2 || i == 0))
    ref = 1. / (pt * pt);
  /*  //Used for debugging, to be later removed  
    uint16_t p=(*schemas.find(schema)).second(i,j).pack(value,ref);
    float up=(*schemas.find(schema)).second(i,j).unpack(p,ref);
    std::cout << "check " << i << " " << j << " " << value << " " << up << " " << p << " " << ref << " " << schema<< std::endl;*/
  return (*schemas.find(schema)).second(i, j).pack(value, ref);
}

void CovarianceParameterization::readFile ( TFile &  f )

private

Definition at line 129 of file CovarianceParameterization.cc.

References addTheHistogram(), cov_elements_noPixelHit, cov_elements_pixelHit, mps_fire::i, dqmiolumiharvest::j, and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by load().

                                                  {
  for (int i = 0; i < 5; i++) {
    for (int j = i; j < 5; j++) {
      std::string String_first_positive = "_pixel_";
      std::string String_second_positive = "_noPixel_";

      addTheHistogram(&cov_elements_pixelHit, String_first_positive, i, j, f);
      addTheHistogram(&cov_elements_noPixelHit, String_second_positive, i, j, f);
    }
  }
}

float CovarianceParameterization::unpack	(	uint16_t	packed,
		int	schema,
		int	i,
		int	j,
		float	pt,
		float	eta,
		int	nHits,
		int	pixelHits,
		float	cii = 1.,
		float	cjj = 1.
	)		const

Definition at line 196 of file CovarianceParameterization.cc.

References alignCSCRings::e, meanValue(), DiDispStaMuonMonitor_cfi::pt, schemas, edm::second(), and std::swap().

Referenced by pat::PackedCandidate::unpackCovarianceElement().

          {
  if (i > j)
    std::swap(i, j);
  float ref = meanValue(i, j, 1., pt, eta, nHits, pixelHits, cii, cjj);
  if (ref == 0) {
    schema = 0;
  }
  if (schema == 0 && i == j && (i == 2 || i == 0))
    ref = 1. / (pt * pt);
  if (i == j && (*schemas.find(schema)).second(i, j).unpack(packed, ref) == 0)
    return 1e-9;
  else
    return (*schemas.find(schema)).second(i, j).unpack(packed, ref);
}

Member Data Documentation

std::vector<TH3D *> CovarianceParameterization::cov_elements_noPixelHit

private

Definition at line 73 of file CovarianceParameterization.h.

Referenced by meanValue(), and readFile().

std::vector<TH3D *> CovarianceParameterization::cov_elements_pixelHit

private

Definition at line 72 of file CovarianceParameterization.h.

Referenced by meanValue(), and readFile().

TFile* CovarianceParameterization::fileToRead_

private

Definition at line 70 of file CovarianceParameterization.h.

Referenced by load().

int CovarianceParameterization::loadedVersion_

private

Definition at line 69 of file CovarianceParameterization.h.

Referenced by isValid(), load(), and loadedVersion().

std::unordered_map<uint16_t, CompressionSchema> CovarianceParameterization::schemas

private

Definition at line 71 of file CovarianceParameterization.h.

Referenced by load(), pack(), and unpack().