# get unique stations and map them

# map:
width = 2000000; 
height=4500000
lon_0 = -130; lat_0 = 51.6
plt.figure(figsize=(10,10))
m = Basemap(width=width,height=width,projection='aeqd', resolution='i',
            lat_0=lat_0,lon_0=lon_0)
# fill background.
m.drawmapboundary()
# draw coasts and fill continents.
m.drawcoastlines(linewidth=0.5)
m.drawrivers()
#m.shadedrelief()
# 20 degree graticule.
m.drawparallels(np.arange(40,60,2))
m.drawmeridians(np.arange(-130,-110,2))


#xpt, ypt = m(-123.26137, 52.07197)
#xpt2, ypt2 = m(lon_0+.04, lat_0)
#m.plot([xpt],[ypt],'ko')
#plt.text(xpt2,ypt2,'station',fontsize=12,fontweight='bold',
#                    ha='left',va='center',color='k')

plt.figure(figsize=(8,8))
width = 2000000; 
height=2000000
lon_0 = -125; lat_0 = 50.0
            
            
# create the figure and axes instances.
fig = plt.figure()
fig.patch.set_facecolor('none')
ax = fig.add_axes([0.1,0.1,0.8,0.8])
# setup of basemap ('lcc' = lambert conformal conic).
# use major and minor sphere radii from WGS84 ellipsoid.
m = Basemap(width=width,height=width,projection='aeqd', resolution='i',
            lat_0=lat_0,lon_0=lon_0,ax=ax)
# transform to nx x ny regularly spaced 5km native projection grid
nx = int((m.xmax-m.xmin)/5000.)+1; ny = int((m.ymax-m.ymin)/5000.)+1
topodat = m.transform_scalar(topoin,lons,lats,nx,ny)
# plot image over map with imshow.
im = m.imshow(topodat,cm.GMT_relief,clim=(-8000, 8000))
# draw coastlines and political boundaries.
m.drawcoastlines()
m.drawcountries()
m.drawstates()
# draw parallels and meridians.
# label on left and bottom of map.
parallels = np.arange(30.,70.,10.)
m.drawparallels(parallels,labels=[1,0,0,1],size=16)
meridians = np.arange(-150.,-80.,10.)
m.drawmeridians(meridians,labels=[0,0,0,1],size=16)
lonmin=-128
lonmax=-121
latmin=47
latmax=52
m.plot((lonmin,lonmax,lonmax,lonmin,lonmin),(latmin,latmin,latmax,latmax,latmin),'r-',latlon=True)
# add colorbar
#cb = m.colorbar(im,"right", size="5%", pad='2%')
#ax.set_title('ETOPO5 Topography - Lambert Conformal Conic')
#plt.show()
fig.savefig('/home/eolson/pyCode/notebooks/figs/mapLarge.png')
fig.savefig('/home/eolson/pyCode/notebooks/figs/mapLarge.pdf')

fig=plt.figure(figsize=(8,8))
m = Basemap(projection='merc',llcrnrlat=latmin,urcrnrlat=latmax,\
            llcrnrlon=lonmin,urcrnrlon=lonmax,resolution='h')
# fill background.
m.drawmapboundary(fill_color=(.6,.8,1))
m.fillcontinents(color=(1,1,.6),lake_color=(.6,.8,1))
# draw coasts and fill continents.
#m.drawcoastlines(linewidth=0.5)
m.drawrivers(color=(.6,.8,1),linewidth=2)
#m.shadedrelief()
# 20 degree graticule.
m.drawparallels(np.arange(int(latmin)+1,int(latmax)+1,2),labels=[1,0,0,1],size=16)
m.drawmeridians(np.arange(int(lonmin),int(lonmax)+1,2),labels=[1,0,0,1],size=16)
m.drawcountries()
m.drawstates()
m.plot(-123.1207,49.2827,'ko',latlon=True)
x,y = m(-123.85, 49.4)
plt.text(x,y,'Vancouver',size=16)
fig.savefig('/home/eolson/pyCode/notebooks/figs/mapInset.png')

fig = plt.figure(figsize=(8,8))
fig.patch.set_facecolor('none')
ax = fig.add_axes([0.1,0.1,0.8,0.8])
# setup of basemap ('lcc' = lambert conformal conic).
# use major and minor sphere radii from WGS84 ellipsoid.
m = Basemap(projection='merc',llcrnrlat=latmin,urcrnrlat=latmax-.4,\
            llcrnrlon=lonmin,urcrnrlon=lonmax,resolution='h')
# transform to nx x ny regularly spaced 5km native projection grid
nx = int((m.xmax-m.xmin)/5000.)+1; ny = int((m.ymax-m.ymin)/5000.)+1
topodat = m.transform_scalar(topoin,lons,lats,nx,ny)
# plot image over map with imshow.
topo2=np.where(topodat<0,np.log10(-topodat+10),np.log10(10))
im = m.imshow(topo2,cmap='Blues',clim=(np.log10(10), np.log10(3000)))#
# draw coastlines and political boundaries.
#m.drawcoastlines()
# draw parallels and meridians.
# label on left and bottom of map.
#parallels = np.arange(30.,70.,10.)
#m.drawparallels(parallels,labels=[1,0,0,1],size=16)
#meridians = np.arange(-150.,-80.,10.)
#m.drawmeridians(meridians,labels=[0,0,0,1],size=16)
#m.fillcontinents(color=(.6,.9,.6),lake_color=(.9,1,.9))
#m.drawrivers(color=(.9,1,.9),linewidth=2)
x,y = m(-121.9, 49.7)
plt.text(x,y,'Fraser\n River',size=20,fontname='Times New Roman')
x,y = m(-124.8, 48.1)
plt.text(x,y,'Strait of\n Juan de\n Fuca',size=20,fontname='Times New Roman')
x,y = m(-124.2, 49.1)
plt.text(x,y,'Strait of\n Georgia',size=20,fontname='Times New Roman')
x,y = m(-123.5, 48.25)
plt.text(x,y,' Gulf/\n San Juan\n Islands',size=20,fontname='Times New Roman')
x,y = m(-126.1, 50.5)
xt,yt=m(-126.5,50.8)
ax.annotate('Johnstone\n Strait', xy=(x, y), xytext=(xt, yt), size=20,
            arrowprops=dict(facecolor='black'),fontname='Times New Roman')#, shrink=0.05

# add colorbar
#cb = m.colorbar(im,"right", size="5%", pad='2%')
#ax.set_title('ETOPO5 Topography - Lambert Conformal Conic')
#plt.show()

fig.savefig('/home/eolson/pyCode/notebooks/figs/mapLabeled.png')
fig.savefig('/home/eolson/pyCode/notebooks/figs/mapLabeled.pdf')