File: remove-body.py

package info (click to toggle)
yade 2025.2.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 33,308 kB
  • sloc: cpp: 93,298; python: 50,409; sh: 577; makefile: 162
file content (56 lines) | stat: -rw-r--r-- 1,543 bytes parent folder | download | duplicates (2)
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
# -*- encoding=utf-8 -*-
# © Václav Šmilauer <eudoxos@arcig.cz>
#
# Test case for sphere-facet interaction.

#O.bodyContainer="BodyVector"

O.engines = [
        ForceResetter(),
        InsertionSortCollider([Bo1_Sphere_Aabb(), Bo1_Facet_Aabb()], nBins=5, sweepLength=5e-3),
        #SpatialQuickSortCollider(),
        InteractionLoop(
                [Ig2_Facet_Sphere_ScGeom()],
                [Ip2_FrictMat_FrictMat_FrictPhys()],
                [Law2_ScGeom_FrictPhys_CundallStrack()],
        ),
        NewtonIntegrator(damping=0.01, gravity=[0, 0, -10]),
]
O.bodies.append([facet([[-1, -1, 0], [1, -1, 0], [0, 1, 0]], fixed=True, color=[1, 0, 0]),
                 facet([[1, -1, 0], [
                         0,
                         1,
                         0,
                 ], [1, .5, .5]], fixed=True)])

#Gl1_Facet(normals=True)

import random, sys


def addRandomSphere():
	return O.bodies.append(sphere([random.gauss(0, 1), random.gauss(0, 1), random.uniform(1, 2)], random.uniform(.02, .05)))
	O.bodies[len(O.bodies) - 1].state.vel = [random.gauss(0, .1), random.gauss(0, .1), random.gauss(0, .1)]


for i in range(0, 100):
	addRandomSphere()

O.dt = 1e-4
#O.run()
O.saveTmp('init')
from yade import qt

qt.Controller()
qt.View()
if 1:
	for i in range(0, 1000):
		O.run(50, True)
		if random.choice([True, False]):
			idOld = random.randint(2, len(O.bodies) - 1)
			O.bodies.erase(idOld)
			print("-%d" % idOld, end=' ')
		else:
			idNew = addRandomSphere()
			print("+%d" % idNew, end=' ')
		sys.stdout.flush()