import copy
from pylmgc90.pre_lmgc import *

import os
if not os.path.isdir('./DATBOX'):
  os.mkdir('./DATBOX')

# definition des conteneurs:
#   * de corps
bodies = avatars()
#   * de modeles
mods = models()
#   * de materiaux
mats = materials()
#   * pour les tables de visibilite
svs = see_tables()
#   * pour les lois de contact
tacts = tact_behavs()

# on se place en 3D
dim = 3

# modele de rigide
mR3D = model(name='rigid', type='MECAx', element='Rxx3D', dimension=3)

# materiau rigide
tdur = material(name='TDURx', type='RIGID', density=2500.)
mats.addMaterial(tdur)

# modele elastique 
m3Dl = model(name='M3DH8', type='MECAx', element='H8xxx', dimension=3, 
     external_model='yes__', kinematic='small', material='elas_', 
     anisotropy='iso__', mass_storage='coher')
mods.addModel(m3Dl)

# materiau elastique
stone = material(name='stone', type='ELAS', density=2500., elas='standard',
   anisotropy='isotropic', young=3.5e10, nu=0.2)  
mats.addMaterial(stone)

# maillage du barreau
mesh_cube=buildMeshH8(x0=-0.5, y0=-0.5, z0=0., lx=1., ly=1., lz=2., nb_elem_x=4, nb_elem_y=4, nb_elem_z=8)

cube=buildMeshedAvatar(model=m3Dl, material=stone,mesh=mesh_cube)

## contacteurs :
cube.addContactors(group='down', type='CSpx4', color='BLEUx')
bodies += cube

# construction d'un polyedre rigide pour la fondation

floor_mesh = lecture('../../../gmsh/fondation.msh', dim)
floor=volumicMeshToRigid3D(volumic_mesh=floor_mesh, model=mR3D, material=tdur, color='FLOOR')

# on bloque la fondation
floor.imposeDrivenDof(component=[1, 2, 3, 5, 6], dofty='vlocy')
floor.imposeDrivenDof(component=4, dofty='vlocy',ct=0.05,rampi=1.)
floor.translate(dz=-0.2)

# ajout de la fondation dans le conteneur de corps
bodies.addAvatar(floor)

# gestion des interactions :

lcspr=tact_behav('gapc1', 'GAP_SGR_CLB', fric=0.5)
tacts+=lcspr

svcspr = see_table(CorpsCandidat='MAILx', candidat='CSxxx',
   colorCandidat='BLEUx', behav=lcspr, CorpsAntagoniste='RBDY3', 
   antagoniste='POLYR', colorAntagoniste='FLOOR', alert=0.1, halo=0.5)
svs+=svcspr

# Ecriture des fichiers pour LMGC
writeBodies(bodies, chemin='./DATBOX/')
writeDofIni(bodies, chemin='./DATBOX/')
writeDrvDof(bodies, chemin='./DATBOX/')
writeModels(mods,chemin='./DATBOX/')
writeGPVIni(bodies, chemin='./DATBOX/')
writeBulkBehav(mats, chemin='./DATBOX/', dim=dim)
writeTactBehav(tacts,svs,chemin='DATBOX/')
writeVlocRlocIni(chemin='DATBOX/')

### postpro
post = postpro_commands()
set_cube1 = [(cube, "up")]
set_cube2 = [(cube, "down")]
cube_sets = postpro_command(type='NEW MECAx SETS',mecax_sets=[set_cube1,set_cube2])
post.addCommand(cube_sets)
# on suit leur deplacement
cube_disp = postpro_command(type='Dep EVOLUTION', step=1)
post.addCommand(cube_disp)
cube_fint = postpro_command(type='Fint EVOLUTION', step=1)
post.addCommand(cube_fint)
# suivi du sol :
#   * cinematique de la brique
floor_disp = postpro_command(type='BODY TRACKING', step=1,rigid_set=[floor])
post.addCommand(floor_disp)
#   * efforts subis par la brique
floor_torque = postpro_command(type='TORQUE EVOLUTION', step=1,rigid_set=[floor])
post.addCommand(floor_torque)
#
energy = postpro_command(type='KINETIC ENERGY', step=1)
post.addCommand(energy)
#
writePostpro(post, bodies, path='DATBOX/')

try:
  visuAvatars(bodies)
except:
  pass