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
