Separate draw_samples and draw_sequence

pulser-core/pulser/sampler/sampler.py

@@ -56,5 +56,6 @@ def sample(
         list(seq.declared_channels.keys()),
         samples_list,
         seq.declared_channels,
+        seq._basis_ref,
         **optionals,
     )

pulser-core/pulser/sampler/samples.py

@@ -10,9 +10,10 @@
 from pulser.channels.base_channel import Channel
 from pulser.channels.eom import BaseEOM
 from pulser.register import QubitId
+from pulser.sequence._basis_ref import _QubitRef
 
 if TYPE_CHECKING:
-    from pulser.sequence._schedule import _EOMSettings
+    from pulser.sequence._schedule import _EOMSettings, _TimeSlot
 
 """Literal constants for addressing."""
 _GLOBAL = "Global"
@@ -91,7 +92,7 @@ class ChannelSamples:
     phase: np.ndarray
     slots: list[_TargetSlot] = field(default_factory=list)
     eom_blocks: list[_EOMSettings] = field(default_factory=list)
-    initial_targets: set[QubitId] = field(default_factory=set)
+    target_time_slots: list[_TimeSlot] = field(default_factory=list)
 
     def __post_init__(self) -> None:
         assert len(self.amp) == len(self.det) == len(self.phase)
@@ -102,6 +103,15 @@ def __post_init__(self) -> None:
         for t1, t2 in zip(self.slots, self.slots[1:]):
             assert t1.tf <= t2.ti  # no overlaps on a given channel
 
+    @property
+    def initial_targets(self) -> set[QubitId]:
+        """Returns the initial targets."""
+        return (
+            self.target_time_slots[0].targets
+            if self.target_time_slots
+            else set()
+        )
+
     def extend_duration(self, new_duration: int) -> ChannelSamples:
         """Extends the duration of the samples.
 
@@ -160,6 +170,27 @@ def _generate_std_samples(self) -> ChannelSamples:
 
         return replace(self, **new_samples)
 
+    def get_eom_mode_intervals(self) -> list[tuple[int, int]]:
+        """Returns EOM mode intervals."""
+        return [
+            (
+                block.ti,
+                block.tf if block.tf is not None else self.duration,
+            )
+            for block in self.eom_blocks
+        ]
+
+    def in_eom_mode(
+        self, slot: Optional[_TimeSlot | _TargetSlot] = None
+    ) -> bool:
+        """States if a time slot is inside an EOM mode block."""
+        if slot is None:
+            return bool(self.eom_blocks) and (self.eom_blocks[-1].tf is None)
+        return any(
+            start <= slot.ti < end
+            for start, end in self.get_eom_mode_intervals()
+        )
+
     def modulate(
         self, channel_obj: Channel, max_duration: Optional[int] = None
     ) -> ChannelSamples:
@@ -292,6 +323,9 @@ class SequenceSamples:
     channels: list[str]
     samples_list: list[ChannelSamples]
     _ch_objs: dict[str, Channel]
+    _basis_ref: dict[str, dict[QubitId, _QubitRef]] = field(
+        default_factory=dict
+    )
     _slm_mask: _SlmMask = field(default_factory=_SlmMask)
     _magnetic_field: np.ndarray | None = None
     _measurement: str | None = None

pulser-core/pulser/sequence/_schedule.py

@@ -138,7 +138,9 @@ def get_samples(
         dt = self.get_duration()
         amp, det, phase = np.zeros(dt), np.zeros(dt), np.zeros(dt)
         slots: list[_TargetSlot] = []
-        initial_targets = self.slots[0].targets if self.slots else set()
+        target_time_slots: list[_TimeSlot] = [
+            s for s in self.slots if s.type == "target"
+        ]
 
         for ind, s in enumerate(channel_slots):
             pulse = cast(Pulse, s.type)
@@ -183,7 +185,7 @@ def get_samples(
             phase[t_start:] = pulse.phase
 
         return ChannelSamples(
-            amp, det, phase, slots, self.eom_blocks, initial_targets
+            amp, det, phase, slots, self.eom_blocks, target_time_slots
         )
 
     @overload