Homec4science
Diffusion Nanopore Tools (master) Browse

Nanopore-tools/SaveAllEvents.py
2bb51f0f8ceemaster

Last Change
Raw File

SaveAllEvents.py
View Options

## Cross-correlation of two channels
import AnalysisParameters as pm
import numpy as np
import scipy
import scipy.signal as sig
import Functions as fu
import os
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages
from matplotlib.font_manager import FontProperties
import pandas as pd
import MiscParameters as pm
from matplotlib.ticker import EngFormatter
import h5py
from tkinter.filedialog import askopenfilenames
from matplotlib import rc
rc('mathtext', default='regular')
pm.init(LoadFiles = 0)
baseline = 250
fitOn = 0
percentage = 1
plotPoints=0
filenames = {'/Users/migraf/Desktop/Chan/Data/R43_KCl gradient_100mMcis_1mM trans_4_20180405_182358_OriginalDB.hdf5'}
#filenames = askopenfilenames()
for filename in filenames:
print(filename)
f = h5py.File(filename, 'r')
out = fu.OpenFile(f['General/FileName'].value)
file = os.sep + str(os.path.split(filename)[1][:-5])
# Graphene Or Not??
if not out['graphene']:
## Only need to plot i1 and i1_Up and v1
#toPlot = ['i1', 'i1_Up']
toPlot = ['i1']
for k in toPlot:
NumberOfEvents=f['LowPassSegmentation/' + k + '/NumberOfEvents'].value
if NumberOfEvents is not 0:
pp = PdfPages(pm.OutputFolder + file + '_Only_' + k + '_EventPlots.pdf')
fig = plt.figure()
ax1 = fig.add_subplot(211)
ax2 = fig.add_subplot(212, sharex=ax1)
ax1.set_title('Ionic Current Trace "' + k + '"')
ax2.set_title('Ionic Voltage')
ax1.set_ylabel('Current [nA]')
ax2.set_ylabel('Voltage [V]')
ax1.set_xlabel('Time [s]')
ax2.set_xlabel('Time [s]')
ax1.yaxis.set_major_formatter(EngFormatter(unit='A', places=1))
ax2.yaxis.set_major_formatter(EngFormatter(unit='V', places=1))
ax1.xaxis.set_major_formatter(EngFormatter(unit='s', places=1))
ax2.xaxis.set_major_formatter(EngFormatter(unit='s', places=1))
line1, = ax1.plot([])
line2, = ax2.plot([], 'c')
if fitOn:
line1f, = ax1.plot([], 'r')
for i in np.linspace(0, NumberOfEvents-1, NumberOfEvents*percentage, dtype=np.uint64):
startp_i1 = np.int64(
f['LowPassSegmentation/' + k + '/' + 'StartPoints'].value[i] - baseline)
endp_i1 = np.int64(
f['LowPassSegmentation/' + k + '/' + 'EndPoints'].value[i] + baseline)
if startp_i1 < 0:
startp_i1 = 0
if endp_i1 > len(out['i1']):
endp_i1 = len(out['i1'])
if fitOn:
if plotPoints:
fit1 = np.concatenate([np.ones(baseline) *
f['LowPassSegmentation/' + k + '/LocalBaseline'].value[i],
f['LowPassSegmentation/' + k + '/AllEvents/' + '{:09d}'.format(i)].value[i],
np.ones(baseline) *
f['LowPassSegmentation/' + k + '/LocalBaseline'].value[i]])
else:
fit1 = np.concatenate([np.ones(baseline) *
f['LowPassSegmentation/' + k + '/LocalBaseline'].value[i],
np.ones(endp_i1 - startp_i1 - 2 * baseline) *
f['LowPassSegmentation/' + k + '/FitLevel'].value[i],
np.ones(baseline) *
f['LowPassSegmentation/' + k + '/LocalBaseline'].value[i]])
line1f.set_data(np.arange(startp_i1, endp_i1)/out['samplerate'], fit1)
line1.set_data(np.arange(startp_i1, endp_i1)/out['samplerate'], out['i1'][startp_i1:endp_i1])
line2.set_data(np.arange(startp_i1, endp_i1)/out['samplerate'], out['v1'][startp_i1:endp_i1])
ax1.set_title('{} Event {}\n Ionic Current Trace'.format(filter, i))
ax1.relim()
ax2.relim()
ax1.autoscale_view(True, True, True)
ax2.autoscale_view(True, True, True)
fig.canvas.draw()
pp.savefig(fig)
print('{}: {} out of {} saved!'.format(toPlot, str(i), NumberOfEvents-1))
pp.close()
else:
toPlot = []
#Initialize all the cases
toPlot.append(['i1', '', np.uint64(f['LowPassSegmentation/i1/OnlyIndexes'].value), np.uint64(f['LowPassSegmentation/i1/OnlyIndexes'].value)])
toPlot.append(['i1_Up', '', np.uint64(f['LowPassSegmentation/i1_Up/OnlyIndexes'].value), np.uint64(f['LowPassSegmentation/i1_Up/OnlyIndexes'].value)])
toPlot.append(['', 'i2', np.uint64(f['LowPassSegmentation/i2/OnlyIndexes'].value), np.uint64(f['LowPassSegmentation/i2/OnlyIndexes'].value)])
toPlot.append(['', 'i2_Up', np.uint64(f['LowPassSegmentation/i2_Up/OnlyIndexes'].value), np.uint64(f['LowPassSegmentation/i2_Up/OnlyIndexes'].value)])
toPlot.append(['i1', 'i2', np.uint64(f['LowPassSegmentation/i1/CommonIndexesWithi2'].value), np.uint64(f['LowPassSegmentation/i2/CommonIndexesWithi1'].value)])
toPlot.append(['i1_Up', 'i2', np.uint64(f['LowPassSegmentation/i1_Up/CommonIndexesWithi2'].value), np.uint64(f['LowPassSegmentation/i2/CommonIndexesWithi1_Up'].value)])
toPlot.append(['i1_Up', 'i2_Up', np.uint64(f['LowPassSegmentation/i1_Up/CommonIndexesWithi2_Up'].value), np.uint64(f['LowPassSegmentation/i2_Up/CommonIndexesWithi1_Up'].value)])
toPlot.append(['i1', 'i2_Up', np.uint64(f['LowPassSegmentation/i1/CommonIndexesWithi2_Up'].value), np.uint64(f['LowPassSegmentation/i2_Up/CommonIndexesWithi1'].value)])
fig = plt.figure()
ax1 = fig.add_subplot(311)
ax2 = fig.add_subplot(312, sharex=ax1)
ax3 = fig.add_subplot(313, sharex=ax1)
ax4 = ax3.twinx()
ax4.set_title('Voltage Trace')
ax1.set_ylabel('Current [nA]')
ax2.set_ylabel('Current [nA]')
ax3.set_ylabel('Ionic Voltage [V]')
ax3.set_xlabel('Time [s]')
ax1.set_xlabel('Time [s]')
ax2.set_xlabel('Time [s]')
ax1.yaxis.set_major_formatter(EngFormatter(unit='A', places=1))
ax2.yaxis.set_major_formatter(EngFormatter(unit='A', places=1))
ax3.yaxis.set_major_formatter(EngFormatter(unit='V', places=1))
ax4.yaxis.set_major_formatter(EngFormatter(unit='V', places=1))
ax3.xaxis.set_major_formatter(EngFormatter(unit='s', places=1))
ax1.xaxis.set_major_formatter(EngFormatter(unit='s', places=1))
ax2.xaxis.set_major_formatter(EngFormatter(unit='s', places=1))
ax4.set_ylabel('Tr. Voltage [V]')
line1, = ax1.plot([])
line2, = ax2.plot([])
line3, = ax3.plot([], 'c')
line4, = ax4.plot([], 'y')
ax4.tick_params(axis='y', colors='y')
ax4.yaxis.label.set_color('y')
ax3.tick_params(axis='y', colors='c')
ax3.yaxis.label.set_color('c')
if fitOn:
line1f, = ax1.plot([], 'r')
line2f, = ax2.plot([], 'r')
for cases in toPlot:
ax2.set_title('Transverse Current Trace "' + cases[1] + '"')
pp = PdfPages(pm.OutputFolder + file + '_' + cases[0] + '_vs_' + cases[1] + '_EventPlots.pdf')
NumberOfEvents = len(cases[2])
for i in np.linspace(0, NumberOfEvents-1, NumberOfEvents*percentage, dtype=np.uint64):
if not cases[0] == '':
startp_i1 = np.int64(
f['LowPassSegmentation/' + cases[0] + '/' + 'StartPoints'].value[cases[2][i]] - baseline)
endp_i1 = np.int64(
f['LowPassSegmentation/' + cases[0] + '/' + 'EndPoints'].value[cases[2][i]] + baseline)
else:
startp_i1 = np.int64(
f['LowPassSegmentation/' + cases[1] + '/' + 'StartPoints'].value[cases[3][i]] - baseline)
endp_i1 = np.int64(
f['LowPassSegmentation/' + cases[1] + '/' + 'EndPoints'].value[cases[3][i]] + baseline)
if not cases[1] == '':
startp_i2 = np.int64(
f['LowPassSegmentation/' + cases[1] + '/' + 'StartPoints'].value[cases[3][i]] - baseline)
endp_i2 = np.int64(
f['LowPassSegmentation/' + cases[1] + '/' + 'EndPoints'].value[cases[3][i]] + baseline)
else:
startp_i2 = np.int64(
f['LowPassSegmentation/' + cases[0] + '/' + 'StartPoints'].value[cases[2][i]] - baseline)
endp_i2 = np.int64(
f['LowPassSegmentation/' + cases[0] + '/' + 'EndPoints'].value[cases[2][i]] + baseline)
if fitOn:
if not cases[0] == '':
if plotPoints:
fit1 = np.concatenate([np.ones(baseline) *
f['LowPassSegmentation/' + cases[0] + '/LocalBaseline'].value[cases[2][i]],
f['LowPassSegmentation/' + cases[0] + '/AllEvents/' + '{:09d}'.format(cases[2][i])].value,
np.ones(baseline) *
f['LowPassSegmentation/' + cases[0] + '/LocalBaseline'].value[cases[2][i]]])
else:
fit1 = np.concatenate([np.ones(baseline) *
f['LowPassSegmentation/' + cases[0] + '/LocalBaseline'].value[cases[2][i]],
np.ones(endp_i1 - startp_i1 - 2 * baseline) * f[
'LowPassSegmentation/' + cases[0] + '/FitLevel'].value[cases[2][i]],
np.ones(baseline) *
f['LowPassSegmentation/' + cases[0] + '/LocalBaseline'].value[cases[2][i]]])
line1f.set_data(np.arange(startp_i1, endp_i1) / out['samplerate'], fit1)
else:
line1f.set_data([], [])
if not cases[1] == '':
if plotPoints:
fit2 = np.concatenate([np.ones(baseline) *
f['LowPassSegmentation/' + cases[1] + '/LocalBaseline'].value[cases[3][i]],
f['LowPassSegmentation/' + cases[1] + '/AllEvents/' + '{:09d}'.format(
cases[3][i])].value,
np.ones(baseline) *
f['LowPassSegmentation/' + cases[1] + '/LocalBaseline'].value[cases[3][i]]])
else:
fit2 = np.concatenate([np.ones(baseline) *
f['LowPassSegmentation/' + cases[1] + '/LocalBaseline'].value[cases[3][i]],
np.ones(endp_i2 - startp_i2 - 2 * baseline) * f['LowPassSegmentation/' + cases[1] + '/FitLevel'].value[cases[3][i]],
np.ones(baseline) *
f['LowPassSegmentation/' + cases[1] + '/LocalBaseline'].value[cases[3][i]]])
line2f.set_data(np.arange(startp_i2, endp_i2) / out['samplerate'], fit2)
else:
line2f.set_data([], [])
if startp_i1 < 0:
startp_i1 = 0
if startp_i2 < 0:
startp_i2 = 0
if endp_i1 > len(out['i1']):
endp_i1 = len(out['i1'])
if endp_i2 > len(out['i2']):
endp_i2 = len(out['i2'])
i1 = out['i1'][startp_i1:endp_i1]
v1 = out['v1'][startp_i1:endp_i1]
t1 = np.arange(len(i1)) / out['samplerate']
i2 = out['i2'][startp_i2:endp_i2]
v2 = out['v2'][startp_i1:endp_i1]
t2 = np.arange(len(i2)) / out['samplerate']
line1.set_data(t1, i1)
line3.set_data(t1, v1)
line4.set_data(t1, v2)
line2.set_data(t2, i2)
ax1.set_title('Event {} of {}\n Ionic Current Trace "{}"'.format(i, NumberOfEvents, cases[0]))
ax1.relim()
ax2.relim()
ax3.relim()
ax4.relim()
ax1.autoscale_view(True, True, True)
ax2.autoscale_view(True, True, True)
# voltage plots
ax3.autoscale_view(True, True, False)
ax4.autoscale_view(True, True, False)
ax3.set_ylim(np.min(v1) - np.mean(v1) * 5 / 10, np.max(v1) + np.mean(v1) * 1 / 10)
ax4.set_ylim([np.min(v2) - np.mean(v2) * 2 / 10, np.max(v2) + np.mean(v2) * 5 / 10])
fig.tight_layout()
fig.canvas.draw()
pp.savefig(fig)
print('{} out of {} saved!'.format(str(i), NumberOfEvents - 1))
pp.close()
plt.close()
c4science · Help