Accelerator, radiation and x-ray optics simulation framework
Ocelot is a multiphysics simulation toolkit designed for studying FEL and storage ring-based light sources. Ocelot is written in Python. Its central concept is the writing of python's scripts for simulations with the usage of Ocelot's modules and functions and the standard Python libraries.
Ocelot includes following main modules:
- Charged particle beam dynamics module (CPBD)
- optics
- tracking
- matching
- collective effects (description can be found here )
- Space Charge (3D Laplace solver)
- CSR (Coherent Synchrotron Radiation) (1D model with arbitrary number of dipoles).
- Wakefields (Taylor expansion up to second order for arbitrary geometry).
- MOGA (Multi Objective Genetics Algorithm). (ref1)
- Native module for spontaneous radiation calculation
- FEL calculations: interface to GENESIS and pre/post-processing
- Modules for online beam control and online optimization of accelerator performances. ref1, ref2, ref3, ref4.
Ocelot extensively uses Python's NumPy (Numerical Python) and SciPy (Scientific Python) libraries, which enable efficient in-core numerical and scientific computation within Python and give you access to various mathematical and optimization techniques and algorithms. To produce high quality figures Python's matplotlib library is used.
It is an open source project and it is being developed by physicists from The European XFEL, DESY (Germany), NRC Kurchatov Institute (Russia).
We still have no documentation but you can find a lot of examples in /demos/ folder including this tutorial
Ocelot is designed for researchers who want to have the flexibility that is given by high-level languages such as Matlab, Python (with Numpy and SciPy) or Mathematica. However if someone needs a GUI it can be developed using Python's libraries like a PyQtGraph or PyQt.
For example, you can see GUI for SASE optimization (uncomment and run next block)
The tutorial includes 7 simple examples dediacted to beam dynamics and optics. However, you should have a basic understanding of Computer Programming terminologies. A basic understanding of Python language is a plus.
- Python 3.4-3.6 (python 2.7 can work as well but not guaranteed)
numpy
version 1.8 or later: http://www.numpy.org/scipy
version 0.15 or later: http://www.scipy.org/matplotlib
version 1.5 or later: http://matplotlib.org/ipython
version 2.4 or later, with notebook support: http://ipython.org
Optional, but highly recommended for speeding up calculations
- numexpr (version 2.6.1)
- pyfftw (version 0.10)
- numba
The easiest way to get these is to download and install the (large) Anaconda software distribution.
Alternatively, you can download and install miniconda. The following command will install all required packages:
$ conda install numpy scipy matplotlib jupyter
The easiest way to install OCELOT is to use Anaconda cloud. In that case use command:
$ conda install -c ocelot-collab ocelot
Clone OCELOT from GitHub:
$ git clone https://github.com/ocelot-collab/ocelot.git
or download last release zip file - recomended. Now you can install pyqtgraph from the source:
$ python setup.py install
Another way is download ocelot from GitHub
-
you have to download from GitHub zip file.
-
Unzip ocelot-master.zip to your working folder /your_working_dir/.
-
Add ../your_working_dir/ocelot-master to PYTHONPATH
- Windows 7: go to Control Panel -> System and Security -> System -> Advance System Settings -> Environment Variables. and in User variables add /your_working_dir/ocelot-master/ to PYTHONPATH. If variable PYTHONPATH does not exist, create it
Variable name: PYTHONPATH
Variable value: ../your_working_dir/ocelot-master/
- Linux:
$ export PYTHONPATH=/your_working_dir/ocelot-master:$PYTHONPATH
in command line run following commands:
$ ipython notebook
or
$ ipython notebook --notebook-dir="path_to_your_directory"
or
$ jupyter notebook --notebook-dir="path_to_your_directory"
You can download OCELOT jupyter tutorials (release v18.02) using GitHub link zip file.
- Preliminaries: Setup & introduction
- Beam dynamics
- Tutorial N1. Linear optics.. Web version.
- Linear optics. Double Bend Achromat (DBA). Simple example of usage OCELOT functions to get periodic solution for a storage ring cell.
- Tutorial N2. Tracking.. Web version.
- Linear optics of the European XFEL Injector.
- Tracking. First and second order.
- Artificial beam matching - BeamTransform
- Tutorial N3. Space Charge.. Web version.
- Tracking through RF cavities with SC effects and RF focusing.
- Tutorial N4. Wakefields.. Web version.
- Tracking through corrugated structure (energy chirper) with Wakefields
- Tutorial N5. CSR.. Web version.
- Tracking trough bunch compressor with CSR effect.
- Tutorial N6. RF Coupler Kick.. Web version.
- Coupler Kick. Example of RF coupler kick influence on trajjectory and optics.
- Tutorial N7. Lattice design.. Web version.
- Lattice design, twiss matching, twiss backtracking