% Check if the 'Communications Toolbox Library for the Bluetooth Protocol'
% support package is installed.
commSupportPackageCheck('BLUETOOTH');

% Configure the number of Slaves in the piconet.
numSlaves = 1;

% Configure the logical links between the Master and Slaves.

% Each element represents the logical link between the Master and the
% respective Slave. If the Master is connected to multiple Slaves, this
% value must be a vector.
linkTraffic = 1;

% Configure the frequency hopping sequence as 'Connection basic' or
% 'Connection adaptive'.
sequenceType = 'Connection adaptive';

% To enable an ACL logical transport, set linkTraffic to 1 or 3. Specify
% the ACL packet type as 'DM1', 'DH1', 'DM3', 'DH3', 'DM5', or 'DH5'.
aclPacketType = 'DM1';

% To enable a SCO logical transport, set linkTraffic to 2 or 3. Specify the
% SCO packet type as 'HV1', 'HV2', or 'HV3' for the respective Slave that
% has SCO link traffic. Here, 1 represents the Slave number, and 'HV3'
% represents the corresponding SCO packet type used by Slave 1.
scoPacketType = {1, 'HV3'};

% Configure wireless channel parameters.
EbNo = 22; % Ratio of energy per bit (Eb) to spectral noise density (No) in dB.

% Configure the WLAN interference.

% Specify the WLAN interference as 'Generated', 'BasebandFile', or 'None'.
% To use the 'wlanBBFilename' option, set wlanInterference to
% 'BasebandFile'.
wlanInterference = 'BasebandFile';
wlanBBFilename =  'WLANNonHTDSSS.bb';

% Signal to interference ratio in dB.
SIR = -15;

% Reset the random number generator.
rng('default');

% Set the total number of nodes in the piconet (one Master and multiple
% Slaves).
numNodes = numSlaves + 1;

% Initialize the cell array to store the Bluetooth nodes.
btNodes = cell(1, numNodes);

% Specify the distance (in meters) between two Bluetooth nodes.
interNodeDistance = 10;

% Configure the nodes as Master or Slave.
for nodeIdx = 1:numNodes
    if nodeIdx == 1
        % Create a baseband Master instance.
        basebandInstance = helperBluetoothBaseband('Role','Master');
    else
        % Create a baseband Slave instance.
        basebandInstance = helperBluetoothBaseband('Role','Slave');
    end
    % Configure the node position.
    nodePosition = [nodeIdx*interNodeDistance 0 0];
    
    % Create and configure the channel of the node.
    channelInstance = helperBluetoothChannel('EbNo',EbNo, ...
                                             'SIR',SIR, ...
                                             'NodePosition',nodePosition, ...
                                             'RxRange',numNodes*interNodeDistance);
    if ~strcmpi(wlanInterference, 'None')
        % Generate the WLAN waveform.
        wlanWaveform = helperBluetoothGenerateWLANWaveform(wlanInterference, wlanBBFilename);
        % Add the WLAN interference to Bluetooth channel.
        addWLANWaveform(channelInstance, wlanWaveform);
    end
    
    % Create and configure the node with a baseband layer and channel.
    node = helperBluetoothFullDuplexNode('Baseband',basebandInstance, ...
                                         'Channel',channelInstance);
    % Update the node ID.
    node.NodeID = nodeIdx;
    
    % Update the node position.
    node.NodePosition = nodePosition;
    
    % Add Bluetooth nodes in the cell array.
    btNodes{nodeIdx} = node;
end

% Create a Bluetooth piconet.
helperBluetoothCreatePiconet(btNodes, numSlaves, linkTraffic, ...
    sequenceType, scoPacketType);

% View the Bluetooth waveforms using the spectrum analyzer.
spectrumAnalyzer = dsp.SpectrumAnalyzer(...
    'Name','Bluetooth Full Duplex Communication', ...
    'ViewType','Spectrum and spectrogram', ...
    'SpectrumType','Power density', ...
    'TimeResolutionSource','Property', ...
    'TimeResolution',0.0005, ...
    'SampleRate',btNodes{1}.PHY.SamplesPerSymbol*1e6, ...
    'TimeSpanSource','Property', ...
    'TimeSpan',0.05, ...
    'FrequencyResolutionMethod','WindowLength', ...
    'WindowLength',512, ...
    'AxesLayout','Horizontal');

% Set the simulation time in seconds.
simulationTime = 2;

