from OCC.AIS import AIS_Shape
from OCC.TopAbs import TopAbs_EDGE
from OCC.TopExp import TopExp_Explorer
from OCC.TopoDS import topods, TopoDS_Shape
from OCC.Quantity import Quantity_NOC_BLACK


def colorTheEdges(shp, display,color,width):
    shapeList  =  []
    Ex = TopExp_Explorer(shp,TopAbs_EDGE)
    ctx = display.Context
    while Ex.More():
        aEdge = topods.Edge(Ex.Current())
        ais_shape = AIS_Shape(aEdge).GetHandle()
        ctx.SetColor(ais_shape,color, False)
        ctx.SetWidth(ais_shape,width, False)
        ctx.Display(ais_shape, False)
        Ex.Next()
        
def setDefaultEdgeStyle(shp, display):
    colorTheEdges(shp, display, Quantity_NOC_BLACK, 1.5)
    #return shps
    
    
def osdagDisplayShape (display, shapes,material=None, texture=None, color=None, transparency=None, update=False):
    setDefaultEdgeStyle(shapes, display)
    display.DisplayShape(shapes,material,texture,color,transparency,update = True)