|
@@ -94,13 +94,6 @@
|
|
|
" # make time from session start\n",
|
|
|
" pos_at_freq[:, 0] = pos_at_freq[:, 0] - pos_at_freq[0][0]\n",
|
|
|
"\n",
|
|
|
- " # convert positions from pixels to meters and center\n",
|
|
|
- " #arena_d = parameters['position']['arena_diameter']\n",
|
|
|
- " #arena_d = 0.92\n",
|
|
|
- " #pixel_size = arena_d / (2 * float(parameters['position']['arena_radius']))\n",
|
|
|
- " #pos_at_freq[:, 1] = (parameters['position']['arena_x'] - pos_at_freq[:, 1]) * pixel_size\n",
|
|
|
- " #pos_at_freq[:, 2] = (parameters['position']['arena_y'] - pos_at_freq[:, 2]) * pixel_size\n",
|
|
|
- "\n",
|
|
|
" width = 100 # 100 points ~= 1 sec with at 100Hz\n",
|
|
|
" kernel = signal.gaussian(width, std=(width) / 7.2)\n",
|
|
|
"\n",
|
|
@@ -124,12 +117,9 @@
|
|
|
" for i in range(t_count):\n",
|
|
|
" t_start_idx = (np.abs(pos_at_freq[:, 0] - events[2*i][0])).argmin()\n",
|
|
|
" t_end_idx = (np.abs(pos_at_freq[:, 0] - events[2*i + 1][0])).argmin()\n",
|
|
|
- " x_in_m = (parameters['position']['arena_x'] - events[2*i][1]) * pixel_size\n",
|
|
|
- " y_in_m = (parameters['position']['arena_y'] - events[2*i][2]) * pixel_size\n",
|
|
|
- " r_in_m = events[2*i][3] * pixel_size\n",
|
|
|
" state = 0 if events[2*i + 1][5] > 1 else 1\n",
|
|
|
"\n",
|
|
|
- " trials[i] = (t_start_idx, t_end_idx, x_in_m, y_in_m, r_in_m, state)\n",
|
|
|
+ " trials[i] = (t_start_idx, t_end_idx, events[2*i][1], events[2*i][2], events[2*i][3], state)\n",
|
|
|
"\n",
|
|
|
" trial_idxs = proc.create_dataset('trial_idxs', data=trials)\n",
|
|
|
" trial_idxs.attrs['headers'] = 't_start_idx, t_end_idx, target_x, target_y, target_r, fail_or_success'\n",
|