1import ase.parallel as parallel 

2from ase.visualize.external import viewers 

3 

4 

5def view(atoms, data=None, viewer='ase', repeat=None, block=False): 

6 if parallel.world.size > 1: 

7 return 

8 

9 vwr = viewers[viewer.lower()] 

10 handle = vwr.view(atoms, data=data, repeat=repeat) 

11 

12 if block and hasattr(handle, 'wait'): 

13 status = handle.wait() 

14 if status != 0: 

15 raise RuntimeError(f'Viewer "{vwr.name}" failed with status ' 

16 '{status}') 

17 

18 return handle