Merge pull request #22 from GinoMellino/develop

Add support for mosaic datasets on Pro

Author: DavidWhittingham

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_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))