Problem with the combined engine

Asked by Przemek

Hi,

I'm new with Yade, and I have problem with combined engine. I want to change the translationAxis and the velocity after a few iterations but I still have an issue about the global variables.
Could you help me?

#!/usr/bin/python
# -*- coding: utf-8 -*-

from __future__ import print_function
from builtins import range
import random
from yade import ymport
from yade import pack, qt

## PhysicalParameters

## Import geometry
rod = O.bodies.append(ymport.stl('tool.stl',wire=False,color=[0.31,0.47,0.47]))

# Plate dimension
thickness = 1.0 # cm
width = 5.0 # cm
length = 11.0 # cm
boxMax = (2,2.5,0)

# Spheres
sphereRadius = 0.05 # cm
ndiv_x = length/(2*sphereRadius)
ndiv_y = width/(2*sphereRadius)
ndiv_z = thickness/(2*sphereRadius)
#nbSpheres = (ndiv_x,ndiv_y,ndiv_z)
nbSpheres = (40,40,10)#(110,50,10)
print("Creating %d spheres..."%(nbSpheres[0]*nbSpheres[1]*nbSpheres[2]), end=' ')
for i in range(nbSpheres[0]):
 for j in range(nbSpheres[1]):
  for k in range(nbSpheres[2]):
   x = boxMax[0]-2.0*sphereRadius*(i)
   y = boxMax[1]-2.0*sphereRadius*(j)
   z = boxMax[2]-2.0*sphereRadius*(k)
   r = random.uniform(sphereRadius,sphereRadius)
   fixed = False
   color=[0.21,0.22,0.1]
   if (i==0 or i==nbSpheres[0]-1 or k==nbSpheres[2]-1 or j==0 or j==nbSpheres[1]-1):
    fixed = True
    color=[0.21,0.22,0.3]
   O.bodies.append(sphere([x,y,z],r,color=color,fixed=fixed))
print("done\n")

## Estimate time step
#O.dt=PWaveTimeStep()
O.dt=0.0001

def updateParameters():
 aTime = (O.iter*O.dt)
 if aTime>0.005: # fist part
  velVec = Vector3(-1,0,0)
  tranVel = 1.0
 transEngine.translationAxis = velVec
 transEngine.velocity = tranVel

## Engines
O.engines=[
 ForceResetter(), # 0
 InsertionSortCollider([ # 1
  Bo1_Sphere_Aabb(),
  Bo1_Facet_Aabb(),
 ]),
 InteractionLoop( # 2
  [Ig2_Sphere_Sphere_ScGeom(),Ig2_Facet_Sphere_ScGeom()],
  [Ip2_FrictMat_FrictMat_FrictPhys()],
  [Law2_ScGeom_FrictPhys_CundallStrack()]
 ),
 CombinedKinematicEngine(ids=rod,label='combEngine') + TranslationEngine(translationAxis=(0,0,-1),velocity=0.6) + RotationEngine(rotationAxis=(0,0,-1), angularVelocity=20, rotateAroundZero=True, zeroPoint=(0,0,0)),
 NewtonIntegrator(damping=2.0,gravity=[0,0,-9.81]), # 3
 PyRunner(iterPeriod=1, command='updateParameters()'),
 # save data from Yade's own 3d view
 #qt.SnapshotEngine(fileBase='3d-',iterPeriod=1,label='snapshot'),
 #PyRunner(command='finish()',iterPeriod=10)
]

import sys,time

print("Start simulation: ")
nbIter=60

from yade import qt
v=qt.View()
v.eyePosition=(6,2,2); v.upVector=(-.4,-.4,.8); v.viewDir=(-.9,-.25,-.3); v.axes=False; v.sceneRadius=20

O.stopAtIter=nbIter
O.run()

# this function is called when the simulation is finished
def finish():
 # snapshot is label of qt.SnapshotEngine
 # the 'snapshots' attribute contains list of all saved files
 makeVideo(snapshot.snapshots,'3d.mpeg',fps=1,bps=10)
 O.pause()

#for t in xrange(2):
# start=time.time();O.run(nbIter);O.wait();finish=time.time()
# speed=nbIter/(finish-start); print '%g iter/sec\n'%speed
#print "FINISH"
#quit()

BR
Przemek

Question information

Language:
English Edit question
Status:
Answered
For:
Yade Edit question
Assignee:
No assignee Edit question
Last query:
Last reply:
Revision history for this message
Jérôme Duriez (jduriez) said :
#1

What is your "issue about the global variables" ? (the error message)

Revision history for this message
Jan Stránský (honzik) said :
#2

Hello,

> I'm new with Yade

welcome :-)

> I still have an issue about the global variables

please read [1] and next time try:
- to be more specific (like including the error message in the question)
- include a MWE (W=working). If external files (like too.stl file) is involved, try to include it in the question, or preferably not use it and e.g. create a few facets manually instead (if the problem is not related to the stl file). Without it, we cannot try the code ourselves..

I guess the problem is that transEngine is not defined in updateParameters() function. You call it, but you do not define it anywhere in the script and Python interpreter has no clue what it is..

Try one of these:

a) set label to TranslationEngine when it is created
... TranslationEngine(..., label='transEngine')

b) create TranslationEngine before O.engines and then add the instance to CombinedKinematicEngine
transEngine = TranslationEngine(...)
O.engines = [
   ...
   CombinedKinematicEngine(...) + transEngine + ...

c) define transEngine in updateParameters() from combEngine content
transEngine = combEngine.comb[0]
transEngine.translationAxis = velVec
transEngine.velocity = tranVel

cheers
Jan

[1] https://www.yade-dem.org/wiki/Howtoask

Revision history for this message
Przemek (przemekn) said :
#3

Sorry, next time I'll adjust to guidelines.

Here is the error:

UnboundLocalError Traceback (most recent call last)
/usr/bin/yade in <module>()

/usr/bin/yade in updateParameters()
     52 tranVel = 1.0
     53 transEngine = combEngine.comb[0]
---> 54 transEngine.translationAxis = velVec
     55 transEngine.velocity = tranVel
     56

UnboundLocalError: local variable 'velVec' referenced before assignment

I tried to modify the script by yous suggestions but it still doesn't work.

STL file: https://www.dropbox.com/s/a7dvy3khefjpl3c/tool.stl?dl=0

Revision history for this message
Jan Stránský (honzik) said :
#4

Hello,

velVec is assigned a value inside an if condition (if aTime>0.005).
This condition is false (you can easily check, O.dt=0.0001, iterPeriod=1) and velVec therefore has no value.
Then you try to assign it to translation engine, but it is currently just nothing, so Python comes with this error.

cheers
Jan

PS: have look at [1], section "Please, no external links!" ;-)

PPS: aTime = (O.iter*O.dt) ... you can use directly aTime = O.time

[1] https://www.yade-dem.org/wiki/Howtoask

Revision history for this message
Przemek (przemekn) said :
#5

Hi,

now it's working :) Thanks a lot!

Can you help with this problem?

Provide an answer of your own, or ask Przemek for more information if necessary.

To post a message you must log in.