Added some cmap functions.

Selected species-specific  colors.
Quite some progress on fig_invariance_thresh_lp_species.pdf.

python/save_inv_data_log-hp.py

@@ -1,19 +1,19 @@
-import glob
 import numpy as np
 from thunderhopper.modeltools import load_data, save_data
-from thunderhopper.filetools import crop_paths
+from thunderhopper.filetools import search_files, crop_paths
 from thunderhopper.filters import decibel, sosfilter
 from IPython import embed
 
 # GENERAL SETTINGS:
-target = 'Omocestus_rufipes'
-data_paths = glob.glob(f'../data/processed/{target}*.npz')
+target = ['Omocestus_rufipes', '*'][0]
+data_paths = search_files(target, excl='noise', dir='../data/processed/')
 save_path = '../data/inv/log_hp/'
 
 # ANALYSIS SETTINGS:
 add_noise = False
+save_snippets = target == 'Omocestus_rufipes'
 example_scales = np.array([0.1, 1, 10, 30, 100, 300])
-scales = np.geomspace(0.1, 10000, 1000)
+scales = np.geomspace(0.1, 10000, 500)
 scales = np.unique(np.concatenate((scales, example_scales)))
 
 # EXECUTION:
@@ -60,13 +60,16 @@ for data_path, name in zip(data_paths, crop_paths(data_paths)):
         data = dict(
             scales=scales,
             example_scales=example_scales,
-            snip_env=mix[:, save_inds],
-            snip_log=mix_log[:, save_inds],
-            snip_inv=mix_inv[:, save_inds],
             measure_env=measure_env,
             measure_log=measure_log,
             measure_inv=measure_inv,
             )
+        if save_snippets:
+            data.update(
+                snip_env=mix[:, save_inds],
+                snip_log=mix_log[:, save_inds],
+                snip_inv=mix_inv[:, save_inds],
+            )
         file_name = save_path + name
         if add_noise:
             file_name += '_noise'