python:test.py

readdosedata.py

#!/usr/bin/python

import matplotlib
import numpy as np
import matplotlib.cm as cm
import matplotlib.mlab as mlab
import matplotlib.pyplot as plt
import sys,getopt

#class dosedata:
#	def __init__(self,):

def readdosedata(filename):
	datafile = open(filename)
	N = np.fromstring(datafile.readline(),dtype=int, sep = ' ')
	N[0],N[2] = N[2],N[0]
	bx = np.fromstring(datafile.readline(),dtype=np.double, sep = ' ')
	by = np.fromstring(datafile.readline(),dtype=np.double, sep = ' ')
	bz = np.fromstring(datafile.readline(),dtype=np.double, sep = ' ')
	dose = np.fromstring(datafile.readline(),dtype=np.double, sep = ' ')
	dose.shape = N
	doseerr = np.fromstring(datafile.readline(),dtype=np.double, sep = ' ')
	doseerr.shape = N
	datafile.close()
	return [N,bx,by,bz,dose,doseerr]

def plotPDD(dosedata):
	dose=dosedata[4]
	bz = dosedata[3]
	N=dosedata[0]
	plt.plot(bz[:-1],dose[:,N[1] /2,N[2] /2]/((dose[7,N[1] /2,N[2] /2]+dose[8,N[1] /2,N[2] /2])/2))
	plt.xlabel("depth(cm)")
	#plt.ylabel("dose(Gy/particle)")
	plt.ylabel("nomalized dose")
	plt.title('PDD of 60mm Collimator')
	locs = plt.yticks()[0]
	labels = ["%g" % l for l in locs]
	plt.yticks(locs,labels)
	plt.show()

def plotOARx(dosedata):
	dose=dosedata[4]
	bx = dosedata[1]
	N=dosedata[0]
	plt.plot(bx[:-1],dose[N[0] /2,N[1] /2,:]) #/((dose[7,N[1] /2,N[2] /2]+dose[8,N[1] /2,N[2] /2])/2))
	plt.xlabel("x(cm)")
	#plt.ylabel("dose(Gy/particle)")
	plt.ylabel("nomalized dose")
	plt.title('OAR in x direction of 60mm Collimator')
	locs = plt.yticks()[0]
	#labels = ["%g" % l for l in locs]
	#plt.yticks(locs,labels)
	plt.show()

def plotOARy(dosedata):
	dose=dosedata[4]
	by = dosedata[2]
	N=dosedata[0]
	plt.plot(by[50:72],dose[N[0] /2,50:72,N[2] /2])#/((dose[7,N[1] /2,N[2] /2]+dose[8,N[1] /2,N[2] /2])/2))
	plt.xlabel("y(cm)")
	#plt.ylabel("dose(Gy/particle)")
	plt.ylabel("nomalized dose")
	plt.title('OAR in y direction of 60mm Collimator')
	#locs = plt.yticks()[0]
	#labels = ["%g" % l for l in locs]
	#plt.yticks(locs,labels)
	plt.show()

def plotProfile(dosedata,depth):
	matplotlib.rcParams['xtick.direction'] = 'out'
	matplotlib.rcParams['ytick.direction'] = 'out'
	X = dosedata[1][0:-1]
	Y = dosedata[2][0:-1]
	Z = dosedata[3]
	depthIndex = 0
	while Z[depthIndex] < depth:
		depthIndex += 1
	plt.figure()
	CS = plt.contourf(X,Y,dosedata[4][depthIndex,:,:])
	# plt.clabel(CS, inline = 1,fontsize = 10)
	plt.show()

if __name__ == '__main__':
	opts,args = getopt.getopt(sys.argv[1:],"xyz","")
	dosedata = readdosedata(args[0])
	opt = opts[0]
	if opt[0] == '-z':
		plotPDD(dosedata)
	if opt[0] == '-x':
		plotOARx(dosedata)
	if opt[0] == '-y':
		plotOARy(dosedata)

test.py

def swap(a,b):
	a,b = b,a