plotandwriteexample.m

function void = plotandwriteexample(comPort, captureDuration, fileName)
%PLOTANDWRITEEXAMPLE - Demonstrate basic features of ShimmerHandleClass
%
%  PLOTANDWRITEEXAMPLE(COMPORT, CAPTUREDURATION, FILENAME) plots 3
%  accelerometer signals, 3 gyroscope signals and 3 magnetometer signals,
%  from the Shimmer paired with COMPORT. The function
%  will stream data for a fixed duration of time defined by the constant
%  CAPTUREDURATION. The function also writes the data in a tab ddelimited
%  format to the file defined in FILENAME.
%
%  SYNOPSIS: plotandwriteexample(comPort, captureDuration, fileName)
%
%  INPUT: comPort - String value defining the COM port number for Shimmer
%  INPUT: captureDuration - Numerical value defining the period of time
%                           (in seconds) for which the function will stream
%                           data from  the Shimmers.
%  INPUT : fileName - String value defining the name of the file that data
%                     is written to in a comma delimited format.
%  OUTPUT: none
%
%  EXAMPLE: plotandwriteexample('COM3', 30, 'testdata.dat')
%
%  See also ShimmerDeviceHandler

%% definitions

deviceHandler = ShimmerDeviceHandler();                                   % Define a handler
firsttime = true;
configured = 0;
% Note: these constants are only relevant to this examplescript and are not used
% by the ShimmerDevice Handler
NO_SAMPLES_IN_PLOT = 500;                                                  % Number of samples that will be displayed in the plot at any one time
DELAY_PERIOD = 0.2;                                                        % A delay period of time in seconds between data read operations
numSamples = 0;

addpath('./Resources/')                                                    % directory containing supporting functions

%%
deviceHandler.bluetoothManager.setVerbose(false);
deviceHandler.bluetoothManager.connectShimmerThroughCommPort(comPort);
cleaner = onCleanup(@() deviceHandler.bluetoothManager.getShimmerDeviceBtConnected(comPort).disconnect());  % Ensure disconnection on cleanup
addlistener(deviceHandler, 'DeviceConnected', @(src,evt) onConnected(src, evt));
addlistener(deviceHandler, 'DeviceDisconnected',    @(src,evt) disp("Script: Disconnected"));
addlistener(deviceHandler, 'DeviceConnectionLost',  @(src,evt) disp("Script: Lost connection"));
% Ensure disconnection happens properly even if the workspace is cleared or the script is interrupted

plotData = [];
timeStamp = [];

h.figure1=figure('Name','Shimmer 1 signals' );                     % Create a handle to figure for plotting data from shimmer
set(h.figure1, 'Position', [100, 500, 800, 400]);

while(isempty(deviceHandler.obj.receiveData(comPort)))                                  % we wait here for the device to start streaming
    pause(0.1);
