genetic correlation plot

src/plotrg.jl

@@ -1,7 +1,7 @@
 function plotrg!(
     ax::Axis,
     r::AbstractMatrix,
-    cols::AbstractVector; 
+    colnames::AbstractVector; 
     p::Union{AbstractMatrix, Nothing} = nothing,
     circle::Bool = true
 )
@@ -37,8 +37,8 @@ function plotrg!(
         counter = 1
         for i in 1:n
             for j in (i + 1):n
-                csl[counter] = Circle(Point2f(i - 0.5, 0.5 - j), r[i, j] * 0.45)
-                csu[counter] = Circle(Point2f(j - 0.5, 0.5 - i), r[j, i] * 0.45)
+                csl[counter] = Circle(Point2f(j - 0.5, 0.5 - i), r[i, j] * 0.45)
+                csu[counter] = Circle(Point2f(i - 0.5, 0.5 - j), r[j, i] * 0.45)
                 colorl[counter] = r[i, j]
                 coloru[counter] = r[j, i]
                 counter += 1
@@ -54,24 +54,38 @@ function plotrg!(
             for j in (i + 1):n
                 widthl = r[i, j] * 0.95
                 widthu = r[j, i] * 0.95
-                csl[counter] = Rect(i - 1 + (1 - widthl) / 2, 1 - j - (1 - widthl) / 2, widthl, -widthl)
-                csu[counter] = Rect(j - 1 + (1 - widthu) / 2, 1 - i - (1 - widthu) / 2, widthu, -widthu)
+                csl[counter] = Rect(j - 1 + (1 - widthl) / 2, 1 - i - (1 - widthl) / 2, widthl, -widthl)
+                csu[counter] = Rect(i - 1 + (1 - widthu) / 2, 1 - j - (1 - widthu) / 2, widthu, -widthu)
                 colorl[counter] = r[i, j]
                 coloru[counter] = r[j, i]
                 counter += 1
             end
         end
     end
-    poly!(ax, ps, color = :white, strokewidth = 0.75, strokecolor = "#C3C3C3")
-    poly!(ax, psd, color = "#C3C3C3", strokewidth = 0.75, strokecolor = "#C3C3C3")
+    poly!(ax, ps, color = :white, strokewidth = 0.5, strokecolor = "#C3C3C3")
+    # poly!(ax, psd, color = "#C3C3C3", strokewidth = 0.5, strokecolor = "#C3C3C3")
     poly!(ax, csl, color = colorl, colorrange = (-1, 1), colormap = :RdBu_10)
     poly!(ax, csu, color = coloru, colorrange = (-1, 1), colormap = :RdBu_10)
-    ax.xticks = (0.5:(n - 0.5), cols)
-    ax.yticks = (-0.5:-1:-(n - 0.5), cols)
+    if !isnothing(p)
+        pvalues = Matrix{Float64}(undef, n^2, 3)
+        counter = 1
+        for i in 1:n
+            for j in 1:n
+                j == i ? pvalues[counter, 1] = 1 : pvalues[counter, 1] = p[j, i]
+                pvalues[counter, 2] = i - 0.5
+                pvalues[counter, 3] = -j + 0.5
+                counter += 1
+            end
+        end
+        ind = findall(x -> x < 0.05, pvalues[:, 1]) 
+        scatter!(ax, pvalues[ind, 2], pvalues[ind, 3], marker = '✳', markersize = 7, color = :black)
+    end
+    ax.xticks = (0.5:(n - 0.5), colnames)
+    ax.yticks = (-0.5:-1:-(n - 0.5), colnames)
     ax.xticklabelsize = 6
     ax.yticklabelsize = 6
-    ax.xticklabelpad = 1
-    ax.yticklabelpad = 2
+    ax.xticklabelpad = 0
+    ax.yticklabelpad = 0
     ax.xaxisposition = :top
     ax.yaxisposition = :left
     hidespines!(ax)