% Copyright (c) 2023 SpinCore Technologies, Inc.
% http://www.spincore.com
% 
% This software is provided 'as-is', without any express or implied warranty. 
% In no event will the authors be held liable for any damages arising from the 
% use of this software.
% 
% Permission is granted to anyone to use this software for any purpose, 
% including commercial applications, and to alter it and redistribute it
% freely, subject to the following restrictions:
% 
% 1. The origin of this software must not be misrepresented; you must not
% claim that you wrote the original software. If you use this software in a
% product, an acknowledgment in the product documentation would be appreciated
% but is not required.
% 2. Altered source versions must be plainly marked as such, and must not be
% misrepresented as being the original software.
% 3. This notice may not be removed or altered from any source distribution.
%

%
% \file pbesr_pro_loop_test.m adapted from pbesr_pro_loop_test.c
% \brief PulseBlasterESR-PRO example program
% 
% The following program tests the functionality of the loop instruction.
% It takes two optional command-line arguments.
% The first is the clock frequency, the default is 400MHz.  
% The second is the number of loops, the default is 3 loops.  
% 
% BNCs 0-3 will output a pulse train with period 100ns. The number of pulses
% in this train will be determined by the number of loops used. This pulse
% train will be followed by ground level for 400ns, after which the entire
% program will repeat.
%
% \ingroup ESRPRO
% 
clc; clear;

global SPINAPI_DLL_NAME

SPINAPI_DLL_NAME = 'spinapi64';

addpath('../MATLAB_SpinAPI'); % MATLAB_SpinAPI Folder path address

SpinAPI_Drive = 'C:'; % Modify to the hard drive letter that contains SpinAPI

% Loads spinapi64 library and headers
% All files are located in SpinAPI folder
if ~libisloaded('spinapi64')  % Needs the MinGW-w64 add-on
    loadlibrary([SpinAPI_Drive,'\SpinCore\SpinAPI\lib\spinapi64.dll'], ...
                 [SpinAPI_Drive,'\SpinCore\SpinAPI\include\spinapi.h'], 'addheader',...
                 [SpinAPI_Drive,'\SpinCore\SpinAPI\include\pulseblaster.h'], 'addheader',...
                 [SpinAPI_Drive,'\SpinCore\SpinAPI\include\radioprocessor.h'], 'addheader',...
                 [SpinAPI_Drive,'\SpinCore\SpinAPI\include\dds.h']);
end

% Load constants to the workspace as variables
load('CONST.mat');

argv = [0];
argc = arg(argv);
clock_freq = 500;
NUM_LOOPS = 3;

% Uncommenting the line below will generate a debug log in your current
% directory that can help debug any problems that you may be experiencing   
% pb_set_debug(1);

fprintf('Copyright (c) 2023 SpinCore Technologies, Inc.\n\n');

fprintf('Using SpinAPI library version %s \n', pb_get_version());


numBoards = detect_boards(); % If there is more than one board in the system, have the user specify.
if numBoards > 1             % Request the board number to use from the user
    select_boards(numBoards)
end

if pb_init() ~= 0
  error('Error initializing board: %s\n', pb_get_error());
end


% Set the clock frequency.
if length(argv) > 1
    clock_freq =(argv(1));
else
    % User input clock
    clock_freq = input_clock();
    fprintf('\n');
end
 	
if (length(argv) > 2)
    num_loops = (argv(2));
else
    fprintf('Please enter the number of loops for the loop instruction.\n');
    num_loops = input('Number of loops: ');
    fprintf('\n')
end 

fprintf('Clock frequency: %.2fMHz\n\n', clock_freq);
fprintf('BNCs 0-3 will output %d successive pulses, each with a period of 100ns, followed by ground level for 400ns and then the program will repeat\n\n\n',num_loops);

% Tell driver what clock frequency the board uses
pb_core_clock(clock_freq);

% Begin pulse program
pb_start_programming(PULSE_PROGRAM);

% Instruction format
% int pb_inst(int flags, int inst, int inst_data, int length)

% Instruction 0 - continue to inst 1 in 200ns
start = pb_inst_pbonly(0, CONTINUE, 0, 200.0 * ns);

% Instruction 1 - Beginning of Loop (Loop num_loop times). Continue to
% next instruction in 50ns
loop = pb_inst_pbonly(bitor(hex2dec(ON),hex2dec('F')), LOOP, num_loops, 50.0 * ns);

% Instruction 2 - End of Loop.  Return to beginning of loop (instruction 1)
% or continue to next instruction in 50ns
pb_inst_pbonly(0, END_LOOP, loop, 50.0 * ns);

% Instruction 3 - branch to start
pb_inst_pbonly(0, BRANCH, start, 200.0 * ns);

% End of programming registers and pulse program
pb_stop_programming();

%Trigger pulse program
pb_reset();
pb_start();

pb_close();

fprintf('Press enter to continue...\n')
pause;

% Unload library and free memory
unloadlibrary('spinapi64');


function argc = arg(varargin)
    argc = nargin;
end

function numBoards = detect_boards()
    numBoards = pb_count_boards();
    if numBoards < 1
        error(['No Boards were detected in your system. Verify that the board is', ...
            ' firmly secured in the PCI slot.']);
    end
end

function select_boards(numBoards)
    while true
        fprintf('Found %d boards in your system. ', numBoards);
        fprintf('Which board should be used? (0-%d): ', (numBoards-1));
        choice = input('');
        if choice < 0 || choice > numBoards-1
            fprintf('Invalid Board Number (%d).\n', choice);
        else 
            pb_select_board(choice);
            fprintf('Board %d selected.\n',choice);
            break
        end 
    end 
end 

function clock = input_clock()
    clock = input('\nPlease enter internal clock frequency (MHz): ');
end