Add colorbar into loop png

LiCSBAS_lib/LiCSBAS_loop_lib.py

@@ -8,6 +8,9 @@
 =========
 Changelog
 =========
+v1.5.2 20210303 Yu Morishita, GSI
+ - Use get_cmap in make_loop_png
+ - Add colorbar in make_loop_png
 v1.5.1 20201119 Yu Morishita, GSI
  - Change default cmap for wrapped phase from insar to SCM.romaO
 v1.5 20201006 Yu Morishita, GSI
@@ -27,8 +30,8 @@
 
 import os
 import numpy as np
-import SCM
 import LiCSBAS_io_lib as io_lib
+import LiCSBAS_tools_lib as tools_lib
 
 os.environ['QT_QPA_PLATFORM']='offscreen'
 import warnings
@@ -43,7 +46,7 @@
 def make_loop_matrix(ifgdates):
  """
  Make loop matrix (containing 1, -1, 0) from ifgdates.
- 
+
  Inputs:
  ifgdates : Unwrapped phase vector at a point without nan (n_ifg)
 
@@ -58,7 +61,8 @@ def make_loop_matrix(ifgdates):
  for ix_ifg12, ifgd12 in enumerate(ifgdates):
  primary12 = ifgd12[0:8]
  secondary12 = ifgd12[9:17]
- ifgdates23 = [ ifgd for ifgd in ifgdates if ifgd.startswith(secondary12)] # all candidates of ifg23
+ ifgdates23 = [ ifgd for ifgd in ifgdates if
+ ifgd.startswith(secondary12)] # all candidates of ifg23
 
  for ifgd23 in ifgdates23: # for each candidate of ifg23
  secondary23 = ifgd23[9:17]
@@ -101,7 +105,7 @@ def read_unw_loop_ph(Aloop1, ifgdates, ifgdir, length, width, bad_ifg=[]):
  unw13file = os.path.join(ifgdir, ifgd13, ifgd13+'.unw')
  unw13 = io_lib.read_img(unw13file, length, width)
  unw13[unw13 == 0] = np.nan # Fill 0 with n
- 
+
  return unw12, unw23, unw13, ifgd12, ifgd23, ifgd13
 
 
@@ -115,15 +119,16 @@ def identify_bad_ifg(bad_ifg_cand, good_ifg):
 
  bad_ifg = list(set(bad_ifg_cand)-set(good_ifg)) # difference
  bad_ifg.sort()
- 
+
  return bad_ifg
 
 
-#%% 
+#%%
 def make_loop_png(unw12, unw23, unw13, loop_ph, png, titles4, cycle):
- cmap_wrap = SCM.romaO
+ cmap_wrap = tools_lib.get_cmap('SCM.romaO')
+ cmap_loop = tools_lib.get_cmap('SCM.vik')
 
- ### Settings 
+ ### Settings
  plt.rcParams['axes.titlesize'] = 10
  data = [unw12, unw23, unw13]
 
@@ -136,25 +141,30 @@ def make_loop_png(unw12, unw23, unw13, loop_ph, png, titles4, cycle):
  figsize_x = 10
  figsize_y = int(figsize_x*length/width+1)
  if figsize_y < 3: figsize_y = 3
- 
+
  ### Plot
  fig = plt.figure(figsize = (figsize_x, figsize_y))
 
  ## 3 ifgs
  for i in range(3):
  data_wrapped = np.angle(np.exp(1j*(data[i]/cycle))*cycle)
  ax = fig.add_subplot(2, 2, i+1) #index start from 1
- ax.imshow(data_wrapped, vmin=-np.pi, vmax=+np.pi, cmap=cmap_wrap, interpolation='nearest')
+ im = ax.imshow(data_wrapped, vmin=-np.pi, vmax=+np.pi, cmap=cmap_wrap,
+ interpolation='nearest')
  ax.set_title('{}'.format(titles4[i]))
  ax.set_xticklabels([])
  ax.set_yticklabels([])
+ cax = plt.colorbar(im)
+ cax.set_ticks([])
 
  ## loop phase
  ax = fig.add_subplot(2, 2, 4) #index start from 1
- ax.imshow(loop_ph, vmin=-np.pi, vmax=+np.pi, cmap=SCM.vik, interpolation='nearest')
+ im = ax.imshow(loop_ph, vmin=-np.pi, vmax=+np.pi, cmap=cmap_loop,
+ interpolation='nearest')
  ax.set_title('{}'.format(titles4[3]))
  ax.set_xticklabels([])
  ax.set_yticklabels([])
+ cax = plt.colorbar(im)
 
  plt.tight_layout()
  plt.savefig(png)