Issue#13

Fixes issue #13 (closed)

mbtrack2/tracking/element.py

@@ -444,6 +444,10 @@ def transverse_map_sector_generator(ring, positions):
     """
     N_sec = len(positions)
     sectors = []
+    if hasattr(ring, "adts") and ring.adts is not None:
+        adts = np.array([val / N_sec for val in ring.adts])
+    else:
+        adts = None
     if ring.optics.use_local_values:
         for i in range(N_sec):
             sectors.append(
@@ -455,9 +459,8 @@ def transverse_map_sector_generator(ring, positions):
                                     ring.optics.local_beta,
                                     ring.optics.local_dispersion,
                                     2 * np.pi * ring.tune / N_sec,
-                                    ring.chro / N_sec,
-                                    adts=ring.adts /
-                                    N_sec if ring.adts else None))
+                                    np.asarray(ring.chro) / N_sec,
+                                    adts=adts))
     else:
         import at
 
@@ -496,15 +499,14 @@ def transverse_map_sector_generator(ring, positions):
             phase_diff = mu1 - mu0
             chro_diff = chro1 - chro0
             sectors.append(
-                TransverseMapSector(
-                    ring,
-                    alpha0,
-                    beta0,
-                    dispersion0,
-                    alpha1,
-                    beta1,
-                    dispersion1,
-                    phase_diff,
-                    chro_diff,
-                ))
+                TransverseMapSector(ring,
+                                    alpha0,
+                                    beta0,
+                                    dispersion0,
+                                    alpha1,
+                                    beta1,
+                                    dispersion1,
+                                    phase_diff,
+                                    chro_diff,
+                                    adts=adts))
     return sectors