https://github.com/pasqualemainolfi/msdnet
Build complex mass-spring-damper networks in python
https://github.com/pasqualemainolfi/msdnet
Last synced: 2 months ago
JSON representation
Build complex mass-spring-damper networks in python
- Host: GitHub
- URL: https://github.com/pasqualemainolfi/msdnet
- Owner: PasqualeMainolfi
- License: other
- Created: 2022-11-21T23:51:08.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2023-03-03T11:08:55.000Z (about 2 years ago)
- Last Synced: 2023-03-10T14:23:35.452Z (about 2 years ago)
- Language: Python
- Size: 3.63 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
## **MSDNet MASS-SPRING-DAMPER NETWORK**
MSDNet is a tool for building complex and modular mass-spring-damper networks in python (use Verlet integration).
1. Import and create network obj
```python
from msdnet import MSDNet
import pygame as pg
net = MSDNet()
```
2. Define network params
```python
pg.init()
w, h = 800, 800
win = (w, h)
screen = pg.display.set_mode(win)
clock = pg.time.Clock()
fps = 60
net = MSDNet()
N_MASS = 30
N_SPRING = N_MASS - 1
N_DAMPER = N_SPRING
DT = 1
l = 1/(N_MASS + 1)
L = l
M = 50
D = 0.981
K = 10
C = 5
R = 5
g = (0, 0.00001, 0)
net.add_gravity(g=g)
net.add_dt(DT)
# masses level 1
p = l
for m in range(N_MASS):
net.add_mass(name=f"l1m{m}", m=M, pos=[p, 0.1, 0], r=R, d=D, anchored=False)
net.lock_unlock_mass(name=f"l1m{m}", anchored=True)
p += l
# masse level 2
p = l
for m in range(N_MASS):
net.add_mass(name=f"l2m{m}", m=M, pos=[p, 0.15, 0], r=R, d=D, anchored=False)
p += l
p = l
for m in range(N_MASS):
net.add_mass(name=f"l3m{m}", m=M, pos=[p, 0.20, 0], r=R, d=D, anchored=False)
p += l
# springs level 1
for s in range(N_SPRING):
net.add_spring(name=f"l1s{s}", k=K, length=L, m1=f"l1m{s}", m2=f"l1m{s+1}")
# springs level 2
for s in range(N_SPRING):
net.add_spring(name=f"l2s{s}", k=K, length=L, m1=f"l2m{s + 1}", m2=f"l2m{s}")
# springs level 3
for s in range(N_SPRING):
net.add_spring(name=f"l3s{s}", k=K, length=L, m1=f"l3m{s + 1}", m2=f"l3m{s}")
# springs level inter 1
for s in range(N_MASS):
net.add_spring(name=f"li1s{s}", k=K, length=0.05, m1=f"l1m{s}", m2=f"l2m{s}")
# springs level inter 2
for s in range(N_MASS):
net.add_spring(name=f"li2s{s}", k=K, length=0.05, m1=f"l2m{s}", m2=f"l3m{s}")
# dampers
for d in range(N_DAMPER):
net.add_damper(name=f"l1d{d}", c=C, spring=f"l1s{d}")
# dampers
for d in range(N_DAMPER):
net.add_damper(name=f"l2d{d}", c=C, spring=f"l2s{d}")
# dampers
for d in range(N_DAMPER):
net.add_damper(name=f"l3d{d}", c=C, spring=f"l3s{d}")
# dampers
for d in range(N_MASS):
net.add_damper(name=f"li1d{d}", c=C, spring=f"li1s{d}")
# dampers
for d in range(N_MASS):
net.add_damper(name=f"li2d{d}", c=C, spring=f"li2s{d}")
N = N_MASS * 3
pos_mass_motion = [None for _ in range(N)]
run = True
while run:
for event in pg.event.get():
if event.type == pg.QUIT:
run = False
pg.quit()
# return network in motion -> current position of all the masses
network = net.run_network(acc_is_costant=False, clip_pos=(0, 1))
for mass in network:
m = network[mass]
x = m["x"] * w
y = m["y"] * h
pg.draw.circle(surface=screen, color=(255, 255, 0), center=(x, y), radius=R)
clock.tick(fps)
pg.display.update()
screen.fill((0, 0, 0))
```
Run and interact with network, using render method:
move (left button pressed) and free (right button pressed) the masses
...use msdnet_tools.shape (see source)
```python
from msdnet_tools.shapes import Cloth
M = 50
D = 0.981
K = 10
C = 10
R = 5
g = (0, 0.000015, 0)
cloth = Cloth(n_masses=30, levels=10, origin=(0, 0.3), scale=(1, 0.1), g=g, dt=1)
net = cloth.generate_cloth(m=M, d=D, k=K, c=C, r=R)
net.render(canvas_size=(800, 800), clip_pos=(0, 1), fps=60)
```
for any questions: [email protected]
© PasqualeMainolfi2022