1 files changed, 21 insertions, 0 deletions
diff --git a/sourcecodes/bnt-master/KPMtools/setdiag.m b/sourcecodes/bnt-master/KPMtools/setdiag.m
new file mode 100644
index 00000000..30bfa60a
--- /dev/null
+++ b/sourcecodes/bnt-master/KPMtools/setdiag.m
@@ -0,0 +1,21 @@
+function M = setdiag(M, v)
+% SETDIAG Set the diagonal of a matrix to a specified scalar/vector.
+% M = set_diag(M, v)
+
+n = length(M);
+if length(v)==1
+  v = repmat(v, 1, n);
+end
+
+% e.g., for 3x3 matrix,  elements are numbered
+% 1 4 7 
+% 2 5 8 
+% 3 6 9
+% so diagnoal = [1 5 9]
+
+
+J = 1:n+1:n^2;
+M(J) = v;
+
+%M = triu(M,1) + tril(M,-1) + diag(v);
+