Fix type casting issue

QEDjl-project · Jun 6, 2024 · 310edb2 · 310edb2
1 parent c71bd8a
commit 310edb2
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/interfaces/lorentz_vectors/arithmetic.jl b/src/interfaces/lorentz_vectors/arithmetic.jl
@@ -1,3 +1,16 @@
+# TODO: these overloads shouldn't be necessary, but are for some reason
+@inline function *(L::TL, NUM::Number) where {TL<:AbstractLorentzVector}
+    return constructorof(TL)(L[1] * NUM, L[2] * NUM, L[3] * NUM, L[4] * NUM)
+end
+
+@inline function *(NUM::Number, L::TL) where {TL<:AbstractLorentzVector}
+    return constructorof(TL)(L[1] * NUM, L[2] * NUM, L[3] * NUM, L[4] * NUM)
+end
+
+@inline function /(L::TL, NUM::Number) where {TL<:AbstractLorentzVector}
+    return constructorof(TL)(L[1] / NUM, L[2] / NUM, L[3] / NUM, L[4] / NUM)
+end
+
 function dot(p1::T1, p2::T2) where {T1<:AbstractLorentzVector,T2<:AbstractLorentzVector}
     return mdot(p1, p2)
 end