def fetch(filename, nevents, nsequence):
    '''
    Fetch and save waveform traces from the oscilloscope.
    '''
    scope = LeCroyScope(config.ip, timeout=config.timeout)
    scope.clear()
    scope.set_sequence_mode(nsequence)
    channels = scope.get_channels()
    # print(channels)
    settings = scope.get_settings()

    # print(settings)

    if b'ON' in settings['SEQUENCE']:
        sequence_count = int(settings['SEQUENCE'].split(',')[1])
    else:
        sequence_count = 1

    if nsequence != sequence_count:
        print('Could not configure sequence mode properly')
    if sequence_count != 1:
        print('Using sequence mode with %i traces per aquisition' %
              sequence_count)

    current_dim = {}
    for channel in channels:
        wave_desc = scope.get_wavedesc(channel)
        current_dim[channel] = wave_desc['wave_array_count'] // sequence_count

    try:
        i = 0
        while i < nevents:
            print('\rfetching event: %i' % i)
            sys.stdout.flush()
            try:
                scope.trigger()
                for channel in channels:
                    wave_desc, wave_array = scope.get_waveform(channel)
                    print("wave_array = ", wave_array)
                    num_samples = wave_desc[
                        'wave_array_count'] // sequence_count
                    if current_dim[channel] < num_samples:
                        current_dim[channel] = num_samples
                    traces = wave_array.reshape(
                        sequence_count, wave_array.size // sequence_count)
                    #necessary because h5py does not like indexing and this is the fastest (and man is it slow) way
                    scratch = numpy.zeros((current_dim[channel], ),
                                          dtype=wave_array.dtype)

            except (socket.error, struct.error) as e:
                print('\n' + str(e))
                scope.clear()
                continue
            i += sequence_count
    except KeyboardInterrupt:
        print('\rUser interrupted fetch early')
    finally:
        print('\r')
        scope.clear()
        return i
Exemple #2
0
def fetch(filename, nevents, nsequence):
    '''
    Fetch and save waveform traces from the oscilloscope.
    '''
    scope = LeCroyScope(config.ip, timeout=config.timeout)
    scope.clear()
    scope.set_sequence_mode(nsequence)
    channels = scope.get_channels()
    settings = scope.get_settings()

    if 'ON' in settings['SEQUENCE']:
        sequence_count = int(settings['SEQUENCE'].split(',')[1])
    else:
        sequence_count = 1

    if nsequence != sequence_count:
        print 'Could not configure sequence mode properly'
    if sequence_count != 1:
        print 'Using sequence mode with %i traces per aquisition' % sequence_count

    f = h5py.File(filename, 'w')
    for command, setting in settings.items():
        f.attrs[command] = setting
    current_dim = {}
    for channel in channels:
        wave_desc = scope.get_wavedesc(channel)
        current_dim[channel] = wave_desc['wave_array_count'] // sequence_count
        f.create_dataset("c%i_samples" % channel,
                         (nevents, current_dim[channel]),
                         dtype=wave_desc['dtype'],
                         compression='gzip',
                         maxshape=(nevents, None))
        for key, value in wave_desc.items():
            try:
                f["c%i_samples" % channel].attrs[key] = value
            except ValueError:
                pass
        f.create_dataset("c%i_vert_offset" % channel, (nevents, ), dtype='f8')
        f.create_dataset("c%i_vert_scale" % channel, (nevents, ), dtype='f8')
        f.create_dataset("c%i_horiz_offset" % channel, (nevents, ), dtype='f8')
        f.create_dataset("c%i_horiz_scale" % channel, (nevents, ), dtype='f8')
        f.create_dataset("c%i_num_samples" % channel, (nevents, ), dtype='f8')

    try:
        i = 0
        while i < nevents:
            print '\rfetching event: %i' % i,
            sys.stdout.flush()
            try:
                scope.trigger()
                for channel in channels:
                    wave_desc, wave_array = scope.get_waveform(channel)
                    num_samples = wave_desc[
                        'wave_array_count'] // sequence_count
                    if current_dim[channel] < num_samples:
                        current_dim[channel] = num_samples
                        f['c%i_samples' % channel].resize(
                            current_dim[channel], 1)
                    traces = wave_array.reshape(
                        sequence_count, wave_array.size // sequence_count)
                    #necessary because h5py does not like indexing and this is the fastest (and man is it slow) way
                    scratch = numpy.zeros((current_dim[channel], ),
                                          dtype=wave_array.dtype)
                    for n in xrange(0, sequence_count):
                        scratch[0:num_samples] = traces[
                            n]  #'fast' copy to right size
                        f['c%i_samples' %
                          channel][i + n] = scratch  #'fast' add to dataset
                        f['c%i_num_samples' % channel][i + n] = num_samples
                        f['c%i_vert_offset' %
                          channel][i + n] = wave_desc['vertical_offset']
                        f['c%i_vert_scale' %
                          channel][i + n] = wave_desc['vertical_gain']
                        f['c%i_horiz_offset' %
                          channel][i + n] = -wave_desc['horiz_offset']
                        f['c%i_horiz_scale' %
                          channel][i + n] = wave_desc['horiz_interval']

            except (socket.error, struct.error) as e:
                print '\n' + str(e)
                scope.clear()
                continue
            i += sequence_count
    except KeyboardInterrupt:
        print '\rUser interrupted fetch early'
    finally:
        print '\r',
        f.close()
        scope.clear()
        return i
Exemple #3
0
def fetch(filename, nevents, nsequence):
    '''
    Fetch and save waveform traces from the oscilloscope.
    '''
    scope = LeCroyScope(config.ip, timeout=config.timeout)
    scope.clear()
    scope.set_sequence_mode(nsequence)
    channels = scope.get_channels()
    settings = scope.get_settings()

    if 'ON' in settings['SEQUENCE']:
        sequence_count = int(settings['SEQUENCE'].split(',')[1])
    else:
        sequence_count = 1
        
    if nsequence != sequence_count:
        print 'Could not configure sequence mode properly'
    if sequence_count != 1:
        print 'Using sequence mode with %i traces per aquisition' % sequence_count 
    
    f = h5py.File(filename, 'w')
    for command, setting in settings.items():
        f.attrs[command] = setting
    current_dim = {}
    for channel in channels:
        wave_desc = scope.get_wavedesc(channel)
        current_dim[channel] = wave_desc['wave_array_count']//sequence_count
        f.create_dataset("c%i_samples"%channel, (nevents,current_dim[channel]), dtype=wave_desc['dtype'], compression='gzip', maxshape=(nevents,None))
        for key, value in wave_desc.items():
            try:
                f["c%i_samples"%channel].attrs[key] = value
            except ValueError:
                pass
        f.create_dataset("c%i_vert_offset"%channel, (nevents,), dtype='f8')
        f.create_dataset("c%i_vert_scale"%channel, (nevents,), dtype='f8')
        f.create_dataset("c%i_horiz_offset"%channel, (nevents,), dtype='f8')
        f.create_dataset("c%i_horiz_scale"%channel, (nevents,), dtype='f8')
        f.create_dataset("c%i_num_samples"%channel, (nevents,), dtype='f8')
        
    try:
        i = 0
        while i < nevents:
            print '\rfetching event: %i' % i,
            sys.stdout.flush()
            try:
                scope.trigger()
                for channel in channels:
                    wave_desc,wave_array = scope.get_waveform(channel)
                    num_samples = wave_desc['wave_array_count']//sequence_count
                    if current_dim[channel] < num_samples:
                        current_dim[channel] = num_samples
                        f['c%i_samples'%channel].resize(current_dim[channel],1)
                    traces = wave_array.reshape(sequence_count, wave_array.size//sequence_count)
                    #necessary because h5py does not like indexing and this is the fastest (and man is it slow) way
                    scratch = numpy.zeros((current_dim[channel],),dtype=wave_array.dtype)
                    for n in xrange(0,sequence_count):
                        scratch[0:num_samples] = traces[n] #'fast' copy to right size
                        f['c%i_samples'%channel][i+n] = scratch #'fast' add to dataset
                        f['c%i_num_samples'%channel][i+n] = num_samples
                        f['c%i_vert_offset'%channel][i+n] = wave_desc['vertical_offset']
                        f['c%i_vert_scale'%channel][i+n] = wave_desc['vertical_gain']
                        f['c%i_horiz_offset'%channel][i+n] = -wave_desc['horiz_offset']
                        f['c%i_horiz_scale'%channel][i+n] = wave_desc['horiz_interval']
                    
            except (socket.error, struct.error) as e:
                print '\n' + str(e)
                scope.clear()
                continue
            i += sequence_count
    except KeyboardInterrupt:
        print '\rUser interrupted fetch early'
    finally:
        print '\r', 
        f.close()
        scope.clear()
        return i