main.m

clc;
clear;
close all;

%% Problem Definition
P = load('Point.mat');

model=CreateModel(P.X);

CostFunction=@(tour) TourCost(tour,model);

nVar=model.n;


%% ACO Parameters

MaxIt=350;      % Maximum Number of Iterations

nAnt=50;       % Number of Ants (Population Size)

Q=1;

tau0=10*Q/(nVar*mean(model.D(:)));	% Initial Phromone

alpha=1;        % Phromone Exponential Weight
beta=1;         % Heuristic Exponential Weight

rho=0.05;       % Evaporation Rate


%% Initialization

eta=1./model.D;             % Heuristic Information Matrix, 1/distance

tau=tau0*ones(nVar,nVar);   % Phromone Matrix

BestCost=zeros(MaxIt,1);    % Array to Hold Best Cost Values

% Empty Ant
empty_ant.Tour=[];
empty_ant.Cost=[];

% Ant Colony Matrix
ant=repmat(empty_ant,nAnt,1);

% Best Ant
BestSol.Cost=inf;

% Read the 3D data
stl = ReadSTLModel1();
% stl = 'model1.stl';
writeObjPath = VideoWriter('PathVideo.avi');
open(writeObjPath);
% figCost = figure(2);
writeObjCost = VideoWriter('CostVideo.avi');
open(writeObjCost);

%% ACO Main Loop

for it=1:MaxIt
    
    % Move Ants
    for k=1:nAnt
        
        ant(k).Tour=randi([1 nVar]); % returns 1 random value between 1 and no. of cities}
        ant(k).Tour = 1;
        for l=2:nVar
            
            i=ant(k).Tour(end);
            
            P=tau(i,:).^alpha.*eta(i,:).^beta; % the probability of next point
            
            P(ant(k).Tour)=0;
            
            P=P/sum(P);
            
            j=RouletteWheelSelection(P);
            
            ant(k).Tour=[ant(k).Tour j];
            
        end
        
        ant(k).Cost=CostFunction(ant(k).Tour);
        
        if ant(k).Cost<BestSol.Cost
            BestSol=ant(k);
        end
        
    end
    
    % Update Phromones
    for k=1:nAnt
        
        tour=ant(k).Tour;
        
        tour=[tour tour(1)]; %#ok
        
        for l=1:nVar
            
            i=tour(l);
            j=tour(l+1);
            
            tau(i,j)=tau(i,j)+Q/ant(k).Cost;
            
        end
        
    end
    
    % Evaporation
    tau=(1-rho)*tau;
    
    % Store Best Cost
    BestCost(it)=BestSol.Cost;
    
    % Show Iteration Information
    disp(['Iteration ' num2str(it) ': Best Cost = ' num2str(BestCost(it))]);
    
    figPath = figure(1);

    PlotSTL(stl);

    hold on;
    PlotSolution(BestSol.Tour,model);
    f1 = getframe(figPath);
    % imshow(f1.cdata);
    writeVideo(writeObjPath,f1)
    hold off;

    fig2 = figure(2);
    plot(BestCost,'LineWidth',2);
    xlabel('Iteration');
    ylabel('Best Cost');
    grid on;
    f2 = getframe(fig2);
    writeVideo(writeObjCost,f2);
end
close(writeObjPath);
close(writeObjCost);

%% Results
% PlotSTL(stl);
% PlotSolution(BestSol.Tour,model);
% % f1 = getframe(figPath);s
% % writeVideo(writeObjPath,f1)
% hold off;

% figure(2);
% plot(BestCost,'LineWidth',2);
% xlabel('Iteration');
% ylabel('Best Cost');
% grid on;
% f2 = getframe(figCost);
% writeVideoScenes(writeObjCost,f2)