1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
'''
Created on 12-Dec-2014
NUT COMMENT
@author: deepa
'''
import math
import numpy
from OCC.BRepAlgoAPI import BRepAlgoAPI_Cut
from ModelUtils import getGpPt,getGpDir,makeEdgesFromPoints,makeWireFromEdges,makePrismFromFace,makeFaceFromWire
from OCC.BRepPrimAPI import BRepPrimAPI_MakeCylinder
from OCC.gp import gp_Ax2
class Nut(object):
'''
a3 X-------------------+ a2
X X|X
X X | X
X X | X
X X | X
X X | X
X X | X
X X 60 | X
a4 X XXXXXXXXXXXXXXXXX a1
X X
X X
XX X
X X
X X
X X
X X
X-------------------X
a6
a5
'''
def __init__(self,R,T,H,innerR1):
self.R = R
self.H = H
self.T = T
self.r1 = innerR1
#self.r2 = outerR2
self.secOrigin = numpy.array([0, 0, 0])
self.uDir = numpy.array([1.0, 0, 0])
self.wDir = numpy.array([0.0, 0, 1.0])
self.computeParams()
def place(self, secOrigin, uDir, wDir):
self.secOrigin = secOrigin
self.uDir = uDir
self.wDir = wDir
self.computeParams()
def getPoint(self,theta):
theta = math.radians(theta)
point = self.secOrigin + (self.R * math.cos(theta)) * self.uDir + (self.R * math.sin(theta)) * self.vDir
return point
def computeParams(self):
self.vDir = numpy.cross(self.wDir, self.uDir)
self.a1 = self.getPoint(0)
self.a2 = self.getPoint(60)
self.a3 = self.getPoint(120)
self.a4 = self.getPoint(180)
self.a5 = self.getPoint(240)
self.a6 = self.getPoint(300)
self.points = [self.a1, self.a2, self.a3, self.a4, self.a5, self.a6]
def createModel(self):
edges = makeEdgesFromPoints(self.points)
wire = makeWireFromEdges(edges)
aFace = makeFaceFromWire(wire)
extrudeDir = self.T * self.wDir # extrudeDir is a numpy array
prism = makePrismFromFace(aFace, extrudeDir)
cylOrigin = self.secOrigin
innerCyl = BRepPrimAPI_MakeCylinder(gp_Ax2(getGpPt(cylOrigin), getGpDir(self.wDir)), self.r1, self.H).Shape()
result_shape = BRepAlgoAPI_Cut(prism, innerCyl).Shape()
return result_shape
|