scretch of recording code. Has to be improved alot. For example inclide fishgrid.cfg
This commit is contained in:
parent
6a877bec5c
commit
165d359fee
196
rasp_grid.py
Normal file
196
rasp_grid.py
Normal file
@ -0,0 +1,196 @@
|
||||
from __future__ import print_function
|
||||
from time import time, sleep
|
||||
import RPi.GPIO as GPIO
|
||||
import subprocess
|
||||
import numpy as np
|
||||
import matplotlib.pyplot as plt
|
||||
from time import sleep, time
|
||||
from os import system
|
||||
from sys import stdout
|
||||
from IPython import embed
|
||||
|
||||
from uldaq import (get_daq_device_inventory, DaqDevice, AInScanFlag, ScanStatus,
|
||||
ScanOption, create_float_buffer, InterfaceType, AiInputMode)
|
||||
|
||||
def GPIO_setup():
|
||||
# LED output pins
|
||||
GPIO.setmode(GPIO.BOARD)
|
||||
|
||||
GPIO.setup(11, GPIO.OUT) # 1
|
||||
GPIO.output(11, GPIO.LOW)
|
||||
GPIO.setup(13, GPIO.OUT) # 2
|
||||
GPIO.output(13, GPIO.LOW)
|
||||
|
||||
LED_status = [False, False]
|
||||
|
||||
# switch controlled input
|
||||
GPIO.setup(16, GPIO.IN)
|
||||
GPIO.setup(18, GPIO.IN)
|
||||
|
||||
return LED_status
|
||||
|
||||
|
||||
def main():
|
||||
LED_status = GPIO_setup()
|
||||
|
||||
# DAQ setup
|
||||
if True:
|
||||
channels = 16
|
||||
|
||||
status = ScanStatus.IDLE
|
||||
|
||||
descriptor_index = 0 # ToDo: ????
|
||||
range_index = 0 # ToDo: ????
|
||||
|
||||
interface_type = InterfaceType.USB
|
||||
low_channel = 0
|
||||
high_channel = channels
|
||||
|
||||
samples_per_channel = 20000 # ToDo: ???? Buffer size ?!
|
||||
rate = 20000
|
||||
scan_options = ScanOption.CONTINUOUS
|
||||
flags = AInScanFlag.DEFAULT
|
||||
|
||||
# Get descriptors for all of the available DAQ devices.
|
||||
devices = get_daq_device_inventory(interface_type)
|
||||
number_of_devices = len(devices)
|
||||
if number_of_devices == 0:
|
||||
raise Exception('Error: No DAQ devices found')
|
||||
|
||||
print('Found', number_of_devices, 'DAQ device(s):')
|
||||
for i in range(number_of_devices):
|
||||
print(' ', devices[i].product_name, ' (', devices[i].unique_id, ')', sep='')
|
||||
|
||||
# Create the DAQ device object associated with the specified descriptor index.
|
||||
daq_device = None
|
||||
daq_device = DaqDevice(devices[descriptor_index])
|
||||
|
||||
# Get the AiDevice object and verify that it is valid.
|
||||
ai_device = None
|
||||
ai_device = daq_device.get_ai_device()
|
||||
if ai_device is None:
|
||||
raise Exception('Error: The DAQ device does not support analog input')
|
||||
|
||||
# Verify that the specified device supports hardware pacing for analog input.
|
||||
ai_info = ai_device.get_info()
|
||||
if not ai_info.has_pacer():
|
||||
raise Exception('\nError: The specified DAQ device does not support hardware paced analog input')
|
||||
|
||||
# Establish a connection to the DAQ device.
|
||||
descriptor = daq_device.get_descriptor()
|
||||
print('\nConnecting to', descriptor.dev_string, '- please wait...')
|
||||
daq_device.connect()
|
||||
|
||||
# The default input mode is SINGLE_ENDED.
|
||||
input_mode = AiInputMode.SINGLE_ENDED
|
||||
# If SINGLE_ENDED input mode is not supported, set to DIFFERENTIAL.
|
||||
if ai_info.get_num_chans_by_mode(AiInputMode.SINGLE_ENDED) <= 0:
|
||||
input_mode = AiInputMode.DIFFERENTIAL
|
||||
|
||||
# Get the number of channels and validate the high channel number.
|
||||
number_of_channels = ai_info.get_num_chans_by_mode(input_mode)
|
||||
if high_channel >= number_of_channels:
|
||||
high_channel = number_of_channels - 1
|
||||
channel_count = high_channel - low_channel + 1
|
||||
|
||||
# Get a list of supported ranges and validate the range index.
|
||||
ranges = ai_info.get_ranges(input_mode)
|
||||
if range_index >= len(ranges):
|
||||
range_index = len(ranges) - 1
|
||||
|
||||
# Allocate a buffer to receive the data.
|
||||
data = create_float_buffer(channel_count, samples_per_channel)
|
||||
system('clear')
|
||||
|
||||
# Start the acquisition.
|
||||
rate = ai_device.a_in_scan(low_channel, high_channel, input_mode, ranges[range_index], samples_per_channel,
|
||||
rate, scan_options, flags, data)
|
||||
last_idx = 0
|
||||
f = open('/media/pi/data1/test_file.raw', 'wb')
|
||||
|
||||
|
||||
# LED on when here ... wait for switch to start data aquisition
|
||||
GPIO.output(11, GPIO.HIGH)
|
||||
LED_status[0] = True
|
||||
while GPIO.input(16) == GPIO.LOW:
|
||||
sleep(.1)
|
||||
GPIO.output(11, GPIO.LOW)
|
||||
LED_status[0] = False
|
||||
|
||||
LED_t = time()
|
||||
LED_t_interval = 2
|
||||
|
||||
disp_eth_power = True
|
||||
|
||||
try:
|
||||
while GPIO.input(16) == GPIO.HIGH:
|
||||
# blinking LED
|
||||
if time() - LED_t < .1 and LED_status[0] == False:
|
||||
LED_status[0] = True
|
||||
GPIO.output(11, GPIO.HIGH)
|
||||
if time() - LED_t >= .1 and LED_status[0] == True:
|
||||
LED_status[0] = False
|
||||
GPIO.output(11, GPIO.LOW)
|
||||
if time() - LED_t >= LED_t_interval:
|
||||
LED_t = time()
|
||||
|
||||
# dist & eth0 controll
|
||||
if GPIO.input(18) == GPIO.HIGH:
|
||||
if disp_eth_power == True:
|
||||
subprocess.run(['tvservice', '-o'])
|
||||
subprocess.run(['vcgencmd', 'display_power', '0'])
|
||||
|
||||
subprocess.run(['sudo', 'ip', 'link', 'set', 'eth0', 'down'])
|
||||
disp_eth_power = False
|
||||
else:
|
||||
if disp_eth_power == False:
|
||||
subprocess.run(['tvservice', '-p'])
|
||||
subprocess.run(['vcgencmd', 'display_power', '1'])
|
||||
subprocess.run(['sudo', '/bin/chvt', '6'])
|
||||
subprocess.run(['sudo', '/bin/chvt', '7'])
|
||||
|
||||
subprocess.run(['sudo', 'ip', 'link', 'set', 'eth0', 'up'])
|
||||
disp_eth_power = True
|
||||
|
||||
# Get the status of the background operation
|
||||
status, transfer_status = ai_device.get_scan_status()
|
||||
|
||||
index = transfer_status.current_index
|
||||
|
||||
if (last_idx > index) and (index != -1):
|
||||
np.array(data[last_idx:], dtype=np.float32).tofile(f)
|
||||
np.array(data[:index], dtype=np.float32).tofile(f)
|
||||
if index == -1:
|
||||
pass
|
||||
else:
|
||||
np.array(data[last_idx:index], dtype=np.float32).tofile(f)
|
||||
|
||||
if index == -1:
|
||||
last_idx= len(data)
|
||||
else:
|
||||
last_idx = index
|
||||
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
|
||||
f.close()
|
||||
if daq_device:
|
||||
# Stop the acquisition if it is still running.
|
||||
if status == ScanStatus.RUNNING:
|
||||
ai_device.scan_stop()
|
||||
if daq_device.is_connected():
|
||||
daq_device.disconnect()
|
||||
daq_device.release()
|
||||
|
||||
if disp_eth_power == False:
|
||||
subprocess.run(['tvservice', '-p'])
|
||||
subprocess.run(['vcgencmd', 'display_power', '1'])
|
||||
subprocess.run(['sudo', '/bin/chvt', '6'])
|
||||
subprocess.run(['sudo', '/bin/chvt', '7'])
|
||||
|
||||
# subprocess.run(['sudo', 'ip', 'link', 'set', 'eth0', 'up'])
|
||||
GPIO.cleanup()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Reference in New Issue
Block a user