Replace ZhatCross with UnitVectors

src/Elements.jl

@@ -11,7 +11,7 @@ using Printf: @sprintf
 import LibGEOS # difference, readgeom, Polygon, MultiPolygon
 import GeoInterface # nhole, ngeom, coordinates, getexterior, gethole
 import PolygonOps
-using ..ZhatCross: ẑ
+using ..UnitVectors: ẑ
 
 macro testpos(var)
     return :(all($(esc(var)) .> 0) || error($(esc(string(var))) * " must be positive!"))

src/Modes.jl

@@ -4,7 +4,7 @@ using Unitful: ustrip, @u_str
 using LinearAlgebra: norm, ⋅, ×
 using StaticArrays: SVector
 using ..Constants: twopi, η₀, dbmin
-using ..ZhatCross: ẑ
+using ..UnitVectors: ẑ
 using ..Layers: Layer, TEorTM, TE, TM
 using ..Sheets: RWGSheet, find_unique_periods
 using ..GSMs: Gblock

src/Outputs.jl

@@ -2,7 +2,7 @@ module Outputs
 export @outputs, Result, append_result_data, read_result_file, extract_result_file, extract_result
 
 using LinearAlgebra: ⋅, norm, ×
-using ..ZhatCross: ẑ
+using ..UnitVectors: ẑ
 using ..Constants: c₀, twopi
 using ..GSMs: GSM
 using ..Layers: TEorTM, TE, TM

src/PSSFSS.jl

@@ -29,7 +29,7 @@ using ProgressMeter
 using PrecompileTools
 
 include("Constants.jl")
-include("ZhatCross.jl")
+include("UnitVectors.jl")
 include("Log.jl")
 include("PSSFSSLen.jl")
 include("Rings.jl")
@@ -46,7 +46,7 @@ include("Modes.jl")
 include("Outputs.jl")
 include("FastSweep.jl")
 
-using .ZhatCross: ẑ
+using .UnitVectors: ẑ
 using .Rings
 @reexport using .Sheets: Sheet, RWGSheet, read_sheet_data, nodecount, facecount, edgecount, 
                          export_sheet, STL_ASCII, STL_BINARY

src/RWG.jl

@@ -19,7 +19,7 @@ export RWGData, setup_rwg, edge_current_unit_vector, rwgbfft!
 using ..Sheets: RWGSheet, SV2
 using StaticArrays: SVector, MVector, SArray, @SVector
 using LinearAlgebra: ⋅, norm, ×
-using ..ZhatCross: ẑ
+using ..UnitVectors: ẑ
 using NearestNeighbors: KDTree, inrange
 using OffsetArrays
 using Statistics: mean

src/UnitVectors.jl

@@ -0,0 +1,41 @@
+module UnitVectors
+export x̂, ŷ, ẑ
+
+using StaticArrays: SVector
+import LinearAlgebra: ×, dot
+
+struct UnitVector{N} end # N = 1, 2, or 3 denotes x, y, or z unit vector
+const x̂ = UnitVector{1}()
+const ŷ = UnitVector{2}()
+const ẑ = UnitVector{3}()
+
+dot(::UnitVector{T}, v::AbstractVector) where {T} = v[t]
+dot(v::AbstractVector, ::UnitVector{T}) where {T} = v[t]
+
+
+×(::UnitVector{3}, t::SVector{2,T}) where {T} = SVector{2,T}(-t[2], t[1])
+
+×(::UnitVector{1}, t::SVector{3,T}) where {T} = SVector{3,T}(zero(T), -t[3], t[2])
+×(::UnitVector{2}, t::SVector{3,T}) where {T} = SVector{3,T}(t[3], zero(T), -t[1])
+×(::UnitVector{3}, t::SVector{3,T}) where {T} = SVector{3,T}(-t[2], t[1], zero(T))
+
+@inline function ×(::UnitVector{1}, t::AbstractVector)
+    length(t) == 3 && return [zero(eltype(t)), -t[3], t[2]]
+    throw(ArgumentError("Vector length not 3"))
+end
+
+@inline function ×(::UnitVector{2}, t::AbstractVector)
+    length(t) == 3 && return [t[3], zero(eltype(t)), -t[1]]
+    throw(ArgumentError("Vector length not 3"))
+end
+
+@inline function ×(::UnitVector{3}, t::AbstractVector)
+    length(t) == 2 && return [-t[2], t[1]]
+    length(t) == 3 && return [-t[2], t[1], zero(eltype(t))]
+    throw(ArgumentError("Vector length not 2 or 3"))
+end
+
+×(v::AbstractVector, û::UnitVector) = ×(û, -v)
+
+
+end # module

src/Zint.jl

@@ -6,7 +6,7 @@ using ..Layers: Layer
 using ..RWG: RWGData
 using ..PGF: jksums, jkringmax
 using LinearAlgebra: ⋅, norm, ×
-using ..ZhatCross: ẑ
+using ..UnitVectors: ẑ
 using Statistics: mean
 using OhMyThreads: @tasks, @set, DynamicScheduler, StaticScheduler