Exemplo n.º 1
0
def get_arrays(casenum):

	dbz,vvel, ht, dayt = read_sprof.retrieve_arrays(base_directory, casenum)

	reqdates = precip.request_dates_significant(casenum)
	idx_st=find_index(dayt,reqdates['ini'])
	idx_end=find_index(dayt,reqdates['end'])

	dayt=dayt[idx_st:idx_end+1]
	vvel=vvel[:,idx_st:idx_end+1]
	dbz=dbz[:,idx_st:idx_end+1]

	with np.errstate(invalid='ignore'):
		vvel[vvel<=-10.]=np.nan
		vvel[vvel>=4.]=np.nan

	' add total seconds to array'
	ts=[]
	for d in dayt:
		ts.append((d-dayt[0]).total_seconds())

	' create equally spaced time grid '
	day = dayt[0]
	end = dayt[-1]
	step = timedelta(seconds=40)
	ts2 = []
	while day < end:
		result=(day-dayt[0]).total_seconds()
		ts2.append(result)
		day += step

	return dbz,vvel,ht,ts,ts2,dayt
Exemplo n.º 2
0
def get_arrays(casenum):

    dbz, vvel, ht, dayt = read_sprof.retrieve_arrays(base_directory, casenum)

    reqdates = precip.request_dates_significant(casenum)
    idx_st = find_index(dayt, reqdates['ini'])
    idx_end = find_index(dayt, reqdates['end'])

    dayt = dayt[idx_st:idx_end + 1]
    vvel = vvel[:, idx_st:idx_end + 1]
    dbz = dbz[:, idx_st:idx_end + 1]

    with np.errstate(invalid='ignore'):
        vvel[vvel <= -10.] = np.nan
        vvel[vvel >= 4.] = np.nan

    ' add total seconds to array'
    ts = []
    for d in dayt:
        ts.append((d - dayt[0]).total_seconds())

    ' create equally spaced time grid '
    day = dayt[0]
    end = dayt[-1]
    step = timedelta(seconds=40)
    ts2 = []
    while day < end:
        result = (day - dayt[0]).total_seconds()
        ts2.append(result)
        day += step

    return dbz, vvel, ht, ts, ts2, dayt
Exemplo n.º 3
0
	def get_index(self,time):

		begdayt=self.get_begdayt()
		enddayt=self.get_enddayt()

		beg_aux = np.asarray([datetime(t.year, t.month, t.day, t.hour,0,0) for t in begdayt])
		end_aux = np.asarray([datetime(t.year, t.month, t.day, t.hour,0,0) for t in enddayt])
		dayt_aux=np.asarray([datetime(t.year, t.month, t.day, t.hour,0,0) for t in time])

		beg_index = np.where(beg_aux==dayt_aux[0])[0][0]
		end_index = np.where(end_aux==dayt_aux[-1])[0][1]

		begdayt = begdayt[beg_index:end_index]
		enddayt = enddayt[beg_index:end_index]

		beg = [[t.year, t.month, t.day, t.hour,t.minute] for t in begdayt]
		idx_be=np.asarray([find_index(time, x) for x in beg])
		end = [[t.year, t.month, t.day, t.hour,t.minute] for t in enddayt]
		idx_en=np.asarray([find_index(time, x) for x in end])
		idx_bbht=(idx_be+idx_en)/2

		return range(beg_index,end_index), idx_bbht
Exemplo n.º 4
0
def main():

	global extent
	global dayt

	casenum = raw_input('\nIndicate case number (i.e. 1): ')

	dbz,vvel, ht, dayt = read_sprof.retrieve_arrays(base_directory, casenum)

	extent=[0, len(dayt), 0, len(ht)]

	''' set index for time zooming '''
	try:
		idx_st=find_index(dayt,reqdates[casenum]['ini'])
		idx_end=find_index(dayt,reqdates[casenum]['end'])
	except KeyError:
		idx_st=0
		idx_end=len(dayt)-1


	''' Precipitation partition 
	**************************'''
	partition=read_partition.partition(dayt[0].year)
	bbht=partition.get_bbht()
	rtype=partition.get_rtype()
	begdayt=partition.get_begdayt()
	enddayt=partition.get_enddayt()
	
	beg_aux = np.asarray([datetime(t.year, t.month, t.day, t.hour,0,0) for t in begdayt])
	end_aux = np.asarray([datetime(t.year, t.month, t.day, t.hour,0,0) for t in enddayt])
	dayt_aux=np.asarray([datetime(t.year, t.month, t.day, t.hour,0,0) for t in dayt])

	beg_index = np.where(beg_aux==dayt_aux[0])[0][0]
	end_index = np.where(end_aux==dayt_aux[-1])[0][1]

	bbht = bbht[beg_index+1:end_index] 
	rtype = rtype[beg_index+1:end_index]
	begdayt = begdayt[beg_index:end_index]
	enddayt = enddayt[beg_index:end_index]

	d = [[t.year, t.month, t.day, t.hour,t.minute] for t in begdayt]
	idx_be=np.asarray([find_index(dayt,x) for x in d[1:]])
	d = [[t.year, t.month, t.day, t.hour,t.minute] for t in enddayt]
	idx_en=np.asarray([find_index(dayt,x) for x in d[1:]])
	idx_bbht=(idx_be+idx_en)/2


	''' 	Plots
	**************'''
	fig,ax=plt.subplots(2,1,sharex=True, figsize=(12,8))
	
	plot_reflectivity(ax[0],dbz,ht, cmap='nipy_spectral',vmin=-20,vmax=60)
	plot_velocity(ax[1],vvel,ht, cmap='bwr',vmin=-3,vmax=3)

	f = interp1d(ht, range(len(ht)))
	ax[0].plot(idx_bbht, f(bbht),marker='o',linestyle='--',color='black')
	for n, s in enumerate(rtype):
		if s[0] != 'NaN':
			ax[0].text(idx_bbht[n], -5, s[0][0].upper(), horizontalalignment='center',clip_on=True)
	ax[1].plot(idx_bbht, f(bbht),marker='o',linestyle='--',color='black')

	' it has to be before set_xlim'
	# format_xaxis(ax[1],dayt,2,casenum)
	format_xaxis(ax[1],dayt,labels=True,format='%d\n%H')

	ax[0].set_xlim([idx_st,idx_end])
	ax[0].set_ylim([-10,len(ht)])
	ax[1].set_ylim([-10,len(ht)])


	fig.subplots_adjust(hspace=.07)	
	
	ax[1].invert_xaxis()
	ax[1].set_xlabel(r'$\Leftarrow$'+' Time [UTC]')
	# ax[1].set_xlabel(' Time [UTC]'+r'$\Rightarrow$')

	# ax[0].grid(b=True, which='major', color='b', linestyle='-')
	plt.suptitle('SPROF observations. Date: '+dayt[0].strftime('%Y-%b'))

	plt.subplots_adjust(left=0.05, right=0.95, top=0.95, bottom=0.1)
	plt.draw()
	# plt.show()
	plt.show(block=False)