From 4bf95f452e3275c12026b16b51dc646846598f19 Mon Sep 17 00:00:00 2001
From: Anthony LaTorre <tlatorre9@gmail.com>
Date: Wed, 3 Aug 2011 14:32:49 -0400
Subject: add a GPU class to handle both the gpu context and module; since the
 geometry requires global device pointers, there should be a one to one
 correspondence between modules and contexts. the current plan is to perform
 all gpu operations within this class. also add a simple color map to display
 hit pmt charge and timing information.

---
 color/__init__.py |  1 +
 color/colormap.py | 27 +++++++++++++++++++++++++++
 2 files changed, 28 insertions(+)
 create mode 100644 color/colormap.py

(limited to 'color')

diff --git a/color/__init__.py b/color/__init__.py
index c5fda6a..a8e7550 100644
--- a/color/__init__.py
+++ b/color/__init__.py
@@ -1 +1,2 @@
 from chromaticity import map_wavelength
+from colormap import map_to_color
diff --git a/color/colormap.py b/color/colormap.py
new file mode 100644
index 0000000..b3761ae
--- /dev/null
+++ b/color/colormap.py
@@ -0,0 +1,27 @@
+import numpy as np
+
+import matplotlib.pyplot as plt
+
+def map_to_color(a, range=None):
+    a = np.asarray(a)
+    if range is None:
+        range = (a.min(), a.max())
+
+    x = np.linspace(0, np.pi, 100)
+
+    yr = np.cos(x)**2
+    yr[x > np.pi/2] = 0.0
+    yg = np.sin(x)**2
+    yb = np.cos(x)**2
+    yb[x < np.pi/2] = 0.0
+
+    #plt.plot(x, yr, 'r-', x, yb, 'b-', x, yg, 'g-')
+    #plt.show()
+
+    ax = np.pi*(a - range[0])/(range[1]-range[0])
+
+    r = (np.interp(ax, x, yr)*255).astype(np.uint32)
+    g = (np.interp(ax, x, yg)*255).astype(np.uint32)
+    b = (np.interp(ax, x, yb)*255).astype(np.uint32)
+
+    return r << 16 | g << 8 | b
-- 
cgit