MuonMillepedeAlgorithm.h

Go to the documentation of this file.

#ifndef Alignment_MuonStandaloneAlgorithm_MuonMillepedeAlgorithm_h
#define Alignment_MuonStandaloneAlgorithm_MuonMillepedeAlgorithm_h

#include "Alignment/CommonAlignmentAlgorithm/interface/AlignmentAlgorithmBase.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"

#include "TH1D.h"
#include "TTree.h"
#include "TFile.h"
#include "TMatrixD.h"

class AlignableNavigator;
class TFile;
class TTree;

class MuonMillepedeAlgorithm : public AlignmentAlgorithmBase {
public:
  MuonMillepedeAlgorithm(const edm::ParameterSet& cfg);

  ~MuonMillepedeAlgorithm() override{};

  void initialize(const edm::EventSetup& setup,
                  AlignableTracker* tracker,
                  AlignableMuon* muon,
                  AlignableExtras* extras,
                  AlignmentParameterStore* store) override;

  void terminate(const edm::EventSetup& setup) override;

  void run(const edm::EventSetup& setup, const EventInfo& eventInfo) override;

  void updateInfo(const AlgebraicMatrix&, const AlgebraicMatrix&, const AlgebraicMatrix&, std::string);

  void toTMat(AlgebraicMatrix*, TMatrixD*);

  void collect();

private:
  // private data members

  void printM(const AlgebraicMatrix&);

  AlignmentParameterStore* theAlignmentParameterStore;
  align::Alignables theAlignables;
  AlignableNavigator* theAlignableDetAccessor;

  // verbosity flag
  bool verbose;

  //Store residuals
  std::map<std::string, TH1D*> histoMap;

  std::map<std::string, AlgebraicMatrix*> map_invCov;
  std::map<std::string, AlgebraicMatrix*> map_weightRes;
  std::map<std::string, AlgebraicMatrix*> map_N;

  double ptCut, chi2nCut;

  //Service for histograms
  edm::Service<TFileService> fs;

  std::string collec_f;
  std::string outputCollName;
  bool isCollectionJob;
  std::string collec_path;
  int collec_number;
};

#endif