clear

if(isunix)
    executable = '"../../build/bin/FeatureExtraction"';
else
    executable = '"../../x64/Release/FeatureExtraction.exe"';
end

output = './output_features_vid/';

if(~exist(output, 'file'))
    mkdir(output)
end
    
in_files = dir('../../videos/2015-10-15-15-14.avi');
% some parameters
verbose = true;

command = executable;
command = cat(2, command, ' -verbose  -no2Dfp -no3Dfp -noMparams -noPose -noAUs ');

% add all videos to single argument list (so as not to load the model anew
% for every video)
for i=1:numel(in_files)
    
    inputFile = ['../../videos/', in_files(i).name];
    [~, name, ~] = fileparts(inputFile);
    
    % where to output tracking results
    outputFile_gaze = [output name '_gaze.txt'];
    
    if(~exist([output name], 'file'))
        mkdir([output name]);
    end
    
    command = cat(2, command, ['-fx 700 -fy 700 -f "' inputFile '" -of "' outputFile_gaze '"']);
               
end

if(isunix)
    unix(command);
else
    dos(command);
end

%% Demonstrating reading the output files
filename = [output name];

% Read gaze (x,y,z) for one eye and (x,y,z) for another
gaze  = dlmread([filename, '_gaze.txt'], ',', 1, 0);

% This indicates which frames were succesfully tracked
valid_frames = gaze(:,4);

% only picking left, right and up down views for visualisation

% These are gaze angles with respect to head pose
gaze_angle = gaze(:,[11,12]);

figure;
plot(smooth(gaze_angle(:,1)), 'DisplayName', 'Left - right');
hold on;
plot(smooth(gaze_angle(:,2)), 'DisplayName', 'Up - down');
xlabel('Frame') % x-axis label
ylabel('Radians') % y-axis label
legend('show');
hold off;

% These are gaze direction vectors
gaze = gaze(:,[5,6,7,8,9,10]);

gaze = (gaze(:,[1,2,3]) + gaze(:,[4,5,6]))/2;
gaze(:,1) = smooth(gaze(:,1));
gaze(:,2) = smooth(gaze(:,2));
gaze(:,3) = smooth(gaze(:,3));

figure;
plot(gaze(:,1), 'DisplayName', 'Left - right');
hold on;
plot(gaze(:,2), 'DisplayName', 'Up - down');
xlabel('Frame') % x-axis label
ylabel('Gaze vector size') % y-axis label
legend('show');
hold off;