Changed # of shuffles & penalty weight

1 files changed, 3 insertions, 5 deletions

diff --git a/fitness.m b/fitness.m
index 00cbada..3516ac7 100644
--- a/fitness.m
+++ b/fitness.m
@@ -8,13 +8,11 @@ function [Energy, Waist, Penalty] = fitness( q_0, q_final, x_final, optics_posit
     [Waist_trial_forward, Radius_trial_forward] = q2wr(q_f_trial_forward, lambda);

     q_f_trial_backward = gbeam_propagation(x,q_final,x_final,optics_placer(optics_positions, optics_focal_length));

     [Waist_trial_backward, Radius_trial_backward] = q2wr(q_f_trial_backward, lambda);

-      

+

     Energy = 0;

     Penalty_waist_mismatch = sum(abs((Waist_trial_forward-Waist_trial_backward)./min(Waist_trial_forward, Waist_trial_backward)))/Np;

-    Penalty_waist_mismatch = 0*Penalty_waist_mismatch + 1e9*sum(abs((Waist_trial_forward-Waist_trial_backward)))/Np;

-    Penalty_radius_mismatch = sum(abs((Radius_trial_forward-Radius_trial_backward)./max(Radius_trial_forward, Radius_trial_backward)))/Np;

-    

-    Energy = 0*Penalty_radius_mismatch + Penalty_waist_mismatch;

+    Penalty_waist_mismatch

+    Energy = 1e-2*Penalty_waist_mismatch;

     

     % penalty calculation

     % do not put lenses too close to each other and end positions