diff --git a/doc/source/reference/kwant.lattice.rst b/doc/source/reference/kwant.lattice.rst index cd5c9b54d8498270235e82ed5dce238cb9956f0d..f4eb8fed900f5eae03b5cdc675856a0ddc9ec57e 100644 --- a/doc/source/reference/kwant.lattice.rst +++ b/doc/source/reference/kwant.lattice.rst @@ -20,6 +20,7 @@ Library of lattices chain square + cubic triangular honeycomb kagome diff --git a/kwant/lattice.py b/kwant/lattice.py index 7183fb5e0ab31a9c303fa0d6205bf49f922d333f..e57a6e27a06830e4face8d5ef48c79dd827abe45 100644 --- a/kwant/lattice.py +++ b/kwant/lattice.py @@ -7,7 +7,7 @@ # http://kwant-project.org/authors. __all__ = ['TranslationalSymmetry', 'general', 'Polyatomic', 'Monatomic', - 'chain', 'square', 'triangular', 'honeycomb', 'kagome'] + 'chain', 'square', 'cubic', 'triangular', 'honeycomb', 'kagome'] from math import sqrt from itertools import product @@ -745,6 +745,12 @@ def square(a=1, name='', norbs=None): return Monatomic(((a, 0), (0, a)), name=name, norbs=norbs) +def cubic(a=1, name='', norbs=None): + """Make a cubic lattice.""" + return Monatomic(((a, 0, 0), (0, a, 0), (0, 0, a)), + name=name, norbs=norbs) + + tri = ta.array(((1, 0), (0.5, 0.5 * sqrt(3)))) def triangular(a=1, name='', norbs=None):