simple_example.py

from CREPE import CREPE, CrepeModus, get_queue, is_poison_pill, QueueService

import time
import os,sys,inspect 

# Find the path to the test_data folder.
__currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
path_to_test_data_folder = __currentdir + "/test_data/"

# A simple way to get data
def main():
    # Get the absolute path for the test file
    file_path = path_to_test_data_folder + "4.h5"

    #Create a crepe object and start it
    crep = CREPE(modus=CrepeModus.FILE, file_path=file_path)    
    """ choose one of the different examples: """ 
    #get_data_from_meame_direct_from_queue(crep)
    get_data_from_meame_with_helper(crep)
    #get_data_from_meame_with_easier_helper(crep)

# uses the queue directly, we would not recomend to do it this way tho
def get_data_from_meame_direct_from_queue(crep):
    # Get the hdf5 queue
    queue = crep.get_first_queue()
        
    # Our simple experiment is just to count how many data points we have in a channel :)
    len_of_stream = 0
     
    # Now we want to iterate 
    while True:
        try: 
            # get the next element in the queue
            data = queue.get(timeout=1) #1 seconds timeout
            if is_poison_pill(data):
                break
        except: # .get raises error on timeout
            break
      s   # Add the length of this data
        len_of_stream += len(data[0])

    print("The length of a row is ", len_of_stream)
    
    # Shutdown crepe
    crep.shutdown()

# gets the data with help from QueueService class (Recomended)
def get_data_from_meame_with_helper(crep):
    # create a QueueService object
    # crep.get_first_queue gets the HDF5Reader/NeuroProcessing/MeameListener queue
    # If you want to get the queue from MeameSpeaker then use crep.get_last_queue()
    helper = QueueService(name="HELPER", queue_in=crep.get_first_queue())
    i = 0
    while True:
        # get the next element
        data = helper.get()
        # check if the element was a poisonous pill (aka the last element)
        if data is False:
            crep.shutdown()
            break
        i += len(data[0])
    print("Got ", i, " elements in each row")

# Alternative way to read from the last queue. (easy, but not that customizable)
def get_data_from_meame_with_easier_helper(crep):
    # make a function that will do something with the data
    def data_func(data):
        #do something with data, for example print it
        print(data)

    #this function reads from the last queue untill it's end
    crep.wait(data_func)

if __name__ == "__main__":
    main()