mirror of
https://github.com/triqs/dft_tools
synced 2024-11-01 03:33:50 +01:00
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
|
|
################################################################################
|
|
#
|
|
# TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
|
#
|
|
# Copyright (C) 2011 by M. Ferrero, O. Parcollet
|
|
#
|
|
# TRIQS 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.
|
|
#
|
|
# TRIQS 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
|
|
# TRIQS. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
################################################################################
|
|
|
|
#from pytriqs import *
|
|
from super_lattice import *
|
|
from pytriqs.dos import DOS
|
|
|
|
class BZPatch:
|
|
"""Description of a Patch of the BZ"""
|
|
def __init__(self, name, polygons):
|
|
""" TO BE WRITTEN: MICHEL! """
|
|
# Cut the patch in triangles (this is what is asked by the C-code)
|
|
self.weight, self.name = 0, name
|
|
self._triangles = []
|
|
self._weights = []
|
|
for polygon in polygons:
|
|
pnt = [0, 0, 0]
|
|
for np, point in enumerate(polygon):
|
|
if np > 1:
|
|
pnt[2] = point
|
|
self._triangles += pnt
|
|
self._weights += [ 0.5*abs((pnt[1][0]-pnt[0][0])*(pnt[2][1]-pnt[0][1])
|
|
-(pnt[1][1]-pnt[0][1])*(pnt[2][0]-pnt[0][0])) ]
|
|
self.weight += 0.5*abs((pnt[1][0]-pnt[0][0])*(pnt[2][1]-pnt[0][1])
|
|
-(pnt[1][1]-pnt[0][1])*(pnt[2][0]-pnt[0][0]))
|
|
pnt[1] = pnt[2]
|
|
else:
|
|
pnt[np%3] = point
|
|
|
|
def dos(self, lattice, n_bins, n_div_triangle):
|
|
""" Compute the partial dos of the Patch for the Lattice lattice"""
|
|
assert isinstance(lattice, (Lattice, TBSuperLattice))
|
|
return lattice.dos_patch(self._triangles, n_bins, n_div_triangle, self.name)
|
|
|
|
|
|
|