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
89
90
91
92
93
94
95
|
'''
Created on 29-Nov-2014
@author: deepa
'''
import numpy
from ModelUtils import getGpPt,getGpDir,makeEdgesFromPoints,makeWireFromEdges,makePrismFromFace,makeFaceFromWire
import math
from OCC.BRepPrimAPI import BRepPrimAPI_MakeCylinder
from OCC.gp import gp_Ax2
from OCC.BRepAlgoAPI import BRepAlgoAPI_Fuse
class Bolt(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,r):
self.R = R
self.H = H
self.T = T
self.r = r
self.origin = None
self.uDir = None
self.shaftDir = None
self.vDir = None
self.a1 = None
self.a2 = None
self.a3 = None
self.a4 = None
self.a5 = None
self.a6 = None
self.points = []
def place(self, origin, uDir, shaftDir):
self.origin = origin
self.uDir = uDir
self.shaftDir = shaftDir
self.computeParams()
def getPoint(self,theta):
theta = math.radians(theta)
point = self.origin + (self.R * math.cos(theta)) * self.uDir + (self.R * math.sin(theta)) * self.vDir
return point
def computeParams(self):
self.vDir = numpy.cross(self.shaftDir, 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.shaftDir # extrudeDir is a numpy array
boltHead = makePrismFromFace(aFace, extrudeDir)
cylOrigin = self.origin
boltCylinder = BRepPrimAPI_MakeCylinder(gp_Ax2(getGpPt(cylOrigin), getGpDir(self.shaftDir)), self.r, self.H).Shape()
whole_Bolt = BRepAlgoAPI_Fuse(boltHead,boltCylinder).Shape()
return whole_Bolt
|
