Generate examples in test folder

Author: guilgautier

test/hard_core_graph.jl

@@ -0,0 +1,31 @@
+using PartialRejectionSampling
+const PRS = PartialRejectionSampling
+
+using LightGraphs
+const LG = LightGraphs
+
+width, height = 5, 5
+g = LG.grid([width, height])
+β = 1.0
+
+hcg = PRS.HardCoreGraph(g, β)
+seed = -1
+@time sample = PRS.generate_sample_prs(hcg; rng=seed)
+
+using GraphPlot
+using Colors
+
+pos = collect(Iterators.product(1:height, 1:width))[:]
+locs_x, locs_y = map(x->x[1], pos), map(x->x[2], pos)
+
+col_nodes = [Colors.colorant"turquoise" for _ in 1:LG.nv(g)]
+col_nodes[sample] .= Colors.colorant"red"
+
+p = gplot(g,
+    locs_x, locs_y,
+    nodelabel=LG.vertices(g),
+    nodefillc=col_nodes,
+#     arrowlengthfrac=0.05
+#     edgestrokec=col_edges
+)
+display(p)

test/hard_core_spatial.jl

@@ -0,0 +1,16 @@
+using PartialRejectionSampling
+const PRS = PartialRejectionSampling
+
+β₀ = 0.1
+r = 0.01  # interaction range = 2*radius
+b = β₀ / (π * (r/2)^2)
+
+c, w = [0.0, 0.0], 1.0
+win = PRS.SquareWindow(c, w)
+
+hc = PRS.HardCorePointProcess(b, r, win)
+
+@time sample = PRS.generate_sample_prs(hc; rng=-1)
+# Todo @time sample = PRS.generate_sample_grid_prs(hc; rng=-1)
+
+p = PRS.plot(hc, sample; title="")

test/ising.jl

@@ -0,0 +1,14 @@
+import PartialRejectionSampling
+const PRS = PartialRejectionSampling
+
+dims = [15, 15] # if > (14, 14) the display becomes all black, don't know why !
+periodic = false
+H, J = 0.0, 0.01
+
+seed = -1
+
+ising = PRS.Ising(dims, periodic, H, J)
+
+@time sample = PRS.generate_sample_gibbs_perfect(ising; rng=seed)
+
+PRS.plot(ising, sample)

test/ising_gibbs_perfect.jl

@@ -0,0 +1,15 @@
+using PartialRejectionSampling
+const PRS = PartialRejectionSampling
+
+dims = [15, 15] # if > (14, 14) the display becomes all black, don't know why !
+periodic = false
+H, J = 0.0, 0.01
+
+ising = PRS.Ising(dims, periodic, H, J)
+
+println("\nPartial Rejection Sampling\n")
+
+seed = -1
+@time sample = PRS.generate_sample_grid_prs(ising; rng=seed)
+
+PRS.plot(ising, sample)

test/ising_grid_prs.jl

@@ -0,0 +1,9 @@
+using PartialRejectionSampling
+const PRS = PartialRejectionSampling
+
+n = 1000
+alph = ["A", "C", "G", "T"]
+pattern = "ACAC"
+
+seed = -1
+@time PRS.generate_pattern_free_string(n, alph, pattern; rng=seed)

test/pattern_free_string_prs.jl

@@ -0,0 +1,33 @@
+using PartialRejectionSampling
+const PRS = PartialRejectionSampling
+
+using LightGraphs
+const LG = LightGraphs
+
+width, height = 5, 5
+g = LG.grid([width, height])
+roots = [1, 5]
+
+rsf = PRS.RootedSpanningForest(g, roots)
+seed = -1
+@time sample = PRS.generate_sample_prs(rsf; rng=seed)
+
+# Display
+
+using GraphPlot
+using Colors
+
+pos = collect(Iterators.product(1:height, 1:width))[:]
+locs_x, locs_y = map(x->x[1], pos), map(x->x[2], pos)
+
+col_nodes, col_edges = PRS.color_cycles(sample)
+col_nodes[roots] .= Colors.colorant"orange"
+
+p = GraphPlot.gplot(sample,
+    locs_x, locs_y,
+    nodelabel=LG.vertices(sample),
+    nodefillc=col_nodes,
+    edgestrokec=col_edges,
+    arrowlengthfrac=0.05,
+)
+display(p)

test/rooted_spanning_forest_prs.jl

@@ -0,0 +1,29 @@
+using PartialRejectionSampling
+const PRS = PartialRejectionSampling
+
+using LightGraphs
+const LG = LightGraphs
+
+width, height = 5, 5
+g = LG.grid([width, height])
+
+sfg = PRS.SinkFreeGraph(g)
+seed = -1
+@time sample = PRS.generate_sample_prs(sfg; rng=seed)
+
+using GraphPlot
+using Colors
+
+pos = collect(Iterators.product(1:height, 1:width))[:]
+locs_x, locs_y = map(x->x[1], pos), map(x->x[2], pos)
+
+sample = PRS.generate_sample_prs(sfg; rng=seed)
+cols = [LG.outdegree(sample, v) == 0 ? colorant"red" : colorant"turquoise"
+        for v in LG.vertices(sample)]
+p = GraphPlot.gplot(sample,
+    locs_x, locs_y,
+    nodelabel=LG.vertices(sample),
+    nodefillc=cols,
+    arrowlengthfrac=0.05
+)
+display(p)

test/sink_free_graph.jl

@@ -0,0 +1,14 @@
+using PartialRejectionSampling
+const PRS = PartialRejectionSampling
+
+# Experimental setup refers to slide 9 of [E. Rubak](https://www-ljk.imag.fr/membres/Jean-Francois.Coeurjolly/documents/lecture4.pdf)
+β, γ, r = 2.0, 0.2, 0.7  # γ = 0 ≡ Hard core
+
+c, w = [0.0, 0.0], 10.0
+win = PRS.SquareWindow(c, w)
+
+strauss = PRS.StraussPointProcess(β, γ, r, win)
+
+@time sample = PRS.generate_sample_dcftp(strauss; rng=-1)
+
+p = PRS.plot(strauss, sample; title="")

test/strauss_dcftp.jl

@@ -0,0 +1,15 @@
+using PartialRejectionSampling
+const PRS = PartialRejectionSampling
+
+β₀ = 0.1
+r = 0.01  # interaction range = 2*radius
+β, γ = β₀ / (π * (r/2)^2), 0.1  # γ = 0 ≡ Hard core
+
+c, w = [0.0, 0.0], 1.0
+win = PRS.SquareWindow(c, w)
+
+strauss = PRS.StraussPointProcess(β, γ, r, win)
+
+@time sample = PRS.generate_sample_grid_prs(strauss; rng=-1)
+
+p = PRS.plot(strauss, sample)