mobilab/dataSourceSET.m at master · sccn/mobilab · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
% Definition of the class dataSourceSET. This class imports EEGLAB set files into MoBILAB.
%
% Author: Alejandro Ojeda, SCCN, INC, UCSD, Oct-2012

classdef dataSourceSET < dataSource
    methods
        function obj = dataSourceSET(varargin)
            % Creates a dataSourceSET object.
            %
            % Input arguments:
            %       file:              EEGLAB set file
            %       mobiDataDirectory: path to the directory where the collection of
            %                          files will be stored
            %
            % Output arguments:
            %       obj:               dataSource object (handle)
            %
            % Usage:
            %        % file: EEGLAB set file
            %        % mobiDataDirectory: path to the directory where the collection of
            %        %                    files are or will be stored.
            %
            %        obj = dataSourceSET( file,  mobiDataDirectory);

            if nargin==1, varargin = varargin{1};end
            if length(varargin) < 2, error('Not enough input arguments.');end
            sourceFileName = varargin{1};
            mobiDataDirectory = varargin{2};
            [path,~,ext] = fileparts(sourceFileName);
            if ~strcmpi(ext,'.set'), error('prog:input',['dataSourceSET cannot read ''' ext ''' format.']);end
            if strcmp(path,mobiDataDirectory), error('Source and destiny folders must be different.');end

            uuid = generateUUID;
            obj@dataSource(mobiDataDirectory,uuid);
            obj.checkThisFolder(mobiDataDirectory);

            obj.container.lockGui;
            EEG = pop_loadset(sourceFileName);
            dim = size(EEG.data);
            dim(end+1) = 1;
            uuid = generateUUID;
            binFile = [obj.mobiDataDirectory filesep EEG.setname '_' uuid '_' obj.sessionUUID '.bin'];
            header  = [obj.mobiDataDirectory filesep EEG.setname '_' uuid '_' obj.sessionUUID '.hdr'];

            eventObj = event;
            eventObj = eventObj.addEvent(round(cell2mat({EEG.event.latency})),{EEG.event.type});
            fiducials = [];
            if ~isempty(EEG.chanlocs)
                labels = {EEG.chanlocs.labels};
                try
                    [elec,lab] = readMontage(EEG);
                    I = ismember(lab,labels);
                    channelSpace = elec(I,:);
                    I = ismember(lab,'fidnz') | ismember(lab,'nasion') | ismember(lab,'Nz');
                    if any(I), fiducials.nasion = elec(I,:);end

                    I = ismember(lab,'fidt9') | ismember(lab,'lpa') | ismember(lab,'LPA') ;
                    if any(I), fiducials.lpa = elec(I,:);end

                    I = ismember(lab,'fidt10') | ismember(lab,'rpa') | ismember(lab,'RPA');
                    if any(I), fiducials.rpa = elec(I,:);end

                    I = ismember(lab,'fidt11') | ismember(lab,'vertex');
                    if any(I), fiducials.vertex = elec(I,:);end

                catch
                    channelSpace = [cell2mat({EEG.chanlocs.X})' cell2mat({EEG.chanlocs.Y})' cell2mat({EEG.chanlocs.Z})'];
                end
            else
                labels = cell(EEG.nbchan,1);
                channelSpace = nan(EEG.nbchan,3);
                for it=1:EEG.nbchan, labels{it} = num2str(it);end
            end
            if length(EEG.times) == prod(dim(2:3))
                timeStamp = EEG.times;
            else timeStamp = (0:prod(dim(2:3))-1)/EEG.srate;
            end
            parentCommand.commandName = 'dataSourceSET';
            parentCommand.varargin{1} = sourceFileName;
            parentCommand.varargin{2} = mobiDataDirectory;

            metadata = struct('binFile',binFile,'header',header,'name',EEG.setname,'timeStamp',[],'numberOfChannels',EEG.nbchan,'precision','double',...
                'uuid',uuid,'sessionUUID',obj.sessionUUID,'writable',false,'unit','none','hardwareMetaData',[],...
                'parentCommand',parentCommand,'label',[],'event',eventObj.saveobj,'notes','','samplingRate',EEG.srate,'channelSpace',[],'class','eeg','fiducials',fiducials);
            metadata.timeStamp = timeStamp;
            metadata.label = labels;
            metadata.channelSpace = channelSpace;

            fid = fopen(binFile,'w');
            data = reshape(EEG.data,[dim(1) prod(dim(2:3))]);
            if isa(data,'mmo'), data = data(:,:);end
            fwrite(fid,data','double');
            fclose(fid);
            header = metadata2headerFile(metadata);
            obj.addItem(header);
            obj.connect;
            obj.updateLogicalStructure;
            obj.container.lockGui;
        end
    end
end