removed trial code
This commit is contained in:
File diff suppressed because it is too large
Load Diff
@@ -1,102 +0,0 @@
|
||||
"""
|
||||
SolidWorks File Viewer
|
||||
Attempts to visualize SolidWorks files (.sldprt, .sldasm)
|
||||
Note: SLDPRT files are proprietary - converting to STEP is recommended
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
|
||||
def convert_sldprt_info():
|
||||
"""
|
||||
Provide information about converting SLDPRT files
|
||||
"""
|
||||
print("=" * 60)
|
||||
print("SLDPRT File Format Notice")
|
||||
print("=" * 60)
|
||||
print("\n.sldprt files are proprietary SolidWorks binary formats.")
|
||||
print("\nRECOMMENDED APPROACH:")
|
||||
print("1. Open your part in SolidWorks")
|
||||
print("2. File → Save As → STEP (*.step, *.stp)")
|
||||
print("3. Use step_viewer.py to visualize")
|
||||
print("\nAlternatively, you can batch convert using SolidWorks API")
|
||||
print("or use free converters like FreeCAD.\n")
|
||||
print("=" * 60)
|
||||
|
||||
|
||||
def batch_convert_with_freecad(sldprt_file):
|
||||
"""
|
||||
Attempt to convert SLDPRT to STEP using FreeCAD (if installed)
|
||||
|
||||
Args:
|
||||
sldprt_file (str): Path to the SLDPRT file
|
||||
|
||||
Returns:
|
||||
str: Path to the converted STEP file, or None if failed
|
||||
"""
|
||||
try:
|
||||
import FreeCAD
|
||||
import Import
|
||||
|
||||
# Load the SLDPRT file
|
||||
print(f"Attempting to load: {sldprt_file}")
|
||||
doc = FreeCAD.newDocument()
|
||||
Import.insert(sldprt_file, doc.Name)
|
||||
|
||||
# Export as STEP
|
||||
base_name = os.path.splitext(sldprt_file)[0]
|
||||
step_file = f"{base_name}_converted.step"
|
||||
|
||||
print(f"Converting to: {step_file}")
|
||||
Import.export(doc.Objects, step_file)
|
||||
|
||||
FreeCAD.closeDocument(doc.Name)
|
||||
print(f"Successfully converted to STEP format!")
|
||||
return step_file
|
||||
|
||||
except ImportError:
|
||||
print("\nFreeCAD not found. Install it with:")
|
||||
print(" conda install -c conda-forge freecad")
|
||||
print("\nOr download from: https://www.freecad.org/")
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"\nError during conversion: {e}")
|
||||
return None
|
||||
|
||||
|
||||
def main():
|
||||
"""Main entry point"""
|
||||
if len(sys.argv) < 2:
|
||||
print("Usage: python solidworks_viewer.py <path_to_sldprt_file>")
|
||||
print("\nExample:")
|
||||
print(" python solidworks_viewer.py model.sldprt")
|
||||
convert_sldprt_info()
|
||||
return
|
||||
|
||||
sldprt_file = sys.argv[1]
|
||||
|
||||
if not os.path.exists(sldprt_file):
|
||||
print(f"Error: File not found: {sldprt_file}")
|
||||
sys.exit(1)
|
||||
|
||||
# Check file extension
|
||||
ext = os.path.splitext(sldprt_file)[1].lower()
|
||||
if ext not in ['.sldprt', '.sldasm']:
|
||||
print(f"Warning: File extension '{ext}' is not a SolidWorks format")
|
||||
|
||||
print("\nAttempting conversion to STEP format...\n")
|
||||
|
||||
# Try to convert using FreeCAD
|
||||
step_file = batch_convert_with_freecad(sldprt_file)
|
||||
|
||||
if step_file and os.path.exists(step_file):
|
||||
print("\nWould you like to visualize the converted STEP file?")
|
||||
print(f"Run: python PythonSim/step_viewer.py {step_file}")
|
||||
else:
|
||||
print("\n" + "=" * 60)
|
||||
convert_sldprt_info()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -1,162 +0,0 @@
|
||||
"""
|
||||
STEP File Viewer for SolidWorks exports
|
||||
Visualize 3D CAD models in an interactive 3D environment
|
||||
"""
|
||||
|
||||
import numpy as np
|
||||
from OCC.Core.STEPControl import STEPControl_Reader
|
||||
from OCC.Core.IFSelect import IFSelect_RetDone
|
||||
from OCC.Core.BRepMesh import BRepMesh_IncrementalMesh
|
||||
from OCC.Core.TopExp import TopExp_Explorer
|
||||
from OCC.Core.TopAbs import TopAbs_FACE
|
||||
from OCC.Core.BRep import BRep_Tool
|
||||
from OCC.Core.gp import gp_Pnt, gp_Ax2, gp_Dir
|
||||
from OCC.Core.BRepPrimAPI import BRepPrimAPI_MakeSphere
|
||||
from OCC.Core.Quantity import Quantity_Color, Quantity_TOC_RGB
|
||||
from OCC.Display.SimpleGui import init_display
|
||||
import sys
|
||||
|
||||
|
||||
def load_step_file(filename):
|
||||
"""
|
||||
Load a STEP file and return the shape
|
||||
|
||||
Args:
|
||||
filename (str): Path to the STEP file
|
||||
|
||||
Returns:
|
||||
TopoDS_Shape: The loaded 3D shape
|
||||
"""
|
||||
step_reader = STEPControl_Reader()
|
||||
status = step_reader.ReadFile(filename)
|
||||
|
||||
if status != IFSelect_RetDone:
|
||||
raise Exception(f"Error reading STEP file: {filename}")
|
||||
|
||||
# Transfer the contents of the STEP file to the shape
|
||||
step_reader.TransferRoots()
|
||||
shape = step_reader.OneShape()
|
||||
|
||||
print(f"Successfully loaded: {filename}")
|
||||
return shape
|
||||
|
||||
|
||||
def visualize_step(filename, background_color=(0.1, 0.1, 0.1), show_origin=True, origin_size=5.0):
|
||||
"""
|
||||
Visualize a STEP file in an interactive 3D viewer
|
||||
|
||||
Args:
|
||||
filename (str): Path to the STEP file
|
||||
background_color (tuple): RGB values (0-1) for background color
|
||||
show_origin (bool): Whether to show origin marker
|
||||
origin_size (float): Size of the origin marker sphere
|
||||
"""
|
||||
# Load the STEP file
|
||||
shape = load_step_file(filename)
|
||||
|
||||
# Initialize the 3D display
|
||||
display, start_display, add_menu, add_function_to_menu = init_display()
|
||||
|
||||
# Set background color using Quantity_Color
|
||||
bg_color = Quantity_Color(background_color[0], background_color[1], background_color[2], Quantity_TOC_RGB)
|
||||
display.View.SetBackgroundColor(bg_color)
|
||||
|
||||
# Display the shape
|
||||
display.DisplayShape(shape, update=True)
|
||||
|
||||
# Add origin marker
|
||||
if show_origin:
|
||||
origin_point = gp_Pnt(0, 0, 0)
|
||||
origin_marker = BRepPrimAPI_MakeSphere(origin_point, origin_size).Shape()
|
||||
origin_color = Quantity_Color(1.0, 0.0, 0.0, Quantity_TOC_RGB) # Red
|
||||
display.DisplayShape(origin_marker, color=origin_color, update=True)
|
||||
print(f"\nOrigin marker (red sphere) displayed at (0, 0, 0) with radius {origin_size}")
|
||||
|
||||
# Fit the view to show the entire model
|
||||
display.FitAll()
|
||||
|
||||
print("\nControls:")
|
||||
print(" - Left mouse button: Rotate")
|
||||
print(" - Middle mouse button: Pan")
|
||||
print(" - Right mouse button: Zoom")
|
||||
print(" - F: Fit all")
|
||||
print(" - ESC: Exit")
|
||||
|
||||
# Start the display loop
|
||||
start_display()
|
||||
|
||||
|
||||
def get_mesh_data(shape, linear_deflection=0.1):
|
||||
"""
|
||||
Extract mesh data (vertices and triangles) from a shape
|
||||
Useful for custom rendering or analysis
|
||||
|
||||
Args:
|
||||
shape: TopoDS_Shape object
|
||||
linear_deflection (float): Mesh quality (lower = finer mesh)
|
||||
|
||||
Returns:
|
||||
tuple: (vertices, triangles) as numpy arrays
|
||||
"""
|
||||
# Mesh the shape
|
||||
mesh = BRepMesh_IncrementalMesh(shape, linear_deflection)
|
||||
mesh.Perform()
|
||||
|
||||
vertices = []
|
||||
triangles = []
|
||||
vertex_index = 0
|
||||
|
||||
# Explore all faces
|
||||
explorer = TopExp_Explorer(shape, TopAbs_FACE)
|
||||
while explorer.More():
|
||||
face = explorer.Current()
|
||||
location = face.Location()
|
||||
facing = BRep_Tool.Triangulation(face, location)
|
||||
|
||||
if facing:
|
||||
# Get transformation
|
||||
trsf = location.Transformation()
|
||||
|
||||
# Extract vertices
|
||||
for i in range(1, facing.NbNodes() + 1):
|
||||
pnt = facing.Node(i)
|
||||
pnt.Transform(trsf)
|
||||
vertices.append([pnt.X(), pnt.Y(), pnt.Z()])
|
||||
|
||||
# Extract triangles
|
||||
for i in range(1, facing.NbTriangles() + 1):
|
||||
triangle = facing.Triangle(i)
|
||||
n1, n2, n3 = triangle.Get()
|
||||
triangles.append([
|
||||
vertex_index + n1 - 1,
|
||||
vertex_index + n2 - 1,
|
||||
vertex_index + n3 - 1
|
||||
])
|
||||
|
||||
vertex_index += facing.NbNodes()
|
||||
|
||||
explorer.Next()
|
||||
|
||||
return np.array(vertices), np.array(triangles)
|
||||
|
||||
|
||||
def main():
|
||||
"""Main entry point for the STEP viewer"""
|
||||
if len(sys.argv) < 2:
|
||||
print("Usage: python step_viewer.py <path_to_step_file>")
|
||||
print("\nExample:")
|
||||
print(" python step_viewer.py model.step")
|
||||
print(" python step_viewer.py C:\\Models\\assembly.stp")
|
||||
return
|
||||
|
||||
step_file = sys.argv[1]
|
||||
|
||||
try:
|
||||
visualize_step(step_file)
|
||||
except Exception as e:
|
||||
print(f"Error: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user