Coverage for python/gsfit/database_readers/st40_mdsplus/__init__.py: 0%

1import typing 

2 

3from gsfit_rs import BpProbes 

4from gsfit_rs import Coils 

5from gsfit_rs import Dialoop 

6from gsfit_rs import FluxLoops 

7from gsfit_rs import Isoflux 

8from gsfit_rs import IsofluxBoundary 

9from gsfit_rs import Passives 

10from gsfit_rs import Plasma 

11from gsfit_rs import Pressure 

12from gsfit_rs import RogowskiCoils 

13from gsfit_rs import StationaryPoint 

14from gsfit_rs import Tf 

15from gsfit_rs import Wall 

16 

17from ..interface import DatabaseReaderProtocol 

18from .setup_bp_probes import setup_bp_probes 

19from .setup_coils import setup_coils 

20from .setup_dialoop import setup_dialoop 

21from .setup_flux_loops import setup_flux_loops 

22from .setup_isoflux_boundary_sensors import setup_isoflux_boundary_sensors 

23from .setup_isoflux_sensors import setup_isoflux_sensors 

24from .setup_passives import setup_passives 

25from .setup_plasma import setup_plasma 

26from .setup_pressure_sensors import setup_pressure_sensors 

27from .setup_rogowski_coils import setup_rogowski_coils 

28from .setup_stationary_point_sensors import setup_stationary_point_sensors 

29from .setup_tf import setup_tf 

30from .setup_wall import setup_wall 

31 

32if typing.TYPE_CHECKING: 

33 from ..st40_mdsplus_with_digital_filter import DatabaseReader as DigitalFilterDatabaseReader 

34 

35 

36class DatabaseReader(DatabaseReaderProtocol): 

37 """ 

38 This class inherits from the DatabaseReaderProtocol, which defines the inputs and outputs to the class methods. 

39 The methods in this class are used to initialise the Rust implementations. 

40 Here we are binding the methods to the class. 

41 

42 See `python/gsfit/database_readers/interface.py` for the interface definitions. 

43 """ 

44 

45 def setup_bp_probes(self, *args: typing.Any, **kwargs: typing.Any) -> BpProbes: 

46 return setup_bp_probes(self, *args, **kwargs) 

47 

48 def setup_coils(self, *args: typing.Any, **kwargs: typing.Any) -> Coils: 

49 return setup_coils(self, *args, **kwargs) 

50 

51 def setup_dialoop(self, *args: typing.Any, **kwargs: typing.Any) -> Dialoop: 

52 return setup_dialoop(self, *args, **kwargs) 

53 

54 def setup_flux_loops(self, *args: typing.Any, **kwargs: typing.Any) -> FluxLoops: 

55 return setup_flux_loops(self, *args, **kwargs) 

56 

57 def setup_isoflux_boundary_sensors(self, *args: typing.Any, **kwargs: typing.Any) -> IsofluxBoundary: 

58 return setup_isoflux_boundary_sensors(self, *args, **kwargs) 

59 

60 def setup_isoflux_sensors(self, *args: typing.Any, **kwargs: typing.Any) -> Isoflux: 

61 return setup_isoflux_sensors(self, *args, **kwargs) 

62 

63 def setup_stationary_point_sensors(self, *args: typing.Any, **kwargs: typing.Any) -> StationaryPoint: 

64 return setup_stationary_point_sensors(self, *args, **kwargs) 

65 

66 def setup_tf(self, *args: typing.Any, **kwargs: typing.Any) -> Tf: 

67 return setup_tf(self, *args, **kwargs) 

68 

69 def setup_wall(self, *args: typing.Any, **kwargs: typing.Any) -> Wall: 

70 return setup_wall(self, *args, **kwargs) 

71 

72 def setup_passives(self, *args: typing.Any, **kwargs: typing.Any) -> Passives: 

73 return setup_passives(self, *args, **kwargs) 

74 

75 def setup_plasma(self, *args: typing.Any, **kwargs: typing.Any) -> Plasma: 

76 return setup_plasma(self, *args, **kwargs) 

77 

78 def setup_pressure_sensors(self, *args: typing.Any, **kwargs: typing.Any) -> Pressure: 

79 return setup_pressure_sensors(typing.cast("DigitalFilterDatabaseReader", self), *args, **kwargs) 

80 

81 def setup_rogowski_coils(self, *args: typing.Any, **kwargs: typing.Any) -> RogowskiCoils: 

82 return setup_rogowski_coils(self, *args, **kwargs)