clear;clc;

target_fs = 500; %Hz

raw_fs=512;

%% 1. prepare files

% 1.1 load data

data_path='E:\LeftRight_Hand_MI';

data_file = 's01.mat';

fid = [data_path, filesep, data_file];

load(fid);

fs = eeg.srate;

EEG_channels = 1:64;

EEG_data1 = eeg.movement_left; % there are multiple datasets in eeg, make sure use all of them. eg: eeg.movement_right

EEG_data1 = EEG_data1(EEG_channels, :);

EEG_data2 = eeg.movement_right; % there are multiple datasets in eeg, make sure use all of them. eg: eeg.movement_right

EEG_data2 = EEG_data2(EEG_channels, :);

EEG_data3 = eeg.imagery_left; % there are multiple datasets in eeg, make sure use all of them. eg: eeg.movement_right

EEG_data3 = EEG_data3(EEG_channels, :);

EEG_data4 = eeg.imagery_right; % there are multiple datasets in eeg, make sure use all of them. eg: eeg.movement_right

EEG_data4 = EEG_data4(EEG_channels, :);

EEG_data5 = eeg.rest; % there are multiple datasets in eeg, make sure use all of them. eg: eeg.movement_right

EEG_data5 = EEG_data5(EEG_channels, :);

% 1.2 prepare electrodes locations

load('biosemi_template.mat');

% check if the locations are the same to data description

topoplot([], locs, 'electrodes', 'ptslabels', 'plotdisk', 'on');

%% Use the method of ICA and IClabel

EEG_epochs1 = IC_label_artifact_removal( EEG_data1, fs, target_fs,locs,EEG_channels);

EEG_epochs2 = IC_label_artifact_removal( EEG_data2, fs, target_fs,locs,EEG_channels);

EEG_epochs3 = IC_label_artifact_removal( EEG_data3, fs, target_fs,locs,EEG_channels);

EEG_epochs4 = IC_label_artifact_removal( EEG_data4, fs, target_fs,locs,EEG_channels);

EEG_epochs5 = IC_label_artifact_removal( EEG_data5, fs, target_fs,locs,EEG_channels);

EEG_epochsall=[EEG_epochs1;EEG_epochs2;EEG_epochs3;EEG_epochs4;EEG_epochs5;]

%% 3. visual check

EEG_epochsall = visual_check(EEG_epochsall, target_fs);

%% save

output = 'C:\DL_denoising_data_cut\EEG_output';

save([output, filesep, 'EEG_epochs_s01','.mat'], 'EEG_epochsall');

%writeNPY(EEG_epochs, [output, filesep, 'EEG_epochs_Cz', char(film_list(film_id)),'.npy']);