1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
function [A, keep, ii, jj, xx] = mesh3d1 (n)
% create an n-by-n-by-n 3D mesh for the 2nd difference operator
% Example:
% A = mesh3d1 (10) ; % a 10-by-10-by-10 mesh
% See also: cs_demo
% CXSparse, Copyright (c) 2006-2022, Timothy A. Davis. All Rights Reserved.
% SPDX-License-Identifier: LGPL-2.1+
ii = zeros (7*n^3, 1) ;
jj = zeros (7*n^3, 1) ;
xx = zeros (7*n^3, 1) ;
t = 1 ;
for k = 0:n-1
for j = 0:n-1
for i = 0:n-1
s = k*n^2 + j*n+i + 1 ;
ii (t:t+6) = [
(k-1)*n^2 + j*n+i
k*n^2 + (j-1)*n+i
k*n^2 + j*n+(i-1)
k*n^2 + j*n+i
k*n^2 + j*n+(i+1)
k*n^2 + (j+1)*n+i
(k+1)*n^2 + j*n+i ]' + 1 ;
jj (t:t+6) = [s s s s s s s] ;
xx (t:t+6) = [-1 -1 -1 6 -1 -1 -1] ;
t = t + 7 ;
end
end
end
keep = find (ii >= 1 & ii <= n^3 & jj >= 1 & jj <= n^3) ;
A = sparse (ii (keep), jj (keep), xx (keep)) ;
|
