Scripts: edgesum.py

#
# Add volume data values within p planes of boundary of active volume.
# Report number of voxels and sum of volume data values.
#
from VolumeViewer import active_volume
v = active_volume()

p = 5

# Get data matrix and matrix size.
m = v.full_matrix()
ksize, jsize, isize = m.shape

vsum = 0
vcount = 0

# Add z planes
vsum += m[0:p,:,:].sum()
vsum += m[ksize-p:ksize,:,:].sum()
vcount += 2*isize*jsize*p

# Add y planes, but not the part that was already in a z plane.
vsum += m[p:ksize-p,0:p,:].sum()
vsum += m[p:ksize-p,jsize-p:jsize,:].sum()
vcount += 2 * (ksize-2*p) * isize * p

# Add x planes, but not part in z or y planes.
vsum += m[p:ksize-p,p:jsize,0:p].sum()
vsum += m[p:ksize-p,p:jsize,isize-p:isize].sum()
vcount += 2 * (ksize-2*p) * (jsize-2*p) * p

# Print result to reply log.
print 'Edge sum for volume %s, %d planes, vsum = %.5g, vcount = %d' % (v.name, p, vsum, vcount)

# Show the reply log.
from chimera import runCommand
runCommand('ac rl')