g9dZddlmZddlZddlZddlmZd.d Zd/d Z d0d1dZ d0d1dZ d0d1dZ d0d1dZ d0d1dZ d2d3dZ d2d4d!Z d5d6d#Z d2d7d%Zd&Z d2d4d'Z d8d9d)Z d:d;d+Z dz.Module for builtin continuous pulse functions.) annotationsN) PulseErrortimes np.ndarrayampcomplexreturnc^tjt||tjS)zzContinuous constant pulse. Args: times: Times to output pulse for. amp: Complex pulse amplitude. dtype)npfulllen complex128)rrs U/granary/edev/quantum/lib/python3.11/site-packages/qiskit/pulse/library/continuous.pyconstantrs# 73u::s"- 8 8 88c"t|dS)zPContinuous zero pulse. Args: times: Times to output pulse for. r)r)rs rzeror"s E1  rfreqfloatphasec||z|tjz z}|ddtj|ztjd|zz zdztjzS)zContinuous square wave. Args: times: Times to output wave for. amp: Pulse amplitude. Wave range is [-amp, amp]. freq: Pulse frequency. units of 1/dt. phase: Pulse phase. r pifloorastyperrrrrxs rsquarer"+sZ  uru}$A !q28A;;!a%89A=EEbmTT TTrc||z|tjz z}|dz|tjd|zz tjzS)zContinuous sawtooth wave. Args: times: Times to output wave for. amp: Pulse amplitude. Wave range is [-amp, amp]. freq: Pulse frequency. units of 1/dt. phase: Pulse phase. rg?rr s rsawtoothr$8sJ  uru}$A 7a"(519---55bmDD DDrc |dtjt|d||tjdz z dz zdztjzS)zContinuous triangle wave. Args: times: Times to output wave for. amp: Pulse amplitude. Wave range is [-amp, amp]. freq: Pulse frequency. units of 1/dt. phase: Pulse phase. rr)r)r absr$rrrrrrrs rtriangler)Es_ "rvhuaebeai>OST=TUUUVVVYZZbb  rc|tjdtjz|z|z|ztjzSzContinuous cosine wave. Args: times: Times to output wave for. amp: Pulse amplitude. freq: Pulse frequency, units of 1/dt. phase: Pulse phase. r)r cosrrrr(s rr,r,S> BE D(505899@@OO OOrc|tjdtjz|z|z|ztjzSr+)r sinrrrr(s rr/r/_r-rFgaussian_samplescentersigma zeroed_width float | None rescale_ampboolret_scale_factor%np.ndarray | tuple[np.ndarray, float]c|d|dzz}ttj|dz g|d|}||z}d}|r||z dkr|||z z nd}||z}|r||fS|S)aEnforce that the supplied gaussian pulse is zeroed at a specific width. This is achieved by subtracting $\Omega_g(center \pm zeroed_width/2)$ from all samples. amp: Pulse amplitude at `center`. center: Center (mean) of pulse. sigma: Standard deviation of pulse. zeroed_width: Subtract baseline from gaussian pulses to make sure $\Omega_g(center \pm zeroed_width/2)=0$ is satisfied. This is used to avoid large discontinuities at the start of a gaussian pulse. If unsupplied, defaults to $2*(center + 1)$ such that $\Omega_g(-1)=0$ and $\Omega_g(2*(center + 1))=0$. rescale_amp: If True the pulse will be rescaled so that $\Omega_g(center)=amp$. ret_scale_factor: Return amplitude scale factor. Nrrr?)gaussianr array) r0rr1r2r3r5r7 zero_offsetamp_scale_factors r_fix_gaussian_widthr?ks.FQJ' 28\A%5$677aGGK #58-8;k8IQ8N8N3# "344TW,,2!111 rret_x*np.ndarray | tuple[np.ndarray, np.ndarray]ctj|tj}||z |z }|tj|dz dz tjz}|t ||||||}|r||fS|S)a7Continuous unnormalized gaussian pulse. Integrated area under curve is $\Omega_g(amp, sigma) = amp \times np.sqrt(2\pi \sigma^2)$ Args: times: Times to output pulse for. amp: Pulse amplitude at `center`. If `zeroed_width` is set pulse amplitude at center will be $amp-\Omega_g(center \pm zeroed_width/2)$ unless `rescale_amp` is set, in which case all samples will be rescaled such that the center amplitude will be `amp`. center: Center (mean) of pulse. sigma: Width (standard deviation) of pulse. zeroed_width: Subtract baseline from gaussian pulses to make sure $\Omega_g(center \pm zeroed_width/2)=0$ is satisfied. This is used to avoid large discontinuities at the start of a gaussian pulse. rescale_amp: If `zeroed_width` is not `None` and `rescale_amp=True` the pulse will be rescaled so that $\Omega_g(center)=amp$. ret_x: Return centered and standard deviation normalized pulse location. $x=(times-center)/sigma. r rNrr1r2r3r5)r asarrayrexprr?) rrr1r2r3r5r@r!gausss rr;r;s: JuBM 2 2 2E 5 A "&1a41%%,,R];; ;E# %#     ax Lr ret_gaussianc Vt||||||d\}}| |z |z} |r| |fS| S)aContinuous unnormalized gaussian derivative pulse. Args: times: Times to output pulse for. amp: Pulse amplitude at `center`. center: Center (mean) of pulse. sigma: Width (standard deviation) of pulse. ret_gaussian: Return gaussian with which derivative was taken with. zeroed_width: Subtract baseline of pulse to make sure $\Omega_g(center \pm zeroed_width/2)=0$ is satisfied. This is used to avoid large discontinuities at the start of a pulse. rescale_amp: If `zeroed_width` is not `None` and `rescale_amp=True` the pulse will be rescaled so that $\Omega_g(center)=amp$. T)rr1r2r3r5r@)r;) rrr1r2rGr3r5rFr! gauss_derivs rgaussian_derivrJs\. !HE1"u*u$K"E!! r sech_samplesc|d|dzz}ttj|dz g|d|}||z}d}|r||z dkr|||z z nd}||z}|r||fS|S)aEnforce that the supplied sech pulse is zeroed at a specific width. This is achieved by subtracting $\Omega_g(center \pm zeroed_width/2)$ from all samples. amp: Pulse amplitude at `center`. center: Center (mean) of pulse. sigma: Standard deviation of pulse. zeroed_width: Subtract baseline from sech pulses to make sure $\Omega_g(center \pm zeroed_width/2)=0$ is satisfied. This is used to avoid large discontinuities at the start of a sech pulse. If unsupplied, defaults to $2*(center + 1)$ such that $\Omega_g(-1)=0$ and $\Omega_g(2*(center + 1))=0$. rescale_amp: If True the pulse will be rescaled so that $\Omega_g(center)=amp$. ret_scale_factor: Return amplitude scale factor. Nrrrr:)sechr r<) rKrr1r2r3r5r7r=r>s r_fix_sech_widthrNs.FQJ' rx!1 233S!UCCKKL58)8;k8IQ8N8N3# "344TW(( .--- rc2dtj|g|Ri|z S)zHyperbolic secant functionr:)r cosh)r!argskwargss rsech_fnrS s' ,T,,,V,, ,,rctj|tj}||z |z }|t|tjz}|t ||||||}|r||fS|S)aContinuous unnormalized sech pulse. Args: times: Times to output pulse for. amp: Pulse amplitude at `center`. center: Center (mean) of pulse. sigma: Width (standard deviation) of pulse. zeroed_width: Subtract baseline from pulse to make sure $\Omega_g(center \pm zeroed_width/2)=0$ is satisfied. This is used to avoid large discontinuities at the start and end of the pulse. rescale_amp: If `zeroed_width` is not `None` and `rescale_amp=True` the pulse will be rescaled so that $\Omega_g(center)=amp$. ret_x: Return centered and standard deviation normalized pulse location. $x=(times-center)/sigma$. r NrC)r rDrrSrrN) rrr1r2r3r5r@r!sech_outs rrMrMs0 JuBM 2 2 2E 5 AWQZZ&&r}555H" %#     { Orret_sechcvt||||d\}}| tj|z|z }|r||fS|S)a2Continuous unnormalized sech derivative pulse. Args: times: Times to output pulse for. amp: Pulse amplitude at `center`. center: Center (mean) of pulse. sigma: Width (standard deviation) of pulse. ret_sech: Return sech with which derivative was taken with. T)rr1r2r@)rMr tanh)rrr1r2rVrUr!sech_out_derivs r sech_derivrZ=sRu#fENNNKHaY+e3N(x'' r square_widthc ||dz z }||dz z}|r||krtd||z }nd}tjt||||dtjt||||dtjt|g} ||k||kg} t j|t j| | S)aContinuous gaussian square pulse. Args: times: Times to output pulse for. amp: Pulse amplitude. center: Center of the square pulse component. square_width: Width of the square pulse component. sigma: Standard deviation of Gaussian rise/fall portion of the pulse. zeroed_width: Subtract baseline of gaussian square pulse to enforce $\OmegaSquare(center \pm zeroed_width/2)=0$. Raises: PulseError: if zeroed_width is not compatible with square_width. rz1zeroed_width cannot be smaller than square_width.NTrC)r) r functoolspartialr;rr piecewiserr) rrr1r[r2r3 square_start square_stopgaussian_zeroed_widthfunclistcondlists rgaussian_squarerePs,L1,,Lrns954""""""......9999 U U U U U E E E E E      P P P P P P P P P P""&"#####V"&-----j!%#####V"&"#####L---"&(((((XTY2"& 3I3I3I3I3Ix"&(+(+(+(+(+(+(+r