-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathscenario_grille_carre.py
32 lines (27 loc) · 1.1 KB
/
scenario_grille_carre.py
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
#!/usr/bin/env python
# -*- coding: utf8 -*-
"""
Dans une grille en tore (pacman) privilégie les co-linéarités à angle droit.
"""
import elasticite as el
import numpy as np
class EdgeGrid(el.EdgeGrid):
def champ(self):
force = np.zeros_like(self.lames[2, :])
damp = lambda t: 0.001
noise = lambda t: 2* np.exp((np.cos(2*np.pi*(t-0.) / 6.)-1.)/ 1.**2)
colin_t = lambda t: -2.*np.exp((np.cos(2*np.pi*(t-4.) / 6.)-1.)/ 1.**2)
colin_d = lambda d: np.exp(-d/.05)
#colin_d = lambda d: -1. + 2*np.exp(-np.log((d+1.e-12)/.1)**2/2/1.5**2)
delta_angle = self.angle_relatif()-np.pi/2.
#delta_angle *= np.sign(delta_angle)
force += colin_t(self.t) * np.sum(np.sin(4*delta_angle)*colin_d(self.distance(do_torus=True)), axis=1)
force += noise(self.t)*np.pi*np.random.randn(self.N_lame)
force -= damp(self.t) * self.lames[3, :]/self.dt
return 1. * force
if __name__ == "__main__":
import sys
if len(sys.argv)>1: mode = sys.argv[1]
else: mode = 'both'
e = el.EdgeGrid(N_lame=72, mode=mode, verb=False)
el.main(e)