Revision 6 as of 2003-10-09 19:04:48

Clear message

Points & Rectangles

A pair of classes to provide points and rectangles.

Surprisingly, I haven't been able to find a single Python module providing such primitive support.

WxPython supports wxPoint and wxRect, but it lacks many basic functions (such as, say, adding two points together to produce a third point..!)

This code is lacking a zillion essential features (such as "Length"). I only put in the ones I needed immediately. Please add, refactor, optimize, rename stuff to be more standard, etc., as you see fit..!

If there's an actual, accessible, easy-to-include Python module, not tied to a graphics library, that does this stuff already, please write about it here! No sense in reinventing the wheel. I've looked, but haven't found one. Hence this.

   1 # Code is Public Domain.
   2 def normalize(x1, y1, x2, y2):
   3     return min(x1,x2), min(y1,y2), max(x1,x2), max(y1,y2)
   4 
   5 class Point:
   6     def __init__(self, x, y):
   7         self.x = x
   8         self.y = y
   9     def __add__(self, other):
  10         return Point(self.x+other.x, self.y+other.y)
  11     def __sub__(self, other):
  12         return Point(self.x-other.x, self.y-other.y)
  13     def __mul__( self, scalar ):
  14         return Point(self.x*scalar, self.y*scalar)
  15     def __div__(self, scalar):
  16         return Point(self.x/scalar, self.y/scalar)
  17     def __str__(self):
  18         return "(%s, %s)" % (self.x, self.y)
  19     def __repr__(self):
  20         return "%s(%r, %r)" % (self.__class__.__name__, self.x, self.y)
  21     def XY(self):
  22         return self.x,self.y
  23     def Clone(self):
  24         return Point(self.x, self.y)
  25     def Integerize(self):
  26         self.x = int(self.x)
  27         self.y = int(self.y)
  28     def Floatize(self):
  29         self.x = float(self.x)
  30         self.y = float(self.y)
  31 
  32 class Rect:
  33     def __init__(self, pt1, pt2):
  34         self.Set(pt1, pt2)
  35     def Contains(self, pt):
  36         x,y = pt.XY()
  37         return self.left <= x <= self.right and self.top <= y <= self.bottom
  38     def Set( self, pt1, pt2 ):
  39         extrema = normalize(pt1.x, pt1.y, pt2.x, pt2.y)
  40         self.left, self.top, self.right, self.bottom = extrema
  41     def Overlaps(self, other):
  42         return (self.right > other.left and self.left < other.right
  43                 and self.top < other.bottom and self.bottom > other.top)
  44     def GetTL(self):
  45         return Point(self.left, self.top)
  46     def GetBR(self):
  47         return Point(self.right, self.bottom)
  48     def ExpandedBy(self, n):
  49         p1 = Point(self.left-n, self.top+n)
  50         p2 = Point(self.right+n, self.bottom+n)
  51         return Rect(p1, p2)
  52     def TransformedByFunction(self, foo):
  53         p1 = Point(self.left,  self.top)
  54         p2 = Point(self.right, self.bottom)
  55         return Rect(foo(p1), foo(p2))
  56     def __str__( self ):
  57         return "<Rect (%s,%s)-(%s,%s)>" % (self.left,self.top,
  58                                            self.right,self.bottom)
  59     def __repr__(self):
  60         return "%s(%r, %r)" % (self.__class__.__name__,
  61                                Point(self.left, self.top),
  62                                Point(self.right, self.bottom))

Unable to edit the page? See the FrontPage for instructions.