-
Notifications
You must be signed in to change notification settings - Fork 2
/
mk_2D_lattice_slow.m
135 lines (122 loc) · 3.1 KB
/
mk_2D_lattice_slow.m
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
function G = mk_2D_lattice_slow(nrows, ncols, wrap_around)
% MK_2D_LATTICE Return adjacency matrix for 4-nearest neighbor connected 2D lattice
% G = mk_2D_lattice(nrows, ncols, wrap_around)
% G(k1, k2) = 1 iff k1=(i1,j1) is connected to k2=(i2,j2)
%
% If wrap_around = 1, we use toroidal boundary conditions (default = 0)
%
% Nodes are assumed numbered as in the following 3x3 lattice
% 1 4 7
% 2 5 8
% 3 6 9
%
% e.g., G = mk_2D_lattice(3, 3, 0) returns
% 0 1 0 1 0 0 0 0 0
% 1 0 1 0 1 0 0 0 0
% 0 1 0 0 0 1 0 0 0
% 1 0 0 0 1 0 1 0 0
% 0 1 0 1 0 1 0 1 0
% 0 0 1 0 1 0 0 0 1
% 0 0 0 1 0 0 0 1 0
% 0 0 0 0 1 0 1 0 1
% 0 0 0 0 0 1 0 1 0
% so find(G(5,:)) = [2 4 6 8]
% but find(G(1,:)) = [2 4]
%
% Using wrap around, G = mk_2D_lattice(3, 3, 1), we get
% 0 1 1 1 0 0 1 0 0
% 1 0 1 0 1 0 0 1 0
% 1 1 0 0 0 1 0 0 1
% 1 0 0 0 1 1 1 0 0
% 0 1 0 1 0 1 0 1 0
% 0 0 1 1 1 0 0 0 1
% 1 0 0 1 0 0 0 1 1
% 0 1 0 0 1 0 1 0 1
% 0 0 1 0 0 1 1 1 0
% so find(G(5,:)) = [2 4 6 8]
% and find(G(1,:)) = [2 3 4 7]
if nargin < 3, wrap_around = 0; end
% M contains the number of each cell e.g.
% 1 4 7
% 2 5 8
% 3 6 9
% North neighbors (assuming wrap around) are
% 3 6 9
% 1 4 7
% 2 5 8
% Without wrap around, they are
% 1 4 7
% 1 4 7
% 2 5 8
% The first row is arbitrary, since pixels at the top have no north neighbor.
if nrows==1
G = zeros(1, ncols);
for i=1:ncols-1
G(i,i+1) = 1;
G(i+1,i) = 1;
end
if wrap_around
G(1,ncols) = 1;
G(ncols,1) = 1;
end
return;
end
npixels = nrows*ncols;
N = 1; E = 2; S = 3; W = 4;
if wrap_around
rows{N} = [nrows 1:nrows-1]; cols{N} = 1:ncols;
rows{E} = 1:nrows; cols{E} = [2:ncols 1];
rows{S} = [2:nrows 1]; cols{S} = 1:ncols;
rows{W} = 1:nrows; cols{W} = [ncols 1:ncols-1];
else
rows{N} = [1 1:nrows-1]; cols{N} = 1:ncols;
rows{E} = 1:nrows; cols{E} = [1 1:ncols-1];
rows{S} = [2:nrows nrows]; cols{S} = 1:ncols;
rows{W} = 1:nrows; cols{W} = [2:ncols ncols];
end
M = reshape(1:npixels, [nrows ncols]);
nbrs = cell(1, 4);
for i=1:4
nbrs{i} = M(rows{i}, cols{i});
end
G = zeros(npixels, npixels);
if wrap_around
for i=1:4
if 0
% naive
for p=1:npixels
G(p, nbrs{i}(p)) = 1;
end
else
% vectorized
ndx2 = sub2ind([npixels npixels], 1:npixels, nbrs{i}(:)');
G(ndx2) = 1;
end
end
else
i = N;
mask = ones(nrows, ncols);
mask(1,:) = 0; % pixels in row 1 have no nbr to the north
ndx = find(mask);
ndx2 = sub2ind([npixels npixels], ndx, nbrs{i}(ndx));
G(ndx2) = 1;
i = E;
mask = ones(nrows, ncols);
mask(:,ncols) = 0;
ndx = find(mask);
ndx2 = sub2ind([npixels npixels], ndx, nbrs{i}(ndx));
G(ndx2) = 1;
i = S;
mask = ones(nrows, ncols);
mask(nrows,:)=0;
ndx = find(mask);
ndx2 = sub2ind([npixels npixels], ndx, nbrs{i}(ndx));
G(ndx2) = 1;
i = W;
mask = ones(nrows, ncols);
mask(:,1)=0;
ndx = find(mask);
ndx2 = sub2ind([npixels npixels], ndx, nbrs{i}(ndx));
G(ndx2) = 1;
end
G = setdiag(G, 0);