File: tempfactor.py

package info (click to toggle)
ball 1.4.2%2B20140406-1
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 316,064 kB
  • ctags: 38,021
  • sloc: cpp: 341,838; ansic: 4,096; python: 2,645; yacc: 1,778; lex: 1,089; xml: 964; sh: 688; sql: 316; makefile: 150; awk: 118
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)