Learn PSO by racing the flock for food

Mohammad Aryayi
/
Learn PSO by racing the flock for food
about 3 hours ago
0
4
1
1
9993
game.m
% Flock Hunt
% Copyright (c) 2026 Mohammad Aryayi
function g=game
g.init=@in;g.update=@up;g.render=@rn;
end
function s=in(A)
set(A,'XLim',[0 160],'YLim',[0 120],'CLim',[0 1],'Colormap',(.55:.01:1)'.^[2 1 4],'DefaultPatchEdgeColor','none','DefaultTextFontUnits','normalized','DefaultTextHorizontalAlignment','center','DefaultTextFontWeight','bold','DefaultTextColor','w');
x=[0 160 160 0];patch(x,[0 0 120 120],[.1 .28 .26],'Parent',A);patch(x,[100 100 120 120],[.06 .18 .17],'Parent',A);
s.G=hgtransform('Parent',A);G=s.G;[x,y]=meshgrid(0:2:160,0:2:100);s.gx=[x(:)';y(:)'];s.gs=size(x);
s.im=image('Parent',G,'XData',[0 160],'YData',[0 100],'CData',zeros(s.gs),'CDataMapping','scaled','AlphaData',.85);
s.sd=patch('Parent',G,'Vertices',[0 0],'Faces',1,'FaceVertexCData',0,'FaceColor','none','Marker','o','MarkerFaceColor','flat');
s.gl=line(0,0,'Parent',G,'Color',[1 .85 .2],'LineWidth',2);
for i=2:8
s.B(i)=hgtransform('Parent',G);dr(s.B(i),"偤傠储傼俹俚信俚倔倥倔倃偤傠储傼倣倈俲倊倗倗偎偊偤傠储傼倣倈俲倊們們俚俞值僠僥僯倔倔倸値值僠僥僯倴倔倢倡偐儓僆债偀偐偀倚倔倎");
end
s.B(1)=hgtransform('Parent',G);dr(s.B(1),"偤僭偔假修係俌係倔倬倔俼偤僭偔假倪倃俤倆候候偦偡偤僭偔假倪倃俤倆倏倏係俇值儓傇偠倔倔偆倫值儓傇偠偂倔倨倦偐儓僆债偒偨偒倜倔倌");
s.pb=line(0,0,.55,'Parent',G,'Color',[.34 .71 .91],'LineWidth',2,'Marker','o','MarkerSize',14);c=[.9 .62 0];s.gb=line(0,0,.55,'Parent',G,'Color',c,'Marker','p','MarkerSize',16,'MarkerFaceColor',c);
s.lb=[tx(G,0,0,.024) tx(G,0,0,.024)];set(s.lb,{'Color','String'},{[.34 .71 .91],"my best";c,"flock's best"});
s.ol=line(0,0,'Parent',G,'Color',[.02 .02 .08],'LineWidth',5.5);c={[.75 .75 .75],[.34 .71 .91],c};
for i=1:3
s.ar(i)=line(0,0,'Parent',G,'Color',c{i},'LineWidth',3);
end
s.gm=line(0,0,.55,'Parent',G,'Color','w','LineWidth',3,'Marker','d','MarkerSize',13);
c=[1 .85 .2];s.pt=line(0,0,'Parent',G,'Color',c,'LineWidth',2,'Marker','o','MarkerIndices',2,'MarkerFaceColor',c);Z=["值偿偙倸倔俺偰倬值儆傠傲俘倡倧値值僭偿傔俱倌倧値值儆傠傲倊們倧値值僭偿傔倣倧倧値值儆傠傲值倥倧値值儆傠傲偐候倧倥值倮倮倮偗倞倘倘","值傔偨债倔倔偂倾值偠偄倨倄倞倠倞值偠偄倨倦倈倢借值偠偄倨倚倨倛倚偤偛偂倨們倗倝倗债债偒偒偐傇傔偔倚倸倮偄偎倾值僭僴儆俴偢倝倘值倮倮倮俴停候倘值僭僴儆倰偮倝倘值倮倮倮倰偨候倘"];
for i=1:3
s.fd(i)=hgtransform('Parent',G);dr(s.fd(i),Z(1+(i>1)));s.ml(i)=tx(G,0,0,.028);
end
set(s.ml,{'String'},{"FOOD";"TRAP";"TRAP"},'Color','k');
s.f=[tx(A,80,114,.034) tx(A,3,105,.027) tx(A,157,105,.027)];set(s.f(2:3),{'HorizontalAlignment','Color'},{'left','w';'right',[1 .85 .2]});
s.tp=tx(A,80,9,.026);s.tp.BackgroundColor='k';s.ot=[tx(A,80,110,.09) tx(A,80,88,.04) tx(A,80,66,.03) tx(A,80,38,.03) tx(A,80,14,.04)];set(s.ot([1 5]),{'Color'},{[1 .8 .35];[.6 .85 1]});
N=newline;s.fm="PSO says = \color[rgb]{.75 .75 .75}keep going\color{white} + \color[rgb]{.34 .71 .91}my best spot\color{white} + \color{orange}flock's best spot";
set(s.ot,{'String'},{"Flock Hunt";"Find the food before the other birds!";"See how PSO (Particle Swarm Optimization) finds the best spot."+N+"Every bird adds 3 arrows:"+N+s.fm;"TAP: choose direction. TAP again: choose speed."+N+"Bright dots = strong smell = near the food.";"TAP TO START"});
s.tt=["Birds share the flock's best spot (the star). This makes PSO fast.","Well done! You searched smarter than the flock.","The flock is stuck at a TRAP (a local best): a weak point of PSO."+N+"Random numbers help PSO explore, but not always."+N+"TIP: when the flock is stuck, do not follow it. Search other places!","","First the flock was stuck at a TRAP: a weak point of PSO."+N+"At last a bird found the food!"+N];
[s.T,s.w,s.bs,s.e,s.sp]=deal(0);s=rs(s);s.st=0;
rn(0,s);
end
function f=sm(C,P)
f=max([1;.8;.8].*exp(-((C(1,:)'-P(1,:)).^2+(C(2,:)'-P(2,:)).^2)./[6050;800;800]),[],1);
end
function [V,Q,g]=ps(s)
[~,j]=max(s.F);g=s.P(:,j);Q=cat(3,.8*s.V,1.5*s.r(1,:).*(s.P-s.X),1.5*s.r(2,:).*(g-s.X));k=min(1,14./(vecnorm(sum(Q,3))+eps));Q=Q.*k;V=sum(Q,3);
end
function s=rs(s)
C=zeros(2,3);
while min(vecnorm(C(:,2:3)-C(:,1)))<35
C=[20;25]+[130;67].*rand(2,3);
end
X=C;
while min(vecnorm(X-C(:,1)))<50
X=5+[150;90].*rand(2,8);
end
a=2*pi*rand(1,8);f=sm(C,X);
[s.C,s.X,s.X0,s.V,s.P,s.F,s.D,s.r,s.h,s.n,s.st,s.t0,s.an,s.fb,s.ls,s.wa]=deal(C,X,X,(7+7*rand(1,8)).*[cos(a);sin(a)],X,f,[X;f],rand(2,8),false(1,8),0,1,s.T,a(1),max(f),0,0);
end
function s=up(s,d,p)
s.T=s.T+d;t=s.T;e=t-s.t0;st=s.st;
if p&&(st==0||st==4&&e>1)
s.w=s.w*(st==4&s.e==2);s=rs(s);
elseif p&&st==1
[s.st,s.t0]=deal(2,t);
elseif p&&st==2
V=ps(s);V(:,1)=s.sp*[cos(s.an);sin(s.an)];q=s.C(:,1)-s.X;u=min(max(sum(q.*V)./(sum(V.^2)+eps),0),1);h=vecnorm(q-u.*V)<4;X=min(max(s.X+V,0),[160;100]);
[s.X0,s.X,s.V,s.h,s.st,s.t0,s.n]=deal(s.X,X+h.*(s.X+max(u-5./(vecnorm(V)+eps),0).*V-X),V,h,3,t,s.n+1);
elseif st==3&&e>.7
f=sm(s.C,s.X);k=f>s.F;s.P(:,k)=s.X(:,k);s.F(k)=f(k);[s.D,s.r,s.st,s.t0]=deal([s.D [s.X;f]],rand(2,8),1,t);
b=max(s.F);
if b>s.fb+.01
[s.fb,s.ls]=deal(b,s.n);
end
s.wa=s.wa|s.n-s.ls>4;
if any(s.h)
s.e=1+s.h(1);s.w=s.w+s.h(1);s.bs=max(s.bs,s.w);s.st=4;
elseif s.n-s.ls>14
[s.e,s.st]=deal(3,4);
end
end
s.an=s.an+(s.st==1)*4*d;
if s.st==2
s.sp=7-7*cos(2.2*(t-s.t0));
end
end
function rn(~,s)
t=s.T;st=s.st;e=t-s.t0;a=st==1|st==2;m=st==4;
s.G.Visible=st>0;set(s.ot,'Visible',st==0);set(s.f,'Visible',st>0);
X=s.X+(s.X0-s.X)*(st==3)*(1-min(1,e/.7));[~,Q,g]=ps(s);p=X(:,1);
for i=1:8
b=atan2(s.V(2,i),s.V(1,i));
if i==1&&a
b=s.an;
end
s.B(i).Matrix=makehgtform('translate',[X(:,i);.3+.1*(i==1)],'zrotate',b,'scale',[1 1+.3*sin(25*t)*(st==3) 1]);
end
c=cumsum([p squeeze(Q(:,1,:))],2);O=[];
for i=1:3
O=[O ar(s.ar(i),c(:,i),c(:,i+1)) [nan;nan]];
end
set(s.ol,'XData',O(1,:),'YData',O(2,:));set(s.gm,'XData',c(1,4),'YData',c(2,4));q=[cos(s.an);sin(s.an)];w=[0 14-(st==2)*(14-s.sp) 14];
set(s.pt,'XData',p(1)+w*q(1),'YData',p(2)+w*q(2),'ZData',[.5 .5 .5]);b=linspace(0,2*pi,40);r=4.5+.8*sin(6*t);set(s.gl,'XData',p(1)+r*cos(b),'YData',p(2)+r*sin(b));set([s.ar s.ol s.gm s.pt],'Visible',a);
set(s.pb,'XData',s.P(1,1),'YData',s.P(2,1));set(s.gb,'XData',g(1),'YData',g(2));set(s.lb,{'Position'},{[s.P(:,1)'-[0 6.5] 1];[g'+[0 5.5] 1]});
set(s.sd,'Vertices',s.D(1:2,:)','Faces',1:size(s.D,2),'FaceVertexCData',s.D(3,:)');
s.im.CData=reshape(sm(s.C,s.gx),s.gs);
for i=1:3
s.fd(i).Matrix=makehgtform('translate',[s.C(:,i);.7]);
end
set(s.ml,{'Position'},num2cell([s.C'-[0 6] [1;1;1]],2));set([s.im s.fd s.ml s.tp],'Visible',m);set(s.lb,'Visible',~m);s.tp.String=s.tt(4+(s.wa&s.e<3))+s.tt(max(s.e,1));
k=sm(s.C,p);g=s.n-s.ls>4&~m;h=["" "TAP: choose direction" "TAP: choose speed" "THE FLOCK FOUND IT FIRST!" "YOU FOUND THE FOOD!" "OUT OF STEPS!" "TAP: play again" "TAP: next round"];
u="WON "+s.w+"   BEST "+s.bs+"   STEP";
if m
s.f(1).String=h(3+s.e);s.f(2).String=u+"S "+s.n;s.f(3).String=h(1+(e>1)*(6+(s.e==2)));
else
o=[u+" "+(s.n+1),"THE FLOCK IS STUCK!   STEPS LEFT "+(15-s.n+s.ls)];s.f(1).String=s.fm;s.f(2).String=o(1+g)+"   SMELL "+round(100*k)+"%";s.f(3).String=h(1+st*(st<3));
end
s.f(2).Color=[1 1 1]-g*[0 .5 .7];
end
function P=ar(h,a,b)
v=b-a;u=2*v/(norm(v)+eps);n=.8*[-u(2);u(1)];P=[a b b-u+n b b-u-n];set(h,'XData',P(1,:),'YData',P(2,:),'ZData',.5+0*P(1,:));
end
function dr(h,z)
r=double(char(z));d=r/20-1025;i=1;a=0:.16:6.4;
while i<numel(r)
k=d(i);n=max(4,2*k);v=d(i+4:i+3+n);x=v(1:k);y=v(k+1:2*k);
if k==2
x=v(1)+v(3)*cos(a);y=v(2)+v(4)*sin(a);
end
patch(x,y,(r(i+1:i+3)-20500)/255,'Parent',h);i=i+4+n;
end
end
function h=tx(A,x,y,z)
h=text(x,y,2,"",'Parent',A,'FontSize',z);
end
Play Game
Preview

Play Game opens this in MATLAB Drive. Click Run this app there to play.