# -*- coding: utf-8 -*-
from OCC.gp import *
from OCC.BRepPrimAPI import *
from OCC.TopExp import *
from OCC.TopAbs import *
from OCC.BRep import *
from OCC.Geom import *
from OCC.GCE2d import *
from OCC.GC import *
from OCC.Geom2d import *
from OCC.BRepLib import *
from OCC.BRepFeat import *
from OCC.Utils.Topology import Topo
from OCC.BRepBuilderAPI import *
from OCC.BRepAlgoAPI import *
from OCC.BRepFilletAPI import *
from OCC.TopoDS import *
from OCC.StlAPI import *
from math import radians

from OCC.Display.SimpleGui import *
from OCC.Graphic3d import *
from OCC.BRepFeat import *

R = 31
L = 120
p = gp_Pnt(0.,0.,0)
d = gp_Dir(1. ,0., 0.) 
myAxes1 = gp_Ax2(p,d) 
myPrim1 = BRepPrimAPI_MakeCylinder(myAxes1, R, 100)

mkFillet = BRepFilletAPI_MakeChamfer(myPrim1.Shape())
ex = TopExp_Explorer(myPrim1.Shape(), TopAbs_EDGE)
ex2 = TopExp_Explorer(myPrim1.Shape(), TopAbs_FACE)
edge = TopoDS_edge(ex.Current())
face = TopoDS_face(ex2.Current())
mkFillet.Add(15.0, 3.0, edge, face)

myAxes2 = gp_Ax2(p, gp_Dir(0,-1,0))
myPrim2 = BRepPrimAPI_MakeCylinder(myAxes1, R-5, L)
myShp1 = BRepAlgoAPI_Cut(mkFillet.Shape(),myPrim2.Shape())

mySphere =  BRepPrimAPI_MakeSphere(p, R)

rotateAxis = gp_Ax1(p, gp_Dir(0.0,0.0,1.0)) #Os rotacije
transfRot1 = gp_Trsf()
transfRot1.SetRotation(rotateAxis, radians(90.0))
myShp2 = BRepBuilderAPI_Transform(myShp1.Shape(), transfRot1)
myShp3 = BRepAlgoAPI_Fuse(myShp1.Shape(),myShp2.Shape())
myShp4 = BRepAlgoAPI_Fuse(myShp3.Shape(), mySphere.Shape())


# Izdelamo navadno rebro
rib_thickness = 6.0 # polovica debeline rebra
rib1 = gp_Pnt(2*R,R,0)
rib2 = gp_Pnt(R,2*R,0)
aPlane = Geom_Plane(0,0,1.0,0.0)
edge1 = BRepBuilderAPI_MakeEdge(rib1, rib2).Edge()
aWire = BRepBuilderAPI_MakeWire(edge1)
rib = BRepFeat_MakeLinearForm( myShp4.Shape(), aWire.Wire(), aPlane.GetHandle(), \
                gp_Vec(0.,0.,rib_thickness), gp_Vec(0.,0.,-rib_thickness), 1, True )
rib.Perform()

# Poiščemo vse štiri robove na obeh debelinah rebra, ki imajo začetno in končno Z isto
# in so vzporedni z osjo X ali Y ter jim dodamo spremenljivo debelino zaokrožitve
TOL=1E-6
with_fillet = BRepFilletAPI_MakeFillet(rib.Shape())
topology_traverser = Topo(rib.Shape())
for aEdge in topology_traverser.edges():  
    first, last = TopExp().FirstVertex(aEdge), TopExp().LastVertex(aEdge)
    first_vert, last_vert = BRep_Tool().Pnt(first), BRep_Tool().Pnt(last)
    if abs(abs(first_vert.Z())-rib_thickness) < TOL and \
       abs(first_vert.Z() - last_vert.Z()) < TOL and \
      (abs(first_vert.X()-last_vert.X())<TOL or abs(first_vert.Y()-last_vert.Y())<TOL):
        with_fillet.Add(25, 0.0, aEdge)

myPart = with_fillet

stl_writer = StlAPI()
stl_writer.Write(myPart.Shape(), "elbow.stl", False)

display, start_display, add_menu, add_function_to_menu = init_display()
material = Graphic3d_MaterialAspect(Graphic3d_NOM_SILVER)
display.DisplayShape(myPart.Shape(), material)
start_display()