adding number_of_monomials function

singular-gpispace · Jun 25, 2024 · e60eefc · e60eefc
1 parent d6838d3
commit e60eefc
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/src/GromovWitten.jl b/src/GromovWitten.jl
@@ -59,7 +59,7 @@ export substitute
 export sum_of_coeff
 export sum_of_divisor_powers
 export vector_to_monomial
-
+export number_of_monomials
 include("graph.jl")
 include("coeftermV.jl")
 include("coefterm.jl")

diff --git a/src/quasimodular.jl b/src/quasimodular.jl
@@ -1,3 +1,14 @@
+function number_of_monomials(w, d)
+    dp = zeros(Int, d + 1)
+    dp[1] = 1
+    for weight in w
+        for j in weight:d
+            dp[j+1] += dp[j-weight+1]
+        end
+    end
+
+    return dp[d+1]
+end
 
 #=function filter_term(pols::Union{QQMPolyRingElem, Int64}, variables::Vector{QQMPolyRingElem}, power::Vector{Int64})
     T = parent(variables[1])

diff --git a/test/quasimod.jl b/test/quasimod.jl
@@ -1,4 +1,7 @@
 @testset "quasimod.jl" begin
+    w = [2, 4, 6]
+    d = 12
+    @test number_of_monomials(w, d) == 7
     import Nemo: QQFieldElem
     S, (E2, E4, E6) = polynomial_ring(QQ, ["E2", "E4", "E6"])
     R, q = polynomial_ring(QQ, ["q"])