def unitPowerBall(ballName, xcen, ycen, zcen,
xradius, yradius, zradius,
par, needplot, flatplot=False):
''' Pwoerball with Gaussian shape, unit maximum magnitude'''
wplot.param(par)
centers = [
[m.sqrt(3.0), 1.0], [0.0, 2.0], [-m.sqrt(3.0), 1.0],
[-m.sqrt(3.0), -1.0], [0.0, -2.0], [m.sqrt(3.0), -1.0]]
centers = map(lambda (x,y): [xcen + x * xradius, ycen + y * xradius], centers)
xsig = 0.512 * xradius
ysig = 0.512 * yradius
zsig = 0.512 * zradius
ball_list = []
for i in range(6):
xcoord = centers[i][0]
ycoord = centers[i][1]
zcoord = zcen
name = ballName +'_%d'%i
ball_list += [name]
gauss3d(name, xcoord, ycoord, zcoord, xsig, ysig, zsig, par)
xframe = xcoord/par['dx']
yframe = ycoord/par['dy']
if needplot == True:
plotstring = wplot.igrey3d('flat=n point1=0.618 point2=0.618 frame2=%d frame3=%d color=j '%(xframe,yframe),par)
if flatplot: plotstring += ' flat=y '
Result(name, 'byte pclip=100 gainpanel=a | '+ plotstring)
# --------- cat plots -----------
extra = ' wantaxis1=n wantaxis2=n wantaxis3=n '
if i == 0 or i == 3: extra += ' wantaxis1=y '
if i == 2 or i == 5: extra += ' wantaxis3=y '
if i > 2 and i < 6: extra += ' wantaxis2=y '
Plot(name, 'byte pclip=100 gainpanel=a | '+ plotstring+extra)
from rsf.recipes import pplot
pplot.p1x3(ballName+'_merge_up',ball_list[0],ball_list[1],ball_list[2],1,1,-9.10)
pplot.p1x3(ballName+'_merge_down',ball_list[3],ball_list[4],ball_list[5],1,1,-9.10)
pplot.p2x3(ballName+'_merge',
ball_list[0], ball_list[1], ball_list[2],
ball_list[3], ball_list[4], ball_list[5],
0.5,0.5,-6.1,-9.1)
Flow(ballName+'_all', ball_list, 'add ${SOURCES[1:-1]}')
if needplot == True:
yframe = centers[5][1] / par['dy']
plotstring = wplot.igrey3d('flat=n point1=0.618 point2=0.618 frame3=%d color=j '%yframe,par)
# ratio = (par['nz']*par['dz']+par['oz']) / (par['nx']*par['dx']+par['ox'])
# heit = 12.0 * ratio
# plotstring += ' screenratio=%f screenht=%f '%(ratio, heit)
if flatplot: plotstring += ' flat=y '
Result(ballName+'_all', 'byte pclip=100 gainpanel=a mean=y | '+plotstring)
if needplot == True:
yframe = centers[5][1] / par['dy']
plotstring = wplot.igrey3d('flat=n color=i frame3=%d'%yframe,par)
if flatplot: plotstring += ' flat=y '
Plot(ballName+'_all', 'byte pclip=100 gainpanel=a mean=y | '+plotstring)
Plot('box_source_zSlice_scaled', 'box_source_zSlice','Overlay', vppen='yscale=0.6 xscale=0.6 xcenter=-1.652')
pplot.p2x1(ballName+'_all_source',
ballName+'_all' , 'box_source_zSlice_scaled',
1,1,-1.518)
def plt3x2(par, filename, pltname='', pclip=100, flat=False, needBound=False):
from copy import deepcopy
pltpar = deepcopy(par)
wplot.param(pltpar)
pltpar['pclip'] = pclip
boundstr = ''
if needBound:
boundstr = ' minval=%(xlb)g maxval=%(xub)g ' % pltpar
if not pltname:
pltname = filename
bytefile = filename + '_byte'
barfile = filename + '_bar'
Flow([bytefile, barfile], filename, '''
byte bar=${TARGETS[1]} gainpanel=a pclip=%(pclip)g mean=n bias=2.3
''' % pltpar + boundstr)
plist = []
for i in range(6):
basic = wplot.igrey3d(
'''
bar=${SOURCES[1]} bartype=h barwidth=0.256 barmove=n
color=j flat=n point1=0.618 point2=0.618 mean=n bias=2.3
''' + boundstr, pltpar)
if flat: pltstr += ' flat=y '
extra = ' wantaxis1=n wantaxis2=n wantaxis3=n '
if i == 0 or i == 3: extra += ' wantaxis1=y '
if i == 2 or i == 5: extra += ' wantaxis3=y '
if i > 2 and i < 6: extra += ' wantaxis2=y '
pltstr = basic + extra
plist += [pltname + '_%d' % i]
Plot(pltname + '_%d' % i, [bytefile, barfile], '''
window n4=1 f4=%d | ''' % i + pltstr)
pplot.p2x3(pltname + '_byte', plist[0], plist[1], plist[2], plist[3],
plist[4], plist[5], 0.5, 0.5, -7.4, -9.2)
Plot(pltname + '_bar_', [bytefile, barfile], '''
window n4=1 |''' + pltstr + ' scalebar=y')
barX = 10.20
barY = 0.22
barUnits = "(km\s75 /\s100 s)"
Plot(
pltname + '_bar_units', None, '''
box x0=%g y0=%g label="%s"
boxit=n pointer=n lab_fat=3 size=0.18
''' % (barX, barY, barUnits))
Plot(pltname + '_bar__',
pltname + '_bar_',
'Overlay',
vppen='ywmin=7.2 yshift=-7.2')
Plot(pltname + '_bar', [pltname + '_bar__', pltname + '_bar_units'],
'Overlay')
Plot(pltname + '_byte_yshift',
pltname + '_byte',
'Overlay',
vppen='ycenter=-1.0')
pplot.p2x1(pltname + '_noText', pltname + '_byte_yshift', pltname + '_bar',
1, 1, 0.0)
addText(pltname, pltname + '_noText', [])
def plots(par):
Result('icomp', 'imgCC migCC-FFD migCC-F60', 'Movie')
Result('imgCC', ['imgCC', 'cos'], 'Overlay')
pplot.p3x2('iCCall', 'migCC-SSF', 'migCC-PSC', 'migCC-FFD', 'migCC-F15',
'migCC-F45', 'migCC-F60', 0.3, 0.3, -8, -12)
pplot.p3x2('iRCall', 'migRC-SSF', 'migRC-PSC', 'migRC-FFD', 'migRC-F15',
'migRC-F45', 'migRC-F60', 0.3, 0.3, -10, -12)
Plot('imgCC', 'window | transp |' + cgrey('pclip=99', par))
Plot('imgRC', 'migCC-FFD', ' transp |' + cgrey('pclip=99', par))
Plot('imgRC-ovl', ['imgRC', 'cos'], 'Overlay')
pplot.p2x1('CCvsRC', 'imgRC-ovl', 'imgCC', 0.5, 0.5, -9)
def plots(par):
pplot.p3x2(par['prefix'] + 'iCCall', 'migCC-SSF', 'migCC-PSC', 'migCC-FFD',
'migCC-F15', 'migCC-F45', 'migCC-F60', 0.3, 0.3, -8, -12)
pplot.p3x2(par['prefix'] + 'iRCall', 'migRC-SSF', 'migRC-PSC', 'migRC-FFD',
'migRC-F15', 'migRC-F45', 'migRC-F60', 0.3, 0.3, -10, -12)
Result(par['prefix'] + 'icomp', ['imgCC', 'migCC-FFD', 'migCC-F60'],
'Movie')
Result(par['prefix'] + 'imgCC', ['imgCC', 'cos'], 'Overlay')
Plot('imgCC', 'window | transp |' + fdmod.cgrey('pclip=100', par))
Plot('imgRC', 'migCC-FFD',
' transp |' + fdmod.cgrey('pclip=100', par))
Plot('imgRC-ovl', ['imgRC', 'cos'], 'Overlay')
pplot.p2x1(par['prefix'] + 'CCvsRC', 'imgRC-ovl', 'imgCC', 0.5, 0.5, -9)
def eframe(frame,movie,index,custom,axis,par,xscale=0.75,yscale=0.75,shift=-8.25):
Flow([movie+'-plt',movie+'-bar'],movie,
'byte bar=${TARGETS[1]} gainpanel=a pclip=100 %s' % custom)
for i in range(2):
Plot(frame+'-'+str(i),movie+'-plt',
'window n3=1 f3=%d n4=1 f4=%d |' % (i,index)
+ cgrey('',par))
# Result(frame+'-'+str(i),movie+'-plt',
# 'window n3=1 f3=%d n4=1 f4=%d |' % (i,index)
# + cgrey('',par))
if(axis==1):
pplot.p2x1(frame,frame+'-1',frame+'-0',yscale,xscale,shift)
else:
pplot.p1x2(frame,frame+'-0',frame+'-1',yscale,xscale,shift)
def eframe(frame,movie,index,custom,axis,par,xscale=0.75,yscale=0.75,shift=-8.25):
Flow([movie+'-plt',movie+'-bar'],movie,
'byte bar=${TARGETS[1]} gainpanel=a pclip=100 %s' % custom)
for i in range(2):
Plot(frame+'-'+str(i),movie+'-plt',
'window n3=1 f3=%d n4=1 f4=%d |' % (i,index)
+ cgrey('',par))
# Result(frame+'-'+str(i),movie+'-plt',
# 'window n3=1 f3=%d n4=1 f4=%d |' % (i,index)
# + cgrey('',par))
if(axis==1):
pplot.p2x1(frame,frame+'-1',frame+'-0',yscale,xscale,shift)
else:
pplot.p1x2(frame,frame+'-0',frame+'-1',yscale,xscale,shift)