diff --git a/demos/getting_started_1.py b/demos/getting_started_1.py index 56fdcb33d5610f32f4565a7f28bbcc4c280393df..8b6d59dcfc5b3a964b5fe70a341851164457d997 100644 --- a/demos/getting_started_1.py +++ b/demos/getting_started_1.py @@ -22,18 +22,19 @@ if __name__ == '__main__': # FIXME description of the tutorial # Choose problem geometry and masking - - # One dimensional regular grid - position_space = ift.RGSpace([1024]) - mask = np.ones(position_space.shape) - - # # Two dimensional regular grid with chess mask - # position_space = ift.RGSpace([128, 128]) - # mask = make_chess_mask(position_space) - - # # Sphere with half of its locations randomly masked - # position_space = ift.HPSpace(128) - # mask = make_random_mask() + mode = 0 + if mode == 0: + # One dimensional regular grid + position_space = ift.RGSpace([1024]) + mask = np.ones(position_space.shape) + elif mode == 1: + # Two dimensional regular grid with chess mask + position_space = ift.RGSpace([128, 128]) + mask = make_chess_mask(position_space) + else: + # Sphere with half of its locations randomly masked + position_space = ift.HPSpace(128) + mask = make_random_mask() harmonic_space = position_space.get_default_codomain() HT = ift.HarmonicTransformOperator(harmonic_space, target=position_space)