CMSSW_4_4_3_patch1/src/Alignment/HIPAlignmentAlgorithm/interface/HIPAlignmentAlgorithm.h

Go to the documentation of this file.

#ifndef Alignment_HIPAlignmentAlgorithm_HIPAlignmentAlgorithm_h
#define Alignment_HIPAlignmentAlgorithm_HIPAlignmentAlgorithm_h

#include "Alignment/CommonAlignmentAlgorithm/interface/AlignmentAlgorithmBase.h"
#include "Alignment/CommonAlignment/interface/AlignableDetOrUnitPtr.h"
#include "Alignment/CommonAlignmentAlgorithm/interface/AlignmentIORoot.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "Riostream.h"

#include "DataFormats/Alignment/interface/AlignmentClusterFlag.h"        
#include "DataFormats/Alignment/interface/AliClusterValueMap.h"          
#include "Utilities/General/interface/ClassName.h"       
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit1D.h"       
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2D.h"       
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHit.h"         
#include "RecoTracker/TransientTrackingRecHit/interface/TSiStripRecHit1D.h"      
#include "RecoTracker/TransientTrackingRecHit/interface/TSiStripRecHit2DLocalPos.h"      
#include "RecoTracker/TransientTrackingRecHit/interface/TSiPixelRecHit.h"

class AlignableNavigator;
class TFile;
class TTree;

class HIPAlignmentAlgorithm : public AlignmentAlgorithmBase
{

 public:
  
  HIPAlignmentAlgorithm(const edm::ParameterSet& cfg);

  ~HIPAlignmentAlgorithm() {};

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

  void terminate(void);

  void startNewLoop(void);

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

 private:

  // private member functions

  bool processHit1D(const AlignableDetOrUnitPtr alidet,
                    const Alignable* ali,
                    const TrajectoryStateOnSurface & tsos,
                    const TransientTrackingRecHit* hit);

  bool processHit2D(const AlignableDetOrUnitPtr alidet,
                    const Alignable* ali,
                    const TrajectoryStateOnSurface & tsos,
                    const TransientTrackingRecHit* hit);
  
  int readIterationFile(std::string filename);
  void writeIterationFile(std::string filename, int iter);
  void setAlignmentPositionError(void);
  double calcAPE(double* par, int iter, double function);
  void bookRoot(void);
  void fillRoot(void);
  bool calcParameters(Alignable* ali);
  void collector(void);
  int  fillEventwiseTree(const char *filename, int iter, int ierr);
  // private data members

  AlignmentParameterStore* theAlignmentParameterStore;
  std::vector<Alignable*> theAlignables;
  AlignableNavigator* theAlignableDetAccessor;

  AlignmentIORoot theIO;
  int ioerr;
  int theIteration;

  // steering parameters

  // verbosity flag
  bool verbose;
  // names of IO root files
  std::string outfile,outfile2,outpath,suvarfile,sparameterfile;
  std::string struefile,smisalignedfile,salignedfile,siterationfile,ssurveyfile;

  // alignment position error parameters
  bool theApplyAPE;
  std::vector<edm::ParameterSet> theAPEParameterSet;
  std::vector<std::pair<std::vector<Alignable*>, std::vector<double> > > theAPEParameters;
  // max allowed pull (residual / uncertainty) on a hit used in alignment
  double theMaxAllowedHitPull;
  // min number of hits on alignable to calc parameters
  int theMinimumNumberOfHits;
  // max allowed rel error on parameter (else not used)
  double theMaxRelParameterError;
  // collector mode (parallel processing)
  bool isCollector;
  int theCollectorNJobs;
  std::string theCollectorPath;
  int theEventPrescale,theCurrentPrescale;
  bool theFillTrackMonitoring;

  std::vector<align::StructureType> theLevels; // for survey residuals

  // root tree variables
  TFile* theFile;
  TTree* theTree; // event-wise tree
  TFile* theFile2;
  TTree* theTree2; // alignable-wise tree
  TFile* theFile3;
  TTree* theTree3; // survey tree

  // variables for event-wise tree
  static const int MAXREC = 99;
  //int m_Run,m_Event;
  int m_Ntracks,m_Nhits[MAXREC],m_nhPXB[MAXREC],m_nhPXF[MAXREC];
  float m_Pt[MAXREC],m_Eta[MAXREC],m_Phi[MAXREC],m_Chi2n[MAXREC],m_P[MAXREC],m_d0[MAXREC],m_dz[MAXREC];

  // variables for alignable-wise tree
  int m2_Nhit,m2_Type,m2_Layer;
  float m2_Xpos, m2_Ypos, m2_Zpos, m2_Eta, m2_Phi; 
  align::ID m2_Id;
  align::StructureType m2_ObjId;

  // variables for survey tree 
  align::ID m3_Id;
  align::StructureType m3_ObjId;
  float m3_par[6];
};

#endif