Coverage for /builds/ase/ase/ase/visualize/paraview_script.py: 10.53%

1# fmt: off 

2 

3def main(): 

4 import paraview.simple as para 

5 version_major = para.servermanager.vtkSMProxyManager.GetVersionMajor() 

6 source = para.GetActiveSource() 

7 renderView1 = para.GetRenderView() 

8 atoms = para.Glyph( 

9 Input=source, 

10 GlyphType='Sphere', 

11 Scalars='radii', 

12 ScaleMode='scalar', 

13 ) 

14 para.RenameSource('Atoms', atoms) 

15 atomsDisplay = para.Show(atoms, renderView1) 

16 if version_major <= 4: 

17 atoms.SetScaleFactor = 0.8 

18 atomicnumbers_PVLookupTable = para.GetLookupTableForArray( 

19 'atomic numbers', 1) 

20 atomsDisplay.ColorArrayName = ('POINT_DATA', 'atomic numbers') 

21 atomsDisplay.LookupTable = atomicnumbers_PVLookupTable 

22 else: 

23 atoms.ScaleFactor = 0.8 

24 para.ColorBy(atomsDisplay, 'atomic numbers') 

25 atomsDisplay.SetScalarBarVisibility(renderView1, True) 

26 para.Render() 

27 

28 

29if __name__ == '__main__': 

30 main()