Unique API for single/multi-patch Domain objects

sympde/topology/domain.py

@@ -6,7 +6,7 @@
 import yaml
 import os
 
-from collections import abc
+from collections import abc, OrderedDict
 from typing import Union as TypeUnion, Optional, List, Dict, Iterable, TYPE_CHECKING
 # Union clashes with core.basic.Union
 
@@ -180,6 +180,20 @@ def mapping(self) -> Optional[Mapping]:
         """The mapping that maps the logical domain to the physical domain"""
         return self.args[3]
 
+    @property
+    def subdomains(self) -> tuple:
+        """returns subdomains as tuple of Domains"""
+        if isinstance( self.interior, iterable_types):
+            subs = self.interior
+        else:
+            subs = [self.interior]
+        return tuple(subs)
+
+    @property
+    def mappings(self) -> OrderedDict:
+        return OrderedDict([(P.logical_domain, P.mapping)
+                           for P in self.subdomains])
+
     @property
     def logical_domain(self) -> Domain:
         """The domain is the image of the logical_domain under the mapping"""
@@ -481,6 +495,11 @@ def join(cls, patches, connectivity, name):
         assert isinstance(connectivity, (tuple, list))
         assert isinstance(name, str)
 
+        if len(patches) == 1:
+            # single patch domain: return the patch
+            assert len(connectivity) == 0
+            return patches[0]
+
         assert all(p.dim==patches[0].dim for p in patches)
         dim = int(patches[0].dim)