tool_path_editor.py

#!/usr/bin/env python2.7
'''
do testing here
'''
from numpy import *
import numpy as np
import pylab as p
import matplotlib.axes as a
import mpl_toolkits.mplot3d.axes3d as a3d
from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits.mplot3d.art3d import Poly3DCollection


#open a obj file and read faces and vertices
vertices = []
faces = []
material = None
 
read_f = ("python.obj")
for i, line in enumerate(open(read_f, "r")):
    if line.startswith('#'): continue
    values = line.split()
    if not values: continue
    if values[0] == 'v':
        v = map(float, values[1:4])
        vertices.append(v)
    elif values[0] == 'f':
        face = []
        texcoords = []
        norms = []
        for v in values[1:]:
            w = v.split('/')
            face.append(int(w[0]))
        faces.append(face)  
        
print vertices             
'''
all_v = []
for v in faces:
    for w in v:
        all_v.append(w)
print all_v
v_set = (set(all_v))
print v_set
per = []   
for f in v_set:
    per.append(vertices[f-1])
print per
'''
xs,ys,zs=[],[],[] 
xs1,ys1,zs1=[],[],[] 

for x1,y1,z1  in (vertices):
    xs.append(x1)
    ys.append(y1) 
    zs.append(z1)   

    

quad = [] 
for f in (faces):
    quad.append([vertices[v - 1] for v in f])
print quad
   
def pointdistance(p1,p2):
    #print "p1,p2",p1,p2
    p1=p1[:3]
    p2=p2[:3]
    x,y,z = [list(a) for a in zip(p1, p2)]
    xd = x[1]-x[0]
    yd = y[1]-y[0]
    zd = z[1]-z[0]
    distance = math.sqrt((xd*xd) + (yd*yd) + (zd*zd))
    return distance

def save_obj(xs,ys,zs):
    write_f = open('C:/Users/Phil Williammee/Desktop/pyjunk/pyThon.obj', 'w+')   
    for xx,yy,zz in zip(xs,ys,zs):
            line = "v "+str(xx)+" "+str(yy)+" "+str(zz)+"\n"
            write_f.write(line)
    write_f.close()   
               
x=[]
y=[]
z=[]   
b=[]
last_v= [0,0,0] 
    
def filter(i,v):
    b.append((i,v))
    x.append(vertices[i-1][0])
    y.append(vertices[i-1][2])
    z.append(80)
    #go over
    x.append(v[0])
    y.append(v[2])
    z.append(80)
    #go up
    #go down
    x.append(v[0])
    y.append(v[2])
    z.append(65.50)
    
def search():
    found=[]
    bad =       [0, 360, 366, 372, 1142, 1160, 1179, 1683, 1689, 1695, 1696, 1701, 1707, 1719, 1731]
    some_good = [0, 99, 360, 366, 367, 372, 950, 1142, 1160, 1179, 1491, 1683, 1689, 1690, 1695, 1696, 1701, 1702, 1707, 1709, 1719, 1720, 1721, 1731]
    for g in some_good:
        if g not in bad:
            found.append(g)
    print "good?",found

def draw_vert():
    global last_v
    skip=[]#1, 361, 373, 1684, 1708, 1749]
    for i, v in enumerate(vertices):
        distance = pointdistance(v, last_v)
        if distance > 2.0 and i not in skip:
            filter(i,v)
        else:
            x.append(v[0])
            y.append(v[2])
            z.append(65.5)
            
        last_v = v
    a=[b1[0] for b1 in b]
    print "bad", a
    #search()
    
draw_vert()
# Create main figure
fig=p.figure()
fig.suptitle('Object', fontsize=14, fontweight='bold')
#ax = a3d.Axes3D(fig)
x = np.multiply(7,x)
y = np.multiply(-7,y)
x=np.add(-200,x)
y=np.add(100,y)
ax = Axes3D(fig)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.plot3D(x,y,z)
save_obj(x,y,z)
p.show()