function to determine the scene type has been implemented

scene.py

@@ -6,10 +6,10 @@ from glob import glob
 import read_data as rd
 
 # lsf: land sea flag
-_scene_list = ['ocean_day', 'ocean_night', 'land_day', 'land_night', 'snow_day', 'coast_day',
-               'desert_day', 'antarctic_day', 'polar_day_snow', 'polar_day_desert',
+_scene_list = ['ocean_day', 'ocean_night', 'land_day', 'land_night', 'snow_day', 'snow_night', 'coast_day',
+               'desert_day', 'antarctic_day', 'polar_day_snow', 'polar_day_desert', 'polar_day_ocean',
                'polar_day_desert_coast', 'polar_day_coast', 'polar_day_land', 'polar_night_snow',
-               'polar_night_land', 'polar_ocean_night']
+               'polar_night_land', 'polar_night_ocean', 'land_day_desert_coast']
 _flags = ['day', 'night', 'land', 'coast', 'sh_lake', 'sh_ocean', 'water', 'polar', 'sunglint',
           'greenland', 'high_elevation', 'antarctica', 'desert', 'visusd', 'vrused', 'map_snow', 'map_ice',
           'ndsi_snow', 'snow', 'ice', 'new_zealand']
@@ -109,7 +109,6 @@ def find_scene(data, sunglint_angle):
     # tmp[day == 1] = day
     # tmp[day == 0] = night
 
-    scene = {scn: np.zeros((dim1, dim2)) for scn in _scene_list}
     scene_flag = {flg: np.zeros((dim1, dim2)) for flg in _flags}
 
     scene_flag['day'][sza <= 85] = 1
@@ -307,4 +306,129 @@ def find_scene(data, sunglint_angle):
     return scene_flag
 
 
+def scene_id(scene_flag):
+
+    dim1, dim2 = scene_flag['day'].shape[0], scene_flag['day'].shape[1]
+    scene = {scn: np.zeros((dim1, dim2)) for scn in _scene_list}
 
