Add x_sep equals 0 in cone shaped

rest-for-physics · Feb 21, 2024 · a402d7f · a402d7f
1 parent cc7e1e4
commit a402d7f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/TRestAxionWolterOptics.cxx b/src/TRestAxionWolterOptics.cxx
@@ -169,15 +169,22 @@ void TRestAxionWolterOptics::Initialize() {
     SetLibraryVersion(LIBRARY_VERSION);
 
     fR1 = GetR1();
+    fR2 = GetR2();
     fR3 = GetR3();
+    fR4 = GetR4();
     fR5 = GetR5();
     fAlpha = GetAlpha();
     fThickness = GetThickness();
 
     fXSep.clear();
     for (unsigned int n = 0; n < fAlpha.size(); n++)
-        fXSep.push_back(2 * (fR1[n] - fR3[n] - fMirrorLength * TMath::Sin(fAlpha[n])) /
+        if (fR2[n]==fR3[n]){
+            fXSep.push_back(0);
+        }
+        else {
+            fXSep.push_back(2 * (fR1[n] - fR3[n] - fMirrorLength * TMath::Sin(fAlpha[n])) /
                         TMath::Tan(fAlpha[n]));
+        }
 
     if (fAlpha.size() == 0) return;