Tools borrowed and create for use in academic research througout Kyle Cole's PhD research.
streams.py - Module for generating streamfiles
fourier_diffraction.py - Module for modeling diffraction in the far field regime with hologram gratings represented by numpy arrays
grating_generation.py - Module for generating numpy arrays which represent hologram gratings
diffractionEfficiencyCalculator.py - Code using fourier_diffraction.py and grating_generation.py to calculate the diffraction efficiency for a binary, or blazed grating (can easily add other kinds of gratings by modifying the inital code block)
BinaryStreamFileGen.ipynb - Notebook demonstrating the basic workflow for generating streamfiles using the streams module and the grating_generation module
https://github.com/0x23/WaveSimulator2D
This repository contains a lightweight 2D wave simulator running on the GPU using CuPy library (probably requires a NVIDIA GPU). It can be used for 2D light and sound simulations. A simple visualizer shows the field and its intensity on the screen and writes a movie file for each to disks. The goal is to provide a fast, easy to use but still felxible wave simulator.
- Scene objects can now draw to a visualization layer (most of them do not yet, feel free to contribute) !
- Example 4 now shows a two-mirror optical cavity and how standing waves emerge.
- Added new Line Sources
- Added Refractive index Polygon object (StaticRefractiveIndexPolygon)
- Added Refractive index Box object (StaticRefractiveIndexBox)
- Fixed some issues with the examples
- Refactored the code to support a more flexible scene description. A simulation scene now consists of a list of objects that add their contribution to the fields. They can be combined to build complex and time dependent simulations. The refactoring also made the core simulation code even simpler.
- Added a few new custom colormaps that work well for wave simulations.
- Added new examples, which should make it easier to understand the usage of the program and how you can setup your own simulations: examples.
The old image based scene description is still available as a scene object. You can continue to use the convenience of an image editing software and create simulations without much programming.
When using the 'StaticImageScene' class the simulation scenes can given as an 8Bit RGB image with the following channel semantics:
- Red: The Refractive index times 100 (for refractive index 1.5 you would use value 150)
- Green: Each pixel with a green value above 0 is a sinusoidal wave source. The green value defines its frequency.
- Blue: Absorbtion field. Larger values correspond to higher dampening of the waves, use graduated transitions to avoid reflections
WARNING: Do not use anti-aliasing for the green channel ! The shades produced are interpreted as different source frequencies, which yields weird results.
- Install Python and PyCharm IDE
- Clone the Project to you hard disk
- Open the folder as a Project using PyCharm
- If prompted to install requirements, accept (or install requirements using pip -r requirements.txt)
- Right click on one of the examples in wave_sim2d/examples and select run





