calculate_behaviors.m

function success = calculate_behaviors(folder_name)
% given tracks that has embeddings, classify every time point into a
% behavior
    addpath(genpath(pwd))
    %set up parameters
    parameters = load_parameters(folder_name);
    if parameters.TrackOnly
        success = true;
        return
    end
    load('reference_embedding.mat')
    number_of_behaviors = max(L(:)-1);
    relevant_track_fields = {'Embeddings'};

    %% Load tracks
    Tracks = load_single_folder(folder_name, relevant_track_fields);
    if isempty(Tracks)
        error('Empty Tracks');
    end


    %get the stereotyped behaviors
    Tracks = find_stereotyped_behaviors(Tracks, L, xx);

    % Get binary array of when certain behaviors start
    Tracks(1).Behaviors = [];

    for track_index = 1:length(Tracks)
        triggers = false(number_of_behaviors, size(Tracks(track_index).Embeddings,1)); %a binary array of when behaviors occur
        for behavior_index = 1:number_of_behaviors
            transition_indecies = Tracks(track_index).BehavioralTransition(:,1) == behavior_index;
            %transition into of
            transition_start_frames = Tracks(track_index).BehavioralTransition(transition_indecies,2);
            triggers(behavior_index,transition_start_frames) = true;
%                 %transition out of
%                 transition_end_frames = Tracks(track_index).BehavioralTransition(transition_indecies,3);
%                 triggers(behavior_index,transition_end_frames) = true;
        end
        Tracks(track_index).Behaviors = triggers(:,1:size(Tracks(track_index).Embeddings,1));
    end

    %save
    savetracks(Tracks, folder_name);
    success = true;    
 end