Abaqus_Dome_Calculator/Edge.py at master · chrishieh/Abaqus_Dome_Calculator · GitHub

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
# Class for Edges

import math as M
from numbers import Number
from decimal import Decimal


class Edge:
    # Edge between two coordinates
    def __init__(self, n):
        self.name = n

        self.edge_number = 0

    def Set_Points(self, a, b):
        self.x1 = a
        self.x2 = b


    def Set_Edge_Number(self, n):
      self.edge_number = n

    def Get_Length(self):

        t1 = M.pow((self.x1.x - self.x2.x),2)
        t2 = M.pow((self.x1.y - self.x2.y),2)
        t3 = M.pow((self.x1.z - self.x2.z),2)

        ans = M.sqrt( t1 + t2 + t3 )
        dans = Decimal(str(ans)).quantize(Decimal(10) ** -2)

        return dans


    def __repr__(self):
        return "\n" + self.name + " --- \n" + str(self.x1.name) + "\n - \n" + str(self.x2.name) #+ " : " + str(self.length)


    def Get_CATIA_Desc(self):

        # Create the VB script code to generate a line between two points

        self.ret_string = "Set hybridShapePointCoord1 = hybridShapes1.Item(\"Point." + str(self.x1.point_number) + "\")\n"
        self.ret_string += "Set reference1 = part1.CreateReferenceFromObject(hybridShapePointCoord1)\n"
        self.ret_string += "Set hybridShapePointCoord2 = hybridShapes1.Item(\"Point." + str(self.x2.point_number) + "\")\n"
        self.ret_string += "Set reference2 = part1.CreateReferenceFromObject(hybridShapePointCoord2)\n"
        #self.ret_string += "Dim hybridShapeLinePtPt" + str(self.edge_number) +" As HybridShapeLinePtPt\n"
        self.ret_string += "Set hybridShapeLinePtPt" + str(self.edge_number) +" = hybridShapeFactory1.AddNewLinePtPt(reference1, reference2)\n"
        self.ret_string += "body1.InsertHybridShape hybridShapeLinePtPt" + str(self.edge_number) +"\n"
        self.ret_string += "part1.InWorkObject = hybridShapeLinePtPt" + str(self.edge_number) +"\n"
        self.ret_string += "part1.Update\n"

        return self.ret_string


    def Get_Edge_Coordinates(self):
        return self.name + ": " + str(self.x1) + " - " + str(self.x2)

    def Get_Edge_Number(self):
        node_number_one = str(self.x1)[:str(self.x1).find('=')-1]
        node_number_two = str(self.x2)[:str(self.x1).find('=')-1]
        node_number_one = node_number_one[node_number_one.find('t') + 1:]
        node_number_two = node_number_two[node_number_two.find('t') + 1:]

        #print(node_number_two)
        return (self.x1.Get_Point_Number(), self.x2.Get_Point_Number())

    def Get_Node_1(self):
        return str(self.x1)

    def Get_Node_2(self):
        return str(self.x2)

    def Update_Point(self, x, pt):
        if self.x1 == pt:
            self.x1 = x
        if self.x2 == pt:
            self.x2 = x


    def __eq__(self, other):

        fnd = False

        if isinstance(other, Edge):
            # Check if both the points are the same regardless of order
            if (self.x1 == other.x1 and self.x2 == other.x2) or (self.x1 == other.x2 and self.x2 == other.x1):
                fnd = True

        return fnd