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
|
import copy
import unittest
import meep as mp
def traverse_tree(bp=None, min_corner=None, max_corner=None):
process_ids = []
chunk_areas = []
def _traverse_tree(bp=None, min_corner=None, max_corner=None):
if (min_corner.x > max_corner.x) or (min_corner.y > max_corner.y):
raise RuntimeError("min_corner/max_corner have been incorrectly defined.")
## reached a leaf
if bp.left is None and bp.right is None:
process_ids.append(bp.proc_id)
chunk_area = (max_corner.x - min_corner.x) * (max_corner.y - min_corner.y)
chunk_areas.append(chunk_area)
## traverse the left branch
if bp.left is not None:
new_max_corner = copy.deepcopy(max_corner)
if bp.split_dir == mp.X:
new_max_corner.x = bp.split_pos
else:
new_max_corner.y = bp.split_pos
_traverse_tree(bp.left, min_corner, new_max_corner)
## traverse the right branch
if bp.right is not None:
new_min_corner = copy.deepcopy(min_corner)
if bp.split_dir == mp.X:
new_min_corner.x = bp.split_pos
else:
new_min_corner.y = bp.split_pos
_traverse_tree(bp.right, new_min_corner, max_corner)
_traverse_tree(bp=bp, min_corner=min_corner, max_corner=max_corner)
return process_ids, chunk_areas
class TestChunkLayoutBinaryPartition(unittest.TestCase):
def test_chunk_layout_binary_partition(self):
chunk_layout = mp.BinaryPartition(
data=[
(mp.X, -2.0),
0,
[(mp.Y, 1.5), [(mp.X, 3.0), 1, [(mp.Y, -0.5), 4, 3]], 2],
]
)
cell_size = mp.Vector3(10.0, 5.0, 0)
sim = mp.Simulation(
cell_size=cell_size, resolution=10, chunk_layout=chunk_layout
)
sim.init_sim()
owners = sim.structure.get_chunk_owners()
areas = [
v.surroundings().full_volume() for v in sim.structure.get_chunk_volumes()
]
process_ids, chunk_areas = traverse_tree(
chunk_layout, -0.5 * cell_size, 0.5 * cell_size
)
self.assertListEqual(
[int(f) for f in owners], [f % mp.count_processors() for f in process_ids]
)
self.assertListEqual(areas, chunk_areas)
def test_meep_default_chunk_layout(self):
cell_size = mp.Vector3(10.0, 5.0, 0)
sim = mp.Simulation(cell_size=cell_size, resolution=10)
sim.init_sim()
owners = sim.structure.get_chunk_owners()
areas = [
v.surroundings().full_volume() for v in sim.structure.get_chunk_volumes()
]
chunk_layout = sim.chunk_layout
process_ids, chunk_areas = traverse_tree(
chunk_layout, -0.5 * cell_size, 0.5 * cell_size
)
self.assertListEqual(
[int(f) for f in owners], [f % mp.count_processors() for f in process_ids]
)
self.assertListEqual(areas, chunk_areas)
if __name__ == "__main__":
unittest.main()
|
