def test_scalar_lin(lin, field='tens', cmap=plt.cm.jet): #{{{ import lineament satfile="input/ConvectingEuropa_GlobalDiurnalNSR.ssrun" europa = ss.Satellite(open(satfile,'r')) NSR = ss.StressCalc([ss.NSR(europa),]) mp_lons, mp_lats = lin.seg_midpoints() seg_lengths = lin.seg_lengths() min_lat = 0 max_lat = np.pi/2.0 min_lon = 0.0 max_lon = np.pi the_fig = plt.figure(figsize=(10,10)) map_ax = the_fig.add_subplot(1,1,1) map_ax.set_title("Point by Point scalar stresses field (%s)" % (field,) ) scalar_basemap = basemap.Basemap(llcrnrlon = np.degrees(min_lon),\ llcrnrlat = np.degrees(min_lat),\ urcrnrlon = np.degrees(max_lon),\ urcrnrlat = np.degrees(max_lat),\ ax = map_ax) scalar_points(stresscalc=NSR, lons=np.mod(mp_lons,np.pi), lats=np.mod(mp_lats,np.pi/2.0), symb_sizes=5000*seg_lengths, field=field, cmap=cmap, basemap_ax=scalar_basemap) junk = lineament.plotlinmap([lin,], map=scalar_basemap, lat_mirror=True, lon_cyc=np.pi) scalar_basemap.drawmeridians(np.degrees(np.linspace(min_lon, max_lon, 7)), labels=[1,0,0,1], linewidth=0.5, color='gray') scalar_basemap.drawparallels(np.degrees(np.linspace(min_lat, max_lat, 7)), labels=[1,0,0,1], linewidth=0.5, color='gray') scalar_basemap.drawmapboundary()
def test_vector_lin(lin, t=0.0, w_stress=False, w_length=False, scale=5e7, plot_tens=True, plot_comp=False, plot_greater=True, plot_lesser=False): #{{{ import lineament satfile="input/ConvectingEuropa_GlobalDiurnalNSR.ssrun" europa = ss.Satellite(open(satfile,'r')) NSR = ss.StressCalc([ss.NSR(europa),]) best_b = lin.best_fit()[1] mp_lons, mp_lats = lin.seg_midpoints() if w_length is True: scale_arr = len(mp_lons)*lin.seg_lengths()/lin.length else: scale_arr = np.ones(np.shape(mp_lons)) min_lat = 0 max_lat = np.pi/2.0 min_lon = 0.0 max_lon = np.pi vector_lin_fig = plt.figure() vector_lin_ax = vector_lin_fig.add_subplot(1,1,1) vector_lin_ax.set_title("Vectors at arbitrary locations (e.g. along a feature)") vector_lin_basemap = basemap.Basemap(llcrnrlon = np.degrees(min_lon),\ llcrnrlat = np.degrees(min_lat),\ urcrnrlon = np.degrees(max_lon),\ urcrnrlat = np.degrees(max_lat),\ ax = vector_lin_ax) junk = lineament.plotlinmap([lin,], map=vector_lin_basemap, lat_mirror=True, lon_cyc=np.pi) vector_lin_basemap.drawmeridians(np.degrees(np.linspace(min_lon, max_lon,13)), labels=[1,0,0,1], linewidth=0.5, color='gray') vector_lin_basemap.drawparallels(np.degrees(np.linspace(min_lat, max_lat, 7)), labels=[1,0,0,1], linewidth=0.5, color='gray') vector_lin_basemap.drawmapboundary() vector_points(stresscalc=NSR, lons=np.mod(mp_lons,np.pi), lats=mp_lats, time_t=0.0, lonshift=best_b,\ plot_tens=plot_tens, plot_greater=plot_greater, plot_comp=plot_comp, plot_lesser=plot_lesser, basemap_ax=vector_lin_basemap, scale=scale, scale_arr=scale_arr, w_stress=w_stress)