hiro/two_qubit_model
1
0
Fork 0
mirror of https://github.com/vale981/two_qubit_model synced 2025-03-05 17:51:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

support time dependent system

Browse source
This commit is contained in:
Valentin Boettcher 2022-04-12 18:22:54 +02:00
parent 329df8de64
commit bc49789a11
2 changed files with 16 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
hiro_models/model_base.py
View file

@ -32,8 +32,8 @@ class Model(ABC):
description: str = "" description: str = ""
"""A free-form description of the model instance.""" """A free-form description of the model instance."""
# t: NDArray[np.float64] = np.linspace(0, 10, 1000) t: NDArray[np.float64] = np.linspace(0, 10, 1000)
# """The simulation time points.""" """The simulation time points."""
__base_version__: int = 1 __base_version__: int = 1
"""The version of the model base.""" """The version of the model base."""
@ -146,7 +146,7 @@ class Model(ABC):
@property @property
@abstractmethod @abstractmethod
def system(self) -> qt.Qobj: def system(self) -> DynamicMatrix:
"""The system hamiltonian.""" """The system hamiltonian."""
pass pass
@ -230,7 +230,7 @@ class Model(ABC):
########################################################################### ###########################################################################
def system_expectation( def system_expectation(
self, data: HIData, operator: qt.Qobj, **kwargs self, data: HIData, operator: DynamicMatrix, **kwargs
) -> EnsembleValue: ) -> EnsembleValue:
"""Calculates the expectation value of ``operator`` from the """Calculates the expectation value of ``operator`` from the
hierarchy data ``data``. hierarchy data ``data``.
@ -246,7 +246,8 @@ class Model(ABC):
return hopsflow.util.operator_expectation_ensemble( return hopsflow.util.operator_expectation_ensemble(
data.valid_sample_iterator(data.stoc_traj), # type: ignore data.valid_sample_iterator(data.stoc_traj), # type: ignore
operator.full(), operator,
self.t,
normalize=True, # always nonlinear normalize=True, # always nonlinear
save=f"{operator_hash}_{self.hexhash}", save=f"{operator_hash}_{self.hexhash}",
N=N, N=N,

13
hiro_models/one_qubit_model.py
View file

@ -129,6 +129,13 @@ class QubitModel(Model):
processes. processes.
""" """
H: DynamicMatrix = field(
default_factory=lambda: ConstantMatrix(1 / 2 * qt.sigmaz().full())
) # type: ignore
"""
The system hamiltonian :math:`H` with shape ``(2, 2)``.
"""
@property @property
def coupling_operators(self) -> list[DynamicMatrix]: def coupling_operators(self) -> list[DynamicMatrix]:
"""The bath coupling operators :math:`L`.""" """The bath coupling operators :math:`L`."""
@ -136,10 +143,10 @@ class QubitModel(Model):
return [self.L] return [self.L]
@property @property
def system(self) -> qt.Qobj: def system(self) -> DynamicMatrix:
"""The system hamiltonian.""" """The system hamiltonian."""
return 1 / 2 * qt.sigmaz() # type: ignore return self.H # type: ignore
@property @property
def bcf_norm(self) -> float: def bcf_norm(self) -> float:
@ -318,7 +325,7 @@ class QubitModel(Model):
g, w = self.bcf_coefficients(self.bcf_terms) g, w = self.bcf_coefficients(self.bcf_terms)
system = params.SysP( system = params.SysP(
H_sys=self.system.full(), H_sys=self.system,
L=self.coupling_operators, L=self.coupling_operators,
g=g, g=g,
w=w, w=w,