def __init__(self, wavetype):
QThread.__init__(self)
self.wavetype = wavetype
self.wave = Wave(self.wavetype)
self.enable = True
class WaveGen(QThread):
def __init__(self, wavetype):
QThread.__init__(self)
self.wavetype = wavetype
self.wave = Wave(self.wavetype)
self.enable = True
def run(self):
self.rampeddown = False
self.cleared = False
self.making = False
# Compute the voltage time series associated with the wave
self.wave.gen_ts_volts()
# Get parameters from the wave object
self.period = self.wave.period
self.height = self.wave.height
self.buffsize = self.wave.sbuffsize
self.sr = self.wave.sr
# Get data to write from the wave object
self.ts_plot = self.wave.ts_elev
self.dataw = self.wave.ts_volts
# If random waves, divide up time series into 120 256 sample parts
if self.wavetype != "Regular":
tsparts = np.reshape(self.dataw, (120, 256))
self.dataw = tsparts[0, :]
# Compute spectrum for plot
if self.wavetype == "Regular":
self.outf, self.outspec = self.wave.comp_spec()
else:
self.outf, self.outspec = self.wave.f, self.wave.spec
# Ramp time series
rampup_ts = ramp_ts(self.dataw, "up")
# Set making variable true
self.making = True
self.AOtaskHandle = daqmx.TaskHandle()
daqmx.CreateTask("", self.AOtaskHandle)
daqmx.CreateAOVoltageChan(self.AOtaskHandle, "Dev1/ao0", "",
-10.0, 10.0, daqmx.Val_Volts, None)
daqmx.CfgSampClkTiming(self.AOtaskHandle, "", self.sr,
daqmx.Val_Rising, daqmx.Val_ContSamps,
self.buffsize)
daqmx.SetWriteRegenMode(self.AOtaskHandle,
daqmx.Val_DoNotAllowRegen)
# Setup a callback function to run once the DAQmx driver finishes
def DoneCallback_py(taskHandle, status, callbackData_ptr):
self.rampeddown = True
return 0
DoneCallback = daqmx.DoneEventCallbackPtr(DoneCallback_py)
daqmx.RegisterDoneEvent(self.AOtaskHandle, 0, DoneCallback, None)
# Output the rampup time series
daqmx.WriteAnalogF64(self.AOtaskHandle, self.buffsize, False, 10.0,
daqmx.Val_GroupByChannel, rampup_ts)
daqmx.StartTask(self.AOtaskHandle)
# Wait a second to allow the DAQmx buffer to empty
if self.wavetype == "Regular":
time.sleep(self.period*0.99) # was self.period*0.4
else:
time.sleep(0.99)
# Set iteration variable to keep track of how many chunks of data
# have been written
i = 1
# Main running loop that writes data to DAQmx buffer
while self.enable:
writeSpaceAvail = daqmx.GetWriteSpaceAvail(self.AOtaskHandle)
if writeSpaceAvail >= self.buffsize:
if self.wavetype != "Regular":
if i >= 120:
self.dataw = tsparts[i % 120, :]
else:
self.dataw = tsparts[i, :]
daqmx.WriteAnalogF64(self.AOtaskHandle, self.buffsize, False,
10.0, daqmx.Val_GroupByChannel, self.dataw)
i += 1
if self.wavetype == "Regular":
time.sleep(0.99*self.period)
else:
time.sleep(0.99) # Was self.period
# After disabled, initiate rampdown time series
if self.wavetype != "Regular":
if i >= 120:
self.rampdown_ts = ramp_ts(tsparts[i % 120, :], "down")
else:
self.rampdown_ts = ramp_ts(tsparts[i, :], "down")
else:
self.rampdown_ts = ramp_ts(self.dataw, "down")
# Write rampdown time series
daqmx.WriteAnalogF64(self.AOtaskHandle, self.buffsize, False, 10.0,
daqmx.Val_GroupByChannel, self.rampdown_ts)
# Keep running, part of PyDAQmx callback syntax
while True:
pass
def stop(self):
self.stopgen = WaveStop(self)
self.stopgen.start()
def __init__(self, wavetype):
QThread.__init__(self)
self.wavetype = wavetype
self.wave = Wave(self.wavetype)
self.enable = True
# -*- coding: utf-8 -*-
"""
Created on Wed Jul 03 22:02:15 2013
@author: Pete
"""
import numpy as np
import matplotlib.pyplot as plt
from wavetsgen import Wave
wave = Wave("Regular")
wave.gen_ts()
y = wave.ts
rampfull = np.ones(len(y))
ramp = np.hanning(len(y))
rampfull[-len(ramp)/2:] = ramp[len(ramp)/2:]
rampedy = y*rampfull
plt.plot(rampedy)
class WaveGen(QThread):
def __init__(self, wavetype):
QThread.__init__(self)
self.wavetype = wavetype
self.wave = Wave(self.wavetype)
self.enable = True
def run(self):
self.rampeddown = False
self.cleared = False
self.making = False
# Compute the voltage time series associated with the wave
self.wave.gen_ts_volts()
# Get parameters from the wave object
self.period = self.wave.period
self.height = self.wave.height
self.buffsize = self.wave.sbuffsize
self.sr = self.wave.sr
# Get data to write from the wave object
self.ts_plot = self.wave.ts_elev
self.dataw = self.wave.ts_volts
# If random waves, divide up time series into 120 256 sample parts
if self.wavetype != "Regular":
tsparts = np.reshape(self.dataw, (120, 256))
self.dataw = tsparts[0, :]
# Compute spectrum for plot
if self.wavetype == "Regular":
self.outf, self.outspec = self.wave.comp_spec()
else:
self.outf, self.outspec = self.wave.f, self.wave.spec
# Ramp time series
rampup_ts = ramp_ts(self.dataw, "up")
# Set making variable true
self.making = True
self.AOtaskHandle = daqmx.TaskHandle()
daqmx.CreateTask("", self.AOtaskHandle)
daqmx.CreateAOVoltageChan(self.AOtaskHandle, "Dev1/ao0", "", -10.0,
10.0, daqmx.Val_Volts, None)
daqmx.CfgSampClkTiming(self.AOtaskHandle, "", self.sr,
daqmx.Val_Rising, daqmx.Val_ContSamps,
self.buffsize)
daqmx.SetWriteRegenMode(self.AOtaskHandle, daqmx.Val_DoNotAllowRegen)
# Setup a callback function to run once the DAQmx driver finishes
def DoneCallback_py(taskHandle, status, callbackData_ptr):
self.rampeddown = True
return 0
DoneCallback = daqmx.DoneEventCallbackPtr(DoneCallback_py)
daqmx.RegisterDoneEvent(self.AOtaskHandle, 0, DoneCallback, None)
# Output the rampup time series
daqmx.WriteAnalogF64(self.AOtaskHandle, self.buffsize, False, 10.0,
daqmx.Val_GroupByChannel, rampup_ts)
daqmx.StartTask(self.AOtaskHandle)
# Wait a second to allow the DAQmx buffer to empty
if self.wavetype == "Regular":
time.sleep(self.period * 0.99) # was self.period*0.4
else:
time.sleep(0.99)
# Set iteration variable to keep track of how many chunks of data
# have been written
i = 1
# Main running loop that writes data to DAQmx buffer
while self.enable:
writeSpaceAvail = daqmx.GetWriteSpaceAvail(self.AOtaskHandle)
if writeSpaceAvail >= self.buffsize:
if self.wavetype != "Regular":
if i >= 120:
self.dataw = tsparts[i % 120, :]
else:
self.dataw = tsparts[i, :]
daqmx.WriteAnalogF64(self.AOtaskHandle, self.buffsize, False,
10.0, daqmx.Val_GroupByChannel,
self.dataw)
i += 1
if self.wavetype == "Regular":
time.sleep(0.99 * self.period)
else:
time.sleep(0.99) # Was self.period
# After disabled, initiate rampdown time series
if self.wavetype != "Regular":
if i >= 120:
self.rampdown_ts = ramp_ts(tsparts[i % 120, :], "down")
else:
self.rampdown_ts = ramp_ts(tsparts[i, :], "down")
else:
self.rampdown_ts = ramp_ts(self.dataw, "down")
# Write rampdown time series
daqmx.WriteAnalogF64(self.AOtaskHandle, self.buffsize, False, 10.0,
daqmx.Val_GroupByChannel, self.rampdown_ts)
# Keep running, part of PyDAQmx callback syntax
while True:
pass
def stop(self):
self.stopgen = WaveStop(self)
self.stopgen.start()
