def read_step_file(filename, return_as_shapes=False, verbosity=False):
""" read the STEP file and returns a compound
filename: the file path
return_as_shapes: optional, False by default. If True returns a list of shapes,
else returns a single compound
verbosity: optionl, False by default.
"""
if not os.path.isfile(filename):
raise FileNotFoundError("%s not found." % filename)
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(filename)
if status == IFSelect_RetDone: # check status
if verbosity:
failsonly = False
step_reader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity)
step_reader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity)
transfer_result = step_reader.TransferRoot(1)
assert transfer_result
_nbs = step_reader.NbShapes()
assert _nbs == 1
shape_to_return = step_reader.Shape(1) # a compound
assert not shape_to_return.IsNull()
else:
raise AssertionError("Error: can't read file.")
if return_as_shapes:
shape_to_return = TopologyExplorer(shape_to_return).solids()
return shape_to_return
def read_step_file(filename, return_as_shapes=False, verbosity=False):
assert os.path.isfile(filename)
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(filename)
if status == IFSelect_RetDone: # check status
if verbosity:
failsonly = False
step_reader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity)
step_reader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity)
shapes = []
nr = 1
try:
while True:
ok = step_reader.TransferRoot(nr)
if not ok:
break
_nbs = step_reader.NbShapes()
shapes.append(step_reader.Shape(nr)) # a compound
#assert not shape_to_return.IsNull()
nr += 1
except:
print("No Shape", nr)
else:
raise AssertionError("Error: can't read file.")
return shapes
def upload_model():
if request.method == "POST":
if request.files:
model = request.files["model"]
modelfile = os.path.join(app.config["MODEL_UPLOADS"],
model.filename)
model.save(modelfile)
filename, file_extension = os.path.splitext(model.filename)
#big_shp = read_step_file(modelfile)
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(modelfile)
if status == IFSelect_RetDone:
ok = step_reader.TransferRoots()
_nbs = step_reader.NbShapes()
big_shp = step_reader.Shape()
tess = ShapeTesselator(big_shp)
tess.Compute(compute_edges=False, mesh_quality=0.5)
jsonfile = filename + ".json"
with open(os.path.join(app.config["MODEL_UPLOADS"], jsonfile),
"w") as text_file:
json_shape = tess.ExportShapeToThreejsJSONString(filename)
json_shape = json_shape.replace("data\\", "data/")
json_shape = json_shape.replace("\\step_postprocessed\\",
"/step_postprocessed/")
text_file.write(json_shape)
else:
raise Exception("error could not read step file")
return redirect(url_for('modelview', modelname=jsonfile))
return render_template("public/upload.html")
def read_step_file(filename, verbosity=True):
""" read the STEP file and returns a compound (based on OCC.Extend.DataExchange)
filename: the file path
return_as_shapes: optional, False by default. If True returns a list of shapes,
else returns a single compound
verbosity: optional, False by default.
"""
if not os.path.isfile(filename):
raise FileNotFoundError("%s not found." % filename)
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(filename)
if status != IFSelect_RetDone: # check status
raise AssertionError("Error: can't read file.")
if verbosity:
failsonly = False
step_reader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity)
step_reader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity)
_nbr = step_reader.TransferRoots()
_nbs = step_reader.NbShapes()
shape_to_return = step_reader.OneShape() # a compound
if shape_to_return is None:
raise AssertionError("Shape is None.")
elif shape_to_return.IsNull():
raise AssertionError("Shape is Null.")
return shape_to_return
def read_step(file_name):
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(file_name)
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)
return aResShape
else:
print("Error: can't read file.")
sys.exit(0)
def read_step(filename):
from OCC.Core.STEPControl import STEPControl_Reader
from OCC.Core.IFSelect import IFSelect_RetDone, IFSelect_ItemsByEntity
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(filename)
if status == IFSelect_RetDone:
failsonly = False
step_reader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity)
step_reader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity)
ok = step_reader.TransferRoot(1)
_nbs = step_reader.NbShapes()
return step_reader.Shape(1)
else:
raise ValueError('Cannot read the file')
def load_model_file(filePath):
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(filePath)
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()
shape = step_reader.Shape(1)
return shape
else:
raise Exception(":Error: can't read file - Method: load_STEP_file")
def read_file(self):
"""
Read the STEP file and stores the result in a _shapes list
"""
stepcontrol_reader = STEPControl_Reader()
status = stepcontrol_reader.ReadFile(self._filename)
if status == IFSelect_RetDone:
stepcontrol_reader.PrintCheckLoad(False, IFSelect_ItemsByEntity)
nb_roots = stepcontrol_reader.NbRootsForTransfer()
logger.info("%i root(s)" % nb_roots)
if nb_roots == 0:
msg = "No root for transfer"
logger.error(msg)
raise StepFileReadException(msg)
stepcontrol_reader.PrintCheckTransfer(False,
IFSelect_ItemsByEntity)
self._number_of_shapes = stepcontrol_reader.NbShapes()
for n in range(1, nb_roots + 1):
logger.info("Root index %i" % n)
ok = stepcontrol_reader.TransferRoot(n)
logger.info("TransferRoots status : %i" % ok)
if ok:
# for i in range(1, self.nb_shapes + 1):
a_shape = stepcontrol_reader.Shape(n)
if a_shape.IsNull():
msg = "At least one shape in IGES cannot be transferred"
logger.warning(msg)
else:
self._shapes.append(a_shape)
logger.info("Appending a %s to list of shapes" %
topo_types_dict[a_shape.ShapeType()])
else:
msg = "One shape could not be transferred"
logger.warning(msg)
warnings.warn(msg)
return True
else:
msg = "Status is not IFSelect_RetDone"
logger.error(msg)
raise StepFileReadException(msg)
def read_step_file(filename, as_compound=True, verbosity=True):
""" read the STEP file and returns a compound
filename: the file path
verbosity: optional, False by default.
as_compound: True by default. If there are more than one shape at root,
gather all shapes into one compound. Otherwise returns a list of shapes.
"""
if not os.path.isfile(filename):
raise FileNotFoundError("%s not found." % filename)
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(filename)
if status == IFSelect_RetDone: # check status
if verbosity:
failsonly = False
step_reader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity)
step_reader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity)
transfer_result = step_reader.TransferRoots()
if not transfer_result:
raise AssertionError("Transfer failed.")
_nbs = step_reader.NbShapes()
if _nbs == 0:
raise AssertionError("No shape to transfer.")
elif _nbs == 1: # most cases
return step_reader.Shape(1)
elif _nbs > 1:
print("Number of shapes:", _nbs)
shps = []
# loop over root shapes
for k in range(1, _nbs + 1):
new_shp = step_reader.Shape(k)
if not new_shp.IsNull():
shps.append(new_shp)
if as_compound:
compound, result = list_of_shapes_to_compound(shps)
if not result:
print("Warning: all shapes were not added to the compound")
return compound
else:
print("Warning, returns a list of shapes.")
return shps
else:
raise AssertionError("Error: can't read file.")
return None
def read_step_file(filename):
""" read the STEP file and returns a compound
"""
print("loading STEP file... ")
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(filename)
if status == IFSelect_RetDone: # check status
failsonly = False
ok = step_reader.TransferRoot(1)
_nbs = step_reader.NbShapes()
aResShape = step_reader.Shape(1)
else:
print("Error: can't read file.")
sys.exit(0)
print("done.")
return aResShape
def read_step_file(filename):
""" read the STEP file and returns a compound
"""
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(filename)
if status == IFSelect_RetDone: # check status
failsonly = False
step_reader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity)
step_reader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity)
ok = step_reader.TransferRoot()
_nbs = step_reader.NbShapes()
aResShape = step_reader.OneShape()
print(_nbs)
else:
print("Error: can't read file.")
sys.exit(0)
return aResShape
def read_step_file(filename):
"""
Read the STEP file and returns a shape. This cann't access any colour/name/layer atribute
If these are used, please use the following method instead
"""
step_reader = STEPControl_Reader()
status = step_reader.ReadFile(filename)
if status == IFSelect_RetDone: # check status
failsonly = False
step_reader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity)
step_reader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity)
step_reader.TransferRoot(1)
step_reader.NbShapes()
aResShape = step_reader.Shape(1)
topo = Topo(aResShape)
else:
print("Error: can't read file.")
sys.exit(0)
return topo
def importStep(fileName):
"""
Accepts a file name and loads the STEP file into a cadquery shape
:param fileName: The path and name of the STEP file to be imported
"""
# Now read and return the shape
reader = STEPControl_Reader()
readStatus = reader.ReadFile(fileName)
if readStatus != OCC.Core.IFSelect.IFSelect_RetDone:
raise ValueError("STEP File could not be loaded")
for i in range(reader.NbRootsForTransfer()):
reader.TransferRoot(i + 1)
occ_shapes = []
for i in range(reader.NbShapes()):
occ_shapes.append(reader.Shape(i + 1))
# Make sure that we extract all the solids
solids = []
for shape in occ_shapes:
solids.append(Shape.cast(shape))
return cadquery.Workplane("XY").newObject(solids)
from OCC.Core.STEPControl import STEPControl_Reader
from OCC.Core.IFSelect import IFSelect_RetDone, IFSelect_ItemsByEntity
from OCC.Display.SimpleGui import init_display
step_reader = STEPControl_Reader()
status = step_reader.ReadFile('./RFQ_noRMS_Spiral_Cut_90+angle_temp.stp')
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)
from OCC.Core.DataExchange import read_iges
display, start_display, add_menu, add_function_to_menu = init_display()
display.DisplayShape(aResShape, update=True)
start_display()