Complex Support

Author: Wimmerer

src/operators/binaryops.jl

@@ -327,6 +327,23 @@ function _load(binary::AbstractBinaryOp)
         "GxB_SECONDJ1",
     ]
 
+    complexes = [
+        "GxB_FIRST",
+        "GxB_SECOND",
+        "GxB_POW",
+        "GxB_PLUS",
+        "GxB_MINUS",
+        "GxB_TIMES",
+        "GxB_DIV",
+        "GxB_RMINUS",
+        "GxB_RDIV",
+        "GxB_PAIR",
+        "GxB_ANY",
+        "GxB_ISEQ",
+        "GxB_ISNE",
+        "GxB_EQ",
+        "GxB_NE",
+    ]
     name = binary.name
     if name ∈ booleans
         binary.pointers[Bool] = load_global(name * "_BOOL")
@@ -353,6 +370,11 @@ function _load(binary::AbstractBinaryOp)
     if name ∈ positionals
         binary.pointers[Any] = load_global(name * "_INT64")
     end
+    name = "GxB_" * name[5:end]
+    if name ∈ complexes
+        binary.pointers[ComplexF32] = load_global(name * "_FC32")
+        binary.pointers[ComplexF64] = load_global(name * "_FC64")
+    end
 end
 
 Base.show(io::IO, ::MIME"text/plain", u::libgb.GrB_BinaryOp) = gxbprint(io, u)

src/operators/monoids.jl

@@ -137,6 +137,7 @@ function _load(monoid::AbstractMonoid)
         "GxB_BXNOR",
     ]
     floats = ["GrB_MIN", "GrB_MAX", "GrB_PLUS", "GrB_TIMES", "GxB_ANY"]
+    complexes = ["GxB_PLUS", "GxB_TIMES", "GxB_ANY"]
     name = monoid.name
 
     if name ∈ booleans
@@ -172,6 +173,12 @@ function _load(monoid::AbstractMonoid)
         monoid.pointers[Float64] =
             load_global(name * (isGxB(name) ? "_FP64_MONOID" : "_MONOID_FP64"))
     end
+    name = "GxB_" * name[5:end]
+    if name ∈ complexes
+        #Complex monoids are always GxB, so "_MONOID" is always at the end.
+        monoid.pointers[ComplexF32] = load_global(name * "_FC32_MONOID")
+        monoid.pointer[ComplexF64] = load_global(name * "_FC64_MONOID")
+    end
 end
 Base.show(io::IO, ::MIME"text/plain", m::libgb.GrB_Monoid) = gxbprint(io, m)
 operator(monoid::libgb.GrB_Monoid) = libgb.GxB_Monoid_operator(monoid)

src/operators/semirings.jl

@@ -814,6 +814,36 @@ function _load(rig::AbstractSemiring)
         "GxB_PLUS_SECONDJ1",
         "GxB_TIMES_SECONDJ1",
     ]
+
+    complexes = [
+        "GxB_PLUS_FIRST",
+        "GxB_TIMES_FIRST",
+        "GxB_ANY_FIRST",
+        "GxB_PLUS_SECOND",
+        "GxB_TIMES_SECOND",
+        "GxB_ANY_SECOND",
+        "GxB_PLUS_PAIR",
+        "GxB_TIMES_PAIR",
+        "GxB_ANY_PAIR",
+        "GxB_PLUS_PLUS",
+        "GxB_TIMES_PLUS",
+        "GxB_ANY_PLUS",
+        "GxB_PLUS_MINUS",
+        "GxB_TIMES_MINUS",
+        "GxB_ANY_MINUS",
+        "GxB_PLUS_TIMES",
+        "GxB_TIMES_TIMES",
+        "GxB_ANY_TIMES",
+        "GxB_PLUS_DIV",
+        "GxB_TIMES_DIV",
+        "GxB_ANY_DIV",
+        "GxB_PLUS_RDIV",
+        "GxB_TIMES_RDIV",
+        "GxB_ANY_RDIV",
+        "GxB_PLUS_RMINUS",
+        "GxB_TIMES_RMINUS",
+        "GxB_ANY_RMINUS",
+    ]
     name = rig.name
     if name ∈ booleans
         rig.pointers[Bool] = load_global(name * "_BOOL")
@@ -840,6 +870,12 @@ function _load(rig::AbstractSemiring)
     if name ∈ positionals
         rig.pointers[Any] = load_global(name * "_INT64")
     end
+    name = replace(name, "GrB_" => "GxB_")
+    name = replace(name, "_SEMIRING" => "")
+    if name ∈ complexes
+        rig.pointers[ComplexF32] = load_global(name * "_FC32")
+        rig.pointers[ComplexF64] = load_global(name * "_FC64")
+    end
 end
 
 Base.show(io::IO, ::MIME"text/plain", s::libgb.GrB_Semiring) = gxbprint(io, s)

src/operators/unaryops.jl

@@ -212,6 +212,45 @@ function _load(unaryop::AbstractUnaryOp)
         "GxB_ISFINITE",
     ]
     positionals = ["GxB_POSITIONI", "GxB_POSITIONI1", "GxB_POSITIONJ", "GxB_POSITIONJ1"]
+    complexes = [
+        "GxB_IDENTITY",
+        "GxB_AINV",
+        "GxB_MINV",
+        "GxB_ONE",
+        "GxB_SQRT",
+        "GxB_LOG",
+        "GxB_EXP",
+        "GxB_LOG2",
+        "GxB_SIN",
+        "GxB_COS",
+        "GxB_TAN",
+        "GxB_ACOS",
+        "GxB_ASIN",
+        "GxB_ATAN",
+        "GxB_SINH",
+        "GxB_COSH",
+        "GxB_TANH",
+        "GxB_ASINH",
+        "GxB_ACOSH",
+        "GxB_ATANH",
+        "GxB_SIGNUM",
+        "GxB_CEIL",
+        "GxB_FLOOR",
+        "GxB_ROUND",
+        "GxB_TRUNC",
+        "GxB_EXP2",
+        "GxB_EXPM1",
+        "GxB_LOG10",
+        "GxB_LOG1P",
+        "GxB_CONJ",
+        "GxB_CREAL",
+        "GxB_CIMAG",
+        "GxB_CARG",
+        "GxB_ABS",
+        "GxB_ISINF",
+        "GxB_ISNAN",
+        "GxB_ISFINITE",
+    ]
     name = unaryop.name
     if name ∈ booleans
         constname = name * "_BOOL"
@@ -239,6 +278,11 @@ function _load(unaryop::AbstractUnaryOp)
     if name ∈ positionals
         unaryop.pointers[Any] = load_global(name)
     end
+    name = "GxB_" * name[5:end]
+    if name ∈ complexes
+        unaryop.pointers[ComplexF32] = load_global(name * "_FC32")
+        unaryop.pointers[ComplexF64] = load_global(name * "_FC64")
+    end
 end
 
 ztype(op::libgb.GrB_UnaryOp) = tojuliatype(ptrtogbtype[libgb.GxB_UnaryOp_ztype(op)])