Philippe 02/10/2024: Philippe 02/10/2024: lagrangian trajectories: Z_TRAJ...

Philippe 02/10/2024: Philippe 02/10/2024: lagrangian trajectories: Z_TRAJ initial positions were not initialized correctly + small optimization

(cherry picked from commit 04b475b7)

src/MNH/compute_r00.f90

@@ -281,21 +281,17 @@ ZX00(:,:,:)=XSVT(:,:,:,NSV_LGBEG)*1.E-3   ! ZX0 in km
 ZY00(:,:,:)=XSVT(:,:,:,NSV_LGBEG+1)*1.E-3 ! ZY0 in km
 ZZ00(:,:,:)=XSVT(:,:,:,NSV_LGEND)*1.E-3   ! ZZ0 in km
 !
-DO JJ = 1, IJU
-  DO JI = 1, IIU-1
-    ZXI(JI,JJ,:) = 0.5 * ( XXHAT(JI) + XXHAT(JI+1) )
-  END DO
-  ZXI(IIU,JJ,:) = 2. * ZXI(IIU-1,JJ,:) - ZXI(IIU-2,JJ,:)
-END DO
-ZXI(:,:,:) = ZXI(:,:,:) * 1.E-3
+ZXI(1:IIU-1,1,1) = 0.5 * ( XXHAT(1:IIU-1) + XXHAT(2:IIU) )
+ZXI(IIU,1,1) = 2. * ZXI(IIU-1,1,1) - ZXI(IIU-2,1,1)
+ZXI(:,:,:) = SPREAD( SPREAD( ZXI(:,1,1), DIM=2, NCOPIES=IJU ), DIM=3, NCOPIES=IKU) * 1.E-3
 !
-DO JI=1,IIU
-  DO JJ=1,IJU-1
-    ZYI(JI,JJ,:)= 0.5 * ( XYHAT(JJ) + XYHAT(JJ+1) )
-  END DO
-  ZYI(JI,IJU,:) = 2. * ZYI(JI,IJU-1,:) - ZYI(JI,IJU-2,:)
-END DO
-ZYI(:,:,:) = ZYI(:,:,:) * 1.E-3
+ZYI(1,1:IJU-1,1) = 0.5 * ( XYHAT(1:IJU-1) + XYHAT(2:IJU) )
+ZYI(1,IJU,1) = 2. * ZYI(1,IJU-1,1) - ZYI(1,IJU-2,1)
+ZYI(:,:,:) = SPREAD( SPREAD( ZYI(1,:,1), DIM=1, NCOPIES=IIU ), DIM=3, NCOPIES=IKU) * 1.E-3
+!
+ZZI(1,1,1:IKU-1) = 0.5 * ( XZHAT(1:IKU-1) + XZHAT(2:IKU) )
+ZZI(1,1,IKU) = 2. * ZZI(1,1,IKU-1) - ZZI(1,1,IKU-2)
+ZZI(:,:,:) = SPREAD( SPREAD( ZZI(1,1,:), DIM=1, NCOPIES=IIU ), DIM=2, NCOPIES=IJU) * 1.E-3
 !
 IF (L2D) THEN
   WHERE ( ZX00<ZXOR .OR. ZX00>ZXMAX .OR. &