# # Copyright (c) 2002, 2003, 2004, 2005, 2006 Art Haas # # This file is part of PythonCAD. # # PythonCAD is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # PythonCAD is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with PythonCAD; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA # # # construction circle class # from __future__ import generators import math from PythonCAD.Generic import point from PythonCAD.Generic import conobject from PythonCAD.Generic import tolerance from PythonCAD.Generic import quadtree from PythonCAD.Generic import util class CCircle(conobject.ConstructionObject): """A class for contruction circles A CCircle has two attributes: center: A Point object radius: The CCircle's radius A CCircle has the following methods: {get/set}Center(): Get/Set the center Point of a CCircle. {get/set}Radius(): Get/Set the radius of a CCircle. move(): Move the CCircle. circumference(): Get the CCircle's circumference. area(): Get the CCircle's area. mapCoords(): Find the nearest Point on the CCircle to a coordinate pair. inRegion(): Returns whether or not a CCircle can be seen in a bounded area. clone(): Return an indentical copy of a CCircle. """ __messages = { 'moved' : True, 'center_changed' : True, 'radius_changed' : True, } def __init__(self, center, radius, **kw): """Initialize a CCircle. CCircle(center, radius) The center should be a Point, or a two-entry tuple of floats, and the radius should be a float greater than 0. """ _cp = center if not isinstance(_cp, point.Point): _cp = point.Point(center) _r = util.get_float(radius) if not _r > 0.0: raise ValueError, "Invalid radius: %g" % _r super(CCircle, self).__init__(**kw) self.__radius = _r self.__center = _cp _cp.connect('moved', self.__movePoint) _cp.connect('change_pending', self.__pointChangePending) _cp.connect('change_complete', self.__pointChangeComplete) _cp.storeUser(self) def __eq__(self, obj): """Compare a CCircle to another for equality. """ if not isinstance(obj, CCircle): return False if obj is self: return True _val = False if self.__center == obj.getCenter(): if abs(self.__radius - obj.getRadius()) < 1e-10: _val = True return _val def __ne__(self, obj): """Compare a CCircle to another for inequality. """ if not isinstance(obj, CCircle): return True if obj is self: return False _val = True if self.__center == obj.getCenter(): if abs(self.__radius - obj.getRadius()) < 1e-10: _val = False return _val def finish(self): self.__center.disconnect(self) self.__center.freeUser(self) self.__center = self.__radius = None super(CCircle, self).finish() def getValues(self): _data = super(CCircle, self).getValues() _data.setValue('type', 'ccircle') _data.setValue('center', self.__center.getID()) _data.setValue('radius', self.__radius) return _data def getCenter(self): """Return the center Point of the CCircle. getCenter() """ return self.__center def setCenter(self, c): """Set the center Point of the CCircle. setCenter(c) The argument must be a Point or a tuple containing two float values. """ if self.isLocked(): raise RuntimeError, "Setting center not allowed - object locked." _cp = self.__center if not isinstance(c, point.Point): raise TypeError, "Invalid center point: " + `type(c)` if _cp is not c: _cp.disconnect(self) _cp.freeUser(self) self.startChange('center_changed') self.__center = c self.endChange('center_changed') self.sendMessage('center_changed', _cp) c.connect('moved', self.__movePoint) c.connect('change_pending', self.__pointChangePending) c.connect('change_complete', self.__pointChangeComplete) c.storeUser(self) if abs(_cp.x - c.x) > 1e-10 or abs(_cp.y - c.y) > 1e-10: self.sendMessage('moved', _cp.x, _cp.y, self.__radius) self.modified() center = property(getCenter, setCenter, None, "CCircle center") def getRadius(self): """Return the radius of the the CCircle. getRadius() """ return self.__radius def setRadius(self, radius): """Set the radius of the CCircle. setRadius(radius) The argument must be float value greater than 0. """ if self.isLocked(): raise RuntimeError, "Setting radius not allowed - object locked." _r = util.get_float(radius) if not _r > 0.0: raise ValueError, "Invalid radius: %g" % _r _cr = self.__radius if abs(_cr - _r) > 1e-10: self.startChange('radius_changed') self.__radius = _r self.endChange('radius_changed') self.sendMessage('radius_changed', _cr) _cx, _cy = self.__center.getCoords() self.sendMessage('moved', _cx, _cy, _cr) self.modified() radius = property(getRadius, setRadius, None, "CCircle radius") def move(self, dx, dy): """Move a CCircle. move(dx, dy) The first argument gives the x-coordinate displacement, and the second gives the y-coordinate displacement. Both values should be floats. """ if self.isLocked(): raise RuntimeError, "Setting radius not allowed - object locked." _dx = util.get_float(dx) _dy = util.get_float(dy) if abs(_dx) > 1e-10 or abs(_dy) > 1e-10: _x, _y = self.__center.getCoords() self.ignore('moved') try: self.__center.move(_dx, _dy) finally: self.receive('moved') self.sendMessage('moved', _x, _y, self.__radius) def circumference(self): """Return the circumference of the CCircle. circumference() """ return 2.0 * math.pi * self.__radius def area(self): """Return the area enclosed by the CCircle. area() """ return math.pi * pow(self.__radius, 2) def mapCoords(self, x, y, tol=tolerance.TOL): """Return the nearest Point on the CCircle to a coordinate pair. mapCoords(x, y[, tol]) The function has two required arguments: x: A Float value giving the x-coordinate y: A Float value giving the y-coordinate There is a single optional argument: tol: A float value equal or greater than 0.0 This function is used to map a possibly near-by coordinate pair to an actual Point on the CCircle. If the distance between the actual Point and the coordinates used as an argument is less than the tolerance, the actual Point is returned. Otherwise, this function returns None. """ _x = util.get_float(x) _y = util.get_float(y) _t = tolerance.toltest(tol) _cx, _cy = self.__center.getCoords() _r = self.__radius _dist = math.hypot((_x - _cx), (_y - _cy)) if abs(_dist - _r) < _t: _angle = math.atan2((_y - _cy),(_x - _cx)) _xoff = _r * math.cos(_angle) _yoff = _r * math.sin(_angle) return (_cx + _xoff), (_cy + _yoff) return None def inRegion(self, xmin, ymin, xmax, ymax, fully=False): """Return whether or not an CCircle exists within a region. inRegion(xmin, ymin, xmax, ymax[, fully]) The first four arguments define the boundary. The optional fifth argument 'fully' indicates whether or not the CCircle must be completely contained within the region or just pass through it. """ _xmin = util.get_float(xmin) _ymin = util.get_float(ymin) _xmax = util.get_float(xmax) if _xmax < _xmin: raise ValueError, "Illegal values: xmax < xmin" _ymax = util.get_float(ymax) if _ymax < _ymin: raise ValueError, "Illegal values: ymax < ymin" util.test_boolean(fully) _xc, _yc = self.__center.getCoords() _r = self.__radius # # cheap test to see if ccircle cannot be in region # if (((_xc - _r) > _xmax) or ((_yc - _r) > _ymax) or ((_xc + _r) < _xmin) or ((_yc + _r) < _ymin)): return False _val = False _bits = 0 # # calculate distances from center to region boundary # if abs(_xc - _xmin) < _r: _bits = _bits | 1 # left edge if abs(_xc - _xmax) < _r: _bits = _bits | 2 # right edge if abs(_yc - _ymin) < _r: _bits = _bits | 4 # bottom edge if abs(_yc - _ymax) < _r: _bits = _bits | 8 # top edge if _bits == 0: # # if the ccircle center is in region then the entire # ccircle is visible since the distance from the center # to any edge is greater than the radius. If the center # is not in the region then the ccircle is not visible in # the region because the distance to any edge is greater # than the radius, and so one of the bits should have been # set ... # if ((_xmin < _xc < _xmax) and (_ymin < _yc < _ymax)): _val = True else: _val = True # # calculate distance to corners of region # if math.hypot((_xc - _xmin), (_yc - _ymax)) < _r: _bits = _bits | 0x10 # upper left if math.hypot((_xc - _xmax), (_yc - _ymin)) < _r: _bits = _bits | 0x20 # lower right if math.hypot((_xc - _xmin), (_yc - _ymin)) < _r: _bits = _bits | 0x40 # lower left if math.hypot((_xc - _xmax), (_yc - _ymax)) < _r: _bits = _bits | 0x80 # upper right # # if all bits are set then distance from ccircle center # to region endpoints is less than radius - ccircle # entirely outside the region # if _bits == 0xff or fully: _val = False return _val def __pointChangePending(self, p, *args): _alen = len(args) if _alen < 1: raise ValueError, "Invalid argument count: %d" % _alen if args[0] == 'moved': self.startChange('moved') def __pointChangeComplete(self, p, *args): _alen = len(args) if _alen < 1: raise ValueError, "Invalid argument count: %d" % _alen if args[0] == 'moved': self.endChange('moved') def __movePoint(self, p, *args): _alen = len(args) if _alen < 2: raise ValueError, "Invalid argument count: %d" % _alen _x = util.get_float(args[0]) _y = util.get_float(args[1]) _cp = self.__center if p is not _cp: raise ValueError, "Point is not ccircle center: " + `p` _x, _y = _cp.getCoords() self.sendMessage('moved', _x, _y, self.__radius) def clone(self): """Create an identical copy of a CCircle clone() """ _cp = self.__center.clone() return CCircle(_cp, self.__radius) def sendsMessage(self, m): if m in CCircle.__messages: return True return super(CCircle, self).sendsMessage(m) # # Quadtree CCircle storage # class CCircleQuadtree(quadtree.Quadtree): def __init__(self): super(CCircleQuadtree, self).__init__() def getNodes(self, *args): _alen = len(args) if _alen != 3: raise ValueError, "Expected 3 arguments, got %d" % _alen _x = util.get_float(args[0]) _y = util.get_float(args[1]) _r = util.get_float(args[2]) _cxmin = _x - _r _cxmax = _x + _r _cymin = _y - _r _cymax = _y + _r _nodes = [self.getTreeRoot()] while len(_nodes): _node = _nodes.pop() _xmin, _ymin, _xmax, _ymax = _node.getBoundary() if ((_cxmin > _xmax) or (_cxmax < _xmin) or (_cymin > _ymax) or (_cymax < _ymin)): continue if _node.hasSubnodes(): _xmid = (_xmin + _xmax)/2.0 _ymid = (_ymin + _ymax)/2.0 _ne = _nw = _sw = _se = True if _cxmax < _xmid: # circle on left side _ne = _se = False if _cxmin > _xmid: # circle on right side _nw = _sw = False if _cymax < _ymid: # circle below _nw = _ne = False if _cymin > _ymid: # circle above _sw = _se = False if _ne: _nodes.append(_node.getSubnode(quadtree.QTreeNode.NENODE)) if _nw: _nodes.append(_node.getSubnode(quadtree.QTreeNode.NWNODE)) if _sw: _nodes.append(_node.getSubnode(quadtree.QTreeNode.SWNODE)) if _se: _nodes.append(_node.getSubnode(quadtree.QTreeNode.SENODE)) else: yield _node def addObject(self, obj): if not isinstance(obj, CCircle): raise TypeError, "Invalid CCircle object: " + `type(obj)` if obj in self: return _x, _y = obj.getCenter().getCoords() _r = obj.getRadius() _node = self.getTreeRoot() _bounds = _node.getBoundary() _xmin = _ymin = _xmax = _ymax = None _cxmin = _x - _r _cxmax = _x + _r _cymin = _y - _r _cymax = _y + _r _resize = False if _bounds is None: # first node in tree _resize = True _xmin = _cxmin - 1.0 _ymin = _cymin - 1.0 _xmax = _cxmax + 1.0 _ymax = _cymax + 1.0 else: _xmin, _ymin, _xmax, _ymax = _bounds if _cxmin < _xmin: _xmin = _cxmin - 1.0 _resize = True if _cxmax > _xmax: _xmax = _cxmax + 1.0 _resize = True if _cymin < _ymin: _ymin = _cymin - 1.0 _resize = True if _cymax > _ymax: _ymax = _cymax + 1.0 _resize = True if _resize: self.resize(_xmin, _ymin, _xmax, _ymax) for _node in self.getNodes(_x, _y, _r): _xmin, _ymin, _xmax, _ymax = _node.getBoundary() if obj.inRegion(_xmin, _ymin, _xmax, _ymax): _node.addObject(obj) super(CCircleQuadtree, self).addObject(obj) obj.connect('moved', self._moveCCircle) def delObject(self, obj): if obj not in self: return _x, _y = obj.getCenter().getCoords() _r = obj.getRadius() _pdict = {} for _node in self.getNodes(_x, _y, _r): _node.delObject(obj) # ccircle may not be in the node ... _parent = _node.getParent() if _parent is not None: _pid = id(_parent) if _pid not in _pdict: _pdict[_pid] = _parent super(CCircleQuadtree, self).delObject(obj) obj.disconnect(self) for _parent in _pdict.values(): self.purgeSubnodes(_parent) def find(self, *args): _alen = len(args) if _alen < 3: raise ValueError, "Invalid argument count: %d" % _alen _x = util.get_float(args[0]) _y = util.get_float(args[1]) _r = util.get_float(args[2]) _t = tolerance.TOL if _alen > 3: _t = tolerance.toltest(args[4]) _xmin = _x - _r - _t _xmax = _x + _r + _t _ymin = _y - _r - _t _ymax = _y + _r + _t _ccircles = [] for _ccirc in self.getInRegion(_xmin, _ymin, _xmax, _ymax): _cx, _cy = _ccirc.getCenter().getCoords() if ((abs(_cx - _x) < _t) and (abs(_cy - _y) < _t) and (abs(_ccirc.getRadius() - _r) < _t)): _ccircles.append(_ccirc) return _ccircles def _moveCCircle(self, obj, *args): if obj not in self: raise ValueError, "CCircle not stored in Quadtree: " + `obj` _alen = len(args) if _alen < 3: raise ValueError, "Invalid argument count: %d" % _alen _x = util.get_float(args[0]) _y = util.get_float(args[1]) _r = util.get_float(args[2]) for _node in self.getNodes(_x, _y, _r): _node.delObject(obj) # ccircle may not be in node ... super(CCircleQuadtree, self).delObject(obj) obj.disconnect(self) self.addObject(obj) def getClosest(self, x, y, tol=tolerance.TOL): _x = util.get_float(x) _y = util.get_float(y) _t = tolerance.toltest(tol) _circ = _tsep = None _bailout = False _cdict = {} _nodes = [self.getTreeRoot()] while len(_nodes): _node = _nodes.pop() _xmin, _ymin, _xmax, _ymax = _node.getBoundary() if ((_x < (_xmin - _t)) or (_x > (_xmax + _t)) or (_y < (_ymin - _t)) or (_y > (_ymax + _t))): continue if _node.hasSubnodes(): _nodes.extend(_node.getSubnodes()) else: for _c in _node.getObjects(): _cid = id(_c) if _cid not in _cdict: _cp = _c.mapCoords(_x, _y, _t) if _cp is not None: _cx, _cy = _cp _sep = math.hypot((_cx - _x), (_cy - _y)) if _tsep is None: _tsep = _sep _circ = _c else: if _sep < _tsep: _tsep = _sep _circ = _c if _sep < 1e-10 and _circ is not None: _bailout = True break if _bailout: break return _circ def getInRegion(self, xmin, ymin, xmax, ymax): _xmin = util.get_float(xmin) _ymin = util.get_float(ymin) _xmax = util.get_float(xmax) if _xmax < _xmin: raise ValueError, "Illegal values: xmax < xmin" _ymax = util.get_float(ymax) if _ymax < _ymin: raise ValueError, "Illegal values: ymax < ymin" _circs = [] if not len(self): return _circs _nodes = [self.getTreeRoot()] _cdict = {} while len(_nodes): _node = _nodes.pop() if _node.hasSubnodes(): for _subnode in _node.getSubnodes(): _sxmin, _symin, _sxmax, _symax = _subnode.getBoundary() if ((_sxmin > _xmax) or (_symin > _ymax) or (_sxmax < _xmin) or (_symax < _ymin)): continue _nodes.append(_subnode) else: for _circ in _node.getObjects(): _cid = id(_circ) if _cid not in _cdict: if _circ.inRegion(_xmin, _ymin, _xmax, _ymax): _circs.append(_circ) _cdict[_cid] = True return _circs # # CCircle history class # class CCircleLog(conobject.ConstructionObjectLog): def __init__(self, c): if not isinstance(c, CCircle): raise TypeError, "Invalid CCircle object: " + `type(c)` super(CCircleLog, self).__init__(c) c.connect('center_changed' ,self._centerChange) c.connect('radius_changed', self._radiusChange) def _radiusChange(self, c, *args): _alen = len(args) if _alen < 1: raise ValueError, "Invalid argument count: %d" % _alen _r = args[0] if not isinstance(_r, float): raise TypeError, "Unexpected type for radius: " + `type(_r)` self.saveUndoData('radius_changed', _r) def _centerChange(self, c, *args): _alen = len(args) if _alen < 1: raise ValueError, "Invalid argument count: %d" % _alen _old = args[0] if not isinstance(_old, point.Point): raise TypeError, "Invalid old center point: " + `type(_old)` self.saveUndoData('center_changed', _old.getID()) def execute(self, undo, *args): util.test_boolean(undo) _alen = len(args) if _alen == 0: raise ValueError, "No arguments to execute()" _c = self.getObject() _cp = _c.getCenter() _op = args[0] if _op == 'radius_changed': if len(args) < 2: raise ValueError, "Invalid argument count: %d" % _alen _r = args[1] if not isinstance(_r, float): raise TypeError, "Unexpected type for radius: " + `type(_r)` _sdata = _c.getRadius() self.ignore(_op) try: if undo: _c.startUndo() try: _c.setRadius(_r) finally: _c.endUndo() else: _c.startRedo() try: _c.setRadius(_r) finally: _c.endRedo() finally: self.receive(_op) self.saveData(undo, _op, _sdata) elif _op == 'center_changed': if _alen < 2: raise ValueError, "Invalid argument count: %d" % _alen _oid = args[1] _parent = _c.getParent() if _parent is None: raise ValueError, "CCircle has no parent - cannot undo" _pt = _parent.getObject(_oid) if _pt is None or not isinstance(_pt, point.Point): raise ValueError, "Center point missing: id=%d" % _oid _sdata = _cp.getID() self.ignore(_op) try: if undo: _c.startUndo() try: _c.setCenter(_pt) finally: _c.endUndo() else: _c.startRedo() try: _c.setCenter(_pt) finally: _c.endRedo() finally: self.receive(_op) self.saveData(undo, _op, _sdata) else: super(CCircleLog, self).execute(undo, *args)