end
elapsedTime = 0;                                                   % Reset to 0
tic;
while (elapsedTime < captureDuration)

    pause(DELAY_PERIOD);                                           % Pause for this period of time on each iteration to allow data to arrive in the buffer

    data = deviceHandler.obj.receiveData(comPort);                                  % Read the latest data from shimmer data buffer, signalFormatArray defines the format of the data and signalUnitArray the unit
    if (isempty(data))
        continue;
    end
    newData = data(1);
    signalNameArray = data(2);
    signalFormatArray = data(3);
    signalUnitArray = data(4);

    signalNameCellArray = cell(numel(signalNameArray), 1);
    for i = 1:numel(signalNameArray)
        signalNameCellArray{i} = char(signalNameArray(i));         % Convert each Java string to a MATLAB char array
    end

    signalFormatCellArray = cell(numel(signalFormatArray), 1);
    for i = 1:numel(signalFormatArray)
        signalFormatCellArray{i} = char(signalFormatArray(i));     % Convert each Java string to a MATLAB char array
    end

    signalUnitCellArray = cell(numel(signalUnitArray), 1);
    for i = 1:numel(signalUnitArray)
        signalUnitCellArray{i} = char(signalUnitArray(i));         % Convert each Java string to a MATLAB char array
    end

    if(~isempty(signalNameCellArray))
        chIndex(1) = find(ismember(signalNameCellArray, 'Timestamp')); % Get signal indices
        chIndex(2) = find(ismember(signalNameCellArray, 'Accel_LN_X'));
        chIndex(3) = find(ismember(signalNameCellArray, 'Accel_LN_Y'));
        chIndex(4) = find(ismember(signalNameCellArray, 'Accel_LN_Z'));
        chIndex(5) = find(ismember(signalNameCellArray, 'Gyro_X'));
        chIndex(6) = find(ismember(signalNameCellArray, 'Gyro_Y'));
        chIndex(7) = find(ismember(signalNameCellArray, 'Gyro_Z'));
        chIndex(8) = find(ismember(signalNameCellArray, 'Mag_X'));
        chIndex(9) = find(ismember(signalNameCellArray, 'Mag_Y'));
        chIndex(10) = find(ismember(signalNameCellArray, 'Mag_Z'));
    end

    if (firsttime==true && isempty(newData)~=1)
        firsttime = newWriteHeadersToFile(fileName,signalNameCellArray(chIndex),signalFormatCellArray(chIndex),signalUnitCellArray(chIndex));
    end

    if ~isempty(newData)                                           % TRUE if new data has arrived

        filtredData = newData(:, chIndex);

        dlmwrite(fileName, double(filtredData), '-append', 'delimiter', '\t', 'precision', 16);

        plotData = [plotData; newData];                            % Update the plotDataBuffer with the new data
        numPlotSamples = size(plotData,1);
        numSamples = numSamples + size(newData,1);
        timeStampNew = newData(:,11);                              % get timestamps
        timeStamp = [timeStamp; timeStampNew];

        if numSamples > NO_SAMPLES_IN_PLOT
            plotData = plotData(numPlotSamples-NO_SAMPLES_IN_PLOT+1:end,:);
        end
        sampleNumber = max(numSamples-NO_SAMPLES_IN_PLOT+1,1):numSamples;

        set(0,'CurrentFigure',h.figure1);
        subplot(2,2,1);                                            % Create subplot
        signalIndex = chIndex(1);
        plot(sampleNumber, plotData(:,signalIndex));               % Plot the time stamp data
        legend([char(signalFormatArray(signalIndex)) ' ' char(signalNameArray(signalIndex)) ' (' char(signalUnitArray(signalIndex)) ')']);
        xlim([sampleNumber(1) sampleNumber(end)]);

        subplot(2,2,2);                                            % Create subplot
        signalIndex1 = chIndex(2);
        signalIndex2 = chIndex(3);
        signalIndex3 = chIndex(4);
        plot(sampleNumber, plotData(:,[signalIndex1 signalIndex2 signalIndex3]));                                 % Plot the accelerometer data
        legendName1 = [char(signalFormatArray(signalIndex1)) ' ' char(signalNameArray(signalIndex1)) ' (' char(signalUnitArray(signalIndex1)) ')'];
        legendName2 = [char(signalFormatArray(signalIndex2)) ' ' char(signalNameArray(signalIndex2)) ' (' char(signalUnitArray(signalIndex2)) ')'];
        legendName3 = [char(signalFormatArray(signalIndex3)) ' ' char(signalNameArray(signalIndex3)) ' (' char(signalUnitArray(signalIndex3)) ')'];
        legend(legendName1,legendName2,legendName3);               % Add legend to plot
        xlim([sampleNumber(1) sampleNumber(end)]);

        subplot(2,2,3);                                            % Create subplot
        signalIndex1 = chIndex(5);
        signalIndex2 = chIndex(6);
        signalIndex3 = chIndex(7);
        plot(sampleNumber, plotData(:,[signalIndex1 signalIndex2 signalIndex3]));                                 % Plot the gyroscope data
        legendName1 = [char(signalFormatArray(signalIndex1)) ' ' char(signalNameArray(signalIndex1)) ' (' char(signalUnitArray(signalIndex1)) ')'];
        legendName2 = [char(signalFormatArray(signalIndex2)) ' ' char(signalNameArray(signalIndex2)) ' (' char(signalUnitArray(signalIndex2)) ')'];
        legendName3 = [char(signalFormatArray(signalIndex3)) ' ' char(signalNameArray(signalIndex3)) ' (' char(signalUnitArray(signalIndex3)) ')'];
        legend(legendName1,legendName2,legendName3);               % Add legend to plot
        xlim([sampleNumber(1) sampleNumber(end)]);

        subplot(2,2,4);                                            % Create subplot
        signalIndex1 = chIndex(8);
        signalIndex2 = chIndex(9);
        signalIndex3 = chIndex(10);
        plot(sampleNumber, plotData(:,[signalIndex1 signalIndex2 signalIndex3]));                                 % Plot the magnetometer data
        legendName1 = [char(signalFormatArray(signalIndex1)) ' ' char(signalNameArray(signalIndex1)) ' (' char(signalUnitArray(signalIndex1)) ')'];
        legendName2 = [char(signalFormatArray(signalIndex2)) ' ' char(signalNameArray(signalIndex2)) ' (' char(signalUnitArray(signalIndex2)) ')'];
        legendName3 = [char(signalFormatArray(signalIndex3)) ' ' char(signalNameArray(signalIndex3)) ' (' char(signalUnitArray(signalIndex3)) ')'];
        legend(legendName1,legendName2,legendName3);               % Add legend to plot
        xlim([sampleNumber(1) sampleNumber(end)]);
    end

    elapsedTime = elapsedTime + toc;                               % Stop timer and add to elapsed time
    tic;                                                           % Start timer

