timer = {}
if USE_MPI:
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
size = comm.Get_size()
root = 0
if USE_MPI is False or rank == root:
timer['detector'] = timeit.default_timer()
if USE_MPI:
if rank == root:
grid = RectGrid(pathname=infile, varname=var, trange=trange)
grid = grid.split(size)
else:
grid = None
grid = comm.scatter(grid, root=root)
else:
grid = RectGrid(pathname=infile, varname=var, trange=trange)
centers = grid.detect()
if USE_MPI:
comm.Barrier()
if USE_MPI is False or rank == root:
timer['detector'] = timeit.default_timer()-timer['detector']
self.tail = new_tail
self.tend = tracks.tend
if __name__ == "__main__":
import timeit
try:
import cPickle as pickle
except ImportError:
import pickle
print("Starting detector...")
timer = timeit.default_timer()
grid = RectGrid(pathname="../slp.2012.nc", varname="slp", trange=(0,120))
centers = grid.detect()
print("Detection time: " + str(timeit.default_timer()-timer))
print("Starting linker...")
timer = timeit.default_timer()
tracks = Tracks()
for c in centers:
tracks.append_center(c)
print("Linking time: " + str(timeit.default_timer()-timer))
self.tend = tracks.tend
if __name__ == "__main__":
import timeit
try:
import cPickle as pickle
except ImportError:
import pickle
print("Starting detector...")
timer = timeit.default_timer()
grid = RectGrid(pathname="../slp.2012.nc", varname="slp", trange=(0, 120))
centers = grid.detect()
print("Detection time: " + str(timeit.default_timer() - timer))
print("Starting linker...")
timer = timeit.default_timer()
tracks = Tracks()
for c in centers:
tracks.append_center(c)
print("Linking time: " + str(timeit.default_timer() - timer))
timer = {}
if USE_MPI:
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
size = comm.Get_size()
root = 0
if USE_MPI is False or rank == root:
timer['detector'] = timeit.default_timer()
if USE_MPI:
if rank == root:
grid = RectGrid(pathname=infile, varname=var, trange=trange)
grid = grid.split(size)
else:
grid = None
grid = comm.scatter(grid, root=root)
else:
grid = RectGrid(pathname=infile, varname=var, trange=trange)
centers = grid.detect()
if USE_MPI:
comm.Barrier()
if USE_MPI is False or rank == root:
timer['detector'] = timeit.default_timer() - timer['detector']
