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