From 7262852e28db7c1c0eec9afebbe9a68630a70718 Mon Sep 17 00:00:00 2001
From: Weijia Wang <9713184+wegank@users.noreply.github.com>
Date: Sun, 12 Apr 2026 01:28:54 +0200
Subject: [PATCH] Add RAGlib wrappers

---
 src/AlgebraicSolving.jl |  2 +
 src/rag/main.jl         | 98 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 100 insertions(+)
 create mode 100644 src/rag/main.jl

diff --git a/src/AlgebraicSolving.jl b/src/AlgebraicSolving.jl
index f48e3d3..c4f113a 100644
--- a/src/AlgebraicSolving.jl
+++ b/src/AlgebraicSolving.jl
@@ -22,6 +22,8 @@ include("siggb/siggb.jl")
 include("progress/main.jl")
 #= interp =#
 include("interp/main.jl")
+#= rag =#
+include("rag/main.jl")
 #= examples =#
 include("examples/katsura.jl")
 include("examples/cyclic.jl")
diff --git a/src/rag/main.jl b/src/rag/main.jl
new file mode 100644
index 0000000..2d80a57
--- /dev/null
+++ b/src/rag/main.jl
@@ -0,0 +1,98 @@
+module RAG
+
+using Nemo
+
+export has_real_solutions, points_per_components
+
+function _flint_get_str_pretty(p::QQMPolyRingElem)::String
+    R = parent(p)
+    x = map(var -> Base.unsafe_convert(Cstring, string(var)), gens(R))
+    str = @ccall Nemo.libflint.fmpq_mpoly_get_str_pretty(p::Ref{QQMPolyRingElem}, x::Ptr{Ref{Cstring}}, R::Ref{QQMPolyRing})::Cstring
+    unsafe_string(str)
+end
+
+function _get_str(f::Vector{QQMPolyRingElem})::String
+    join([_flint_get_str_pretty(p) for p in f], ", ")
+end
+
+function _call_procedure_with_maple(
+    procedure::String;
+    savelibname::String=joinpath(homedir(), "libs"),
+    maple_path::String="maple",
+    msolve_path::String="msolve",
+)::String
+    if !isfile(joinpath(savelibname, "MSolve.mla"))
+        error("Could not find MSolve.mla in $savelibname. Please set savelibname to the correct path.")
+    end
+    if isnothing(Sys.which(maple_path))
+        error("Could not find Maple executable. Please set maple_path to the correct path.")
+    end
+    if isnothing(Sys.which(msolve_path))
+        error("Could not find MSolve executable. Please set msolve_path to the correct path.")
+    end
+    input = tempname()
+    output = tempname()
+    open(input, "w") do io
+        println(io, "savelibname := \"$savelibname\":")
+        println(io, "libname := savelibname, libname:")
+        println(io, "MSolve:-SetMSolvePath(\"$msolve_path\"):")
+        println(io, "f:=proc()")
+        println(io, procedure)
+        println(io, "end proc:")
+        println(io, "fd := fopen(\"$output\", WRITE):")
+        println(io, "fprintf(fd, \"%a\\n\", f()):")
+        println(io, "fclose(fd):")
+    end
+    command = Cmd(`$maple_path -q $input`)
+    run(command)
+    line = open(output, "r") do io
+        readline(io)
+    end
+    line
+end
+
+function has_real_solutions(
+    eqs::Vector{QQMPolyRingElem},
+    pos::Vector{QQMPolyRingElem},
+    ineqs::Vector{QQMPolyRingElem};
+    savelibname::String=joinpath(homedir(), "libs"),
+    maple_path::String="maple",
+    msolve_path::String="msolve",
+    info_level::Int=0
+)::Vector{Vector{Vector{QQFieldElem}}}
+    procedure = """
+        local vars, res:
+        vars := [$(_get_str(gens(parent([eqs; pos; ineqs][1]))))]:
+        res := RAG:-HasRealSolutions([$(_get_str(eqs))], [$(_get_str(pos))], [$(_get_str(ineqs))], {"verb"=$info_level}):
+        return map(point -> subs(point, vars), res):
+    """
+    line = _call_procedure_with_maple(procedure; savelibname, maple_path, msolve_path)
+    line = replace(line, r"(-?\d+)/(\d+)" => s"QQ(\1)/QQ(\2)")
+    line = replace(line, r"(-?\d+)" => s"QQ(\1)")
+    res = eval(Meta.parse(line))
+    res
+end
+
+function points_per_components(
+    eqs::Vector{QQMPolyRingElem},
+    pos::Vector{QQMPolyRingElem},
+    ineqs::Vector{QQMPolyRingElem};
+    savelibname::String=joinpath(homedir(), "libs"),
+    maple_path::String="maple",
+    msolve_path::String="msolve",
+    info_level::Int=0
+)::Vector{Vector{Vector{QQFieldElem}}}
+    procedure = """
+        local vars, res:
+        vars := [$(_get_str(gens(parent([eqs; pos; ineqs][1]))))]:
+        res := RAG:-PointsPerComponents([$(_get_str(eqs))], [$(_get_str(pos))], [$(_get_str(ineqs))], {"verb"=$info_level}):
+        return map(point -> subs(point, vars), res):
+    """
+    line = _call_procedure_with_maple(procedure; savelibname, maple_path, msolve_path)
+    line = replace(line, r"(-?\d+)/(\d+)" => s"QQ(\1)/QQ(\2)")
+    line = replace(line, r"(-?\d+)" => s"QQ(\1)")
+    res = eval(Meta.parse(line))
+    res
+end
+
+end # module RAG