# ================================== LICENSE ===================================
# Wulfric - Cell, Atoms, K-path, visualization.
# Copyright (C) 2023-2025 Andrey Rybakov
#
# e-mail: anry@uv.es, web: adrybakov.com
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
#
# ================================ END LICENSE =================================
from math import cos
from math import pi as PI
from math import sin, sqrt
import numpy as np
from wulfric.constants._numerical import TORADIANS
from wulfric.geometry._geometry import get_angle, parallelepiped_check
# Save local scope at this moment
old_dir = set(dir())
old_dir.add("old_dir")
[docs]
def get_reciprocal(cell):
r"""
Computes reciprocal cell.
Parameters
----------
cell : (3, 3) |array-like|_
Matrix of a direct cell, rows are interpreted as vectors.
Returns
-------
reciprocal_cell : (3, 3) :numpy:`ndarray`
Matrix of a reciprocal cell.
.. code-block:: python
cell = [
[b1_x, b1_y, b1_z],
[b2_x, b2_y, b2_z],
[b3_x, b3_y, b3_z],
]
Examples
--------
.. doctest::
>>> import wulfric
>>> cell = [[1, 1, 0], [0, 1, 0], [0, 0, 1]]
>>> wulfric.cell.get_reciprocal(cell)
array([[ 6.28318531, 0. , 0. ],
[-6.28318531, 6.28318531, 0. ],
[ 0. , 0. , 6.28318531]])
"""
return 2 * PI * np.linalg.inv(cell).T
[docs]
def from_params(a=1.0, b=1.0, c=1.0, alpha=90.0, beta=90.0, gamma=90.0):
r"""
Constructs the cell from lattice parameters.
The spatial orientation of the cell is decided as
* Lattice vector :math:`\boldsymbol{a_1}` has the length ``a`` and oriented along
:math:`{\cal x}` axis.
* Lattice vector :math:`\boldsymbol{a_2}` has the length ``b``, is placed in
:math:`{\cal xy}` plane and form an angle ``gamma`` with vector
:math:`\boldsymbol{a_1}`, positive in a mathematical sense.
* Lattice vector :math:`\boldsymbol{a_3}` has the length ``c`` and form an angle
``alpha`` with the vector :math:`\boldsymbol{a_2}` and an angle ``beta`` with
the vector :math:`\boldsymbol{a_1}`.
Parameters
----------
a : float, default 1.0
Length of the :math:`\boldsymbol{a_1}` vector.
b : float, default 1.0
Length of the :math:`\boldsymbol{a_2}` vector.
c : float, default 1.0
Length of the :math:`\boldsymbol{a_3}` vector.
alpha : float, default 90.0
Angle between vectors :math:`\boldsymbol{a_2}` and :math:`\boldsymbol{a_3}` given
in degrees.
beta : float, default 90.0
Angle between vectors :math:`\boldsymbol{a_1}` and :math:`\boldsymbol{a_3}` given
in degrees.
gamma : float, default 90.0
Angle between vectors :math:`\boldsymbol{a_1}` and :math:`\boldsymbol{a_2}` given
in degrees.
Returns
-------
cell : (3, 3) :numpy:`ndarray`
Matrix of a direct cell, rows are interpreted as vectors.
.. code-block:: python
cell = [
[a1_x, a1_y, a1_z],
[a2_x, a2_y, a2_z],
[a3_x, a3_y, a3_z],
]
Raises
------
ValueError
If parameters could not form a parallelepiped.
See Also
--------
get_params
wulfric.geometry.parallelepiped_check : Check if parameters can form a parallelepiped.
Examples
--------
.. doctest::
>>> import wulfric
>>> import numpy as np
>>> np.around(wulfric.cell.from_params(1, 2, 3, 90, 90, 60), 6)
array([[1. , 0. , 0. ],
[1. , 1.732051, 0. ],
[0. , 0. , 3. ]])
.. doctest::
>>> wulfric.cell.from_params(1, 2, 3, 60, 60, 130)
Traceback (most recent call last):
...
ValueError: Parameters could not form a parallelepiped:
a = 1
b = 2
c = 3
alpha = 60
beta = 60
gamma = 130
Inequality gamma < alpha + beta is not satisfied with numerical tolerance: 0.0001
"""
parallelepiped_check(a, b, c, alpha, beta, gamma, raise_error=True)
alpha = alpha * TORADIANS
beta = beta * TORADIANS
gamma = gamma * TORADIANS
return np.array(
[
[a, 0, 0],
[b * cos(gamma), b * sin(gamma), 0],
[
c * cos(beta),
c / sin(gamma) * (cos(alpha) - cos(beta) * cos(gamma)),
c
/ sin(gamma)
* sqrt(
1
+ 2 * cos(alpha) * cos(beta) * cos(gamma)
- cos(alpha) ** 2
- cos(beta) ** 2
- cos(gamma) ** 2
),
],
],
dtype=float,
)
[docs]
def get_params(cell):
r"""
Computes lattice parameters of the cell.
Parameters
----------
cell : (3, 3) |array-like|_
Matrix of a cell, rows are interpreted as vectors.
Returns
-------
a : float
Length of the :math:`\boldsymbol{a_1}` vector.
b : float
Length of the :math:`\boldsymbol{a_2}` vector.
c : float
Length of the :math:`\boldsymbol{a_3}` vector.
alpha : float
Angle between vectors :math:`\boldsymbol{a_2}` and :math:`\boldsymbol{a_3}`
in degrees.
beta : float
Angle between vectors :math:`\boldsymbol{a_1}` and :math:`\boldsymbol{a_3}`
in degrees.
gamma : float
Angle between vectors :math:`\boldsymbol{a_1}` and :math:`\boldsymbol{a_2}`
in degrees.
See Also
--------
from_params
Examples
--------
.. doctest::
>>> import wulfric
>>> cell = [[1, 0, 0], [0, 2, 0], [0, 0, 3]]
>>> wulfric.cell.get_params(cell)
(1.0, 2.0, 3.0, 90.0, 90.0, 90.0)
"""
return (
float(np.linalg.norm(cell[0])),
float(np.linalg.norm(cell[1])),
float(np.linalg.norm(cell[2])),
get_angle(cell[1], cell[2]),
get_angle(cell[0], cell[2]),
get_angle(cell[0], cell[1]),
)
[docs]
def get_scalar_products(cell):
r"""
Returns pairwise scalar products of the lattice vectors.
Parameters
----------
cell : (3, 3) |array-like|_
Matrix of a cell, rows are interpreted as vectors.
Returns
-------
sp_23 : float
Scalar product of the vectors :math:`\boldsymbol{a}_2\cdot\boldsymbol{a}_3`.
sp_13 : float
Scalar product of the vectors :math:`\boldsymbol{a}_1\cdot\boldsymbol{a}_3`.
sp_12 : float
Scalar product of the vectors :math:`\boldsymbol{a}_1\cdot\boldsymbol{a}_2`.
Examples
--------
.. doctest::
>>> import wulfric
>>> cell = [[1, 0, 0], [0, 1, 0], [0, 0, 1]]
>>> wulfric.cell.get_scalar_products(cell)
(0.0, 0.0, 0.0)
"""
return (
float(np.dot(cell[1], cell[2])),
float(np.dot(cell[0], cell[2])),
float(np.dot(cell[0], cell[1])),
)
# Populate __all__ with objects defined in this file
__all__ = list(set(dir()) - old_dir)
# Remove all semi-private objects
__all__ = [i for i in __all__ if not i.startswith("_")]
del old_dir
