STEP file reading issue in Python.

[dev1intellial]

I am using Python3.4.2 and pythonOCC-0.16.0-win32-py34.exe to draw components. Every components are rendered properly with one defined color but that is not look like a real world component.

Above image is my Python implementation which generate 3D image from STEP file with one color.

Below image is rendered one of my windows software and there I have used Step file. I want to render component same as look like in below image so its look like a real world component.

Is there any way to get correct colored output in Python by read STEP file? I have searched a lot but didn't get a way to implement it. Please help me to go in forward direction.

from future import print_function
import sys

from OCC.STEPCAFControl import STEPCAFControl_Reader

from OCC.STEPControl import STEPControl_Reader
from OCC.IFSelect import IFSelect_RetDone, IFSelect_ItemsByEntity
from OCC.Display.SimpleGui import init_display
from OCC.Display.WebGl import threejs_renderer

step_reader = STEPControl_Reader()
status = step_reader.ReadFile('./models/test.STEP')

if status == IFSelect_RetDone: # check status
failsonly = False
step_reader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity)
step_reader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity)

ok = step_reader.TransferRoot(1)
_nbs = step_reader.NbShapes()
aResShape = step_reader.Shape(1)

else:
print("Error: can't read file.")
sys.exit(0)

display, start_display, add_menu, add_function_to_menu = init_display()

display.DisplayShape(aResShape, update=True)

start_display()

my_renderer = threejs_renderer.ThreejsRenderer(background_color="#123345")
my_renderer.DisplayShape(aResShape)

The above code is used for read STEP file using Python.

[priyankatgit]

@dev1intellial @jf--- @tpaviot
I am too facing exactly same issue as in described question. Does pythonocc supports more then one color for object if its defined in step file?
So far I have tested number of objects, and all have same color where originally they do have more then one color.

[tpaviot]

Yes, OCC supports import colors into STEP files, so pythonocc does. Hey guys, what's the matter with exporting color using STEP? honestly, who cares ?

[dev1intellial]

Thanx for reply. Can you give one example which will display object with exact color?
Right now I am working this below example ..

from future import print_function
import OCC.XCAFDoc

from OCC.TCollection import TCollection_ExtendedString

from OCC.TDocStd import Handle_TDocStd_Document
from OCC.XCAFApp import XCAFApp_Application
from OCC.XCAFDoc import (XCAFDoc_DocumentTool_ShapeTool,
XCAFDoc_DocumentTool_ColorTool,
XCAFDoc_DocumentTool_LayerTool,
XCAFDoc_DocumentTool_MaterialTool)
from OCC.STEPCAFControl import STEPCAFControl_Reader
from OCC.IFSelect import IFSelect_RetDone
from OCC.TDF import TDF_LabelSequence

from OCC.Display.SimpleGui import init_display

filename = './models/lemo.STEP'
_shapes = []

create an handle to a document

h_doc = Handle_TDocStd_Document()

Create the application

app = XCAFApp_Application.GetApplication().GetObject()
app.NewDocument(TCollection_ExtendedString("MDTV-CAF"), h_doc)

Get root assembly

doc = h_doc.GetObject()
h_shape_tool = XCAFDoc_DocumentTool_ShapeTool(doc.Main())
l_colors = XCAFDoc_DocumentTool_ColorTool(doc.Main())
l_layers = XCAFDoc_DocumentTool_LayerTool(doc.Main())
l_materials = XCAFDoc_DocumentTool_MaterialTool(doc.Main())

step_reader = STEPCAFControl_Reader()

step_reader.SetColorMode(True)
step_reader.SetLayerMode(True)
step_reader.SetNameMode(True)
step_reader.SetMatMode(True)

status = step_reader.ReadFile(filename)
if status == IFSelect_RetDone:
step_reader.Transfer(doc.GetHandle())

labels = TDF_LabelSequence()
color_labels = TDF_LabelSequence()

shape_tool = h_shape_tool.GetObject()
h_shape_tool.GetObject().GetFreeShapes(labels)

print("Number of shapes at root :%i" % labels.Length())
for i in range(labels.Length()):
sub_shapes_labels = TDF_LabelSequence()
print("Is Assembly :", shape_tool.IsAssembly(labels.Value(i+1)))
sub_shapes = shape_tool.GetSubShapes(labels.Value(i+1), sub_shapes_labels)
print("Number of subshapes in the assemly :%i" % sub_shapes_labels.Length())
for j in range(sub_shapes_labels.Length()) :
l_colors.GetObject().GetColors(color_labels)
l_colors.GetObject().SetColor(sub_shapes, 4, OCC.XCAFDoc.XCAFDoc_ColorGen)
#print("Number of subshapes in the j assemly :%j" % sub_shapes_labels.Length())

print("Number of colors=%i" % color_labels.Length())
for i in range(color_labels.Length()):
color = color_labels.Value(i+1)
print(color.DumpToString())

for i in range(labels.Length()):
label = labels.Value(i+1)
a_shape = h_shape_tool.GetObject().GetShape(label)
m = l_layers.GetObject().GetLayers(a_shape)
a_shape.Modified
l_colors.GetObject().SetColor(a_shape, color_labels.Value(i+1), OCC.XCAFDoc.XCAFDoc_ColorGen)
_shapes.append(a_shape)

Display

display, start_display, add_menu, add_function_to_menu = init_display()
display.DisplayShape(_shapes, update=True)
start_display()

In the above example, I am trying to show exact color of an object but failure every time. Even can't identify the individual assembly object. Give some idea or example for doing all this things.
Please help.

[jf---]

@tpaviot , lets close the issue...

[tpaviot]

@dev1intellial Please post this thread to the pythonocc google groups if you still fail to get it work.