<s> """<STR_LIT>""" <EOL> import sys <EOL> import numpy as np <EOL> from vispy import app , scene <EOL> canvas = scene . SceneCanvas ( keys = '<STR_LIT>' ) <EOL> view = canvas . central_widget . add_view ( ) <EOL> def psi ( i , j , k , offset = ( <NUM_LIT> , <NUM_LIT> , <NUM_LIT:50> ) ) : <EOL> x = i - offset [ <NUM_LIT:0> ] <EOL> y = j - offset [ <NUM_LIT:1> ] <EOL> z = k - offset [ <NUM_LIT:2> ] <EOL> th = np . arctan2 ( z , ( x ** <NUM_LIT:2> + y ** <NUM_LIT:2> ) ** <NUM_LIT:0.5> ) <EOL> r = ( x ** <NUM_LIT:2> + y ** <NUM_LIT:2> + z ** <NUM_LIT:2> ) ** <NUM_LIT:0.5> <EOL> a0 = <NUM_LIT:1> <EOL> ps = ( ( <NUM_LIT:1.> / <NUM_LIT> ) * <NUM_LIT:1.> / ( <NUM_LIT> * np . pi ) ** <NUM_LIT:0.5> * ( <NUM_LIT:1.> / a0 ) ** ( <NUM_LIT:3> / <NUM_LIT:2> ) * ( r / a0 ) ** <NUM_LIT:2> * <EOL> np . exp ( - r / ( <NUM_LIT:3> * a0 ) ) * ( <NUM_LIT:3> * np . cos ( th ) ** <NUM_LIT:2> - <NUM_LIT:1> ) ) <EOL> return ps <EOL> print ( "<STR_LIT>" ) <EOL> data = np . abs ( np . fromfunction ( psi , ( <NUM_LIT:50> , <NUM_LIT:50> , <NUM_LIT:100> ) ) ) <EOL> surface = scene . visuals . Isosurface ( data , level = data . max ( ) / <NUM_LIT> , <EOL> color = ( <NUM_LIT:0.5> , <NUM_LIT> , <NUM_LIT:1> , <NUM_LIT:1> ) , shading = '<STR_LIT>' , <EOL> parent = view . scene ) <EOL> surface . transform = scene . transforms . STTransform ( translate = ( - <NUM_LIT> , - <NUM_LIT> , - <NUM_LIT:50> ) ) <EOL> axis = scene . visuals . XYZAxis ( parent = view . scene ) <EOL> cam = scene . TurntableCamera ( elevation = <NUM_LIT:30> , azimuth = <NUM_LIT:30> ) <EOL> cam . set_range ( ( - <NUM_LIT:10> , <NUM_LIT:10> ) , ( - <NUM_LIT:10> , <NUM_LIT:10> ) , ( - <NUM_LIT:10> , <NUM_LIT:10> ) ) <EOL> view . camera = cam <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> canvas . show ( ) <EOL> if sys . flags . interactive == <NUM_LIT:0> : <EOL> app . run ( ) </s>