+    # Ocean Day
+    idx = np.nonzero((scene_flag['water'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['ice'] == 0) & (scene_flag['snow'] == 0) &
+                     (scene_flag['polar'] == 0) & (scene_flag['antarctica'] == 0) &
+                     (scene_flag['coast'] == 0) & (scene_flag['desert'] == 0))
+    scene['ocean_day'][idx] = 1
+
+    # Ocean Night
+    idx = np.nonzero((scene_flag['water'] == 1) & (scene_flag['night'] == 1) &
+                     (scene_flag['ice'] == 0) & (scene_flag['snow'] == 0) &
+                     (scene_flag['polar'] == 0) & (scene_flag['antarctica'] == 0) &
+                     (scene_flag['coast'] == 0) & (scene_flag['desert'] == 0))
+    scene['ocean_night'][idx] = 1
+
+    # Land Day
+    idx = np.nonzero((scene_flag['land'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['ice'] == 0) & (scene_flag['snow'] == 0) &
+                     (scene_flag['polar'] == 0) & (scene_flag['antarctica'] == 0) &
+                     (scene_flag['coast'] == 0) & (scene_flag['desert'] == 0))
+    scene['land_day'][idx] = 1
+
+    # Land Night
+    idx = np.nonzero((scene_flag['land'] == 1) & (scene_flag['night'] == 1) &
+                     (scene_flag['ice'] == 0) & (scene_flag['snow'] == 0) &
+                     (scene_flag['polar'] == 0) & (scene_flag['antarctica'] == 0) &
+                     (scene_flag['coast'] == 0))
+    scene['land_night'][idx] = 1
+
+    # Snow Day
+    idx = np.nonzero((scene_flag['day'] == 1) &
+                     ((scene_flag['ice'] == 1) | (scene_flag['snow'] == 1)) &
+                     (scene_flag['polar']) & (scene_flag['antarctica']))
+    scene['snow_day'][idx] = 1
+
+    # Snow Night
+    idx = np.nonzero((scene_flag['night'] == 1) &
+                     ((scene_flag['ice'] == 1) | (scene_flag['snow'] == 1)) &
+                     (scene_flag['polar']) & (scene_flag['antarctica']))
+    scene['snow_night'][idx] = 1
+
+    # Land Day Coast
+    idx = np.nonzero((scene_flag['land'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['coast'] == 1) & (scene_flag['desert'] == 0) &
+                     (scene_flag['polar'] == 0) & (scene_flag['antarctica'] == 0))
+    scene['land_day_coast'][idx] = 1
+
+    # Land Day Desert
+    idx = np.nonzero((scene_flag['land'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['desert'] == 1) & (scene_flag['coast'] == 0) &
+                     (scene_flag['polar'] == 0) & (scene_flag['antarctica'] == 0))
+    scene['desert_day'][idx] = 1
+
+    # Land Day Desert Coast
+    idx = np.nonzero((scene_flag['land'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['desert'] == 1) & (scene_flag['coast'] == 1) &
+                     (scene_flag['polar'] == 0) & (scene_flag['antarctica'] == 0))
+    scene['land_day_desert_coast'][idx] = 1
+
+    # Antarctic Day
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['antarctica'] == 1) & (scene_flag['land'] == 1))
+    scene['antarctic_day'][idx] = 1
+
+    # Polar Day Snow
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['day'] == 1) &
+                     ((scene_flag['snow'] == 1) | (scene_flag['ice'] == 1)) &
+                     (scene_flag['antarctica'] == 0))
+    scene['polar_day_snow'][idx] = 1
+
+    # Polar Day Desert
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['land'] == 1) & (scene_flag['desert'] == 1) &
+                     (scene_flag['coast'] == 0) & (scene_flag['antarctica'] == 0) &
+                     (scene_flag['ice'] == 0) & (scene_flag['snow'] == 0))
+    scene['polar_day_desert'][idx] = 1
+
+    # Polar Day Desert Coast
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['land'] == 1) & (scene_flag['desert'] == 1) &
+                     (scene_flag['coast'] == 1) & (scene_flag['antarctica'] == 0) &
+                     (scene_flag['ice'] == 0) & (scene_flag['snow'] == 0))
+    scene['polar_day_desert_coast'][idx] = 1
+
+    # Polar Day Coast
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['land'] == 1) & (scene_flag['coast'] == 1)
+                     (scene_flag['desert'] == 0) & (scene_flag['antarctica'] == 0) &
+                     (scene_flag['ice'] == 0) & (scene_flag['snow'] == 0))
+    scene['polar_day_coast'][idx] = 1
+
+    # Polar Day Land
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['land'] == 1) & (scene_flag['coast'] == 0)
+                     (scene_flag['desert'] == 0) & (scene_flag['antarctica'] == 0) &
+                     (scene_flag['ice'] == 0) & (scene_flag['snow'] == 0))
+    scene['polar_day_land'][idx] = 1
+
+    # Polar Day Ocean
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['day'] == 1) &
+                     (scene_flag['water'] == 1) & (scene_flag['coast'] == 0)
+                     (scene_flag['desert'] == 0) & (scene_flag['antarctica'] == 0) &
+                     (scene_flag['ice'] == 0) & (scene_flag['snow'] == 0))
+    scene['polar_day_land'][idx] = 1
+
+    # Polar Night Snow
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['night'] == 1) &
+                     ((scene_flag['snow'] == 1) | (scene_flag['ice'] == 1)))
+    scene['polar_night_snow'][idx] = 1
+
+    # Polar Night Land
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['night'] == 1) &
+                     (scene_flag['land'] == 1))
+    scene['polar_night_land'][idx] = 1
+
+    # Polar Night Ocean
+    idx = np.nonzero((scene_flag['polar'] == 1) & (scene_flag['night'] == 1) &
+                     (scene_flag['water'] == 1))
+    scene['polar_night_ocean'][idx] = 1
+
+    return scene
+    scene['polar_day_ocean'][idx] = 1