Translated setdiag()

R/setdiag.R

@@ -1,21 +1,19 @@
 setdiag <- function(M, v) {
-  stop("needs translation")
+  #  SETDIAG Set the diagonal of a matrix to a specified scalar/vector.
-  # function M = setdiag(M, v)
+  #  M <- set_diag(M, v)
-  # % SETDIAG Set the diagonal of a matrix to a specified scalar/vector.
-  # % M = set_diag(M, v)
 
-  # n = length(M);
+  n <- length(M)
-  # if length(v)==1
+  if (length(v) == 1) {
-  #   v = repmat(v, 1, n);
+    v <- repmat(v, c(1, n))
-  # end
+  }
 
-  # % e.g., for 3x3 matrix,  elements are numbered
+  #  e.g., for 3x3 matrix,  elements are numbered
-  # % 1 4 7
+  #  1 4 7
-  # % 2 5 8
+  #  2 5 8
-  # % 3 6 9
+  #  3 6 9
-  # % so diagnoal = [1 5 9]
+  #  so diagnoal = [1 5 9]
 
-
+  J <- seq(1, n ^ 2, n + 1)
-  # J = 1:n+1:n^2;
+  M[J] <- v
-  # M(J) = v;
+  return(M)
 }