Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 12:55:04

0001 #include <cppunit/extensions/HelperMacros.h>
0002 
0003 #include "DetectorDescription/DDCMS/interface/DDFilteredView.h"
0004 #include "DetectorDescription/DDCMS/interface/DDDetector.h"
0005 #include "FWCore/ParameterSet/interface/FileInPath.h"
0006 #include "DD4hep/Detector.h"
0007 
0008 #include <string>
0009 #include <memory>
0010 #include <vector>
0011 
0012 #include "cppunit/TestAssert.h"
0013 #include "cppunit/TestFixture.h"
0014 
0015 using namespace cms;
0016 
0017 class testDDFilteredViewGet : public CppUnit::TestFixture {
0018   CPPUNIT_TEST_SUITE(testDDFilteredViewGet);
0019   CPPUNIT_TEST(checkFilteredView);
0020   CPPUNIT_TEST_SUITE_END();
0021 
0022 public:
0023   void setUp() override;
0024   void tearDown() override {}
0025   void checkFilteredView();
0026 
0027 private:
0028   void printMe(const cms::DDFilteredView&);
0029   double refRadLength_ = 0.03142;
0030   double refXi_ = 6.24526e-05;
0031   int refCopyNoTag_ = 1000;
0032   int refCopyNoOffset_ = 100;
0033   int refCopyNo_ = 1;
0034   std::string fileName_;
0035   std::vector<int> refPos_{0, 0, 6, 2, 2};
0036 };
0037 
0038 CPPUNIT_TEST_SUITE_REGISTRATION(testDDFilteredViewGet);
0039 
0040 void testDDFilteredViewGet::setUp() {
0041   fileName_ = edm::FileInPath("Geometry/CMSCommonData/data/dd4hep/cmsExtendedGeometry2021.xml").fullPath();
0042 }
0043 
0044 void testDDFilteredViewGet::checkFilteredView() {
0045   std::unique_ptr<DDDetector> det = std::make_unique<DDDetector>("DUMMY", fileName_);
0046   DDFilteredView fview(det.get(), det->description()->worldVolume());
0047 
0048   int count = 1;
0049   auto testPos = fview.navPos();
0050   while (fview.next(0)) {
0051     std::cout << "#" << count << ": ";
0052     printMe(fview);
0053 
0054     if (count == 45) {
0055       testPos = fview.navPos();
0056     }
0057     if (count == 50) {
0058       break;
0059     }
0060     count++;
0061   }
0062 
0063   // world_volume/OCMS_1/CMSE_1/Tracker_1/PixelBarrel_1/pixbarlayer0:PixelBarrelLayer0_1/PixelBarrelLadderFull0_6/PixelBarrelModuleBoxFull_1/PixelBarrelModuleFullPlus_4/PixelBarrelSensorFull_1/PixelBarrelActiveFull0_1
0064   //
0065   std::vector<int> activeVol{0, 0, 6, 2, 93, 12, 1, 7, 1, 0};
0066   fview.goTo(activeVol);
0067   printMe(fview);
0068   fview.findSpecPar("TrackerRadLength", "TrackerXi");
0069 
0070   double radLength = fview.getNextValue("TrackerRadLength");
0071   double xi = fview.getNextValue("TrackerXi");
0072   CPPUNIT_ASSERT(abs(radLength - refRadLength_) < 10e-6);
0073   CPPUNIT_ASSERT(abs(xi - refXi_) < 10e-6);
0074 
0075   std::cout << "TrackerRadLength = " << radLength << "\nTrackerXi = " << xi << "\n";
0076   auto vals = fview.getValuesNS<std::string>("TrackerRadLength");
0077   for (auto i : vals)
0078     std::cout << "TrackerRadLength = " << i << "\n";
0079 
0080   std::cout << "\n==== Let's go to #45\n";
0081   fview.goTo(testPos);
0082   printMe(fview);
0083 
0084   int i = 0;
0085   for (auto it : fview.navPos()) {
0086     CPPUNIT_ASSERT(it == testPos[i++]);
0087   }
0088   i = 0;
0089   for (auto it : testPos) {
0090     CPPUNIT_ASSERT(it == refPos_[i++]);
0091   }
0092 
0093   // Start with Muon
0094   std::cout << "\n==== Let's go to Muon\n";
0095   fview.goTo({0, 0, 8});
0096   printMe(fview);
0097 
0098   CPPUNIT_ASSERT(fview.fullName() == "muonBase:MUON");
0099 
0100   // Go to the first daughter
0101   fview.next(0);
0102   printMe(fview);
0103 
0104   // Use it as an escape level
0105   int startLevel = fview.level();
0106 
0107   count = 1;
0108 
0109   do {
0110     std::cout << "#" << count++ << ": ";
0111     std::cout << "started at level " << startLevel << "\n";
0112     printMe(fview);
0113 
0114   } while (fview.next(0) && fview.level() < startLevel);
0115 
0116   std::cout << "\n==== Continue iteration\n";
0117 
0118   count = 1;
0119   fview.next(0);
0120   startLevel = fview.level();
0121   printMe(fview);
0122 
0123   do {
0124     std::cout << "#" << count++;
0125     std::cout << " started at level " << startLevel << ":\n";
0126     printMe(fview);
0127   } while (fview.next(0) && fview.level() < startLevel);
0128 
0129   fview.next(0);
0130   printMe(fview);
0131 
0132   std::cout << "\n==== Let's do it again, go to Muon\n";
0133   fview.goTo({0, 0, 8});
0134   printMe(fview);
0135   CPPUNIT_ASSERT(fview.name() == "MUON");
0136 
0137   // Go to the first daughter
0138   fview.next(0);
0139   printMe(fview);
0140 
0141   // Use it as an escape level
0142   startLevel = fview.level();
0143 
0144   count = 1;
0145 
0146   do {
0147     std::cout << "#" << count++ << ": ";
0148     std::cout << "started at level " << startLevel << "\n";
0149     printMe(fview);
0150 
0151   } while (fview.next(0) && fview.level() < startLevel);
0152 
0153   fview.goTo({0, 0, 8, 0});
0154   printMe(fview);
0155   fview.findSpecPar("CopyNoTag", "CopyNoOffset");
0156 
0157   auto tag = fview.getNextValue("CopyNoTag");
0158   auto offset = fview.getNextValue("CopyNoOffset");
0159   std::cout << "CopyNoTag = " << tag << "\n";
0160   std::cout << "CopyNoOffset = " << fview.getNextValue("CopyNoOffset") << "\n";
0161   CPPUNIT_ASSERT(refCopyNoTag_ == tag);
0162   CPPUNIT_ASSERT(refCopyNoOffset_ == offset);
0163 
0164   const auto& nodes = fview.history();
0165   int ctr(0);
0166   for (const auto& t : nodes.tags) {
0167     std::cout << t << ": " << nodes.offsets[ctr] << ", " << nodes.copyNos[ctr] << "\n";
0168     CPPUNIT_ASSERT(refCopyNoTag_ == t);
0169     CPPUNIT_ASSERT(refCopyNoOffset_ == nodes.offsets[ctr]);
0170     CPPUNIT_ASSERT(refCopyNo_ == nodes.copyNos[ctr]);
0171     ctr++;
0172   }
0173 }
0174 
0175 void testDDFilteredViewGet::printMe(const cms::DDFilteredView& fview) {
0176   std::cout << ">>> " << fview.level() << " level: " << fview.name() << " is a "
0177             << cms::dd::name(cms::DDSolidShapeMap, fview.shape()) << "\n";
0178   std::cout << "Full path to it is " << fview.path() << "\n";
0179 
0180   auto copies = fview.copyNos();
0181   std::cout << "    copy Nos: ";
0182   std::for_each(copies.rbegin(), copies.rend(), [](const auto& it) { std::cout << it << ", "; });
0183   std::cout << "\n    levels  : ";
0184   for (auto it : fview.navPos()) {
0185     std::cout << it << ", ";
0186   }
0187   std::cout << "\n";
0188 }