% Convert the simulation time into microseconds.
simulationTimeInUS = simulationTime*1e6;

% Enable or disable the visualization in the example.
enableVisualization = true;

% Current simulation time in microseconds.
curTime = 0;

% Elapsed time in microseconds.
elapsedTime = 0;

% Next invoke times of all of the nodes in microseconds.
nextInvokeTimes = zeros(1, numNodes);

% Transmission flag to specify that the channel is free.
txFlag = false;

% Buffer to store the transmitted waveforms.
txBuffer = cell(1, numNodes);

if enableVisualization
    % Plot the BER.
    berFigure = figure("Name","BER of Each Bluetooth Node",'Tag','BluetoothBERPlot');
    berAxes = axes(berFigure);
    % Add annotations to the figure.
    ylim(berAxes, [0 0.3]);
    xlabel(berAxes, 'Simulation Time (in Microseconds)');
    ylabel(berAxes, 'BER');
    title(berAxes, 'BER of Each Bluetooth Node');
    % Plot the BER line for each Bluetooth node.
    [berPlots, legendStr] = deal(cell(1, numNodes));
    for plotIdx = 1:numNodes
        hold on
        berPlots{plotIdx} = plot(berAxes, curTime, 0);
        if plotIdx == 1
            legendStr{1} = ['\color[rgb]{' num2str(berPlots{plotIdx}.Color) '} Master'];
        else
            legendStr{plotIdx} = ['\color[rgb]{' num2str(berPlots{plotIdx}.Color) '} Slave-' num2str(plotIdx-1)];
        end
    end
    % Add a legend to the figure.
    legend(berAxes, legendStr,'Location','northeastoutside','Box','on');
end

% Run the simulation.
while(curTime < simulationTimeInUS)
    if txFlag
        % When the channel is busy, update the elapsed time as 0.
        elapsedTime = 0;
    else
        % When the channel is free, update the elapsed time as a minimum of
        % all of the next invoke times.
        elapsedTime = min(nextInvokeTimes(nextInvokeTimes ~= -1));
    end
    
    % Simulate the Bluetooth nodes.
    for nodeIdx = 1:numNodes
        % Push the data into the node.
        pushData(btNodes{nodeIdx}, aclPacketType, scoPacketType);
        
        % Run the Bluetooth node instance.
        nextInvokeTimes(nodeIdx) = runNode(btNodes{nodeIdx}, elapsedTime);
        
        if enableVisualization
            if ~isempty(findobj('Tag', 'BluetoothBERPlot'))
                % Update and plot the BER.
                berPlots{nodeIdx}.XData = [berPlots{nodeIdx}.XData curTime];
                if isempty(btNodes{nodeIdx}.PHY.BER)
                    ber = 0;
                else
                    ber = btNodes{nodeIdx}.PHY.BER;
                end
                berPlots{nodeIdx}.YData = [berPlots{nodeIdx}.YData ber];
            end
           if btNodes{nodeIdx}.VisualizeSpectrum
                % Plot the Bluetooth waveform.
                spectrumAnalyzer(btNodes{nodeIdx}.ChannelWaveform);
            end
        end
    end
    
    % Reset the transmission buffer index.
    txBufIdx = 0;
    
    % Reset the transmission flag to specify that the channel is free.
    txFlag = false;

    % Get the data from all of the nodes to be transmitted.
    for nodeIdx = 1:numNodes
        if (btNodes{nodeIdx}.TxBuffer.NumSamples > 0)
            txBufIdx = txBufIdx + 1;
            txBuffer{txBufIdx} = btNodes{nodeIdx}.TxBuffer;
        end
    end
    
    % Pass the transmitted data to the receiving buffers of all of the
    % nodes.
    if txBufIdx > 0
        for nodeIdx = 1:numNodes
            btNodes{nodeIdx}.RxBuffer(1:txBufIdx) = txBuffer(1:txBufIdx);
            btNodes{nodeIdx}.RxBufferIdx = txBufIdx;
        end
        % Set the transmission flag to specify the channel is busy (at
        % least one node is transmitting the waveform).
        txFlag = true;
    end
    
    % Update the current simulation time.
    curTime = curTime + elapsedTime;
end

% Get the baseband layer and channel classification statistics of each Bluetooth node in the piconet
[statisticsAtEachNode, classificationStats] = helperBluetoothFullDuplexStatistics(btNodes)