File: tempfactor.py

package info (click to toggle)
ball 1.5.0%2Bgit20180813.37fc53c-11
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 239,924 kB
  • sloc: cpp: 326,149; ansic: 4,208; python: 2,303; yacc: 1,778; lex: 1,099; xml: 958; sh: 322; javascript: 164; makefile: 88
file content (21 lines) | stat: -rw-r--r-- 639 bytes parent folder | download | duplicates (9)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# color a representation transparent in relation to the temp factor
max_value = 9
rep = getRepresentations()[0]
# enable transparency for the representation
rep.setTransparency(190)
# disable model and coloring recalculation
rep.enableColoringUpdate(false)
rep.enableModelUpdate(false)
gos = rep.getGeometricObjects()
for i in range(len(gos)):
	go = gos[i]
	atoml = all_atoms(go.getComposite())
	if len(atoml) == 0:
		continue
	tf  = 0
	for j in range(len(atoml)):
		tf = tf + cast(atoml[j],PDBAtom).getTemperatureFactor()
	tf = tf * 255/ (len(atoml) * max_value)
	go.getColor().setAlpha(ColorUnit(int(tf)))
	
getMainControl().update(rep)