Add limited caching for color distance

jquast · Jan 12, 2020 · 64a83e7 · 64a83e7
1 parent 031e707
commit 64a83e7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/blessed/color.py b/blessed/color.py
@@ -11,6 +11,11 @@
 
 from math import atan2, cos, exp, sin, sqrt
 
+try:
+    from functools import lru_cache
+except ImportError:
+    from backports.functools_lru_cache import lru_cache
+
 
 def rgb_to_xyz(red, green, blue):
     """
@@ -71,6 +76,7 @@ def xyz_to_lab(x_val, y_val, z_val):
     return cie_l, cie_a, cie_b
 
 
+@lru_cache(maxsize=256)
 def rgb_to_lab(red, green, blue):
     """
     Convert RGB color to CIE-Lab color.

diff --git a/requirements.txt b/requirements.txt
@@ -2,5 +2,7 @@ wcwidth>=0.1.4
 six>=1.9.0
 # support python2.6 by using backport of 'orderedict'
 ordereddict==1.1; python_version < "2.7"
+# support python2.7 by using backport of 'functools.lru_cache'
+backports.functools-lru-cache>=1.2.1; python_version < "3.2"
 # Windows requires jinxed
 jinxed>=0.5.4; platform_system == "Windows"