Exemplo n.º 1
0
import pickle
import time
from utils.data_utils import RNStream

# recording_buffer = pickle.load(open('C:/Recordings/03_21_2021_22_48_25-Exp_Unity.RealityNavigationHotel.EventMarkers-Sbj_someone-Ssn_0.p', 'rb'))

# rns.stream_out(recording_buffer)

rns = RNStream(
    'C:/Users/S-Vec/Dropbox/research/RealityNavigation/Data/Pilot/03_22_2021_17_13_28-Exp_realitynavigation-Sbj_0-Ssn_3.dats'
)
print('String names are {0}'.format(str(rns.get_stream_names())))

start_time = time.time()
reloaded_buffer = rns.stream_in(only_stream=('BioSemi', ))
print('reload with only took {0}'.format(time.time() - start_time))
#
# start_time = time.time()
# reloaded_buffer = rns.stream_in(ignore_stream=('monitor1', '0'))
# print('reload with ignore took {0}'.format(time.time() - start_time))
#
#
# start_time = time.time()
# reloaded_buffer = rns.stream_in()
# print('reload all took {0}'.format(time.time() - start_time))

rns.generate_video(video_stream_name='monitor1')
Exemplo n.º 2
0
import os

import numpy as np
import cv2

import config
from utils.data_utils import RNStream
import matplotlib.pyplot as plt

data_root = 'C:/Users/S-Vec/Dropbox/research/RealityNavigation/Data/Pilot/'
data_fn = '03_22_2021_17_03_52-Exp_realitynavigation-Sbj_0-Ssn_2.dats'

video_stream_label = 'monitor1'

rns = RNStream(os.path.join(data_root, data_fn))
data = rns.stream_in(ignore_stream=('0'))

video_frame_stream = data[video_stream_label][0]
frame_count = video_frame_stream.shape[-1]
frame_size = (data[video_stream_label][0].shape[1],
              data[video_stream_label][0].shape[0])
out_path = os.path.join(
    data_root, '{0}_{1}.avi'.format(data_fn.split('.')[0], video_stream_label))
out = cv2.VideoWriter(out_path, cv2.VideoWriter_fourcc(*'DIVX'),
                      1 / config.CAMERA_SCREENCAPTURE_REFRESH_INTERVAL,
                      frame_size)

for i in range(frame_count):
    print('Creating video progress {}%'.format(
        str(round(100 * i / frame_count, 2))),
          sep=' ',
Exemplo n.º 3
0
from utils.data_utils import RNStream

test_rns = RNStream('C:/Recordings/test.dats')
test_reloaded_data = test_rns.stream_in()

another_rns = RNStream(
    'C:/Recordings/03_22_2021_00_00_55-Exp_myexperiment-Sbj_someone-Ssn_0.dats'
)
another_reloaded_data = another_rns.stream_in()

result_rns = RNStream('C:/Recordings/results.dats')
result_rns.stream_out(test_reloaded_data)
result_rns.stream_out(another_reloaded_data)

results_reloaded_data = result_rns.stream_in()
import numpy as np

from utils.data_utils import RNStream
import matplotlib.pyplot as plt
from matplotlib.pyplot import figure

file_path = 'C:/Users/S-Vec/Dropbox/research/RealityNavigation/Data/Pilot/03_22_2021_16_52_54-Exp_realitynavigation-Sbj_0-Ssn_1.dats'
em_stream_name = 'Unity.RotationWheel.EventMarkers'

rns = RNStream(file_path)
data = rns.stream_in(ignore_stream=('monitor1', '0'))

data_stream = data[em_stream_name][0]
timestamps_stream = data[em_stream_name][1]
event_label_stream = data_stream[-1, :]
item_count = 5
offset = 2
trial_count = 24
trial_started_index = 0
for target_label in range(offset, offset + 5):
    target_onset_em = np.logical_and(
        event_label_stream == target_label,
        np.concatenate([np.array([0]),
                        np.diff(event_label_stream)]) != 0)
    started_em = event_label_stream >= 1
    started = False
    target_count = 0
    target_indices = []
    clean_count = 0
    trial_count = 0
    for i in range(1, len(target_onset_em)):
Exemplo n.º 5
0
from utils.data_utils import RNStream

test_rns = RNStream(
    '/Users/Leo/Dropbox/research/RealityNavigation/Data/Pilot/03_22_2021_17_03_52-Exp_realitynavigation-Sbj_0-Ssn_2.dats'
)
reloaded_data = test_rns.stream_in(ignore_stream=('monitor1', '0'))