marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

bugfix: rays2vtk last point not displayed

Browse Source
This commit is contained in:
Marcel Paffrath 2015-11-11 15:45:56 +01:00
parent 54cb6d2f9a
commit 39a12bd1d1
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pylot/core/active/fmtomo2vtk.py
View File

@ -47,6 +47,9 @@ def vgrids2VTK(inputfile = 'vgrids.in', outputfile = 'vgrids.vtk', absOrRel = 'a
return sR, sTheta, sPhi return sR, sTheta, sPhi
def readVelocity(filename): def readVelocity(filename):
'''
Reads in velocity from vgrids file and returns a list containing all values in the same order
'''
vel = []; count = 0 vel = []; count = 0
fin = open(filename, 'r') fin = open(filename, 'r')
vglines = fin.readlines() vglines = fin.readlines()
@ -59,7 +62,7 @@ def vgrids2VTK(inputfile = 'vgrids.in', outputfile = 'vgrids.vtk', absOrRel = 'a
print("Read %d points out of file: %s" %(count - 4, filename)) print("Read %d points out of file: %s" %(count - 4, filename))
return vel return vel
R = 6371 # earth radius R = 6371. # earth radius
outfile = open(outputfile, 'w') outfile = open(outputfile, 'w')
# Theta, Phi in radians, R in km # Theta, Phi in radians, R in km
@ -167,7 +170,7 @@ def rays2VTK(fnin, fdirout = './vtk_files/', nthPoint = 50):
rays[shotnumber] = {} rays[shotnumber] = {}
rays[shotnumber][raynumber] = [] rays[shotnumber][raynumber] = []
for index in range(nRayPoints): for index in range(nRayPoints):
if index%nthPoint is 0 or index == nRayPoints: if index % nthPoint is 0 or index == (nRayPoints - 1):
rad, lat, lon = infile.readline().split() rad, lat, lon = infile.readline().split()
rays[shotnumber][raynumber].append([getDistance(np.rad2deg(float(lon))), getDistance(np.rad2deg(float(lat))), float(rad) - R]) rays[shotnumber][raynumber].append([getDistance(np.rad2deg(float(lon))), getDistance(np.rad2deg(float(lat))), float(rad) - R])
else: else: