From 16fd651e6f9ce0c670eefe047d337cbe7bbc8296 Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
Date: Thu, 18 Apr 2024 13:11:14 -0400
Subject: [PATCH] Add benchmarks/material_scan (#18)
---
.gitlab-ci.yml | 4 +-
Snakefile | 1 +
benchmarks/material_scan/Snakefile | 21 ++++
benchmarks/material_scan/config.yml | 13 +++
benchmarks/others/.gitignore | 0
benchmarks/others/config.yml | 75 -------------
benchmarks/others/materialScanEta.cxx | 132 -----------------------
benchmarks/others/materialScanEtaPhi.cxx | 74 -------------
8 files changed, 37 insertions(+), 283 deletions(-)
create mode 100644 benchmarks/material_scan/Snakefile
create mode 100644 benchmarks/material_scan/config.yml
delete mode 100644 benchmarks/others/.gitignore
delete mode 100644 benchmarks/others/config.yml
delete mode 100644 benchmarks/others/materialScanEta.cxx
delete mode 100644 benchmarks/others/materialScanEtaPhi.cxx
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 159dd6af..97f28ace 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -135,15 +135,15 @@ include:
- local: 'benchmarks/barrel_hcal/config.yml'
- local: 'benchmarks/zdc/config.yml'
- local: 'benchmarks/material_maps/config.yml'
+ - local: 'benchmarks/material_scan/config.yml'
- local: 'benchmarks/pid/config.yml'
- local: 'benchmarks/timing/config.yml'
- local: 'benchmarks/b0_tracker/config.yml'
- - local: 'benchmarks/others/config.yml'
deploy_results:
stage: deploy
needs:
- - ["collect_results:zdc","collect_results:barrel_ecal","collect_results:barrel_hcal","collect_results:materialscan"]
+ - ["collect_results:zdc","collect_results:barrel_ecal","collect_results:barrel_hcal","collect_results:material_scan"]
script:
- echo "deploy results!"
- find results -print | sort | tee summary.txt
diff --git a/Snakefile b/Snakefile
index 62c31acd..a543b210 100644
--- a/Snakefile
+++ b/Snakefile
@@ -20,6 +20,7 @@ else:
include: "benchmarks/backgrounds/Snakefile"
include: "benchmarks/barrel_ecal/Snakefile"
include: "benchmarks/ecal_gaps/Snakefile"
+include: "benchmarks/material_scan/Snakefile"
rule warmup_run:
diff --git a/benchmarks/material_scan/Snakefile b/benchmarks/material_scan/Snakefile
new file mode 100644
index 00000000..b295715a
--- /dev/null
+++ b/benchmarks/material_scan/Snakefile
@@ -0,0 +1,21 @@
+rule material_scan_fetch_script:
+ output:
+ "material_scan.py",
+ shell: """
+curl -L --output {output} https://github.com/eic/epic/raw/main/scripts/subdetector_tests/material_scan.py
+"""
+
+rule material_scan:
+ input:
+ script="material_scan.py",
+ output:
+ "{DETECTOR_CONFIG}/results/material_scan.png",
+ "{DETECTOR_CONFIG}/results/material_scan_agg.csv",
+ "{DETECTOR_CONFIG}/results/material_scan_details.pdf",
+ log:
+ "{DETECTOR_CONFIG}/results/material_scan.log",
+ shadow: "full" # avoid putting calibrations/fieldmaps to results
+ shell: """
+cd {wildcards.DETECTOR_CONFIG}/results
+python ../../{input.script} $DETECTOR_PATH/{wildcards.DETECTOR_CONFIG}.xml
+"""
diff --git a/benchmarks/material_scan/config.yml b/benchmarks/material_scan/config.yml
new file mode 100644
index 00000000..f8bc0822
--- /dev/null
+++ b/benchmarks/material_scan/config.yml
@@ -0,0 +1,13 @@
+bench:material_scan:
+ extends: .det_benchmark
+ stage: benchmarks
+ script:
+ - snakemake --cores 1 epic_craterlake/results/material_scan_details.pdf
+
+collect_results:material_scan:
+ extends: .det_benchmark
+ stage: collect
+ needs:
+ - ["bench:material_scan"]
+ script:
+ - ls -lrht
diff --git a/benchmarks/others/.gitignore b/benchmarks/others/.gitignore
deleted file mode 100644
index e69de29b..00000000
diff --git a/benchmarks/others/config.yml b/benchmarks/others/config.yml
deleted file mode 100644
index 4a375285..00000000
--- a/benchmarks/others/config.yml
+++ /dev/null
@@ -1,75 +0,0 @@
-variables:
- ETAMIN: "-4.5"
- ETAMAX: "+4.5"
- ETASTEP: "0.01"
- PHIMIN: "0.0"
- PHIMAX: "6.28318530718"
- PHISTEP: "0.01"
- TRACKING_RHOMAX: "103."
- TRACKING_ZNMAX: "191."
- TRACKING_ZPMAX: "350."
- ECAL_RHOMAX: "230."
- ECAL_ZNMAX: "355."
- ECAL_ZPMAX: "380."
-
-bench:materialScanEta:
- stage: benchmarks
- extends: .det_benchmark
- script:
- - echo ".x benchmarks/others/materialScanEta.cxx($ETAMIN, $ETAMAX, $ETASTEP, $PHIMIN)" | materialScan ${DETECTOR_PATH}/${DETECTOR_CONFIG}.xml -interactive
- - mkdir results/images/
- - mv materialScanEta.png results/images/materialScanEta.png
- - mv materialScanEta.pdf results/images/materialScanEta.pdf
-
-bench:materialScanEta:tracking:
- stage: benchmarks
- extends: .det_benchmark
- script:
- - echo ".x benchmarks/others/materialScanEta.cxx($ETAMIN, $ETAMAX, $ETASTEP, $PHIMIN, $TRACKING_RHOMAX, $TRACKING_ZNMAX, $TRACKING_ZPMAX)" | materialScan ${DETECTOR_PATH}/${DETECTOR_CONFIG}.xml -interactive
- - mkdir results/images/
- - mv materialScanEta.png results/images/materialScanEtaTracking.png
- - mv materialScanEta.pdf results/images/materialScanEtaTracking.pdf
-
-bench:materialScanEta:ecal:
- stage: benchmarks
- extends: .det_benchmark
- script:
- - echo ".x benchmarks/others/materialScanEta.cxx($ETAMIN, $ETAMAX, $ETASTEP, $PHIMIN, $ECAL_RHOMAX, $ECAL_ZNMAX, $ECAL_ZPMAX)" | materialScan ${DETECTOR_PATH}/${DETECTOR_CONFIG}.xml -interactive
- - mkdir results/images/
- - mv materialScanEta.png results/images/materialScanEtaEcal.png
- - mv materialScanEta.pdf results/images/materialScanEtaEcal.pdf
-
-bench:materialScanEtaPhi:
- stage: benchmarks
- extends: .det_benchmark
- script:
- - echo ".x benchmarks/others/materialScanEtaPhi.cxx($ETAMIN, $ETAMAX, $ETASTEP, $PHIMIN, $PHIMAX, $PHISTEP)" | materialScan ${DETECTOR_PATH}/${DETECTOR_CONFIG}.xml -interactive
- - mkdir results/images/
- - mv materialScanEtaPhi.png results/images/materialScanEtaPhi.png
- - mv materialScanEtaPhi.pdf results/images/materialScanEtaPhi.pdf
-
-bench:materialScanEtaPhi:tracking:
- stage: benchmarks
- extends: .det_benchmark
- script:
- - echo ".x benchmarks/others/materialScanEtaPhi.cxx($ETAMIN, $ETAMAX, $ETASTEP, $PHIMIN, $PHIMAX, $PHISTEP, $TRACKING_RHOMAX, $TRACKING_ZNMAX, $TRACKING_ZPMAX)" | materialScan ${DETECTOR_PATH}/${DETECTOR_CONFIG}.xml -interactive
- - mkdir results/images/
- - mv materialScanEtaPhi.png results/images/materialScanEtaPhiTracking.png
- - mv materialScanEtaPhi.pdf results/images/materialScanEtaPhiTracking.pdf
-
-bench:materialScanEtaPhi:ecal:
- stage: benchmarks
- extends: .det_benchmark
- script:
- - echo ".x benchmarks/others/materialScanEtaPhi.cxx($ETAMIN, $ETAMAX, $ETASTEP, $PHIMIN, $PHIMAX, $PHISTEP, $ECAL_RHOMAX, $ECAL_ZNMAX, $ECAL_ZPMAX)" | materialScan ${DETECTOR_PATH}/${DETECTOR_CONFIG}.xml -interactive
- - mkdir results/images/
- - mv materialScanEtaPhi.png results/images/materialScanEtaPhiEcal.png
- - mv materialScanEtaPhi.pdf results/images/materialScanEtaPhiEcal.pdf
-
-collect_results:materialscan:
- extends: .det_benchmark
- stage: collect
- needs:
- - ["bench:materialScanEta", "bench:materialScanEta:tracking", "bench:materialScanEta:ecal", "bench:materialScanEtaPhi", "bench:materialScanEtaPhi:tracking", "bench:materialScanEtaPhi:ecal"]
- script:
- - ls -lrht
diff --git a/benchmarks/others/materialScanEta.cxx b/benchmarks/others/materialScanEta.cxx
deleted file mode 100644
index 1dc43c3f..00000000
--- a/benchmarks/others/materialScanEta.cxx
+++ /dev/null
@@ -1,132 +0,0 @@
-#import <iostream>
-#import <vector>
-
-#include <DDRec/Material.h>
-#include <DDRec/MaterialScan.h>
-
-#include <TH1F.h>
-#include <THStack.h>
-
-Color_t color(const Material& m) {
- if (m.name() == std::string("Silicon")) return kGray;
- else if (m.name() == std::string("Aluminum")) return kAzure;
- else if (m.name() == std::string("CarbonFiber")) return kGray;
- else if (m.name() == std::string("Beryllium")) return kGreen;
- else if (m.name() == std::string("Gold")) return kYellow;
- else if (m.name() == std::string("Mylar")) return kGreen;
- else if (m.name() == std::string("Kapton")) return kGreen;
- else if (m.name() == std::string("Copper")) return kGreen;
- else if (m.name() == std::string("C2F6_DRICH")) return kOrange;
- else if (m.name() == std::string("Ar10CO2")) return kOrange;
- else if (m.name() == std::string("Aerogel")) return kPink;
- else if (m.name() == std::string("AerogelOptical")) return kPink;
- else if (m.name() == std::string("Aerogel_DRICH")) return kPink;
- else if (m.name() == std::string("Lead")) return kBlack;
- else if (m.name() == std::string("Steel235")) return kGray+2;
- else if (m.name() == std::string("TungstenDens24")) return kBlack;
- else if (m.name() == std::string("Polystyrene")) return kGray;
- else if (m.name() == std::string("PolystyreneFoam")) return kGray;
- else if (m.name() == std::string("Epoxy")) return kGray;
- else if (m.name() == std::string("PlasticScint")) return kGray;
- else if (m.name() == std::string("AcrylicOptical")) return kGray;
- else if (m.name() == std::string("Acrylic_DRICH")) return kGray;
- else if (m.name() == std::string("Quartz")) return kViolet;
- else if (m.name() == std::string("Air")) return kBlue;
- else if (m.name() == std::string("AirOptical")) return kBlue;
- else if (m.name() == std::string("Vacuum")) return kWhite;
- else {
- std::cout << "Unknown material: " << m.name() << std::endl;
- return kRed;
- }
-}
-
-void materialScanEta(
- double etamin = -2.0, // minimum eta
- double etamax = +1.0, // maximum eta
- double etastep = 0.2, // steps in eta
- double phi = 0.0, // phi angle for material scan
- double rhomax = 10000.0, // maximum distance from z axis
- double znmax = 10000.0, // maximum negative endcap z plane (positive number)
- double zpmax = 10000.0 // maximum positive endcap z plane (positive number)
-) {
- // check inputs
- if (etamin > etamax) {
- std::cout << "Error: ordered eta range required" << std::endl;
- return;
- }
- if (rhomax <= 0.0) {
- std::cout << "Error: positive rhomax required" << std::endl;
- return;
- }
- if (znmax <= 0.0) {
- std::cout << "Error: positive znmax required" << std::endl;
- return;
- }
- if (zpmax <= 0.0) {
- std::cout << "Error: positive zpmax required" << std::endl;
- return;
- }
-
- // get material scans
- size_t total{0};
- std::vector<dd4hep::rec::MaterialVec> scan;
- double x0{0}, y0{0}, z0{0};
- for (double eta = etamin; eta <= etamax + 0.5*etastep; eta += etastep) {
- double theta = 2.0 * (atan(1) - atan(exp(-eta))); // |theta| < 90 deg, cos(theta) > 0, sin(theta) can be 0
- double r = min((theta < 0? -znmax: zpmax) / sin(theta), rhomax / cos(theta)); // theta == 0 results in min(+inf, rhomax)
- double x = r * cos(theta) * cos(phi);
- double y = r * cos(theta) * sin(phi);
- double z = r * sin(theta);
- scan.emplace_back(gMaterialScan->scan(x0,y0,z0,x,y,z));
- total += scan.back().size();
- }
-
- // start creating histograms for stacking:
- // - start with first material layer at central eta bin
- // - pop front layers first positive
- size_t layer = 0;
- std::vector<TH1F> histograms;
- while (total > 0) {
- // find next layer, starting from center bin outwards
- size_t eta0 = static_cast<unsigned int>(std::rint(abs((etamax-etamin)/etastep/2)));
- for (size_t i = 0, j = eta0;
- i < scan.size();
- ++i, j += (2*eta0 < scan.size()? -1: +1) * (i <= 2*min(eta0,scan.size()-eta0-1)? (i%2==0? -i: +i): -1)) {
- if (scan.at(j).size() == 0) continue;
- // define layer
- auto layer_anchor = scan.at(j).at(0);
- histograms.emplace_back(Form("h%zu",layer++),layer_anchor.first.name(),scan.size(),etamin,etamax);
- //histograms.back().SetLineColor(color(layer_anchor.first));
- histograms.back().SetFillColor(color(layer_anchor.first));
- // add all bins to this layer
- for (size_t bin = 0; bin < scan.size(); bin++) {
- double X0{0};
- for (auto& mat_len: scan.at(bin)) {
- if (mat_len.first.name() == layer_anchor.first.name()) {
- X0 += mat_len.second / mat_len.first.radLength();
- scan.at(bin).erase(scan.at(bin).begin()); total--;
- } else {
- break;
- }
- }
- histograms.back().SetBinContent(bin+1, X0); // bins start at 1
- }
- }
- }
- std::cout << histograms.size() << " histograms created" << std::endl;
-
- // plot histograms as stack
- THStack hs("hs",Form("Material Scan (rho < %.0f cm, -%.0f cm < z < %.0f cm)", rhomax, znmax, zpmax));
- for (auto& h: histograms) {
- hs.Add(&h);
- }
- TCanvas cs("cs","Material Scan",1920,1080);
- auto pad = cs.cd();
- pad->SetLogy();
- hs.Draw();
- hs.GetXaxis()->SetTitle("eta");
- hs.GetYaxis()->SetTitle("Fraction X0");
- hs.SetMinimum(2.5e-3);
- cs.SaveAs("materialScanEta.png");
- cs.SaveAs("materialScanEta.pdf");
-}
diff --git a/benchmarks/others/materialScanEtaPhi.cxx b/benchmarks/others/materialScanEtaPhi.cxx
deleted file mode 100644
index 5051ca1b..00000000
--- a/benchmarks/others/materialScanEtaPhi.cxx
+++ /dev/null
@@ -1,74 +0,0 @@
-#import <iostream>
-#import <vector>
-
-#include <DDRec/Material.h>
-#include <DDRec/MaterialScan.h>
-
-#include <TH1F.h>
-#include <THStack.h>
-
-void materialScanEtaPhi(
- double etamin = -2.0, // minimum eta
- double etamax = +1.0, // maximum eta
- double etastep = 0.2, // steps in eta
- double phimin = 0.0, // minimum phi
- double phimax = 2.0 * M_PI, // maximum phi
- double phistep = M_PI / 2, // steps in phi
- double rhomax = 10000.0, // maximum distance from z axis
- double znmax = 10000.0, // maximum negative endcap z plane (positive number)
- double zpmax = 10000.0 // maximum positive endcap z plane (positive number)
-) {
- // check inputs
- if (etamin > etamax) {
- std::cout << "Error: ordered eta range required" << std::endl;
- return;
- }
- if (phimin > phimax) {
- std::cout << "Error: ordered phi range required" << std::endl;
- return;
- }
- if (rhomax <= 0.0) {
- std::cout << "Error: positive rhomax required" << std::endl;
- return;
- }
- if (znmax <= 0.0) {
- std::cout << "Error: positive znmax required" << std::endl;
- return;
- }
- if (zpmax <= 0.0) {
- std::cout << "Error: positive zpmax required" << std::endl;
- return;
- }
-
- // get material scans
- size_t total{0};
- double x0{0}, y0{0}, z0{0};
- TH2F h2("h2","Material Scan Eta vs Phi",
- (etamax-etamin)/etastep+1,etamin,etamax,
- (phimax-phimin)/phistep+1,phimin,phimax
- );
- for (double eta = etamin; eta <= etamax + 0.5*etastep; eta += etastep) {
- for (double phi = phimin; phi <= phimax + 0.5*phistep; phi += phistep) {
- double theta = 2.0 * (atan(1) - atan(exp(-eta))); // |theta| < 90 deg, cos(theta) > 0, sin(theta) can be 0
- double r = min((theta < 0? -znmax: zpmax) / sin(theta), rhomax / cos(theta)); // theta == 0 results in min(+inf, rhomax)
- double x = r * cos(theta) * cos(phi);
- double y = r * cos(theta) * sin(phi);
- double z = r * sin(theta);
- auto scan = gMaterialScan->scan(x0,y0,z0,x,y,z);
- double X0{0};
- for (auto& mat_len: scan)
- X0 += mat_len.second / mat_len.first.radLength();
- h2.Fill(eta,phi,X0);
- }
- }
-
- // plot histograms as stack
- TCanvas cs("cs","Material Scan Eta vs Phi",1920,1080);
- auto pad = cs.cd();
- cs.SetLogz();
- h2.Draw("colz");
- h2.GetXaxis()->SetTitle("eta");
- h2.GetYaxis()->SetTitle("phi");
- cs.SaveAs("materialScanEtaPhi.png");
- cs.SaveAs("materialScanEtaPhi.pdf");
-}
--
GitLab