def test_gen_barrier(self):
"""Test gen_barrier."""
inst_data = types.BarrierInstruction(
5, 0.1, [pulse.DriveChannel(0),
pulse.ControlChannel(0)])
obj = barrier.gen_barrier(inst_data,
formatter=self.formatter,
device=self.device)[0]
# type check
self.assertEqual(type(obj), drawings.LineData)
# data check
self.assertListEqual(obj.channels,
[pulse.DriveChannel(0),
pulse.ControlChannel(0)])
ref_x = [5, 5]
ref_y = [types.AbstractCoordinate.BOTTOM, types.AbstractCoordinate.TOP]
self.assertListEqual(list(obj.xvals), ref_x)
self.assertListEqual(list(obj.yvals), ref_y)
# style check
ref_style = {
'alpha': self.formatter['alpha.barrier'],
'zorder': self.formatter['layer.barrier'],
'linewidth': self.formatter['line_width.barrier'],
'linestyle': self.formatter['line_style.barrier'],
'color': self.formatter['color.barrier']
}
self.assertDictEqual(obj.styles, ref_style)
def _schedule_loader(self, program: pulse.Schedule):
"""Load Schedule instance.
This function is sub-routine of py:method:`load_program`.
Args:
program: `Schedule` to draw.
"""
# initialize scale values
self.chan_scales = {}
for chan in program.channels:
if isinstance(chan, pulse.channels.DriveChannel):
self.chan_scales[chan] = self.formatter[
'channel_scaling.drive']
elif isinstance(chan, pulse.channels.MeasureChannel):
self.chan_scales[chan] = self.formatter[
'channel_scaling.measure']
elif isinstance(chan, pulse.channels.ControlChannel):
self.chan_scales[chan] = self.formatter[
'channel_scaling.control']
elif isinstance(chan, pulse.channels.AcquireChannel):
self.chan_scales[chan] = self.formatter[
'channel_scaling.acquire']
else:
self.chan_scales[chan] = 1.0
# create charts
mapper = self.layout['chart_channel_map']
for name, chans in mapper(channels=program.channels,
formatter=self.formatter,
device=self.device):
chart = Chart(parent=self, name=name)
# add standard pulse instructions
for chan in chans:
chart.load_program(program=program, chan=chan)
# add barriers
barrier_sched = program.filter(
instruction_types=[pulse.instructions.RelativeBarrier],
channels=chans)
for t0, _ in barrier_sched.instructions:
inst_data = types.BarrierInstruction(t0, self.device.dt, chans)
for gen in self.generator['barrier']:
obj_generator = partial(gen,
formatter=self.formatter,
device=self.device)
for data in obj_generator(inst_data):
chart.add_data(data)
# add chart axis
chart_axis = types.ChartAxis(name=chart.name,
channels=chart.channels)
for gen in self.generator['chart']:
obj_generator = partial(gen,
formatter=self.formatter,
device=self.device)
for data in obj_generator(chart_axis):
chart.add_data(data)
self.charts.append(chart)
# add snapshot data to global
snapshot_sched = program.filter(
instruction_types=[pulse.instructions.Snapshot])
for t0, inst in snapshot_sched.instructions:
inst_data = types.SnapshotInstruction(t0, self.device.dt,
inst.label, inst.channels)
for gen in self.generator['snapshot']:
obj_generator = partial(gen,
formatter=self.formatter,
device=self.device)
for data in obj_generator(inst_data):
self.global_charts.add_data(data)
# calculate axis break
self.time_breaks = self._calculate_axis_break(program)
