Index: /fact/tools/pyscripts/sandbox/dneise/gain/test_gain.py
===================================================================
--- /fact/tools/pyscripts/sandbox/dneise/gain/test_gain.py	(revision 13440)
+++ /fact/tools/pyscripts/sandbox/dneise/gain/test_gain.py	(revision 13440)
@@ -0,0 +1,97 @@
+#!/usr/bin/python -tti
+#
+# 
+
+from pyfact   import RawData
+from drs_spikes import DRSSpikes        
+from fir_filter import CFD
+from extractor import ZeroXing
+from fir_filter import SlidingAverage
+
+from plotters import Plotter
+
+import sys
+import cPickle
+import matplotlib.pyplot as plt
+import numpy as np
+
+data_filename = 'data/20111017_009.fits.gz
+calib_filename = 'data/20111017_006.drs.fits.gz'
+out_filename = 'test.pkl'
+
+run = RawData(data_filename, calib_filename, return_dict = True, do_calibration=True)
+despike = DRSSpikes()
+smooth = SlidingAverage(7)
+cfd = CFD()
+find = ZeroXing()
+
+plt.ion()
+
+peak_list = []
+        
+#p = Plotter('data[0]')
+
+
+#fig = plt.figure()
+#plt.grid(True)
+
+stupid_list = []
+thr = 3
+
+fig = plt.figure()
+fig.hold(True)
+
+for event in run:
+    print event['event_id'].value
+    data = event['acal_data']
+    origorig = data.copy()
+    data = despike(data)
+    data_orig = data.copy()
+    data = smooth(data)
+    filtered = cfd(data)
+    filtered = smooth(filtered)
+
+    for dat, fil, orig, zzz in zip(data, filtered, data_orig, origorig):
+        plt.cla()
+        prod = fil[:-1] * fil[1:]
+        cand = np.where( prod <= 0)[0]
+        # zero crossing with rising edge
+        cross = cand[np.where(fil[cand] < 0)[0]]
+        
+        over_thr = cross[np.where(dat[cross-4] > thr)[0]]
+
+        maxima = np.zeros( over_thr.shape )
+        maxima_pos =np.zeros( over_thr.shape )
+        omaxima = np.zeros( over_thr.shape )
+        omaxima_pos =np.zeros( over_thr.shape )
+        
+        for i in range(len(over_thr)):
+            if len(dat[over_thr[i]-12:over_thr[i]]) > 0:
+                maxima_pos[i] = over_thr[i]-12 + np.argmax(dat[over_thr[i]-12:over_thr[i]])
+                maxima[i]     = dat[maxima_pos[i]]
+                omaxima_pos[i] = over_thr[i]-12 + np.argmax(orig[over_thr[i]-12:over_thr[i]])
+                omaxima[i]     = orig[maxima_pos[i]-3]
+
+        x_values = np.array(range(len(dat)))
+
+        plt.plot( x_values-3, dat , 'k:')
+        plt.plot( x_values, orig , 'y:')
+        plt.plot( x_values, zzz , 'r.')
+        
+        plt.plot( maxima_pos-3, maxima, 'bo')
+        plt.plot( maxima_pos-3, omaxima, 'yo')
+        plt.plot( over_thr-3, dat[over_thr], 'ro')
+            
+        
+
+        ret = raw_input('quit?')
+        if ret=='q':
+            break
+
+#    peak_list.append( good_peaks )
+
+
+#output = open(out_filename, 'wb')
+#cPickle.dump(data_filename, output, -1)
+#cPickle.dump(calib_filename, output, -1)
+#cPickle.dump(peak_list, output, -1)