| 27 | float min_mass = 0; |
| 28 | |
| 29 | void initial_conditions_rand(af::array &mass, vector<af::array> &pos, |
| 30 | vector<af::array> &vels, |
| 31 | vector<af::array> &forces) { |
| 32 | for (int i = 0; i < (int)pos.size(); ++i) { |
| 33 | pos[i] = af::randn(total_particles) * width + width; |
| 34 | vels[i] = 0 * af::randu(total_particles) - 0.5; |
| 35 | forces[i] = af::constant(0, total_particles); |
| 36 | } |
| 37 | mass = af::constant(gravity_constant, total_particles); |
| 38 | } |
| 39 | |
| 40 | void initial_conditions_galaxy(af::array &mass, vector<af::array> &pos, |
| 41 | vector<af::array> &vels, |