00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015 #include "FWCore/Framework/interface/ESHandle.h"
00016 #include "FWCore/Framework/interface/ESTransientHandle.h"
00017
00018
00019 #include "Alignment/MuonAlignment/interface/MuonAlignmentOutputXML.h"
00020 #include "Alignment/CommonAlignment/interface/AlignableObjectId.h"
00021 #include "DataFormats/MuonDetId/interface/DTChamberId.h"
00022 #include "DataFormats/MuonDetId/interface/DTSuperLayerId.h"
00023 #include "DataFormats/MuonDetId/interface/DTLayerId.h"
00024 #include "DataFormats/MuonDetId/interface/CSCDetId.h"
00025 #include "Geometry/Records/interface/MuonNumberingRecord.h"
00026 #include "Geometry/DTGeometryBuilder/src/DTGeometryBuilderFromDDD.h"
00027 #include "Geometry/CSCGeometryBuilder/src/CSCGeometryBuilderFromDDD.h"
00028 #include "Geometry/DTGeometry/interface/DTGeometry.h"
00029 #include "Geometry/CSCGeometry/interface/CSCGeometry.h"
00030 #include "Alignment/CommonAlignment/interface/SurveyDet.h"
00031 #include "CondFormats/Alignment/interface/AlignmentErrors.h"
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044 MuonAlignmentOutputXML::MuonAlignmentOutputXML(const edm::ParameterSet &iConfig)
00045 : m_fileName(iConfig.getParameter<std::string>("fileName"))
00046 , m_survey(iConfig.getParameter<bool>("survey"))
00047 , m_rawIds(iConfig.getParameter<bool>("rawIds"))
00048 , m_eulerAngles(iConfig.getParameter<bool>("eulerAngles"))
00049 , m_precision(iConfig.getParameter<int>("precision"))
00050 , m_suppressDTBarrel(iConfig.getUntrackedParameter<bool>("suppressDTBarrel", false))
00051 , m_suppressDTWheels(iConfig.getUntrackedParameter<bool>("suppressDTWheels", false))
00052 , m_suppressDTStations(iConfig.getUntrackedParameter<bool>("suppressDTStations", false))
00053 , m_suppressDTChambers(iConfig.getUntrackedParameter<bool>("suppressDTChambers", false))
00054 , m_suppressDTSuperLayers(iConfig.getUntrackedParameter<bool>("suppressDTSuperLayers", false))
00055 , m_suppressDTLayers(iConfig.getUntrackedParameter<bool>("suppressDTLayers", false))
00056 , m_suppressCSCEndcaps(iConfig.getUntrackedParameter<bool>("suppressCSCEndcaps", false))
00057 , m_suppressCSCStations(iConfig.getUntrackedParameter<bool>("suppressCSCStations", false))
00058 , m_suppressCSCRings(iConfig.getUntrackedParameter<bool>("suppressCSCRings", false))
00059 , m_suppressCSCChambers(iConfig.getUntrackedParameter<bool>("suppressCSCChambers", false))
00060 , m_suppressCSCLayers(iConfig.getUntrackedParameter<bool>("suppressCSCLayers", false))
00061 {
00062 std::string str_relativeto = iConfig.getParameter<std::string>("relativeto");
00063
00064 if (str_relativeto == std::string("none")) {
00065 m_relativeto = 0;
00066 }
00067 else if (str_relativeto == std::string("ideal")) {
00068 m_relativeto = 1;
00069 }
00070 else if (str_relativeto == std::string("container")) {
00071 m_relativeto = 2;
00072 }
00073 else {
00074 throw cms::Exception("BadConfig") << "relativeto must be \"none\", \"ideal\", or \"container\"" << std::endl;
00075 }
00076 }
00077
00078
00079
00080
00081
00082
00083 MuonAlignmentOutputXML::~MuonAlignmentOutputXML()
00084 {
00085 }
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101
00102
00103 void MuonAlignmentOutputXML::write(AlignableMuon *alignableMuon, const edm::EventSetup &iSetup) const {
00104 std::ofstream outputFile(m_fileName.c_str());
00105 outputFile << std::setprecision(m_precision) << std::fixed;
00106
00107 outputFile << "<?xml version=\"1.0\" encoding=\"UTF-8\"?>" << std::endl;
00108 outputFile << "<?xml-stylesheet type=\"text/xml\" href=\"MuonAlignment.xsl\"?>" << std::endl;
00109 outputFile << "<MuonAlignment>" << std::endl << std::endl;
00110
00111 std::map<align::ID, CLHEP::HepSymMatrix> errors;
00112 AlignmentErrors *dtErrors = alignableMuon->dtAlignmentErrors();
00113 AlignmentErrors *cscErrors = alignableMuon->cscAlignmentErrors();
00114 for (std::vector<AlignTransformError>::const_iterator dtError = dtErrors->m_alignError.begin(); dtError != dtErrors->m_alignError.end(); ++dtError) {
00115 errors[dtError->rawId()] = dtError->matrix();
00116 }
00117 for (std::vector<AlignTransformError>::const_iterator cscError = cscErrors->m_alignError.begin(); cscError != cscErrors->m_alignError.end(); ++cscError) {
00118 errors[cscError->rawId()] = cscError->matrix();
00119 }
00120
00121 std::vector<Alignable*> barrels = alignableMuon->DTBarrel();
00122 std::vector<Alignable*> endcaps = alignableMuon->CSCEndcaps();
00123
00124 if (m_relativeto == 1) {
00125 edm::ESTransientHandle<DDCompactView> cpv;
00126 iSetup.get<IdealGeometryRecord>().get(cpv);
00127
00128 edm::ESHandle<MuonDDDConstants> mdc;
00129 iSetup.get<MuonNumberingRecord>().get(mdc);
00130 DTGeometryBuilderFromDDD DTGeometryBuilder;
00131 CSCGeometryBuilderFromDDD CSCGeometryBuilder;
00132
00133 boost::shared_ptr<DTGeometry> dtGeometry(new DTGeometry );
00134 DTGeometryBuilder.build(dtGeometry, &(*cpv), *mdc);
00135
00136 boost::shared_ptr<CSCGeometry> boost_cscGeometry(new CSCGeometry);
00137 CSCGeometryBuilder.build(boost_cscGeometry, &(*cpv), *mdc);
00138
00139 AlignableMuon ideal_alignableMuon(&(*dtGeometry), &(*boost_cscGeometry));
00140
00141 std::vector<Alignable*> ideal_barrels = ideal_alignableMuon.DTBarrel();
00142 std::vector<Alignable*> ideal_endcaps = ideal_alignableMuon.CSCEndcaps();
00143
00144 writeComponents(barrels, ideal_barrels, errors, outputFile, true);
00145 writeComponents(endcaps, ideal_endcaps, errors, outputFile, false);
00146 }
00147 else {
00148 std::vector<Alignable*> empty1, empty2;
00149
00150 writeComponents(barrels, empty1, errors, outputFile, true);
00151 writeComponents(endcaps, empty2, errors, outputFile, false);
00152 }
00153
00154 outputFile << "</MuonAlignment>" << std::endl;
00155 }
00156
00157 void MuonAlignmentOutputXML::writeComponents(std::vector<Alignable*> &alignables, std::vector<Alignable*> &ideals,
00158 std::map<align::ID, CLHEP::HepSymMatrix>& errors, std::ofstream &outputFile, bool DT) const {
00159 std::vector<Alignable*>::const_iterator ideal = ideals.begin();
00160 for (std::vector<Alignable*>::const_iterator alignable = alignables.begin(); alignable != alignables.end(); ++alignable) {
00161 if (m_survey && (*alignable)->survey() == NULL) {
00162 throw cms::Exception("Alignment") << "SurveyDets must all be defined when writing to XML" << std::endl;
00163 }
00164
00165 align::StructureType alignableObjectId = (*alignable)->alignableObjectId();
00166
00167 if ((alignableObjectId == align::AlignableDTBarrel && !m_suppressDTBarrel) ||
00168 (alignableObjectId == align::AlignableDTWheel && !m_suppressDTWheels) ||
00169 (alignableObjectId == align::AlignableDTStation && !m_suppressDTStations) ||
00170 (alignableObjectId == align::AlignableDTChamber && !m_suppressDTChambers) ||
00171 (DT && alignableObjectId == align::AlignableDTSuperLayer && !m_suppressDTSuperLayers) ||
00172 (DT && alignableObjectId == align::AlignableDetUnit && !m_suppressDTLayers) ||
00173 (alignableObjectId == align::AlignableCSCEndcap && !m_suppressCSCEndcaps) ||
00174 (alignableObjectId == align::AlignableCSCStation && !m_suppressCSCStations) ||
00175 (alignableObjectId == align::AlignableCSCRing && !m_suppressCSCRings) ||
00176 (alignableObjectId == align::AlignableCSCChamber && !m_suppressCSCChambers) ||
00177 (!DT && alignableObjectId == align::AlignableDetUnit && !m_suppressCSCLayers)) {
00178
00179 unsigned int rawId = (*alignable)->geomDetId().rawId();
00180 outputFile << "<operation>" << std::endl;
00181
00182 if (DT) {
00183 if (m_rawIds && rawId != 0) {
00184 static AlignableObjectId converter;
00185 std::string typeName = converter.typeToName(alignableObjectId);
00186 if (alignableObjectId == align::AlignableDTSuperLayer) typeName = std::string("DTSuperLayer");
00187 if (alignableObjectId == align::AlignableDetUnit) typeName = std::string("DTLayer");
00188 outputFile << " <" << typeName << " rawId=\"" << rawId << "\" />" << std::endl;
00189 }
00190 else {
00191 if (alignableObjectId == align::AlignableDetUnit) {
00192 DTLayerId id(rawId);
00193 outputFile << " <DTLayer wheel=\"" << id.wheel() << "\" station=\"" << id.station() << "\" sector=\"" << id.sector() << "\" superlayer=\"" << id.superlayer() << "\" layer=\"" << id.layer() << "\" />" << std::endl;
00194 }
00195 else if (alignableObjectId == align::AlignableDTSuperLayer) {
00196 DTSuperLayerId id(rawId);
00197 outputFile << " <DTSuperLayer wheel=\"" << id.wheel() << "\" station=\"" << id.station() << "\" sector=\"" << id.sector() << "\" superlayer=\"" << id.superlayer() << "\" />" << std::endl;
00198 }
00199 else if (alignableObjectId == align::AlignableDTChamber) {
00200 DTChamberId id(rawId);
00201 outputFile << " <DTChamber wheel=\"" << id.wheel() << "\" station=\"" << id.station() << "\" sector=\"" << id.sector() << "\" />" << std::endl;
00202 }
00203
00204 else {
00205 DTChamberId id((*alignable)->id());
00206 if (alignableObjectId == align::AlignableDTStation) {
00207 outputFile << " <DTStation wheel=\"" << id.wheel() << "\" station=\"" << id.station() << "\" />" << std::endl;
00208 }
00209 else if (alignableObjectId == align::AlignableDTWheel) {
00210 outputFile << " <DTWheel wheel=\"" << id.wheel() << "\" />" << std::endl;
00211 }
00212 else if (alignableObjectId == align::AlignableDTBarrel) {
00213 outputFile << " <DTBarrel />" << std::endl;
00214 }
00215 else throw cms::Exception("Alignment") << "Unknown DT Alignable StructureType" << std::endl;
00216 }
00217
00218 }
00219 }
00220
00221 else {
00222 if (m_rawIds && rawId != 0) {
00223 static AlignableObjectId converter;
00224 std::string typeName = converter.typeToName(alignableObjectId);
00225 if (alignableObjectId == align::AlignableDetUnit) typeName = std::string("CSCLayer");
00226 outputFile << " <" << typeName << " rawId=\"" << rawId << "\" />" << std::endl;
00227 }
00228 else {
00229 if (alignableObjectId == align::AlignableDetUnit) {
00230 CSCDetId id(rawId);
00231 outputFile << " <CSCLayer endcap=\"" << id.endcap() << "\" station=\"" << id.station() << "\" ring=\"" << id.ring() << "\" chamber=\"" << id.chamber() << "\" layer=\"" << id.layer() << "\" />" << std::endl;
00232 }
00233 else if (alignableObjectId == align::AlignableCSCChamber) {
00234 CSCDetId id(rawId);
00235 outputFile << " <CSCChamber endcap=\"" << id.endcap() << "\" station=\"" << id.station() << "\" ring=\"" << id.ring() << "\" chamber=\"" << id.chamber() << "\" />" << std::endl;
00236 }
00237 else {
00238 CSCDetId id((*alignable)->id());
00239 if (alignableObjectId == align::AlignableCSCRing) {
00240 outputFile << " <CSCRing endcap=\"" << id.endcap() << "\" station=\"" << id.station() << "\" ring=\"" << id.ring() << "\" />" << std::endl;
00241 }
00242 else if (alignableObjectId == align::AlignableCSCStation) {
00243 outputFile << " <CSCStation endcap=\"" << id.endcap() << "\" station=\"" << id.station() << "\" />" << std::endl;
00244 }
00245 else if (alignableObjectId == align::AlignableCSCEndcap) {
00246 outputFile << " <CSCEndcap endcap=\"" << id.endcap() << "\" />" << std::endl;
00247 }
00248 else throw cms::Exception("Alignment") << "Unknown CSC Alignable StructureType" << std::endl;
00249
00250 }
00251
00252 }
00253 }
00254
00255 align::PositionType pos = (*alignable)->globalPosition();
00256 align::RotationType rot = (*alignable)->globalRotation();
00257
00258 if (m_survey) {
00259 pos = (*alignable)->survey()->position();
00260 rot = (*alignable)->survey()->rotation();
00261 }
00262
00263 std::string str_relativeto;
00264 if (m_relativeto == 0) {
00265 str_relativeto = std::string("none");
00266 }
00267
00268 else if (m_relativeto == 1) {
00269 if (ideal == ideals.end() || (*ideal)->alignableObjectId() != alignableObjectId || (*ideal)->id() != (*alignable)->id()) {
00270 throw cms::Exception("Alignment") << "AlignableMuon and ideal_AlignableMuon are out of sync!" << std::endl;
00271 }
00272
00273 align::PositionType idealPosition = (*ideal)->globalPosition();
00274 align::RotationType idealRotation = (*ideal)->globalRotation();
00275
00276 pos = align::PositionType(idealRotation * (pos.basicVector() - idealPosition.basicVector()));
00277 rot = rot * idealRotation.transposed();
00278
00279 str_relativeto = std::string("ideal");
00280 }
00281
00282 else if (m_relativeto == 2 && (*alignable)->mother() != NULL) {
00283 align::PositionType globalPosition = (*alignable)->mother()->globalPosition();
00284 align::RotationType globalRotation = (*alignable)->mother()->globalRotation();
00285
00286 pos = align::PositionType(globalRotation * (pos.basicVector() - globalPosition.basicVector()));
00287 rot = rot * globalRotation.transposed();
00288
00289 str_relativeto = std::string("container");
00290 }
00291
00292 else assert(false);
00293
00294 outputFile << " <setposition relativeto=\"" << str_relativeto << "\" "
00295 << "x=\"" << pos.x() << "\" y=\"" << pos.y() << "\" z=\"" << pos.z() << "\" ";
00296
00297 if (m_eulerAngles) {
00298 align::EulerAngles eulerAngles = align::toAngles(rot);
00299 outputFile << "alpha=\"" << eulerAngles(1) << "\" beta=\"" << eulerAngles(2) << "\" gamma=\"" << eulerAngles(3) << "\" />" << std::endl;
00300 }
00301
00302 else {
00303
00304
00305 double phix = atan2(rot.yz(), rot.zz());
00306 double phiy = asin(-rot.xz());
00307 double phiz = atan2(rot.xy(), rot.xx());
00308
00309 outputFile << "phix=\"" << phix << "\" phiy=\"" << phiy << "\" phiz=\"" << phiz << "\" />" << std::endl;
00310 }
00311
00312 if (m_survey) {
00313 align::ErrorMatrix err = (*alignable)->survey()->errors();
00314
00315 outputFile << " <setsurveyerr"
00316 << " xx=\"" << err(0,0) << "\" xy=\"" << err(0,1) << "\" xz=\"" << err(0,2) << "\" xa=\"" << err(0,3) << "\" xb=\"" << err(0,4) << "\" xc=\"" << err(0,5)
00317 << "\" yy=\"" << err(1,1) << "\" yz=\"" << err(1,2) << "\" ya=\"" << err(1,3) << "\" yb=\"" << err(1,4) << "\" yc=\"" << err(1,5)
00318 << "\" zz=\"" << err(2,2) << "\" za=\"" << err(2,3) << "\" zb=\"" << err(2,4) << "\" zc=\"" << err(2,5)
00319 << "\" aa=\"" << err(3,3) << "\" ab=\"" << err(3,4) << "\" ac=\"" << err(3,5)
00320 << "\" bb=\"" << err(4,4) << "\" bc=\"" << err(4,5)
00321 << "\" cc=\"" << err(5,5) << "\" />" << std::endl;
00322 }
00323
00324 else if (rawId != 0) {
00325 CLHEP::HepSymMatrix err = errors[(*alignable)->id()];
00326
00327 outputFile << " <setape xx=\"" << err(1,1) << "\" xy=\"" << err(1,2) << "\" xz=\"" << err(1,3)
00328 << "\" yy=\"" << err(2,2) << "\" yz=\"" << err(2,3) << "\" zz=\"" << err(3,3) << "\" />" << std::endl;
00329 }
00330
00331 outputFile << "</operation>" << std::endl << std::endl;
00332
00333 }
00334
00335
00336 if (ideal != ideals.end()) {
00337 std::vector<Alignable*> components = (*alignable)->components();
00338 std::vector<Alignable*> ideal_components = (*ideal)->components();
00339 writeComponents(components, ideal_components, errors, outputFile, DT);
00340 ++ideal;
00341 }
00342 else {
00343 std::vector<Alignable*> components = (*alignable)->components();
00344 std::vector<Alignable*> dummy;
00345 writeComponents(components, dummy, errors, outputFile, DT);
00346 }
00347
00348 }
00349 }
00350
00351
00352
00353
00354
00355
00356
00357