summaryrefslogtreecommitdiff
path: root/color
diff options
context:
space:
mode:
Diffstat (limited to 'color')
-rw-r--r--color/__init__.py1
-rw-r--r--color/chromaticity.py5
2 files changed, 5 insertions, 1 deletions
diff --git a/color/__init__.py b/color/__init__.py
new file mode 100644
index 0000000..c5fda6a
--- /dev/null
+++ b/color/__init__.py
@@ -0,0 +1 @@
+from chromaticity import map_wavelength
diff --git a/color/chromaticity.py b/color/chromaticity.py
index 6229b51..fbd812a 100644
--- a/color/chromaticity.py
+++ b/color/chromaticity.py
@@ -1,6 +1,9 @@
import numpy as np
+import os
-f = open('sbrgb10w.csv')
+dir = os.path.split(os.path.realpath(__file__))[0]
+
+f = open(dir + '/sbrgb10w.csv')
color_map = []
for line in f: