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