MuonErrorMatrix.h

Go to the documentation of this file.

#ifndef MUONERRORMATRIX_H
#define MUONERRORMATRIX_H
 
#include <TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h>
#include <TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h>
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include <TFile.h>
#include <TProfile3D.h>
#include <TString.h>
#include <TAxis.h>
 
class MuonErrorMatrix {
public:
  enum action { use, constructor };
  MuonErrorMatrix(const edm::ParameterSet& pset);
 
  ~MuonErrorMatrix();
 
  void close();
 
  CurvilinearTrajectoryError get(GlobalVector momentum, bool convolute = true);
  CurvilinearTrajectoryError getFast(GlobalVector momentum);
 
  static void multiply(CurvilinearTrajectoryError& initial_error, const CurvilinearTrajectoryError& scale_error);
 
  static bool divide(CurvilinearTrajectoryError& num_error, const CurvilinearTrajectoryError& denom_error);
 
  inline TProfile3D* get(int i, int j) { return Index(i, j); }
  inline unsigned int index(int i, int j) { return Pindex(i, j); }
 
  static const TString vars[5];
 
  static double Term(const AlgebraicSymMatrix55& curv, int i, int j);
 
  int findBin(TAxis* axis, double value);
 
  void simpleTerm(const AlgebraicSymMatrix55& input, AlgebraicSymMatrix55& output);
 
  void complicatedTerm(const AlgebraicSymMatrix55& input, AlgebraicSymMatrix55& output);
 
  void adjust(FreeTrajectoryState& state);
 
  void adjust(TrajectoryStateOnSurface& state);
 
private:
  std::string theCategory;
 
  TDirectory* theD;
  TProfile3D* theData[15];
  TProfile3D* theData_fast[5][5];
 
  enum TermAction { error, scale, assign };
  TermAction theTermAction[15];
 
  inline int Pindex(int i, int j) {
    static const int offset[5] = {0, 5, 5 + 4, 5 + 4 + 3, 5 + 4 + 3 + 2};
    return offset[i] + abs(j - i);
  }
  inline TProfile3D* Index(int i, int j) { return theData[Pindex(i, j)]; }
 
  double Value(GlobalVector& momentum, int i, int j, bool convolute = true);
  double Rms(GlobalVector& momentum, int i, int j);
};
 
#endif