From 91bcd6e01af4a57f1be0da51fc98313944e8315a Mon Sep 17 00:00:00 2001
From: Felix Hagemann <hagemann@mpp.mpg.de>
Date: Sat, 24 May 2025 15:24:36 +0200
Subject: [PATCH] Generalize plot recipe of `AbstractParticleSource`

---
 src/PlotRecipes/ChargeCloudModels.jl | 46 +++++++++++++++-------------
 1 file changed, 24 insertions(+), 22 deletions(-)

diff --git a/src/PlotRecipes/ChargeCloudModels.jl b/src/PlotRecipes/ChargeCloudModels.jl
index a9cf035f0..08fa72ef0 100644
--- a/src/PlotRecipes/ChargeCloudModels.jl
+++ b/src/PlotRecipes/ChargeCloudModels.jl
@@ -291,29 +291,31 @@ end
 
 @recipe function f(m::AbstractParticleSource; length = 0.01)
     
-    if iszero(m.opening_angle)
-        v = m.position
-        vnew = m.position + length * normalize(m.direction)
-        @series begin
-            linewidth --> 2
-            color --> :green
-            label := ""
-            [v.x, vnew.x], [v.y, vnew.y], [v.z, vnew.z]
-        end
-    elseif m.opening_angle <= 90u"°"
-        d = normalize(m.direction)
-        a = normalize(d × (abs(d.x) == 1 ? CartesianVector(0,1,0) : CartesianVector(1,0,0)))
-        b = normalize(a × d)
-        rot = hcat(a,b,d)
-        cone = SolidStateDetectors.ConstructiveSolidGeometry.Cone(r = ((0,0),(0,length*sin(m.opening_angle))), hZ = length*cos(m.opening_angle)/2, 
-        origin = rot * [0,0,length*cos(m.opening_angle)/2] + m.position, 
-        rotation = rot)
+    if hasproperty(m, :opening_angle)
+        if iszero(m.opening_angle)
+            v = m.position
+            vnew = m.position + length * normalize(m.direction)
+            @series begin
+                linewidth --> 2
+                color --> :green
+                label := ""
+                [v.x, vnew.x], [v.y, vnew.y], [v.z, vnew.z]
+            end
+        elseif m.opening_angle <= 90u"°"
+            d = normalize(m.direction)
+            a = normalize(d × (abs(d.x) == 1 ? CartesianVector(0,1,0) : CartesianVector(1,0,0)))
+            b = normalize(a × d)
+            rot = hcat(a,b,d)
+            cone = SolidStateDetectors.ConstructiveSolidGeometry.Cone(r = ((0,0),(0,length*sin(m.opening_angle))), hZ = length*cos(m.opening_angle)/2, 
+            origin = rot * [0,0,length*cos(m.opening_angle)/2] + m.position, 
+            rotation = rot)
 
-        @series begin
-            linewidth := 0
-            color --> :green
-            label := ""
-            cone
+            @series begin
+                linewidth := 0
+                color --> :green
+                label := ""
+                cone
+            end
         end
     end