Added a switch to change between 2D and 3D MOT cases to allow for this code to be expanded and generalized in the future.

This commit is contained in:
Karthik 2021-07-11 06:18:21 +02:00
parent ba621f3c32
commit 1a8975c218

View File

@ -1,14 +1,15 @@
function ret = calculateCaptureVelocity(this, PositionVector, VelocityVector)
VelocityVector = VelocityVector./norm(VelocityVector);
switch this.SimulationMode
case "2D"
VelocityUnitVector = VelocityVector./norm(VelocityVector);
UpperLimit = 500;
LowerLimit = 0;
this.AtomicBeamCollision = false;
for Index = 1:500
InitialVelocity = 0.5 * (UpperLimit + LowerLimit) * VelocityVector;
[~, FinalDynamicalQuantities] = this.solver(PositionVector, InitialVelocity);
FinalPositionVector = FinalDynamicalQuantities(1:3);
InitialVelocity = (0.5 * (UpperLimit + LowerLimit)) * VelocityUnitVector;
ParticleDynamicalQuantities = this.solver(PositionVector, InitialVelocity);
FinalPositionVector = ParticleDynamicalQuantities(end, 1:3);
if rssq(FinalPositionVector) <= this.OvenDistance
LowerLimit = 0.5 * (UpperLimit + LowerLimit);
else
@ -16,9 +17,12 @@ function ret = calculateCaptureVelocity(this, PositionVector, VelocityVector)
end
if UpperLimit - LowerLimit < 1
ret = InitialVelocity;
ret = (0.5 * (UpperLimit + LowerLimit)) * VelocityUnitVector;
break;
end
end
case "3D"
% Development In progress
end
end