Coverage for pySDC/implementations/transfer_classes/TransferMesh_NoCoarse.py: 75%

1from pySDC.core.errors import TransferError 

2from pySDC.core.space_transfer import SpaceTransfer 

3from pySDC.implementations.datatype_classes.mesh import mesh, imex_mesh 

4 

5 

6class mesh_to_mesh(SpaceTransfer): 

7 """ 

8 Custom base_transfer class, implements Transfer.py 

9 

10 This implementation can restrict and prolong between nd meshes with dirichlet-0 or periodic boundaries 

11 via matrix-vector products 

12 

13 Attributes: 

14 Rspace: spatial restriction matrix, dim. Nf x Nc 

15 Pspace: spatial prolongation matrix, dim. Nc x Nf 

16 """ 

17 

18 def restrict(self, F): 

19 """ 

20 Restriction implementation 

21 

22 Args: 

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

24 """ 

25 if isinstance(F, mesh): 

26 G = mesh(F) 

27 elif isinstance(F, imex_mesh): 

28 G = imex_mesh(F) 

29 else: 

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

31 return G 

32 

33 def prolong(self, G): 

34 """ 

35 Prolongation implementation 

36 

37 Args: 

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

39 """ 

40 if isinstance(G, mesh): 

41 F = mesh(G) 

42 elif isinstance(G, imex_mesh): 

43 F = imex_mesh(G) 

44 else: 

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

46 return F