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
|
def main():
import paraview.simple as para
version_major = para.servermanager.vtkSMProxyManager.GetVersionMajor()
source = para.GetActiveSource()
renderView1 = para.GetRenderView()
atoms = para.Glyph(
Input=source,
GlyphType='Sphere',
Scalars='radii',
ScaleMode='scalar',
)
para.RenameSource('Atoms', atoms)
atomsDisplay = para.Show(atoms, renderView1)
if version_major <= 4:
atoms.SetScaleFactor = 0.8
atomicnumbers_PVLookupTable = para.GetLookupTableForArray(
'atomic numbers', 1)
atomsDisplay.ColorArrayName = ('POINT_DATA', 'atomic numbers')
atomsDisplay.LookupTable = atomicnumbers_PVLookupTable
else:
atoms.ScaleFactor = 0.8
para.ColorBy(atomsDisplay, 'atomic numbers')
atomsDisplay.SetScalarBarVisibility(renderView1, True)
para.Render()
if __name__ == '__main__':
main()
|
