|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
|
#ifndef Alignment_HIPAlignmentAlgorithm_HIPAlignmentAlgorithm_h
#define Alignment_HIPAlignmentAlgorithm_HIPAlignmentAlgorithm_h
#include <vector>
#include "Alignment/CommonAlignmentAlgorithm/interface/AlignmentAlgorithmBase.h"
#include "Alignment/CommonAlignment/interface/Alignable.h"
#include "Alignment/CommonAlignment/interface/AlignableDetOrUnitPtr.h"
#include "Alignment/CommonAlignment/interface/AlignableObjectId.h"
#include "Alignment/CommonAlignment/interface/AlignableNavigator.h"
#include "Alignment/CommonAlignmentAlgorithm/interface/AlignmentIORoot.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.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 "Geometry/CommonTopologies/interface/SurfaceDeformation.h"
#include "Geometry/CommonTopologies/interface/SurfaceDeformationFactory.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"
#include "Geometry/Records/interface/IdealGeometryRecord.h"
#include "Alignment/HIPAlignmentAlgorithm/interface/HIPMonitorConfig.h"
#include "Alignment/HIPAlignmentAlgorithm/interface/HIPAlignableSpecificParameters.h"
#include "TFormula.h"
class TFile;
class TTree;
class HIPAlignmentAlgorithm : public AlignmentAlgorithmBase {
public:
/// Constructor
HIPAlignmentAlgorithm(const edm::ParameterSet& cfg, edm::ConsumesCollector& iC);
/// Destructor
~HIPAlignmentAlgorithm() override {}
/// Call at beginning of job
void initialize(const edm::EventSetup& setup,
AlignableTracker* tracker,
AlignableMuon* muon,
AlignableExtras* extras,
AlignmentParameterStore* store) override;
/// Call at end of job
void terminate(const edm::EventSetup& setup) override;
/// Called at start of new loop
void startNewLoop(void) override;
/// Run the algorithm
void run(const edm::EventSetup& setup, const EventInfo& eventInfo) override;
private:
bool processHit1D(const AlignableDetOrUnitPtr& alidet,
const Alignable* ali,
const HIPAlignableSpecificParameters* alispecifics,
const TrajectoryStateOnSurface& tsos,
const TrackingRecHit* hit,
double hitwt);
bool processHit2D(const AlignableDetOrUnitPtr& alidet,
const Alignable* ali,
const HIPAlignableSpecificParameters* alispecifics,
const TrajectoryStateOnSurface& tsos,
const TrackingRecHit* hit,
double hitwt);
int readIterationFile(std::string filename);
void writeIterationFile(std::string filename, int iter);
void setAlignmentPositionError(void);
double calcAPE(double* par, int iter, int function);
void bookRoot(void);
void fillAlignablesMonitor(const edm::EventSetup& setup);
bool calcParameters(Alignable* ali, int setDet, double start, double step);
void collector(void);
void collectMonitorTrees(const std::vector<std::string>& filenames);
HIPAlignableSpecificParameters* findAlignableSpecs(const Alignable* ali);
// private data members
const edm::ESGetToken<TrackerTopology, IdealGeometryRecord> topoToken_;
const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> topoToken2_;
std::unique_ptr<AlignableObjectId> alignableObjectId_;
AlignmentParameterStore* theAlignmentParameterStore;
align::Alignables theAlignables;
std::unique_ptr<AlignableNavigator> theAlignableDetAccessor;
AlignmentIORoot theIO;
int ioerr;
int theIteration;
// steering parameters
// verbosity flag
const bool verbose;
// Monitor configuration
HIPMonitorConfig theMonitorConfig;
const bool doTrackHitMonitoring;
// names of IO root files
std::string outfile2, outpath, suvarfilecore, suvarfile, sparameterfile;
std::string struefile, smisalignedfile, salignedfile, siterationfile, ssurveyfile;
bool themultiIOV;
std::vector<unsigned> theIOVrangeSet;
// alignment position error parameters
bool theApplyAPE;
std::vector<edm::ParameterSet> theAPEParameterSet;
std::vector<std::pair<align::Alignables, std::vector<double> > > theAPEParameters;
// Default alignment specifications
// - min number of hits on alignable to calc parameters
// - max allowed rel error on parameter (else not used)
// - max allowed pull (residual / uncertainty) on a hit used in alignment
HIPAlignableSpecificParameters defaultAlignableSpecs;
bool theApplyCutsPerComponent;
std::vector<edm::ParameterSet> theCutsPerComponent;
std::vector<HIPAlignableSpecificParameters> theAlignableSpecifics;
// collector mode (parallel processing)
bool isCollector;
int theCollectorNJobs;
std::string theCollectorPath;
int theDataGroup; // The data type specified in the cfg
bool trackPs, trackWt, IsCollision, uniEta, rewgtPerAli;
std::string uniEtaFormula;
double Scale, cos_cut, col_cut;
std::vector<double> SetScanDet;
std::unique_ptr<TFormula> theEtaFormula;
const std::vector<std::string> surveyResiduals_;
std::vector<align::StructureType> theLevels; // for survey residuals
// root tree variables
TFile* theTrackHitMonitorIORootFile;
TTree* theTrackMonitorTree; // event-wise tree
TTree* theHitMonitorTree; // hit-wise tree
TFile* theAlignablesMonitorIORootFile;
TTree* theAlignablesMonitorTree; // alignable-wise tree
TFile* theSurveyIORootFile;
TTree* theSurveyTree; // survey tree
// common variables for monitor trees
int m_datatype;
// variables for alignable-wise tree
align::ID m2_Id;
align::StructureType m2_ObjId;
int m2_Nhit, m2_Type, m2_Layer;
int m2_datatype;
float m2_Xpos, m2_Ypos, m2_Zpos;
SurfaceDeformationFactory::Type m2_dtype;
unsigned int m2_nsurfdef;
std::vector<float> m2_surfDef;
// variables for survey tree
align::ID m3_Id;
align::StructureType m3_ObjId;
float m3_par[6];
};
#endif
|