mirror of
https://github.com/nqrduck/nqr-blochsimulator.git
synced 2024-06-21 13:13:21 +00:00
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import numpy as np
|
|
|
|
|
|
class PulseArray:
|
|
"""A class to represent a pulsearray for a NQR sequence."""
|
|
|
|
def __init__(self, pulseamplitude, pulsephase, dwell_time) -> None:
|
|
"""
|
|
Constructs all the necessary attributes for the pulsearray object.
|
|
|
|
Parameters
|
|
----------
|
|
pulseamplitude : float
|
|
The amplitude of the pulse.
|
|
pulsephase : float
|
|
The phase of the pulse.
|
|
dwell_time : float
|
|
The dwell time of the pulse.
|
|
"""
|
|
self.pulseamplitude = pulseamplitude
|
|
self.pulsephase = pulsephase
|
|
self.dwell_time = dwell_time
|
|
|
|
def get_real_pulsepower(self) -> np.array:
|
|
"""Returns the real part of the pulse power."""
|
|
return self.pulseamplitude * np.cos(self.pulsephase)
|
|
|
|
def get_imag_pulsepower(self) -> np.array:
|
|
"""Returns the imaginary part of the pulse power."""
|
|
return self.pulseamplitude * np.sin(self.pulsephase)
|
|
|
|
@property
|
|
def pulseamplitude(self) -> np.array:
|
|
"""Amplitude of the pulse."""
|
|
return self._pulseamplitude
|
|
|
|
@pulseamplitude.setter
|
|
def pulseamplitude(self, pulseamplitude):
|
|
self._pulseamplitude = pulseamplitude
|
|
|
|
@property
|
|
def pulsephase(self) -> np.array:
|
|
"""Phase of the pulse."""
|
|
return self._pulsephase
|
|
|
|
@pulsephase.setter
|
|
def pulsephase(self, pulsephase):
|
|
self._pulsephase = pulsephase
|