Skip to content

Commit

Permalink
added GolfPhaseSpace project
Browse files Browse the repository at this point in the history
  • Loading branch information
@BrianHaidet
BrianHaidet committed Apr 12, 2020
1 parent 9690f3e commit edd47c3
Show file tree
Hide file tree
Showing 58 changed files with 1,478 additions and 0 deletions.
138 changes: 138 additions & 0 deletions Golf_green_Verlet_simulator_phase-space/Copy_of_orbiting.m
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,138 @@
clear
close all
clc;

%time
dt=0.005;

%coeffs
Cg=200.;
Cd=0;
minSpeed=10;

name='greencone.tiff'
sanitizedname=[strrep(name,'.','') '_static_shortstep']
mkdir([sanitizedname 'outputdir'])
%green
%greenheight=double(imread('testgreen.tiff'))/256*1.7;
%greenheight=double(imread('greencone.tiff'))/256*.3;
%greenheight=double(imread('greenconeinv.tiff'))/256*.3;
greenheight=-fliplr(rot90(rot90(double(imread(name))/256*.3))).^-1 * 1000;
greenheightPLOT=max(-fliplr(rot90(rot90(double(imread(name))/256*.3))).^-1 * 500,-100);
[gradX,gradY]=gradient(greenheight(:,:,1));
forceFun=@(R,RL) (RL-R)/dt*Cd + [interp2(-gradX,R(1,:),R(2,:));interp2(-gradY,R(1,:),R(2,:))]*Cg;
%forceFun=@(R,RL) [interp2(-gradX,R(1,:),R(2,:));interp2(-gradY,R(1,:),R(2,:))];
l=size(greenheight,1);

%pin
holeLoc=size(greenheight)'/2;
holeRadius=3;

%graphics
%sph=sphere(30)
alt=30.85+20;
az=-39+20;
cmapgreen=flipud([181,228,138; 160,220,104; 124,216,87; 81,212,76; 52,188,67; 37,167,60; 32,154,61; 1,114,56; 0,86,19]/255);
stickX=[holeLoc(1),holeLoc(1)];
stickY=[holeLoc(2),holeLoc(2)];
stickZ=[greenheightPLOT(floor(holeLoc(1)),floor(holeLoc(1))),greenheightPLOT(floor(holeLoc(1)),floor(holeLoc(1)))+100];
flagX=[stickX;stickX+40];
flagY=[stickY;stickY];
flagZ=[stickZ(2),stickZ(2)-30;stickZ(2),stickZ(2)-30];
flagC=zeros(2,2,3);
flagC(:,:,1)=.9;

% balls
startLoc=[64;64]*3;

%angles=(45)*pi/180;
%speeds=400;
%angles=linspace((45-15)*pi/180,(45+15)*pi/180,10);%-pi/4;
angles=linspace((-45-20)*pi/180,(-45+20)*pi/180,5);%-pi/4;
speeds=linspace(60,80,20);

numParticles=length(speeds)*length(angles);

speeds=reshape(speeds,1,1,length(speeds));
angleVectors=[cos(angles);sin(angles)];
startConditions=bsxfun(@times,speeds,angleVectors);
startConditions=reshape(startConditions,2,length(speeds)*length(angles));

r=repmat(startLoc,1,numParticles);
rl=r-startConditions*dt;

rstart=r;
rlstart=rl;

%sequentialhits staging
%each ball laucnhes after 1/4 second (300 iters/4) = 80 ticks
launchTicks=80

i=0;
while sum(sum(r~=rl))
%energy verlet + drag
i=i+1;
rn=2*r-rl+(forceFun(r,rl))*dt^2;
rl=r;
r=rn;

%%%LAUNCH CONDITIONS
if i<50
r=rstart;
rl=rlstart;
end

