From e24c272856c28f3675496f0635b0d8694e6d7b6c Mon Sep 17 00:00:00 2001
From: Alexander Jentsch <ajentsch@bnl.gov>
Date: Wed, 11 Aug 2021 10:49:10 -0400
Subject: [PATCH] Trying to add occupancy plots for the B0.
---
.../b0_tracker/scripts/b0_tracker_hits.cxx | 34 +++++++++++++++++--
.../scripts/gen_forward_protons.cxx | 10 +++---
2 files changed, 38 insertions(+), 6 deletions(-)
diff --git a/benchmarks/b0_tracker/scripts/b0_tracker_hits.cxx b/benchmarks/b0_tracker/scripts/b0_tracker_hits.cxx
index 41ed7a90..898ecc3a 100644
--- a/benchmarks/b0_tracker/scripts/b0_tracker_hits.cxx
+++ b/benchmarks/b0_tracker/scripts/b0_tracker_hits.cxx
@@ -46,13 +46,43 @@ void b0_tracker_hits(const char* fname = "./sim_output/sim_forward_protons.root"
return result;
};
- auto d1 = d0.Define("hits_theta", hits_theta, {"B0TrackerHits"});
+ auto x_pos_disk_1 = [&](const std::vector<std::array<double, 2>>& xypos) {
+ std::vector<double> result;
+ for (const auto& h : xypos) {
+
+ result.push_back(h.at(0));
+ }
+ return result;
+ };
+
+ auto y_pos_disk_1 = [&](const std::vector<std::array<double, 2>>& xypos) {
+ std::vector<double> result;
+ for (const auto& h : xypos) {
+
+ result.push_back(h.at(1));
+ }
+ return result;
+ };
+
+ auto d1 = d0.Define("nhits", nhits, {"B0TrackerHits"})
+ .Define("xy_hit_pos", local_position, {"B0TrackerHits"})
+ .Define("x_pos", x_pos, {"xy_hit_pos"})
+ .Define("y_pos", y_pos, {"xy_hit_pos"});
- auto h1 = d1.Histo1D({"h1", "hits_theta", 100, 0,20}, "hits_theta");
+ auto h_local_pos = d1.Histo2D({"h_local_pos", ";x [mm]; y [mm] ", 100, -100.0, -200.0, 100, -100.0, 100.0}, "x_pos", "y_pos");
+
+
+ auto d2 = d0.Define("hits_theta", hits_theta, {"B0TrackerHits"});
+
+ auto h1 = d2.Histo1D({"h1", "hits_theta", 100, 0,20}, "hits_theta");
TCanvas* c = new TCanvas();
h1->DrawCopy();
c->SaveAs("results/b0_tracker_hits_theta.png");
c->SaveAs("results/b0_tracker_hits_theta.pdf");
+ h_local_pos->DrawCopy("colz");
+ c->SaveAs("results/b0_tracker_hits_occupancy_disk_1.png");
+ c->SaveAs("results/b0_tracker_hits_occupancy_disk_1.pdf");
+
auto n1 = h1->GetMean();
std::cout << "Polar angle of hits: " << n1 << std::endl;
diff --git a/benchmarks/b0_tracker/scripts/gen_forward_protons.cxx b/benchmarks/b0_tracker/scripts/gen_forward_protons.cxx
index 1773423d..ef69220f 100644
--- a/benchmarks/b0_tracker/scripts/gen_forward_protons.cxx
+++ b/benchmarks/b0_tracker/scripts/gen_forward_protons.cxx
@@ -17,13 +17,15 @@ using namespace HepMC3;
/** Generate electrons in the central region.
* This is for testing detectors in the "barrel" region.
*/
-void gen_forward_protons(int n_events = 100,
+void gen_forward_protons(int n_events = 1000,
const char* out_fname = "forward_protons.hepmc")
{
// generate protons in B0 acceptance - roughly 5 - 20 mrad
- double cos_theta_min = std::cos(0.005*(M_PI/180.0));
- double cos_theta_max = std::cos(0.020*(M_PI/180.0));
+ double cos_theta_min = 0.005; //std::cos(0.005*(M_PI/180.0));
+ double cos_theta_max = 0.020; //std::cos(0.020*(M_PI/180.0));
+
+
double partEnergyMin = 270.0; // xL 0.98
double partEnergyMax = 275.0; // top beam energy
@@ -52,7 +54,7 @@ void gen_forward_protons(int n_events = 100,
Double_t p = r1->Uniform(partEnergyMin, partEnergyMax);
Double_t phi = r1->Uniform(0.0, 2.0 * M_PI);
Double_t costh = r1->Uniform(cos_theta_min, cos_theta_max);
- Double_t th = std::acos(costh);
+ Double_t th = costh; //std::acos(costh);
Double_t px = p * std::cos(phi) * std::sin(th);
Double_t py = p * std::sin(phi) * std::sin(th);
Double_t pz = p * std::cos(th);
--
GitLab