check if the input of marker in scatter_gplot() is a vector or an Nx1 matrix

Author: haotian127

src/helpers.jl

@@ -121,15 +121,20 @@ SCATTER\\_GPLOT!(X; ...) adds a plot to `current` one.
 
 """
 function scatter_gplot(X; marker = nothing, ms = 4, plotOrder = :normal, c = :viridis)
-    dim = size(X,2)
+    N, dim = size(X)
     if marker != nothing && plotOrder != :normal
-        marker = marker[:]  # reshape N x 1 matrix to a N-dim vector
+        if size(marker) == (N,) || size(marker) == (N, 1)
+            marker = marker[:]  # reshape N x 1 matrix to a vector of length N
+        else
+            error("marker only accepts a vector of length N or an N x 1 matrix,
+                where N is the total number of points.")
+        end
         if plotOrder == :s2l
             idx = sortperm(marker)
         elseif plotOrder == :l2s
             idx = sortperm(marker, rev=true)
         else
-            print("Error: plotOrder only supports for :normal, :s2l, or :l2s.")
+            error("plotOrder only supports for :normal, :s2l, or :l2s.")
         end
         X = X[idx,:]
         marker = marker[idx]
@@ -142,20 +147,25 @@ function scatter_gplot(X; marker = nothing, ms = 4, plotOrder = :normal, c = :vi
     elseif dim == 3
         scatter(X[:,1],X[:,2],X[:,3], marker_z = marker, ms = ms, c = c, legend = false, mswidth = 0, cbar = true, aspect_ratio = 1, grid = false)
     else
-        print("Dimension Error: scatter_gplot only supports for 2-dim or 3-dim scatter plots.")
+        error("Dimension Error: scatter_gplot only supports for 2-dim or 3-dim scatter plots.")
     end
 end
 
 function scatter_gplot!(X; marker = nothing, ms = 4, plotOrder = :normal, c = :viridis)
-    dim = size(X,2)
+    N, dim = size(X)
     if marker != nothing && plotOrder != :normal
-        marker = marker[:]  # reshape N x 1 matrix to a N-dim vector
+        if size(marker) == (N,) || size(marker) == (N, 1)
+            marker = marker[:]  # reshape N x 1 matrix to a vector of length N
+        else
+            error("marker only accepts a vector of length N or an N x 1 matrix,
+                where N is the total number of points.")
+        end
         if plotOrder == :s2l
             idx = sortperm(marker)
         elseif plotOrder == :l2s
             idx = sortperm(marker, rev=true)
         else
-            print("Error: plotOrder only supports for :normal, :s2l, or :l2s.")
+            error("plotOrder only supports for :normal, :s2l, or :l2s.")
         end
         X = X[idx,:]
         marker = marker[idx]
@@ -168,7 +178,7 @@ function scatter_gplot!(X; marker = nothing, ms = 4, plotOrder = :normal, c = :v
     elseif dim == 3
         scatter!(X[:,1],X[:,2],X[:,3], marker_z = marker, ms = ms, c = c, legend = false, mswidth = 0, cbar = true, aspect_ratio = 1, grid = false)
     else
-        print("Dimension Error: scatter_gplot! only supports for 2-dim or 3-dim scatter plots.")
+        error("Dimension Error: scatter_gplot! only supports for 2-dim or 3-dim scatter plots.")
     end
 end