StraightTrackAlignment.h

Go to the documentation of this file.

/****************************************************************************
* Authors: 
*  Jan Kašpar (jan.kaspar@gmail.com) 
*  Cristian Baldenegro (crisx.baldenegro@gmail.com)
****************************************************************************/
 
#ifndef CalibPPS_AlignmentRelative_StraightTrackAlignment_h
#define CalibPPS_AlignmentRelative_StraightTrackAlignment_h
 
#include <set>
#include <vector>
#include <string>
 
#include <TMatrixD.h>
#include <TVectorD.h>
#include <TFile.h>
 
#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/CTPPSReco/interface/TotemRPRecHit.h"
#include "DataFormats/CTPPSReco/interface/TotemRPUVPattern.h"
#include "DataFormats/CTPPSReco/interface/CTPPSDiamondRecHit.h"
#include "DataFormats/CTPPSReco/interface/CTPPSPixelRecHit.h"
#include "DataFormats/CTPPSReco/interface/CTPPSPixelLocalTrack.h"
 
#include "CondFormats/PPSObjects/interface/CTPPSRPAlignmentCorrectionsData.h"
 
#include "CalibPPS/AlignmentRelative/interface/AlignmentGeometry.h"
#include "CalibPPS/AlignmentRelative/interface/HitCollection.h"
#include "CalibPPS/AlignmentRelative/interface/AlignmentAlgorithm.h"
#include "CalibPPS/AlignmentRelative/interface/AlignmentConstraint.h"
#include "CalibPPS/AlignmentRelative/interface/AlignmentTask.h"
#include "CalibPPS/AlignmentRelative/interface/LocalTrackFit.h"
#include "CalibPPS/AlignmentRelative/interface/LocalTrackFitter.h"
#include "FWCore/Framework/interface/ESHandle.h"
 
namespace edm {
  class ParameterSet;
  class Event;
  class EventSetup;
}  // namespace edm
 
class TH1D;
class TGraph;
 
class StraightTrackAlignment {
public:
  StraightTrackAlignment(const edm::ParameterSet &);
  virtual ~StraightTrackAlignment();
 
  virtual void begin(edm::ESHandle<CTPPSRPAlignmentCorrectionsData> hRealAlignment,
                     edm::ESHandle<CTPPSGeometry> hRealGeometry,
                     edm::ESHandle<CTPPSGeometry> hMisalignedGeometry);
 
  virtual void processEvent(const edm::EventID &eventId,
                            const edm::DetSetVector<TotemRPUVPattern> &uvPatternsStrip,
                            const edm::DetSetVector<CTPPSDiamondRecHit> &hitsDiamond,
                            const edm::DetSetVector<CTPPSPixelRecHit> &hitsPixel,
                            const edm::DetSetVector<CTPPSPixelLocalTrack> &tracksPixel);
 
  virtual void finish();
 
protected:
  // ---------- input parameters -----------
 
  unsigned int verbosity;
 
  std::vector<unsigned int> rpIds;
 
  std::vector<unsigned int> excludePlanes;
 
  double z0;
 
  std::vector<AlignmentAlgorithm *> algorithms;
 
  enum ConstraintsType { ctFixedDetectors, ctStandard };
 
  ConstraintsType constraintsType;
 
  signed int maxEvents;
 
  // ---------- hit/track selection parameters ----------
 
  bool removeImpossible;
 
  unsigned int requireNumberOfUnits;
 
  bool requireAtLeast3PotsInOverlap;
 
  bool requireOverlap;
 
  bool cutOnChiSqPerNdf;
 
  double chiSqPerNdfCut;
 
  double maxTrackAx;
  double maxTrackAy;
 
  std::vector<std::set<unsigned int> > additionalAcceptedRPSets;
 
  // ---------- output parameters ----------
 
  std::string fileNamePrefix;
 
  std::string cumulativeFileNamePrefix;
 
  std::string expandedFileNamePrefix;
 
  std::string factoredFileNamePrefix;
 
  bool preciseXMLFormat;
 
  bool saveXMLUncertainties;
 
  bool saveIntermediateResults;
 
  std::string taskDataFileName;
 
  TFile *taskDataFile;
 
  // ---------- internal data members ----------
 
  AlignmentTask task;
 
  LocalTrackFitter fitter;
 
  CTPPSRPAlignmentCorrectionsData initialAlignments;
 
  // ---------- diagnostics parameters and plots ----------
 
  bool buildDiagnosticPlots;
 
  std::string diagnosticsFile;
 
  signed int eventsTotal;     
  signed int eventsFitted;    
  signed int eventsSelected;  
 
  std::map<std::set<unsigned int>, unsigned long> fittedTracksPerRPSet;  
  std::map<std::set<unsigned int>, unsigned long>
      selectedTracksPerRPSet;  
 
  std::map<unsigned int, unsigned int> selectedHitsPerPlane;  
 
  TH1D *fitNdfHist_fitted, *fitNdfHist_selected;  
  TH1D *fitPHist_fitted, *fitPHist_selected;      
  TH1D *fitAxHist_fitted, *fitAxHist_selected;    
  TH1D *fitAyHist_fitted, *fitAyHist_selected;    
  TH1D *fitBxHist_fitted, *fitBxHist_selected;    
  TH1D *fitByHist_fitted, *fitByHist_selected;    
 
  struct RPSetPlots {
    std::string name;
 
    TH1D *chisqn_lin_fitted = nullptr, *chisqn_lin_selected = nullptr, *chisqn_log_fitted = nullptr,
         *chisqn_log_selected = nullptr;
 
    TGraph *fitAxVsAyGraph_fitted = nullptr, *fitAxVsAyGraph_selected = nullptr;
 
    TGraph *fitBxVsByGraph_fitted = nullptr, *fitBxVsByGraph_selected = nullptr;
 
    RPSetPlots() {}
 
    RPSetPlots(const std::string &_name);
 
    void free();
 
    void write() const;
  };
 
  RPSetPlots globalPlots;
 
  std::map<std::set<unsigned int>, RPSetPlots> rpSetPlots;
 
  struct ResiduaHistogramSet {
    TH1D *total_fitted, *total_selected;
    TGraph *selected_vs_chiSq;
    std::map<std::set<unsigned int>, TH1D *> perRPSet_fitted, perRPSet_selected;
  };
 
  std::map<unsigned int, ResiduaHistogramSet> residuaHistograms;
 
  // ----------- methods ------------
 
  TH1D *newResiduaHist(const char *name);
 
  void fitLocalTrack(HitCollection &, LocalTrackFit &, bool &failed, bool &selectionChanged);
 
  void removeInsufficientPots(HitCollection &, bool &selectionChanged);
 
  void buildConstraints(std::vector<AlignmentConstraint> &);
 
  void updateDiagnosticHistograms(const HitCollection &selection,
                                  const std::set<unsigned int> &selectedRPs,
                                  const LocalTrackFit &trackFit,
                                  bool trackSelected);
 
  static std::string setToString(const std::set<unsigned int> &);
 
  void printN(const char *str, unsigned int N);
  void printLineSeparator(const std::vector<std::map<unsigned int, AlignmentResult> > &);
  void printQuantitiesLine(const std::vector<std::map<unsigned int, AlignmentResult> > &);
  void printAlgorithmsLine(const std::vector<std::map<unsigned int, AlignmentResult> > &);
 
  void saveDiagnostics() const;
};
 
#endif