#!/usr/bin/env python
# encoding: utf-8
# Anton Gladky <gladk@debian.org>

# If you want to change the number of generated particles,
# change it in line 44. Do not forget yo increase the firstIterRun
# in line 62 not to revolute the hourglass before particles are
# settled down at the bottom.


from yade import ymport
import time

utils.readParamsFromTable(descriptionIn = 'noDescription', 
  frIn = 0.5, enIn=0.01, etIn=0.01, tcIn=0.0001,                 # Frictionangle (rad), strength-parameter
  rhoIn = 1500.0,                                                # Density
  dumpVTKIn = 4000                                               # Periods of dumps in iterations
)

from yade.params.table import *

import shutil

try:
  shutil.rmtree('cpt')
except OSError:
  pass
os.mkdir('cpt')

folderNameBase = 'cpt/' + str(descriptionIn)
folderName = folderNameBase
os.mkdir(folderNameBase)

o = Omega()
o.dt = 0.05*tcIn
rotPeriod = 0.01
rotPeriodIter = int(rotPeriod/o.dt)

mat1 = O.materials.append(ViscElMat(frictionAngle=frIn, density=rhoIn,tc=tcIn, en=enIn, et=etIn,))
id_HourGl = O.bodies.append(ymport.gmsh("hourglass.mesh",scale=1.0, material=mat1,color=(0,0,1),mask=5))

o.engines = [
  ForceResetter(),
  InsertionSortCollider([Bo1_Sphere_Aabb(),Bo1_Facet_Aabb()],verletDist=(2.0e-3),label='collider',ompThreads=1),
  InteractionLoop(
    [Ig2_Sphere_Sphere_ScGeom(),Ig2_Facet_Sphere_ScGeom()],
    [Ip2_ViscElMat_ViscElMat_ViscElPhys()],
    [Law2_ScGeom_ViscElPhys_Basic()],
  ),
  NewtonIntegrator(damping=0,gravity=Vector3(0,0,-90.81), label='newt'),
  CircularFactory(maxParticles=1500,            #Number of generated particles
    center=(0.0e-3, 0.0e-3,20.0e-3),
    radius = (12.0e-3),
    PSDsizes = [0.5e-3],
    PSDcum   = [1.0],
    vMin=0.1,vMax=0.1,vAngle=0,massFlowRate=0.01,
    normal=(0.0,0.0,-1.0),label='factorySpheres',mask=3,materialId=mat1,
    stopIfFailed=False),
  RotationEngine(ids=id_HourGl, angularVelocity = math.pi/rotPeriod,
    rotateAroundZero=True, rotationAxis=Vector3(1,0,0),
    zeroPoint=Vector3(0.,0.,0.), label='rotEng', dead=True),
  VTKRecorder(iterPeriod=dumpVTKIn,fileName=folderName+'/spheres-',mask=2,
    recorders=['spheres','velocity','colors','intr','ids','mask','materialId','stress'], label='VTK_box1'),
  VTKRecorder(iterPeriod=dumpVTKIn,fileName=folderName+'/facetB-',mask=4,
    recorders=['facets'],label='VTK_box2'),
  PyRunner(iterPeriod=rotPeriodIter,initRun=False,firstIterRun=13000,command='startStepRotation()'),
]

def startStepRotation():
  if (rotEng.dead):
    rotEng.dead = False
  else:
    rotEng.dead = True

from yade import qt
qt.View()
r=qt.Renderer()
