Skip to content

Commit 64e8090

Browse files
wkliaowkliao
committed
make function of example codes appearing on top
1 parent 4be774a commit 64e8090

6 files changed

Lines changed: 92 additions & 91 deletions

File tree

‎examples/create_open.py‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -21,18 +21,6 @@
2121
from mpi4py import MPI
2222
import pnetcdf
2323

24-
def parse_help():
25-
help_flag = "-h" in sys.argv or "--help" in sys.argv
26-
if help_flag and rank == 0:
27-
help_text = (
28-
"Usage: {} [-h] | [-q] [file_name]\n"
29-
" [-h] Print help\n"
30-
" [-q] Quiet mode (reports when fail)\n"
31-
" [filename] (Optional) output netCDF file name\n"
32-
).format(sys.argv[0])
33-
print(help_text)
34-
return help_flag
35-
3624
def pnetcdf_io(filename):
3725

3826
# create a new file using file clobber mode, i.e. flag "-w"
@@ -51,6 +39,18 @@ def pnetcdf_io(filename):
5139
f.close()
5240

5341

42+
def parse_help():
43+
help_flag = "-h" in sys.argv or "--help" in sys.argv
44+
if help_flag and rank == 0:
45+
help_text = (
46+
"Usage: {} [-h] | [-q] [file_name]\n"
47+
" [-h] Print help\n"
48+
" [-q] Quiet mode (reports when fail)\n"
49+
" [filename] (Optional) output netCDF file name\n"
50+
).format(sys.argv[0])
51+
print(help_text)
52+
return help_flag
53+
5454
if __name__ == "__main__":
5555
comm = MPI.COMM_WORLD
5656
rank = comm.Get_rank()

‎examples/flexible_api.py‎

Lines changed: 24 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -72,19 +72,6 @@
7272
from mpi4py import MPI
7373
import pnetcdf
7474

75-
def parse_help():
76-
help_flag = "-h" in sys.argv or "--help" in sys.argv
77-
if help_flag and rank == 0:
78-
help_text = (
79-
"Usage: {} [-h] | [-q] [file_name]\n"
80-
" [-h] Print help\n"
81-
" [-q] Quiet mode (reports when fail)\n"
82-
" [-k format] file format: 1 for CDF-1, 2 for CDF-2, 5 for CDF-5\n"
83-
" [filename] (Optional) output netCDF file name\n"
84-
).format(sys.argv[0])
85-
print(help_text)
86-
return help_flag
87-
8875
def pnetcdf_io(filename, file_format):
8976
NY = 5
9077
NX = 5
@@ -124,9 +111,9 @@ def pnetcdf_io(filename, file_format):
124111
array_of_subsizes = np.array([NZ, NY])
125112
array_of_start = np.array([ghost_len, ghost_len])
126113

