def preferred(self, vector):
"""
Returns True if vector is preferred, False if it is not
"""
xy = self.memory["preferred_direction"]
xy = (xy[0], xy[1], 0)
if helpers.compare_tuple(xy, (0, 0, 0)) or helpers.compare_tuple((vector[0], vector[1], 0), (0, 0, 0)):
return True
return helpers.smallest_angle((vector[0], vector[1], 0), xy) <= BConstants.beam["direction_tolerance_angle"]
def exists(self,e1,e2):
'''
Returns whether or not the beam defined by the endpoints e1 -> e2 exists
'''
# Let's get the box
xi, yi, zi = self.__get_indeces(e1)
# Cycle through the box and compare endpoints
for name in self.model[xi][yi][zi]:
beam = self.model[xi][yi][zi][name]
if ((helpers.compare_tuple(beam.endpoints.i,e1,0.5) and helpers.compare_tuple(
beam.endpoints.j,e2,0.5)) or (helpers.compare_tuple(beam.endpoints.i,e2,0.5) and
helpers.compare_tuple(beam.endpoints.j,e1,0.5))):
return True
return False
def exists(self, e1, e2):
'''
Returns whether or not the beam defined by the endpoints e1 -> e2 exists
'''
# Let's get the box
xi, yi, zi = self.__get_indeces(e1)
# Cycle through the box and compare endpoints
for name in self.model[xi][yi][zi]:
beam = self.model[xi][yi][zi][name]
if ((helpers.compare_tuple(beam.endpoints.i, e1, 0.5)
and helpers.compare_tuple(beam.endpoints.j, e2, 0.5))
or (helpers.compare_tuple(beam.endpoints.i, e2, 0.5)
and helpers.compare_tuple(beam.endpoints.j, e1, 0.5))):
return True
return False
def addjoint(self, coord, beam):
'''
Adds a joint (at the specified coordinate), to the beam itself. The beam
variable defines the which crosses this one at the joint
'''
# Verify that the coordinate is on the beam based on endpoints
if not helpers.on_line(self.endpoints.i, self.endpoints.j, coord):
return False
else:
coord = Coord(x=coord[0],y=coord[1],z=coord[2])
# We cycle manually so we can compare using our compare function
for key, beams in self.joints.items():
# We have a key and the beam isn't already there
if helpers.compare_tuple(key,coord) and beam not in beams:
self.joints[key].append(beam)
return True
self.joints[coord] = [beam]
return True
def addjoint(self, coord, beam):
'''
Adds a joint (at the specified coordinate), to the beam itself. The beam
variable defines the which crosses this one at the joint
'''
# Verify that the coordinate is on the beam based on endpoints
if not helpers.on_line(self.endpoints.i, self.endpoints.j, coord):
return False
else:
coord = Coord(x=coord[0], y=coord[1], z=coord[2])
# We cycle manually so we can compare using our compare function
for key, beams in self.joints.items():
# We have a key and the beam isn't already there
if helpers.compare_tuple(key, coord) and beam not in beams:
self.joints[key].append(beam)
return True
self.joints[coord] = [beam]
return True