ParticleSimulation/ParticlesSetup.py at master · FelixFifi/ParticleSimulation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
from random import randint, randrange, choice, uniform
from Particles.GravityParticle import GravityParticle
from Particles.ChargedParticle import ChargedParticle
from Particles.ForceFunctionParticle import ForceFunctionParticle
from ForceFunctions import *
from Particles.Particle import Particle

import numpy as np


def random_position(width, height):
    return np.array([randint(0, width - 1), randint(0, height - 1)], dtype=np.float)


def random_velocity(max_v):
    return np.array([randint(-max_v, max_v), randint(-max_v, max_v)], dtype=np.float)


def random_gravity_particles(num_particles, width, height):
    particles = []
    for i in range(num_particles):
        elasticity = 0.5
        mass = 4000000000
        velocity = np.array([0.0, 0.0], dtype=np.float)
        position = random_position(width, height)
        particle = GravityParticle(position, velocity, width, height,
                                   mass=mass, elasticity=elasticity)
        particles.append(particle)
    return particles


def random_charged_particles(num_particles, width, height):
    particles = []

    charge_strength = 0.00001
    charges = [-charge_strength, charge_strength]
    for i in range(num_particles):
        position = random_position(width, height)
        velocity = np.array([0.0, 0.0], dtype=np.float)

        charge = choice(charges)
        particle = ChargedParticle(position, velocity, charge, width, height, mass=1, elasticity=0.5)
        particles.append(particle)
    return particles


def random_attract_repel_particles(num_particles, width, height):
    particles = []

    force_function = attract_and_repel
    for i in range(num_particles):
        position = random_position(width, height)
        velocity = np.array([0.0, 0.0], dtype=np.float)

        particle = ForceFunctionParticle(position, force_function, width, height, mass=1, elasticity=0.5)
        particles.append(particle)
    return particles


def random_repel_particles(num_particles, width, height):
    particles = []

    force_function = repel_1_div_r2
    for i in range(num_particles):
        position = random_position(width, height)

        particle = ForceFunctionParticle(position, force_function, width, height, mass=1, elasticity=0.5)
        particles.append(particle)
    return particles


def random_identical_parameterized_particles(num_particles, width, height):
    particles = []

    force_function = parameterized_force_function_type1(100, 30, -0.5, 100)
    for i in range(num_particles):
        position = random_position(width, height)

        particle = ForceFunctionParticle(position, force_function, width, height, mass=1, elasticity=0.5)
        particles.append(particle)
    return particles


def random_different_groups_parameterized_particles(num_particles, width, height):
    particles = []

    radius = 5
    friction = 0.3

    num_groups = 5
    for i_group in range(num_groups):
        color = Particle.random_color()

        factor_repelling = float(uniform(5, 20))
        distance_mode_change = float(uniform(radius * 3, radius * 5))
        max_strength_slopes = float(uniform(-1, 1))
        distance_force_end = float(uniform(distance_mode_change * 2, distance_mode_change * 7))
        force_function = parameterized_force_function_type1(factor_repelling, distance_mode_change, max_strength_slopes,
                                                            distance_force_end)
        print(f"{factor_repelling} {distance_mode_change} {max_strength_slopes} {distance_force_end}")
        for i in range(num_particles // num_groups):
            position = random_position(width, height)

            particle = ForceFunctionParticle(position, force_function, width, height,
                                             mass=1, elasticity=0.5, radius=radius, color=color, friction=friction)
            particles.append(particle)
    return particles