x_points() and strike_points() of pulse schedule

Author: orso82

src/physics/boundary.jl

@@ -309,8 +309,10 @@ function x_points(x_points::IMAS.IDSvector{<:IMAS.pulse_schedule__position_contr
     x_points0 = Tuple{T,T}[]
     for x_point in x_points
         Rx = get_time_array(x_point.r, :reference, time0)
-        Zx = get_time_array(x_point.z, :reference, time0)
-        push!(x_points0, (Rx, Zx))
+        if Rx > 0.0 # discard NaN and points with Rx==0
+            Zx = get_time_array(x_point.z, :reference, time0)
+            push!(x_points0, (Rx, Zx))
+        end
     end
     return x_points0
 end
@@ -325,10 +327,12 @@ Beturns vector with tuples of R,Z coordinates of x-points in pulse_schedule at t
 """
 function strike_points(strike_points::IMAS.IDSvector{<:IMAS.pulse_schedule__position_control__strike_point{T}}; time0::Float64=global_time(strike_points)) where {T<:Real}
     strike_points0 = Tuple{T,T}[]
-    for x_point in strike_points
-        Rxx = get_time_array(x_point.r, :reference, time0)
-        Zxx = get_time_array(x_point.z, :reference, time0)
-        push!(strike_points0, (Rxx, Zxx))
+    for strike_point in strike_points
+        Rs = get_time_array(strike_point.r, :reference, time0)
+        if Rs > 0.0 # discard NaN and points with Rx==0
+            Zs = get_time_array(strike_point.z, :reference, time0)
+            push!(strike_points0, (Rs, Zs))
+        end
     end
     return strike_points0
 end

src/plot.jl

@@ -2791,15 +2791,15 @@ end
         markerstrokewidth --> 0
         primary := false
         Xs = x_points(pc.x_point; time0)
-        [x[1] for x in Xs if x[1] != 0.0], [x[2] for x in Xs if x[1] != 0.0]
+        [x[1] for x in Xs], [x[2] for x in Xs]
     end
     @series begin
         seriestype := :scatter
         marker --> :circle
         markerstrokewidth --> 0
         primary := false
         Ss = strike_points(pc.strike_point; time0)
-        [x[1] for x in Ss if x[1] != 0.0], [x[2] for x in Ss if x[1] != 0.0]
+        [x[1] for x in Ss], [x[2] for x in Ss]
     end
     if !ismissing(pc.magnetic_axis.r, :reference) && !ismissing(pc.magnetic_axis.z, :reference)
         @series begin