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