SiPixelPhase1Analyzer.h

Go to the documentation of this file.

#ifndef DQM_TRACKERREMAPPER_SIPIXELPHASE1ANALYZER_H
#define DQM_TRACKERREMAPPER_SIPIXELPHASE1ANALYZER_H

//
// Original Author:  Pawel Jurgielewicz
//         Created:  Tue, 21 Feb 2017 09:42:19 GMT
//
//

// system include files
#include <memory>

// #include <iostream>
#include <fstream>
#include <string>

#include <algorithm>

#include <vector>
#include <map>

// user include files
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "DQM/TrackerRemapper/interface/mat4.h"
#include "DataFormats/GeometrySurface/interface/DiskSectorBounds.h"
#include "DataFormats/GeometrySurface/interface/RectangularPlaneBounds.h"
#include "DataFormats/GeometrySurface/interface/TrapezoidalPlaneBounds.h"
#include "DataFormats/SiPixelDetId/interface/PXBDetId.h"
#include "DataFormats/SiPixelDetId/interface/PXFDetId.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackerCommon/interface/PixelBarrelName.h"
#include "DataFormats/TrackerCommon/interface/PixelEndcapName.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "Geometry/CommonTopologies/interface/PixelGeomDetUnit.h"
#include "Geometry/CommonTopologies/interface/PixelTopology.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"

#include "TH2.h"
#include "TProfile2D.h"
#include "TH2Poly.h"
#include "TGraph.h"

#define CODE_FORWARD(s, d, b) ((unsigned short)((b << 8) + (d << 4) + s))

//#define DEBUG_MODE

//
// class declaration
//

enum OperationMode { MODE_ANALYZE = 0, MODE_REMAP = 1 };

class SiPixelPhase1Analyzer : public edm::one::EDAnalyzer<edm::one::SharedResources> {
public:
  explicit SiPixelPhase1Analyzer(const edm::ParameterSet&);
  ~SiPixelPhase1Analyzer() override;

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

private:
  void analyze(const edm::Event&, const edm::EventSetup&) override;
  void BookHistograms();

  void BookBarrelHistograms(TDirectory* currentDir, const std::string& currentHistoName);
  void BookForwardHistograms(TDirectory* currentDir, const std::string& currentHistoName);

  void BookBins(const TrackerGeometry& theTrackerGeometry, const TrackerTopology* tt);
  void BookBarrelBins(const TrackerGeometry& theTrackerGeometry, const TrackerTopology* tt);
  void BookForwardBins(const TrackerGeometry& theTrackerGeometry, const TrackerTopology* tt);

  void SaveDetectorVertices(const TrackerTopology* tt);

  void FillBins(edm::Handle<reco::TrackCollection>* tracks,
                const TrackerGeometry& theTrackerGeometry,
                const TrackerTopology* tt);

  void FillBarrelBinsAnalyze(const TrackerGeometry& theTrackerGeometry,
                             const TrackerTopology* tt,
                             unsigned rawId,
                             const GlobalPoint& globalPoint);
  void FillForwardBinsAnalyze(const TrackerGeometry& theTrackerGeometry,
                              const TrackerTopology* tt,
                              unsigned rawId,
                              const GlobalPoint& globalPoint);

  void FillBarrelBinsRemap(const TrackerGeometry& theTrackerGeometry, const TrackerTopology* tt);
  void FillForwardBinsRemap(const TrackerGeometry& theTrackerGeometry, const TrackerTopology* tt);

  // ----------member data ---------------------------
  const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> geomToken_;
  const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> topoToken_;

  OperationMode opMode;

  edm::EDGetTokenT<reco::TrackCollection> tracksToken;

  std::string debugFileName;
  std::ofstream debugFile;

  edm::Service<TFileService> fs;

  bool firstEvent;

  std::map<uint32_t, TGraph*> bins, binsSummary;

  std::map<std::string, std::vector<TH2Poly*> > th2PolyBarrel;
  std::map<std::string, TH2Poly*> th2PolyBarrelSummary;

#ifdef DEBUG_MODE
  std::map<std::string, std::vector<TH2*> > th2PolyBarrelDebug;
#endif

  std::map<std::string, std::vector<TH2Poly*> > pxfTh2PolyForward;
  std::map<std::string, TH2Poly*> pxfTh2PolyForwardSummary;

#ifdef DEBUG_MODE
  std::map<std::string, std::vector<TH2*> > pxfTh2PolyForwardDebug;
#endif

  mat4 orthoProjectionMatrix;

  struct complementaryElements {
    mat4 mat[2];
    unsigned rawId[2];
  };
  // used to hold information about elements': ids & matrices which are of the same side, disk and barrel but different panel
  // to build trapezoidal ring elements
  std::map<unsigned short, complementaryElements> mapOfComplementaryElements;

  //Input root file handle;
  TFile* rootFileHandle;

  // read input histograms
  std::vector<unsigned> isBarrelSource;
  std::vector<std::string> analazedRootFileName;
  std::vector<std::string> pathToHistograms;
  std::vector<std::string> baseHistogramName;

  // temporal functionality
  void SaveDetectorData(bool isBarrel, unsigned rawId, int shell_hc, int layer_disk, int ladder_blade) {
    std::ofstream file("det.data", std::ofstream::out);

    file << isBarrel << "\t" << rawId << "\t" << shell_hc << "\t" << layer_disk << "\t" << ladder_blade << std::endl;
  }
};

#endif