def __call__(self, u, b): if self.dim=="1": if self.Neumann: SFTc.UTDMA_1D_Neumann(self.bk, u, b) else: SFTc.UTDMA_1D(self.ak, self.bk, u, b) elif self.dim=="3": if self.Neumann: SFTc.UTDMA_Neumann(self.bk, u, b) else: SFTc.UTDMA(self.ak, self.bk, u, b) return b