127-
subarray = MPI.INT.Create_subarray(array_of_sizes, \
128-
array_of_subsizes, \
129-
array_of_start, \
114+
subarray = MPI.INT.Create_subarray(array_of_sizes,
115+
array_of_subsizes,
116+
array_of_start,
130117
order=MPI.ORDER_C)
131118
subarray.Commit()
132119

@@ -139,12 +126,13 @@ def pnetcdf_io(filename, file_format):
139126
count = np.array([NZ, NY])
140127

141128
# calling a blocking flexible API using put_var_all()
142-
var_zy.put_var_all(buf_zy, start = start, \
143-
count = count, \
144-
bufcount = 1, \
129+
var_zy.put_var_all(buf_zy, start = start,
130+
count = count,
131+
bufcount = 1,
145132
buftype = subarray)
146133

147134

135+
# check if write buffer's contents are altered (should not be).
148136
for i in range(buffer_len):
149137
if buf_zy[i] != rank:
150138
print(f"Error at line {sys._getframe().f_lineno} in {__file__}: put buffer[{i}] is altered")
@@ -158,7 +146,7 @@ def pnetcdf_io(filename, file_format):
158146
bufcount = 1,
159147
buftype = subarray)
160148

161-
# check contents of the get buffer
149+
# check whether contents of the get buffer are expected
162150
for i in range(array_of_sizes[0]):
163151
for j in range(array_of_sizes[1]):
164152
index = i*array_of_sizes[1] + j
@@ -200,7 +188,7 @@ def pnetcdf_io(filename, file_format):
200188
status = [None]
201189
f.wait_all(1, [req_id], status = status)
202190

203-
# check request error msg for each unsuccessful requests
191+
# check request error for each unsuccessful requests
204192
if pnetcdf.strerrno(status[0]) != "NC_NOERR":
205193
print(f"Error on request {i}:", pnetcdf.strerror(status[0]))
206194

@@ -216,11 +204,11 @@ def pnetcdf_io(filename, file_format):
216204
# commit posted pending nonblocking requests
217205
f.wait_all(1, [req_id], status = status)
218206

219-
# check request error msg for each unsuccessful requests
207+
# check request error for each unsuccessful requests
220208
if pnetcdf.strerrno(status[0]) != "NC_NOERR":
221209
print(f"Error on request {i}:", pnetcdf.strerror(status[0]))
222210

223-
# check the contents of iget buffer
211+
# check the contents of read buffer
224212
for i in range(array_of_sizes[0]):
225213
for j in range(array_of_sizes[1]):
226214
index = i * array_of_sizes[1] + j
@@ -236,6 +224,19 @@ def pnetcdf_io(filename, file_format):
236224
f.close()
237225

238226

227+
def parse_help():
228+
help_flag = "-h" in sys.argv or "--help" in sys.argv
229+
if help_flag and rank == 0:
230+
help_text = (
231+
"Usage: {} [-h] | [-q] [file_name]\n"
232+
" [-h] Print help\n"
233+
" [-q] Quiet mode (reports when fail)\n"
234+
" [-k format] file format: 1 for CDF-1, 2 for CDF-2, 5 for CDF-5\n"
235+
" [filename] (Optional) output netCDF file name\n"
236+
).format(sys.argv[0])
237+
print(help_text)
238+
return help_flag
239+
239240
if __name__ == "__main__":
240241
comm = MPI.COMM_WORLD
241242
rank = comm.Get_rank()

‎examples/get_info.py‎

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -38,19 +38,6 @@
3838
from mpi4py import MPI
3939
import pnetcdf
4040

41-
def parse_help():
42-
help_flag = "-h" in sys.argv or "--help" in sys.argv
43-
if help_flag and rank == 0:
44-
help_text = (
45-
"Usage: {} [-h] | [-q] [file_name]\n"
46-
" [-h] Print help\n"
47-
" [-q] Quiet mode (reports when fail)\n"
48-
" [filename] (Optional) output netCDF file name\n"
49-
).format(sys.argv[0])
50-
print(help_text)
51-
return help_flag
52-
53-
5441
def print_info(info_used):
5542
nkeys = info_used.Get_nkeys()
5643
print("MPI File Info: nkeys =", nkeys)
@@ -84,6 +71,19 @@ def pnetcdf_io(filename):
8471
f.close()
8572

8673

74+
def parse_help():
75+
help_flag = "-h" in sys.argv or "--help" in sys.argv
76+
if help_flag and rank == 0:
77+
help_text = (
78+
"Usage: {} [-h] | [-q] [file_name]\n"
79+
" [-h] Print help\n"
80+
" [-q] Quiet mode (reports when fail)\n"
81+
" [filename] (Optional) output netCDF file name\n"
82+
).format(sys.argv[0])
83+
print(help_text)
84+
return help_flag
85+
86+
8787
if __name__ == "__main__":
8888
comm = MPI.COMM_WORLD
8989
rank = comm.Get_rank()

‎examples/put_varn_int.py‎

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -42,20 +42,6 @@
4242
import pnetcdf
4343

4444

45-
def parse_help():
46-
help_flag = "-h" in sys.argv or "--help" in sys.argv
47-
if help_flag and rank == 0:
48-
help_text = (
49-
"Usage: {} [-h] | [-q] [file_name]\n"
50-
" [-h] Print help\n"
51-
" [-q] Quiet mode (reports when fail)\n"
52-
" [-k format] file format: 1 for CDF-1, 2 for CDF-2, 5 for CDF-5\n"
53-
" [filename] (Optional) output netCDF file name\n"
54-
).format(sys.argv[0])
55-
print(help_text)
56-
return help_flag
57-
58-
5945
def pnetcdf_io(file_name, file_format):
6046
NY = 4
6147
NX = 10
@@ -159,6 +145,20 @@ def pnetcdf_io(file_name, file_format):
159145
f.close()
160146

161147

148+
def parse_help():
149+
help_flag = "-h" in sys.argv or "--help" in sys.argv
150+
if help_flag and rank == 0:
151+
help_text = (
152+
"Usage: {} [-h] | [-q] [file_name]\n"
153+
" [-h] Print help\n"
154+
" [-q] Quiet mode (reports when fail)\n"
155+
" [-k format] file format: 1 for CDF-1, 2 for CDF-2, 5 for CDF-5\n"
156+
" [filename] (Optional) output netCDF file name\n"
157+
).format(sys.argv[0])
158+
print(help_text)
159+
return help_flag
160+
161+
162162
if __name__ == "__main__":
163163
comm = MPI.COMM_WORLD
164164
rank = comm.Get_rank()

‎examples/transpose.py‎

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -29,21 +29,6 @@
2929
from mpi4py import MPI
3030
import pnetcdf
3131

32-
def parse_help():
33-
help_flag = "-h" in sys.argv or "--help" in sys.argv
34-
if help_flag and rank == 0:
35-
help_text = (
36-
"Usage: {} [-h] | [-q] [file_name]\n"
37-
" [-h] Print help\n"
38-
" [-q] Quiet mode (reports when fail)\n"
39-
" [-k format] file format: 1 for CDF-1, 2 for CDF-2, 5 for CDF-5\n"
40-
" [-l len] size of each dimension of the local array\n"
41-
" [filename] (Optional) output netCDF file name\n"
42-
).format(sys.argv[0])
43-
print(help_text)
44-
return help_flag
45-
46-
4732
def pnetcdf_io(filename, file_format, length):
4833

4934
NDIMS = 3
@@ -164,6 +149,21 @@ def pnetcdf_io(filename, file_format, length):
164149
f.close()
165150

166151

152+
def parse_help():
153+
help_flag = "-h" in sys.argv or "--help" in sys.argv
154+
if help_flag and rank == 0:
155+
help_text = (
156+
"Usage: {} [-h] | [-q] [file_name]\n"
157+
" [-h] Print help\n"
158+
" [-q] Quiet mode (reports when fail)\n"
159+
" [-k format] file format: 1 for CDF-1, 2 for CDF-2, 5 for CDF-5\n"
160+
" [-l len] size of each dimension of the local array\n"
161+
" [filename] (Optional) output netCDF file name\n"
162+
).format(sys.argv[0])
163+
print(help_text)
164+
return help_flag
165+
166+
167167
if __name__ == "__main__":
168168
comm = MPI.COMM_WORLD
169169
rank = comm.Get_rank()

‎examples/transpose2D.py‎

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -51,20 +51,6 @@
5151
from mpi4py import MPI
5252
import pnetcdf
5353

54-
def parse_help():
55-
help_flag = "-h" in sys.argv or "--help" in sys.argv
56-
if help_flag and rank == 0:
57-
help_text = (
58-
"Usage: {} [-h] | [-q] [file_name]\n"
59-
" [-h] Print help\n"
60-
" [-q] Quiet mode (reports when fail)\n"
61-
" [-k format] file format: 1 for CDF-1, 2 for CDF-2, 5 for CDF-5\n"
62-
" [-l len] size of each dimension of the local array\n"
63-
" [filename] (Optional) output netCDF file name\n"
64-
).format(sys.argv[0])
65-
print(help_text)
66-
return help_flag
67-
6854
def pnetcdf_io(filename, file_format, length):
6955
NDIMS = 2
7056

@@ -147,6 +133,20 @@ def pnetcdf_io(filename, file_format, length):
147133
f.close()
148134

149135

136+
def parse_help():
137+
help_flag = "-h" in sys.argv or "--help" in sys.argv
138+
if help_flag and rank == 0:
139+
help_text = (
140+
"Usage: {} [-h] | [-q] [file_name]\n"
141+
" [-h] Print help\n"
142+
" [-q] Quiet mode (reports when fail)\n"
143+
" [-k format] file format: 1 for CDF-1, 2 for CDF-2, 5 for CDF-5\n"
144+
" [-l len] size of each dimension of the local array\n"
145+
" [filename] (Optional) output netCDF file name\n"
146+
).format(sys.argv[0])
147+
print(help_text)
148+
return help_flag
149+
150150
if __name__ == "__main__":
151151
comm = MPI.COMM_WORLD
152152
rank = comm.Get_rank()

0 commit comments

Comments
 (0)