Add registered classes to namespace, validate cmap/color names

Author: lukelbd

proplot/__init__.py

@@ -1,17 +1,21 @@
 #!/usr/bin/env python3
 """
-A python package for making beautiful, publication-quality graphics.
+A comprehensive matplotlib wrapper for making beautiful, publication-quality graphics.
 """
-# Import everything to top-level
-# NOTE: In future will enable submodule access and stop importing classes to top-level
-import pkg_resources as _pkg
-
-from .config import *  # noqa: F401 F403
-from .internals import timers as _timers
+# SCM versioning
+import pkg_resources as pkg
+name = 'proplot'
+try:
+    version = __version__ = pkg.get_distribution(__name__).version
+except pkg.DistributionNotFound:
+    version = __version__ = 'unknown'
 
-with _timers._benchmark('imports'):
-    from .utils import *  # noqa: F401 F403
+# Import everything to top level
+from .internals import benchmarks, docstring, rcsetup, warnings  # noqa: F401
+with benchmarks._benchmark('imports'):
+    from .config import *  # noqa: F401 F403
     from .crs import *  # noqa: F401 F403
+    from .utils import *  # noqa: F401 F403
     from .colors import *  # noqa: F401 F403
     from .ticker import *  # noqa: F401 F403
     from .scale import *  # noqa: F401 F403
@@ -22,9 +26,46 @@
     from .ui import *  # noqa: F401 F403
     from .demos import *  # noqa: F401 F403
 
-# SCM versioning
-name = 'proplot'
-try:
-    version = __version__ = _pkg.get_distribution(__name__).version
-except _pkg.DistributionNotFound:
-    version = __version__ = 'unknown'
+# Dynamically add registered classes to top-level namespace
+from .constructor import NORMS, LOCATORS, FORMATTERS, SCALES, PROJS
+_globals = globals()
+for _src in (NORMS, LOCATORS, FORMATTERS, SCALES, PROJS):
+    for _key, _cls in _src.items():
+        if isinstance(_cls, type):  # i.e. not a scale preset
+            _globals[_cls.__name__] = _cls  # may overwrite proplot names
+
+# Register objects
+from .config import register_cmaps, register_cycles, register_colors, register_fonts
+with benchmarks._benchmark('cmaps'):
+    register_cmaps(default=True)
+with benchmarks._benchmark('cycles'):
+    register_cycles(default=True)
+with benchmarks._benchmark('colors'):
+    register_colors(default=True)
+with benchmarks._benchmark('fonts'):
+    register_fonts(default=True)
+
+# Validate colormap names and propagate 'cycle' to 'axes.prop_cycle'
+# NOTE: cmap.sequential also updates siblings 'cmap' and 'image.cmap'
+from .config import rc
+rcsetup.VALIDATE_REGISTERED_CMAPS = True
+for _key in ('cycle', 'cmap.sequential', 'cmap.diverging', 'cmap.cyclic', 'cmap.qualitative'):  # noqa: E501
+    try:
+        rc[_key] = rc[_key]
+    except ValueError as err:
+        warnings._warn_proplot(f'Invalid user rc file setting: {err}')
+        rc[_key] = 'Fire'  # fill value
+
+# Validate color names now that colors are registered
+from .config import rc_proplot, rc_matplotlib
+rcsetup.VALIDATE_REGISTERED_COLORS = True
+for _src in (rc_proplot, rc_matplotlib):
+    for _key in _src:  # loop through unsynced properties
+        if 'color' not in _key:
+            continue
+        # Likely has a color validator or derivative thereof; if not, harmless
+        try:
+            _src[_key] = _src[_key]
+        except ValueError as err:
+            warnings._warn_proplot(f'Invalid user rc file setting: {err}')
+            _src[_key] = 'black'  # fill value