diff options
| -rw-r--r-- | beam_tracing/python/drawPrismDisk.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/beam_tracing/python/drawPrismDisk.py b/beam_tracing/python/drawPrismDisk.py index a099096..a18d94a 100644 --- a/beam_tracing/python/drawPrismDisk.py +++ b/beam_tracing/python/drawPrismDisk.py @@ -74,8 +74,9 @@ def drawPrismDisk(prismAngleInDegrees, nDisk, nPrism, diskX, coupDesc): xOutStart = xCross
yOutStart = yCross
xOutStop = 1.5
- xBeamOut = np.linspace(xOutStart, xOutStop)
- yBeamOut = yOutStart + np.tan(thetaAirHorizon) * (xBeamOut - xOutStart)
+ yOutStop = 2.0
+ yBeamOut = np.linspace(yOutStart, yOutStop)
+ xBeamOut = xOutStart + (1/np.tan(thetaAirHorizon)) * (yBeamOut - yOutStart)
# this keeps the beam from going farther than necessary
i = 0
for value in yBeamOut:
|