wavepacket.builder
Functions to assemble wave functions or density operators.
Functions
|
Returns the direct product of wave functions as a density operator. |
|
Given an input wave function, create the corresponding pure density operator. |
|
Builds a product wave function from a set of one-dimensional wave functions. |
Package Contents
- wavepacket.builder.direct_product(ket: wavepacket.grid.State, bra: wavepacket.grid.State) wavepacket.grid.State
Returns the direct product of wave functions as a density operator.
Given two wave functions
, this function returns the
density operator as
.
This operation can be useful to build up a
density operator piece by piece.- Parameters:
- ketwp.grid.State
The ket state

- brawp.grid.State
The bra state
. Note that the function performs a
complex conjugation of this state prior to multiplication.
- Returns:
- wp.grid.State
The direct product of the two states.
- Raises:
- wp.BadStateError
If one of the input states is not a valid wave function.
- wp.BadGridError
If the input states are defined on different grids.
- wavepacket.builder.pure_density(psi: wavepacket.grid.State) wavepacket.grid.State
Given an input wave function, create the corresponding pure density operator.
This function only performs the direct product, it does not apply further modifications like normalizations.
- Parameters:
- psiwp.grid.State
The input wave function
- Returns:
- wp.grid.State
The corresponding density operator.
- Raises:
- wp.BadStateError
If the input is not a valid wave function.
See also
direct_productThis function is identical to direct_product(psi, psi)
- wavepacket.builder.product_wave_function(grid: wavepacket.grid.Grid, generators: wavepacket.typing.Generator | collections.abc.Sequence[wavepacket.typing.Generator], normalize: bool = True) wavepacket.grid.State
Builds a product wave function from a set of one-dimensional wave functions.
- Parameters:
- gridwp.grid.Grid
The grid on which the product wave function is assembled
- generatorsSequence[wp.typing.Generator]
A list of callables that specifies the wave function along each degree of freedom. The generators return the one-dimensional functions in the DVR, i.e., raw function values at the grid points.
- normalizebool, default=true
If the norm is non-zero and this value is set, the resulting product wave function is normalized, otherwise the product is returned directly.
- Returns:
- wp.grid.State
The product wave function in the Wavepacket-default weighted DVR.
- Raises:
- wp.InvalidValueError
If the number of generators does not match the grid dimensions.