Added support for mosiac datasets

Author: GinoMellino

arcpyext/mapping/_cim/factories.py

@@ -11,7 +11,8 @@
 
 import json
 
-from .layers import ProFeatureLayer, ProGroupLayer, ProRasterLayer
+from .layers import ProFeatureLayer, ProGroupLayer, ProRasterLayer, \
+    ProMosaicLayer, ProFeatureMosaicSubLayer, ProImageMosaicSubLayer
 
 
 def create_layer(proj_zip, layer_string):
@@ -30,6 +31,12 @@ def create_layer(proj_zip, layer_string):
             layer_obj = ProGroupLayer(proj_zip, layer_string)
         elif layer_type == "CIMRasterLayer":
             layer_obj = ProRasterLayer(proj_zip, layer_string)
+        elif layer_type == "CIMMosaicLayer":
+            layer_obj = ProMosaicLayer(proj_zip, layer_string)
+        elif layer_type == "CIMFeatureMosaicSubLayer":
+            layer_obj = ProFeatureMosaicSubLayer(proj_zip, layer_string)
+        elif layer_type == "CIMImageMosaicSubLayer":
+            layer_obj = ProImageMosaicSubLayer(proj_zip, layer_string)
     else:
         # layer is probably XML, fallback to that
         if layer_string.startswith("<CIMFeatureLayer"):
@@ -38,5 +45,11 @@ def create_layer(proj_zip, layer_string):
             layer_obj = ProGroupLayer(proj_zip, layer_string)
         elif layer_string.startswith("<CIMRasterLayer"):
             layer_obj = ProRasterLayer(proj_zip, layer_string)
+        elif layer_string.startswith("<CIMMosaicLayer"):
+            layer_obj = ProMosaicLayer(proj_zip, layer_string)
+        elif layer_string.startswith("<CIMFeatureMosaicSubLayer"):
+            layer_obj = ProFeatureMosaicSubLayer(proj_zip, layer_string)
+        elif layer_string.startswith("<CIMImageMosaicSubLayer"):
+            layer_obj = ProImageMosaicSubLayer(proj_zip, layer_string)
 
     return layer_obj

arcpyext/mapping/_cim/layers.py

@@ -19,7 +19,8 @@
 from .tables import ProFeatureTable
 
 # .NET Imports
-from ArcGIS.Core.CIM import CIMFeatureLayer, CIMGroupLayer, CIMRasterLayer
+from ArcGIS.Core.CIM import CIMFeatureLayer, CIMGroupLayer, CIMRasterLayer, \
+    CIMMosaicLayer, CIMImageMosaicSubLayer,CIMFeatureMosaicSubLayer
 
 
 class ProLayerBase(with_metaclass(ABCMeta, object)):
@@ -110,3 +111,33 @@ def __init__(self, proj_zip, layer_string):
 
     def _get_child_paths(self):
         return [cp[8:] for cp in self._cim_obj.Layers]
+
+
+class ProMosaicLayer(ProLayerBase):
+    def __init__(self, proj_zip, layer_string):
+        try:
+            super().__init__(proj_zip, CIMMosaicLayer.FromXml(layer_string))
+        except AttributeError:
+            # probably JSON, attempt that
+            super().__init__(proj_zip, CIMMosaicLayer.FromJson(layer_string))
+
+    def _get_mosaic_child_paths(self):
+        return [self._cim_obj.BoundaryLayer[8:],self._cim_obj.FootprintLayer[8:],self._cim_obj.ImageLayer[8:]]
+
+
+class ProFeatureMosaicSubLayer(ProLayerBase):
+    def __init__(self, proj_zip, layer_string):
+        try:
+            super().__init__(proj_zip, CIMFeatureMosaicSubLayer.FromXml(layer_string))
+        except AttributeError:
+            # probably JSON, attempt that
+            super().__init__(proj_zip, CIMFeatureMosaicSubLayer.FromJson(layer_string))
+
+
+class ProImageMosaicSubLayer(ProLayerBase):
+    def __init__(self, proj_zip, layer_string):
+        try:
+            super().__init__(proj_zip, CIMImageMosaicSubLayer.FromXml(layer_string))
+        except AttributeError:
+            # probably JSON, attempt that
+            super().__init__(proj_zip, CIMImageMosaicSubLayer.FromJson(layer_string))

arcpyext/mapping/_cim/pro_map.py

@@ -10,6 +10,7 @@
 # pylint: enable=wildcard-import,unused-wildcard-import,wrong-import-order,wrong-import-position,import-error,no-name-in-module
 
 # Third-party imports
+import json
 import arcpy
 
 # Local imports
@@ -88,12 +89,26 @@ def _create_layers(self, layer_path):
         layer_obj = create_layer(self._proj_zip, layer_string)
         self._layers.append(layer_obj)
 
-        if layer_obj:
-            for child_path in layer_obj._get_child_paths():
-                child_layer = self._create_layers(child_path)
-
-                # build parent/child relationships
-                child_layer._parent = layer_obj
-                layer_obj._children.append(child_layer)
+        if layer_obj:            
+            
+            # Mosaic imagery datasets have different internal structures and need
+            # to be handled differently from other layers. 
+            layer_json = json.loads(layer_string)
+            if layer_json["type"] == "CIMMosaicLayer":
+                mosaic_children_paths = layer_obj._get_mosaic_child_paths()
+                
+                for mosaic_child_path in mosaic_children_paths:
+                    mosaic_child_layer = self._create_layers(mosaic_child_path)
+
+                    # build parent/child relationships
+                    mosaic_child_layer._parent = layer_obj
+                    layer_obj._children.append(mosaic_child_layer)
+            else:
+                for child_path in layer_obj._get_child_paths():
+                    child_layer = self._create_layers(child_path)
+
+                    # build parent/child relationships
+                    child_layer._parent = layer_obj
+                    layer_obj._children.append(child_layer)
 
         return layer_obj