matplotlib
diff --git a/‎lib/matplotlib/projections/__init__.py
Copy file name to clipboard
+39Lines changed: 39 additions & 0 deletions b/‎lib/matplotlib/projections/__init__.py
Copy file name to clipboard
+39Lines changed: 39 additions & 0 deletions
@@ -0,0 +1,39 @@
+from polar import PolarAxes
+from matplotlib import axes
+
+class ProjectionRegistry(object):
+    def __init__(self):
+        self._all_projection_types = {}
+
+    def register(self, *projections):
+        for projection in projections:
+            name = projection.name
+            self._all_projection_types[name] = projection
+
+    def get_projection_class(self, name):
+        return self._all_projection_types[name]
+
+    def get_projection_names(self):
+        names = self._all_projection_types.keys()
+        names.sort()
+        return names
+projection_registry = ProjectionRegistry()
+
+projection_registry.register(
+    axes.Axes,
+    PolarAxes)
+
+def get_projection_class(projection):
+    if projection is None:
+        projection = 'rectilinear'
+
+    try:
+        return projection_registry.get_projection_class(projection)
+    except KeyError:
+        raise ValueError("Unknown projection '%s'" % projection)
+
+def projection_factory(projection, figure, rect, **kwargs):
+    return get_projection_class(projection)(figure, rect, **kwargs)
+
+def get_projection_names():
+    return projection_registry.get_projection_names()