g?dZddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZmZmZmZmZdd lmZdd lmZmZdd lmZGd d ZddZddZdS)a A convenient way to track reusable subschedules by name and qubit. This can be used for scheduling circuits with custom definitions, for instance:: inst_map = InstructionScheduleMap() inst_map.add('new_inst', 0, qubit_0_new_inst_schedule) sched = schedule(quantum_circuit, backend, inst_map) An instance of this class is instantiated by Pulse-enabled backends and populated with defaults (if available):: inst_map = backend.defaults().instruction_schedule_map ) annotationsN) defaultdict)IterableCallable)circuit)ParameterExpression)CalibrationEntry ScheduleDef CallableDef PulseQobjDefCalibrationPublisher) PulseError)Schedule ScheduleBlock)deprecate_pulse_funcceZdZdZedZd*dZed+dZd,d Z d-dZ d.dZ d/dZ d0dZ d1dZ d2d3dZd4d#Zd/d$Zd0d%Zd5d'Zd(Zd)ZdS)6InstructionScheduleMapaMapping from :py:class:`~qiskit.circuit.QuantumCircuit` :py:class:`qiskit.circuit.Instruction` names and qubits to :py:class:`~qiskit.pulse.Schedule` s. In particular, the mapping is formatted as type:: Dict[str, Dict[Tuple[int], Schedule]] where the first key is the name of a circuit instruction (e.g. ``'u1'``, ``'measure'``), the second key is a tuple of qubit indices, and the final value is a Schedule implementing the requested instruction. These can usually be seen as gate calibrations. cttjtt|_tt |_dS)z=Initialize a circuit instruction to schedule mapper instance.N)r functoolspartialr _mapset_qubit_instructionsselfs [/granary/edev/quantum/lib/python3.11/site-packages/qiskit/pulse/instruction_schedule_map.py__init__zInstructionScheduleMap.__init__Bs>  )+7GHH I I @K3?O?O   returnboolc|jD]$}|D] }|jrdS%dS)z9Return ``True`` if the map has user provided instruction.TF)rvalues user_provided)r qubit_instentrys rhas_custom_gatez&InstructionScheduleMap.has_custom_gateQs_)**,,  J#**,,  & 444  ur list[str]cNt|jS)a#Return all instructions which have definitions. By default, these are typically the basis gates along with other instructions such as measure and reset. Returns: The names of all the circuit instructions which have Schedule definitions in this. )listrkeysrs r instructionsz#InstructionScheduleMap.instructionsYsDINN$$%%%r instruction%str | circuit.instruction.Instructionlist[int | tuple[int, ...]]ct|}||jvrgSdt|j|DS)a/Return a list of the qubits for which the given instruction is defined. Single qubit instructions return a flat list, and multiqubit instructions return a list of ordered tuples. Args: instruction: The name of the circuit instruction. Returns: Qubit indices which have the given instruction defined. This is a list of tuples if the instruction has an arity greater than 1, or a flat list of ints otherwise. Raises: PulseError: If the instruction is not found. cHg|]}t|dkr|dn| S)r)len).0qubitss r zBInstructionScheduleMap.qubits_with_instruction..ys@   V))F1IIv   r)_get_instruction_stringrsortedr*)rr,s rqubits_with_instructionz.InstructionScheduleMap.qubits_with_instructiones_".k:: di ' 'I   ;!7!!>??    rr4int | Iterable[int]ct||jvr't|jt|SgS)aReturn a list of the instruction names that are defined by the backend for the given qubit or qubits. Args: qubits: A qubit index, or a list or tuple of indices. Returns: All the instructions which are defined on the qubits. For 1 qubit, all the 1Q instructions defined. For multiple qubits, all the instructions which apply to that whole set of qubits (e.g. ``qubits=[0, 1]`` may return ``['cx']``). ) _to_tuplerr))rr4s rqubit_instructionsz)InstructionScheduleMap.qubit_instructions~s= V   8 8 8061B1BCDD D rcjt|}||jvot||j|vS)aIs the instruction defined for the given qubits? Args: instruction: The instruction for which to look. qubits: The specific qubits for the instruction. Returns: True iff the instruction is defined. )r6rr;rr,r4s rhaszInstructionScheduleMap.hass7.k:: di'WIf,=,=;AW,WWrNonec pt|}||t|stj5tjdt dd||jvr)td|d| |dtd|d#1swxYwYd Sd S) aError if the given instruction is not defined. Args: instruction: The instruction for which to look. qubits: The specific qubits for the instruction. Raises: PulseError: If the instruction is not defined on the qubits. ignorez".*The entire Qiskit Pulse package*qiskit)categorymessagemodulez Operation 'z)' exists, but is only defined for qubits .z!' is not defined for this system.N) r6r?r;warningscatch_warningsfilterwarningsDeprecationWarningrrr8r>s r assert_hasz!InstructionScheduleMap.assert_hass/.k:: xx Yv%6%677 _(** _ _'/@#  $)++$HkHH77 DDHHH!!]{!]!]!]^^^ _ _ _ _ _ _ _ _ _ _  _ _sA#B))B-0B-paramscomplex | ParameterExpressionkwparamsSchedule | ScheduleBlockcD|||j|i|S)alReturn the defined :py:class:`~qiskit.pulse.Schedule` or :py:class:`~qiskit.pulse.ScheduleBlock` for the given instruction on the given qubits. If all keys are not specified this method returns schedule with unbound parameters. Args: instruction: Name of the instruction or the instruction itself. qubits: The qubits for the instruction. *params: Command parameters for generating the output schedule. **kwparams: Keyworded command parameters for generating the schedule. Returns: The Schedule defined for the input. )_get_calibration_entry get_schedule)rr,r4rMrOs rgetzInstructionScheduleMap.gets.*Mt**;??LfaX`aaarr ct|}||||j|t|S)aReturn the :class:`.CalibrationEntry` without generating schedule. When calibration entry is un-parsed Pulse Qobj, this returns calibration without parsing it. :meth:`CalibrationEntry.get_schedule` method must be manually called with assigned parameters to get corresponding pulse schedule. This method is expected be directly used internally by the V2 backend converter for faster loading of the backend calibrations. Args: instruction: Name of the instruction or the instruction itself. qubits: The qubits for the instruction. Returns: The calibration entry. )r6rLrr;r>s rrRz-InstructionScheduleMap._get_calibration_entrys?*.k::   V,,,y%i&7&788rNscheduleBSchedule | ScheduleBlock | Callable[..., Schedule | ScheduleBlock] argumentslist[str] | Nonect|}t|}|std|dt|tt frt |}nIt|r+|rtj dtt}ntd| |d| |||dS)aAdd a new known instruction for the given qubits and its mapping to a pulse schedule. Args: instruction: The name of the instruction to add. qubits: The qubits which the instruction applies to. schedule: The Schedule that implements the given instruction. arguments: List of parameter names to create a parameter-bound schedule from the associated gate instruction. If :py:meth:`get` is called with arguments rather than keyword arguments, this parameter list is used to map the input arguments to parameter objects stored in the target schedule. Raises: PulseError: If the qubits are provided as an empty iterable. zCannot add definition z with no target qubits.zZArguments are overruled by the callback function signature. Input `arguments` are ignored.zcSupplied schedule must be one of the Schedule, ScheduleBlock or a callable that outputs a schedule.T)r#N)r6r;r isinstancerrr callablerHwarn UserWarningr define_add)rr,r4rVrXr%s raddzInstructionScheduleMap.adds*.k:: 6"" \ZkZZZ[[ [ h= 9 : : &1)&<&>>>>rc"t|}t|}||||j||=|j|s|j|=|j|||j|s |j|=dSdS)zRemove the given instruction from the listing of instructions defined in self. Args: instruction: The name of the instruction to add. qubits: The qubits which the instruction applies to. N)r6r;rLrrremover>s rrgzInstructionScheduleMap.remove5s.k:: 6""  V,,, Ik "6 *y% ' +&  (// <<<'/ 1(000 1 1rctt|}|j||g|Ri|}||||S)aRemove and return the defined schedule for the given instruction on the given qubits. Args: instruction: Name of the instruction. qubits: The qubits for the instruction. *params: Command parameters for generating the output schedule. **kwparams: Keyworded command parameters for generating the schedule. Returns: The Schedule defined for the input. )r6rTrg)rr,r4rMrOrVs rpopzInstructionScheduleMap.popJsM&.k:: 48KE&EEEHEE K(((rtuple[str, ...]ct|}|||tj5tjdt |j|t|}dddn #1swxYwYt|j S)a&Return the list of parameters taken by the given instruction on the given qubits. Args: instruction: Name of the instruction. qubits: The qubits for the instruction. Returns: The names of the parameters required by the instruction. rB)actionrDN) r6rLrHrI simplefilterrKrr; get_signaturetuple parametersr*)rr,r4 signatures rget_parametersz%InstructionScheduleMap.get_parametersbs.k::   V,,,  $ & & R R  !)ritemsr2 __class____name__)rsingle_q_insts multi_q_instsr4instsr+s r__str__zInstructionScheduleMap.__str__ws-5 !5;;== : :MFE6{{a">q ">">U">">">>!9f!9!9!9!9!99 % 5 =4>*==\====rc,t|tsdS|jD]t}||D]\} |j|t ||j|t |krdSJ#t $rYdSwxYwudS)NFT)r[rr+r8rr;KeyError)rotherinstqindss r__eq__zInstructionScheduleMap.__eq__s%!788 5% ! !D55d;; ! !!yy'7'78EJtbbbb.9999>'+ -.-.-.-.-.^????01111*02222* > > >     rrr"r9rrdcH t|S#t$r|fcYSwxYw)zReturn the input as a tuple. Args: values: An integer, or iterable of integers. Returns: The input values as a sorted tuple. )ro TypeError)r"s rr;r;s<V}} ys  !!rr-rcct|tr|S |jS#t$r}t d|d}~wwxYw)NzMInput "inst" has no attribute "name". This should be a circuit "Instruction".)r[rcnameAttributeErrorr)rexs rr6r6s]$  9    _  s ?:?)r"r9rrd)rr-rrc)r __future__rrrH collectionsrcollections.abcrrrCr"qiskit.circuit.parameterexpressionr qiskit.pulse.calibration_entriesr r r r r qiskit.pulse.exceptionsrqiskit.pulse.schedulerrqiskit.utils.deprecate_pulserrr;r6rrrrsm #"""""######........BBBBBB/.....99999999======YYYYYYYYx          r