def get_wave_function_array(self, n, k, s, realspace=True,
cut=True):
psit_G = FDPWWaveFunctions.get_wave_function_array(self, n, k, s,
realspace)
if cut and psit_G is not None and not realspace:
psit_G = psit_G[:self.ng_k[k]].copy()
return psit_G
def get_wave_function_array(self, n, k, s, realspace=True,
cut=True):
psit_G = FDPWWaveFunctions.get_wave_function_array(self, n, k, s,
realspace)
if cut and psit_G is not None and not realspace:
psit_G = psit_G[:self.ng_k[k]].copy()
return psit_G
