|
39 | 39 | import numpy as np
|
40 | 40 | from ocelot.cpbd.magnetic_lattice import MagneticLattice, merger
|
41 | 41 |
|
42 |
| -from ocelot.cpbd.beam import ParticleArray, global_slice_analysis,Particle, Beam, Twiss, get_current, \ |
43 |
| - get_envelope, generate_parray, ellipse_from_twiss |
44 |
| - |
45 |
| -from ocelot.cpbd.optics import lattice_transfer_map, TransferMap, Navigator, twiss, get_map, MethodTM, \ |
46 |
| - SecondTM, KickTM, CavityTM, UndulatorTestTM |
47 |
| - |
| 42 | +from ocelot.cpbd.beam import * |
| 43 | +from ocelot.cpbd.optics import * |
48 | 44 | from ocelot.cpbd.elements import *
|
49 |
| -from ocelot.cpbd.match import match, match_tunes |
| 45 | +from ocelot.cpbd.match import * |
50 | 46 | from ocelot.cpbd.track import *
|
51 |
| -from ocelot.common.globals import pi, m_e_eV, m_e_MeV, m_e_GeV, speed_of_light |
| 47 | +from ocelot.common.globals import * |
52 | 48 | from ocelot.common.ocelog import *
|
53 |
| -from ocelot.cpbd.chromaticity import compensate_chromaticity |
54 |
| -from ocelot.cpbd.beam_params import EbeamParams |
55 |
| -from ocelot.cpbd.io import save_particle_array, load_particle_array, write_lattice |
56 |
| -from ocelot.cpbd.sc import SpaceCharge, LSC |
57 |
| -from ocelot.cpbd.csr import CSR |
58 |
| -from ocelot.cpbd.wake3D import Wake, WakeTable, WakeKick, WakeTableDechirperOffAxis |
59 |
| - |
60 |
| -from ocelot.cpbd.physics_proc import BeamTransform, SmoothBeam, EmptyProc, PhysProc, LaserHeater, \ |
61 |
| - LaserModulator, SpontanRadEffects, PhaseSpaceAperture |
| 49 | +from ocelot.cpbd.chromaticity import * |
| 50 | +from ocelot.cpbd.beam_params import * |
| 51 | +from ocelot.cpbd.io import * |
| 52 | +from ocelot.cpbd.sc import * |
| 53 | +from ocelot.cpbd.csr import * |
| 54 | +from ocelot.cpbd.wake3D import * |
| 55 | +from ocelot.cpbd.physics_proc import * |
62 | 56 |
|
63 | 57 |
|
64 | 58 | print('initializing ocelot...')
|
|
0 commit comments