ENH: add rcParam for ConciseDate and interval_multiples

jklymak · jklymak · commit 0577839e63bc · 2020-04-03T16:33:19.000-07:00
diff --git a/lib/matplotlib/dates.py b/lib/matplotlib/dates.py
@@ -1784,8 +1784,11 @@ class DateConverter(units.ConversionInterface):
     The 'unit' tag for such data is None or a tzinfo instance.
     """
 
-    @staticmethod
-    def axisinfo(unit, axis):
+    def __init__(self, interval_multiples=True):
+        self._interval_multiples = interval_multiples
+        super().__init__()
+
+    def axisinfo(self, unit, axis):
         """
         Return the `~matplotlib.units.AxisInfo` for *unit*.
 
@@ -1794,7 +1797,8 @@ def axisinfo(unit, axis):
         """
         tz = unit
 
-        majloc = AutoDateLocator(tz=tz)
+        majloc = AutoDateLocator(tz=tz,
+                                 interval_multiples=self._interval_multiples)
         majfmt = AutoDateFormatter(majloc, tz=tz)
         datemin = datetime.date(2000, 1, 1)
         datemax = datetime.date(2010, 1, 1)
@@ -1836,17 +1840,19 @@ class ConciseDateConverter(DateConverter):
     # docstring inherited
 
     def __init__(self, formats=None, zero_formats=None, offset_formats=None,
-                 show_offset=True):
+                 show_offset=True, interval_multiples=True):
         self._formats = formats
         self._zero_formats = zero_formats
         self._offset_formats = offset_formats
         self._show_offset = show_offset
+        self._interval_multiples = interval_multiples
         super().__init__()
 
     def axisinfo(self, unit, axis):
         # docstring inherited
         tz = unit
-        majloc = AutoDateLocator(tz=tz)
+        majloc = AutoDateLocator(tz=tz,
+                                 interval_multiples=self._interval_multiples)
         majfmt = ConciseDateFormatter(majloc, tz=tz, formats=self._formats,
                                       zero_formats=self._zero_formats,
                                       offset_formats=self._offset_formats,
@@ -1857,6 +1863,25 @@ def axisinfo(self, unit, axis):
                               default_limits=(datemin, datemax))
 
 
-units.registry[np.datetime64] = DateConverter()
-units.registry[datetime.date] = DateConverter()
-units.registry[datetime.datetime] = DateConverter()
+
+def register_converters():
+    """
+    Helper to register the date converters.
+
+    Useful to call if the :rc:`date.converter` or :rc:`date.interval_multiples`
+    change.
+    """
+    if matplotlib.rcParams['date.converter'] == 'concise':
+        converter = ConciseDateConverter
+    else:
+        converter = DateConverter
+
+    interval_multiples = matplotlib.rcParams['date.interval_multiples']
+    units.registry[np.datetime64] = converter(
+            interval_multiples=interval_multiples)
+    units.registry[datetime.date] = converter(
+            interval_multiples=interval_multiples)
+    units.registry[datetime.datetime] = converter(
+            interval_multiples=interval_multiples)
+
+register_converters()
diff --git a/lib/matplotlib/rcsetup.py b/lib/matplotlib/rcsetup.py
@@ -1266,6 +1266,11 @@ def _convert_validator_spec(key, conv):
     'date.autoformatter.second': ['%H:%M:%S', validate_string],
     'date.autoformatter.microsecond': ['%M:%S.%f', validate_string],
 
+    # 'auto', 'concise', 'auto-noninterval'
+    'date.converter': ['auto', validate_string],
+    # for auto date locator, choose interval_multiples
+    'date.interval_multiples': [True, validate_bool],
+
     #legend properties
     'legend.fancybox': [True, validate_bool],
     'legend.loc': ['best',
diff --git a/matplotlibrc.template b/matplotlibrc.template
@@ -326,7 +326,7 @@
 #mathtext.sf:  sans
 #mathtext.tt:  monospace
 #mathtext.fallback: cm  # Select fallback font from ['cm' (Computer Modern), 'stix'
-                        # 'stixsans'] when a symbol can not be found in one of the 
+                        # 'stixsans'] when a symbol can not be found in one of the
                         # custom math fonts. Select 'None' to not perform fallback
                         # and replace the missing character by a dummy symbol.
 #mathtext.default: it  # The default font to use for math.
@@ -431,7 +431,10 @@
 #date.autoformatter.minute:      %d %H:%M
 #date.autoformatter.second:      %H:%M:%S
 #date.autoformatter.microsecond: %M:%S.%f
-
+## 'auto', 'concise':
+#date.converter:                 'auto'
+## For auto converter whether to use interval_multiples:
+#date.interval_multiples:         True
 
 ## ***************************************************************************
 ## * TICKS                                                                   *
