From 7f8efc35712337c6fc353f0bae4111206a5fed25 Mon Sep 17 00:00:00 2001
From: lintao <lintao51@gmail.com>
Date: Thu, 27 May 2021 21:01:28 +0800
Subject: [PATCH] WIP: calculate the index.
---
.../src/FieldMapFileProvider.cpp | 53 +++++++++++++++++--
.../src/FieldMapFileProvider.h | 3 ++
2 files changed, 51 insertions(+), 5 deletions(-)
diff --git a/Detector/MagneticFieldMap/src/FieldMapFileProvider.cpp b/Detector/MagneticFieldMap/src/FieldMapFileProvider.cpp
index 388ea22e..2865f6bf 100644
--- a/Detector/MagneticFieldMap/src/FieldMapFileProvider.cpp
+++ b/Detector/MagneticFieldMap/src/FieldMapFileProvider.cpp
@@ -9,24 +9,62 @@
#include <sstream>
FieldMapFileProvider::FieldMapFileProvider(const std::string& url_)
- : m_url(url_), rBinMax(-1), zBinMax(-1), drBin(-1), dzBin(-1) {
+ : m_url(url_), nr(-1), nz(-1),
+ rBinMin(-1), zBinMin(-1),
+ rBinMax(-1), zBinMax(-1),
+ drBin(-1), dzBin(-1) {
init();
}
int FieldMapFileProvider::rBinIdx(double r) {
+ //
+ // | --- | --- | --- |
+ // ^ ^ ^
+ // rmin | rmax
+ // r
+ // |
+ // return
+
+
double absr = std::fabs(r);
+
+ int idx = -1;
+
+ if ( rBinMin <= absr && absr < rBinMax) {
+ idx = (absr - rBinMin) / drBin;
+ }
- return -1;
+ return idx;
}
int FieldMapFileProvider::zBinIdx(double z) {
double absz = std::fabs(z);
+
+ int idx = -1;
+
+ if ( zBinMin <= absz && absz < zBinMax) {
+ idx = (absz - zBinMin) / drBin;
+ }
- return -1;
+ return idx;
}
void FieldMapFileProvider::access(int rbin, int zbin, double& Br, double& Bz) {
+ // the valid bin should between [0, n)
+ // if the point is not in the valid region, return 0
+ if ((rbin < 0 || rbin >= nr) || (zbin < 0 || zbin >= nr)) {
+ Br = 0;
+ Bz = 0;
+ }
+ // convert to the internal table (with left col and top row)
+ int ridx = rbin+1;
+ int zidx = zbin+1;
+
+ int globalidx = ridx + zidx*(nr+1);
+
+ Br = Brdata[globalidx];
+ Bz = Bzdata[globalidx];
}
// ======================
@@ -120,14 +158,19 @@ bool FieldMapFileProvider::loadBr(const std::string& fn) {
nr = ncol-1; // skip the top row and left col
nz = nrow-1;
+ rBinMin = Brdata[1];
+ zBinMin = Brdata[ncol];
+
rBinMax = Brdata[ncol-1];
zBinMax = Brdata[ncol*(nrow-1)];
- drBin = rBinMax / (nr-1);
- dzBin = zBinMax / (nz-1);
+ drBin = (rBinMax-rBinMin) / (nr-1);
+ dzBin = (zBinMax-zBinMin) / (nz-1);
std::cout << "nr: " << nr << std::endl;
std::cout << "nz: " << nz << std::endl;
+ std::cout << "rBinMin: " << rBinMin << std::endl;
+ std::cout << "zBinMin: " << zBinMin << std::endl;
std::cout << "rBinMax: " << rBinMax << std::endl;
std::cout << "zBinMax: " << zBinMax << std::endl;
std::cout << "drBin: " << drBin << std::endl;
diff --git a/Detector/MagneticFieldMap/src/FieldMapFileProvider.h b/Detector/MagneticFieldMap/src/FieldMapFileProvider.h
index 3661863b..a6712c66 100644
--- a/Detector/MagneticFieldMap/src/FieldMapFileProvider.h
+++ b/Detector/MagneticFieldMap/src/FieldMapFileProvider.h
@@ -39,6 +39,9 @@ private:
int nr; // include the two endpoints, so nr = nrBin + 1
int nz;
+ double rBinMin;
+ double zBinMin;
+
double rBinMax;
double zBinMax;
--
GitLab