end
fprintf('The percentage of received packets: %d \n',deviceHandler.bluetoothManager.getShimmerDeviceBtConnected(comPort).getPacketReceptionRateOverall()); % Detect loss packets
deviceHandler.bluetoothManager.getShimmerDeviceBtConnected(comPort).stopStreaming();                                       % Stop data streaming                                                       % Stop data streaming
deviceHandler.bluetoothManager.getShimmerDeviceBtConnected(comPort).disconnect();





    function onConnected(deviceHandler, evt)
        disp("Script: Connected");
        if (configured==1) % a connected state is also triggered after configuring, so this differentiates the two
            deviceHandler.bluetoothManager.getShimmerDeviceBtConnected(comPort).startStreaming();
            return
        end
        shimmerClone = deviceHandler.bluetoothManager.getShimmerDeviceBtConnected(comPort).deepClone();
        shimmerClone.setSamplingRateShimmer(51.2);

        shimmerClone.disableAllSensors();                                      % Disables all currently enabled sensors
        shimmerClone.setEnabledAndDerivedSensorsAndUpdateMaps(0, 0);           % Resets configuration on enabled and derived sensors

        sensorIds = javaArray('java.lang.Integer', 3);
        sensorIds(1) = java.lang.Integer(deviceHandler.sensorClass.SHIMMER_ANALOG_ACCEL);
        sensorIds(2) = java.lang.Integer(deviceHandler.sensorClass.SHIMMER_MPU9X50_GYRO);
        sensorIds(3) = java.lang.Integer(deviceHandler.sensorClass.SHIMMER_LSM303_MAG);
        hwid = shimmerClone.getHardwareVersionParsed();
        if hwid.equals('Shimmer3R')
            sensorIds(1) = java.lang.Integer(deviceHandler.sensorClass.SHIMMER_LSM6DSV_ACCEL_LN);
            sensorIds(2) = java.lang.Integer(deviceHandler.sensorClass.SHIMMER_LSM6DSV_GYRO);
            sensorIds(3) = java.lang.Integer(deviceHandler.sensorClass.SHIMMER_LIS2MDL_MAG);
        end
        shimmerClone.setSensorIdsEnabled(sensorIds);

        commType = javaMethod('valueOf', 'com.shimmerresearch.driver.Configuration$COMMUNICATION_TYPE', 'BLUETOOTH');
        com.shimmerresearch.driverUtilities.AssembleShimmerConfig.generateSingleShimmerConfig(shimmerClone, commType);
        deviceHandler.bluetoothManager.getShimmerDeviceBtConnected(comPort).configureFromClone(shimmerClone);
        configured = configured + 1;

    end


end