Coverage for pySDC/implementations/transfer_classes/TransferMesh_FFT.py: 100%

1from scipy.signal import resample 

2 

3from pySDC.core.errors import TransferError 

4from pySDC.core.space_transfer import SpaceTransfer 

5 

6 

7class mesh_to_mesh_fft(SpaceTransfer): 

8 """ 

9 Custom base_transfer class, implements Transfer.py 

10 

11 This implementation can restrict and prolong between 1d meshes with FFT for periodic boundaries 

12 

13 Attributes: 

14 irfft_object_fine: planned FFT for backward transformation, real-valued output 

15 rfft_object_coarse: planned real-valued FFT for forward transformation 

16 """ 

17 

18 def __init__(self, fine_prob, coarse_prob, params): 

19 """ 

20 Initialization routine 

21 

22 Args: 

23 fine_prob: fine problem 

24 coarse_prob: coarse problem 

25 params: parameters for the transfer operators 

26 """ 

27 # invoke super initialization 

28 super().__init__(fine_prob, coarse_prob, params) 

29 

30 self.ratio = int(self.fine_prob.nvars / self.coarse_prob.nvars) 

31 

32 def restrict(self, F): 

33 """ 

34 Restriction implementation 

35 

36 Args: 

37 F: the fine level data (easier to access than via the fine attribute) 

38 """ 

39 G = type(F)(self.coarse_prob.init, val=0.0) 

40 

41 if type(F).__name__ == 'mesh': 

42 G[:] = F[:: self.ratio] 

43 elif type(F).__name__ == 'imex_mesh': 

44 G.impl[:] = F.impl[:: self.ratio] 

45 G.expl[:] = F.expl[:: self.ratio] 

46 else: 

47 raise TransferError('Unknown data type, got %s' % type(F)) 

48 return G 

49 

50 def prolong(self, G): 

51 """ 

52 Prolongation implementation 

53 

54 Args: 

55 G: the coarse level data (easier to access than via the coarse attribute) 

56 """ 

57 F = type(G)(self.fine_prob.init, val=0.0) 

58 

59 def _prolong(coarse): 

60 # Fourier interpolation. `resample` also gets the normalisation and the splitting of the 

61 # Nyquist mode right; zero-padding the spectrum by hand placed the coarse Nyquist mode at 

62 # the *fine* Nyquist wavenumber, which is a different function entirely. 

63 return resample(coarse, self.fine_prob.init[0]) 

64 

65 if type(G).__name__ == 'mesh': 

66 F[:] = _prolong(G) 

67 elif type(G).__name__ == 'imex_mesh': 

68 F.impl[:] = _prolong(G.impl) 

69 F.expl[:] = _prolong(G.expl) 

70 else: 

71 raise TransferError('Unknown data type, got %s' % type(G)) 

72 return F