%%%STOP CONDITIONS
%static friction
dr=rl-r;
s=sqrt(dr(1,:).^2+dr(2,:).^2)/dt;
%walls
haltBallsEdge=(r>l)|(r<1);
%in the hole
distToHole=bsxfun(@minus,holeLoc,rl);%use rl not r to check if the ball is in the hole so it can't escape
distToHole=sqrt(distToHole(1,:).^2+distToHole(2,:).^2);
%stop those in stop conditions
holeBalls=(distToHole<holeRadius);
rl(:,holeBalls)=repmat(holeLoc,1,sum(holeBalls));
haltBalls=(haltBallsEdge(1,:)|haltBallsEdge(2,:))|(s<minSpeed)|(distToHole<holeRadius);
r(:,haltBalls)=rl(:,haltBalls);
if mod(i,20)==0
figure(1)
surf(greenheightPLOT,'edgecolor','none')
colormap(cmapgreen)
set(gca,'DataAspectRatio',[1 1 1])
hold on
scatter3(r(1,:),r(2,:),3+greenheightPLOT(sub2ind(size(greenheightPLOT), floor(r(2,:)),floor(r(1,:)))),'wo','filled')
plot3(stickX,stickY,stickZ,'k-')
surf(flagX,flagY,flagZ,flagC,'edgecolor','none')
hold off
%az=az+.1;
view([az alt])
% imshow(greenheight,[min(min(greenheight)),max(max(greenheight))],'colormap',colormap('parula'))
hold on
%rectangle('Position',[holeLoc(1)-holeRadius holeLoc(2)-holeRadius holeRadius*2 holeRadius*2],'Curvature',[1,1],'FaceColor','black');
%scatter(r(1,:),r(2,:),'w.')
hold off
xlim([0,512])
ylim([0,512])
zlim([-100,0])
%title(num2str(s(1)))
set(gcf, 'Units', 'Normalized', 'OuterPosition', [0, 0.04, 1, 0.96]);
saveas(gcf,[[sanitizedname 'outputdir'] '\' num2str(i/20,'%05i') '.png'])
end
end

%phase space

figure(2)
maxDist=256;
maxColor=reshape([1,0,0],1,1,3);
minColor=reshape([1,1,1],1,1,3);
dists=fliplr(flipud(reshape(distToHole,length(angles),length(speeds))'));
imshow(dists,[holeRadius,maxDist],'colormap',colormap('jet'))
hold on
h=imshow(repmat(ones(size(dists)),1,1,3));
hold off
set(h,'AlphaData',dists<holeRadius);
ylabel('Putt speed')
xlabel('putt angle')
136 changes: 136 additions & 0 deletions Golf_green_Verlet_simulator_phase-space/bankExample.m
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,136 @@
clear
close all
clc;

%time
dt=0.01;

%coeffs
Cg=50.;
Cd=1;
minSpeed=10;

name='greenbankexample.png.tiff'
sanitizedname=[strrep(name,'.','') '_introsequential_1Mgrid']
mkdir([sanitizedname 'outputdir'])
%green
%greenheight=double(imread('testgreen.tiff'))/256*1.7;
%greenheight=double(imread('greencone.tiff'))/256*.3;
%greenheight=double(imread('greenconeinv.tiff'))/256*.3;
greenheight=fliplr(rot90(rot90(double(imread(name))/256*.3)));
[gradX,gradY]=gradient(greenheight(:,:,1));
forceFun=@(R,RL) (RL-R)/dt*Cd + [interp2(-gradX,R(1,:),R(2,:));interp2(-gradY,R(1,:),R(2,:))]*Cg;
%forceFun=@(R,RL) [interp2(-gradX,R(1,:),R(2,:));interp2(-gradY,R(1,:),R(2,:))];
l=size(greenheight,1);

%pin
holeLoc=size(greenheight)'/2;
holeRadius=6;

%graphics
%sph=sphere(30)
alt=30.85;
az=-39;
cmapgreen=flipud([181,228,138; 160,220,104; 124,216,87; 81,212,76; 52,188,67; 37,167,60; 32,154,61; 1,114,56; 0,86,19]/255);
stickX=[holeLoc(1),holeLoc(1)];
stickY=[holeLoc(2),holeLoc(2)];
stickZ=[greenheight(floor(holeLoc(1)),floor(holeLoc(1))),greenheight(floor(holeLoc(1)),floor(holeLoc(1)))+100];
flagX=[stickX;stickX+40];
flagY=[stickY;stickY];
flagZ=[stickZ(2),stickZ(2)-30;stickZ(2),stickZ(2)-30];
flagC=zeros(2,2,3);
flagC(:,:,1)=.9;

% balls
startLoc=[64;64];

% angles=(45)*pi/180;
% speeds=400;
angles=linspace((45+15)*pi/180,(45-15)*pi/180,10);%-pi/4;
speeds=linspace(500,200,10);

numParticles=length(speeds)*length(angles);

speeds=reshape(speeds,1,1,length(speeds));
angleVectors=[cos(angles);sin(angles)];
startConditions=bsxfun(@times,speeds,angleVectors);
startConditions=reshape(startConditions,2,length(speeds)*length(angles));

r=repmat(startLoc,1,numParticles);
rl=r-startConditions*dt;

rstart=r;
rlstart=rl;

%sequentialhits staging
%each ball laucnhes after 1/4 second (300 iters/4) = 80 ticks
launchTicks=80

i=0;
while sum(sum(r~=rl))
%energy verlet + drag
i=i+1;
rn=2*r-rl+(forceFun(r,rl))*dt^2;
rl=r;
r=rn;

%%%LAUNCH CONDITIONS
if i<100
r=rstart;
rl=rlstart;
end

%%%STOP CONDITIONS
%static friction
dr=rl-r;
s=sqrt(dr(1,:).^2+dr(2,:).^2)/dt;
%walls
haltBallsEdge=(r>l)|(r<1);
%in the hole
distToHole=bsxfun(@minus,holeLoc,rl);%use rl not r to check if the ball is in the hole so it can't escape
distToHole=sqrt(distToHole(1,:).^2+distToHole(2,:).^2);
%stop those in stop conditions
holeBalls=(distToHole<holeRadius);
rl(:,holeBalls)=repmat(holeLoc,1,sum(holeBalls));
haltBalls=(haltBallsEdge(1,:)|haltBallsEdge(2,:))|(s<minSpeed)|(distToHole<holeRadius);
r(:,haltBalls)=rl(:,haltBalls);

if mod(i,10)==0
figure(1)
surf(greenheight,'edgecolor','none')
colormap(cmapgreen)
set(gca,'DataAspectRatio',[1 1 1])
hold on
scatter3(r(1,:),r(2,:),3+greenheight(sub2ind(size(greenheight), floor(r(2,:)),floor(r(1,:)))),'wo','filled')
plot3(stickX,stickY,stickZ,'k-')
surf(flagX,flagY,flagZ,flagC,'edgecolor','none')
hold off
%az=az+0.2;
view([az alt])
% imshow(greenheight,[min(min(greenheight)),max(max(greenheight))],'colormap',colormap('parula'))
hold on
%rectangle('Position',[holeLoc(1)-holeRadius holeLoc(2)-holeRadius holeRadius*2 holeRadius*2],'Curvature',[1,1],'FaceColor','black');
%scatter(r(1,:),r(2,:),'w.')
hold off
xlim([0,512])
ylim([0,512])
%title(num2str(s(1)))
set(gcf, 'Units', 'Normalized', 'OuterPosition', [0, 0.04, 1, 0.96]);
%saveas(gcf,[[sanitizedname 'outputdir'] '\' num2str(i/10,'%05i') '.png'])
end
end

%phase space

figure(2)
maxDist=256;
maxColor=reshape([1,0,0],1,1,3);
minColor=reshape([1,1,1],1,1,3);
dists=fliplr(flipud(reshape(distToHole,length(angles),length(speeds))'));
imshow(dists,[holeRadius,maxDist],'colormap',colormap('jet'))
hold on
h=imshow(repmat(ones(size(dists)),1,1,3));
hold off
set(h,'AlphaData',dists<holeRadius);
ylabel('Putt speed')
xlabel('putt angle')
Binary file added Golf_green_Verlet_simulator_phase-space/bankexample.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added Golf_green_Verlet_simulator_phase-space/bimodal phase.png
View file
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading

0 comments on commit edd47c3

Please sign in to comment.