#  GEOM GEOM_SWIG : binding of C++ omplementaion with Python
#
#  Copyright (C) 2003  CEA/DEN, EDF R&D
#
#
#
#  File   : GEOM_Partition1.py
#  Module : GEOM

#%Make geometry (like CEA script (A1)) using Partition algorithm%
# appel:
# import alveole_3D_01_GEOM
# reload(alveole_3D_01_GEOM)

# -- Rayon de la bariere

barier_height = 7.0
barier_radius = 5.6 / 2 # Rayon de la bariere
colis_radius = 1.0 / 2  # Rayon du colis
colis_step = 2.0        # Distance s‰parant deux colis
cc_width = 0.11         # Epaisseur du complement de colisage

# --

cc_radius = colis_radius + cc_width
from math import sqrt
colis_center = sqrt(2.0)*colis_step/2

# --

import geompy
geom = geompy.geom

boolean_common  = 1
boolean_cut     = 2
boolean_fuse    = 3
boolean_section = 4

# --

pnt0 = geompy.MakeVertex(0.,0.,0.)
vecz = geompy.MakeVectorDXDYDZ(0.,0.,1.)

barier = geompy.MakeCylinder(
    pnt0,
    vecz,
    barier_radius,
    barier_height)

# --

colis = geompy.MakeCylinder(pnt0, vecz, colis_radius, barier_height)

cc = geompy.MakeCylinder(pnt0, vecz, cc_radius, barier_height)

colis_cc = geompy.MakeCompound([colis, cc])

colis_cc = geompy.MakeTranslation(colis_cc, colis_center, 0.0, 0.0)

colis_cc_multi = geompy.MultiRotate1D(colis_cc, vecz, 4)

# --

alveole = geompy.MakePartition([colis_cc_multi, barier])

geompy.addToStudy(alveole, "alveole before explode")

subshapes = geompy.SubShapeAll(alveole, geompy.ShapeType["SHAPE"])

## there are 9 subshapes

comp1 = geompy.MakeCompound([subshapes[0], subshapes[1]]);
comp2 = geompy.MakeCompound([subshapes[2], subshapes[3]]);
comp3 = geompy.MakeCompound([subshapes[4], subshapes[5]]);
comp4 = geompy.MakeCompound([subshapes[6], subshapes[7]]);

compGOs = []
compGOs.append(comp1);
compGOs.append(comp2);
compGOs.append(comp3);
compGOs.append(comp4);
comp = geompy.MakeCompound(compGOs);

alveole = geompy.MakeCompound([comp, subshapes[8]]);

geompy.addToStudy(alveole, "alveole")