LaserAlignment.h

Go to the documentation of this file.

#ifndef LaserAlignment_LaserAlignment_H
#define LaserAlignment_LaserAlignment_H

#include <sstream>
#include <iostream>
#include <cmath>

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/one/EDProducer.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"

#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/SiStripDigi/interface/SiStripDigi.h"
#include "DataFormats/SiStripDigi/interface/SiStripRawDigi.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/SiStripDetId/interface/SiStripDetId.h"
#include "DataFormats/SiStripDetId/interface/StripSubdetector.h"

#include "DataFormats/GeometryCommonDetAlgo/interface/ErrorFrameTransformer.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/Alignment/interface/SiStripLaserRecHit2D.h"
#include "DataFormats/Alignment/interface/TkLasBeam.h"

#include "CondFormats/SiStripObjects/interface/SiStripPedestals.h"
#include "CondFormats/DataRecord/interface/SiStripPedestalsRcd.h"
#include "CondFormats/Alignment/interface/DetectorGlobalPosition.h"
#include "CondFormats/AlignmentRecord/interface/GlobalPositionRcd.h"

#include "Geometry/TrackerGeometryBuilder/interface/StripGeomDetUnit.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeomBuilderFromGeometricDet.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/CommonTopologies/interface/StripTopology.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/CommonTopologies/interface/GeometryAligner.h"

#include "Alignment/TrackerAlignment/interface/AlignableTracker.h"

#include "Alignment/LaserAlignment/interface/LASGlobalData.h"
#include "Alignment/LaserAlignment/interface/LASGlobalLoop.h"
#include "Alignment/LaserAlignment/interface/LASModuleProfile.h"
#include "Alignment/LaserAlignment/interface/LASProfileJudge.h"
#include "Alignment/LaserAlignment/interface/LASBarrelAlgorithm.h"
#include "Alignment/LaserAlignment/interface/LASAlignmentTubeAlgorithm.h"
#include "Alignment/LaserAlignment/interface/LASEndcapAlgorithm.h"
#include "Alignment/LaserAlignment/interface/LASPeakFinder.h"
#include "Alignment/LaserAlignment/interface/LASCoordinateSet.h"
#include "Alignment/LaserAlignment/interface/LASGeometryUpdater.h"
#include "Alignment/LaserAlignment/interface/LASConstants.h"

#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"

#include "TH1.h"
#include "TFile.h"
#include "TF1.h"

class LaserAlignment : public edm::one::EDProducer<edm::EndRunProducer> {
public:
  explicit LaserAlignment(edm::ParameterSet const& theConf);
  ~LaserAlignment() override;
  void beginJob() override;
  void produce(edm::Event&, edm::EventSetup const&) override;
  void endJob() override;
  void endRunProduce(edm::Run&, const edm::EventSetup&) override;

  void testRoutine(void);

private:
  void fillDataProfiles(edm::Event const&, edm::EventSetup const&);

  void fillPedestalProfiles(edm::ESHandle<SiStripPedestals>&);

  bool isTECBeam(void);
  bool isATBeam(void);

  double getTIBTOBNominalBeamOffset(unsigned int, unsigned int, unsigned int);

  double getTEC2TECNominalBeamOffset(unsigned int, unsigned int, unsigned int);

  void fillDetectorId(void);

  double ConvertAngle(double);

  void CalculateNominalCoordinates(void);

  void DumpPosFileSet(LASGlobalData<LASCoordinateSet>&);

  void DumpStripFileSet(LASGlobalData<std::pair<float, float> >&);

  void DumpHitmaps(LASGlobalData<int>&);

  void ApplyEndcapMaskingCorrections(LASGlobalData<LASCoordinateSet>&,
                                     LASGlobalData<LASCoordinateSet>&,
                                     LASEndcapAlignmentParameterSet&);

  void ApplyATMaskingCorrections(LASGlobalData<LASCoordinateSet>&,
                                 LASGlobalData<LASCoordinateSet>&,
                                 LASBarrelAlignmentParameterSet&);

  const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> topoToken_;
  const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> geomToken_;
  const edm::ESGetToken<GeometricDet, IdealGeometryRecord> geomDetToken_;
  const edm::ESGetToken<PTrackerParameters, PTrackerParametersRcd> ptpToken_;
  const edm::ESGetToken<Alignments, GlobalPositionRcd> gprToken_;
  const edm::ESGetToken<SiStripPedestals, SiStripPedestalsRcd> stripPedestalsToken_;

  int theEvents;

  bool theDoPedestalSubtraction;

  bool theUseMinuitAlgorithm;

  bool theApplyBeamKinkCorrections;

  double peakFinderThreshold;

  bool enableJudgeZeroFilter;

  unsigned int judgeOverdriveThreshold;

  bool updateFromInputGeometry;

  bool misalignedByRefGeometry;

  bool theStoreToDB;

  // digi producer
  std::vector<edm::ParameterSet> theDigiProducersList;

  bool theSaveHistograms;

  int theCompression;

  std::string theFileName;

  std::vector<unsigned int> theMaskTecModules;
  std::vector<unsigned int> theMaskAtModules;

  bool theSetNominalStrips;

  // this object can judge if
  // a LASModuleProfile is usable for position measurement
  LASProfileJudge judge;

  // colection of constants
  LASConstants theLasConstants;

  // the detector ids for all the modules
  LASGlobalData<unsigned int> detectorId;

  // the detector ids for the doubly hit modules in the TECs
  std::vector<unsigned int> tecDoubleHitDetId;

  // all the 474 profiles for the pedestals
  LASGlobalData<LASModuleProfile> pedestalProfiles;

  LASGlobalData<LASModuleProfile> currentDataProfiles;

  // summed data profiles
  LASGlobalData<LASModuleProfile> collectedDataProfiles;

  // 474 names for retrieving data from the branches
  LASGlobalData<std::string> theProfileNames;

  // number of accepted profiles for each module
  LASGlobalData<int> numberOfAcceptedProfiles;

  // hit map for the current event (int=0,1)
  // which is needed for branching into AT or TEC mode
  LASGlobalData<int> isAcceptedProfile;

  // histograms for the summed profiles;
  // these are needed for the fitting procedure (done by ROOT)
  LASGlobalData<TH1D*> summedHistograms;

  // container for nominal module positions
  LASGlobalData<LASCoordinateSet> nominalCoordinates;

  // a class for easy looping over LASGlobalData objects,
  // avoids nested for-statements
  LASGlobalLoop moduleLoop;

  TFile* theFile;
  TDirectory* singleModulesDir;

  edm::ESHandle<GeometricDet> gD;
  edm::ESHandle<TrackerGeometry> theTrackerGeometry;
  edm::ESHandle<Alignments> theGlobalPositionRcd;

  AlignableTracker* theAlignableTracker;

  std::string theAlignRecordName, theErrorRecordName;

  bool firstEvent_;
};
#endif