def skelDrawer(self, event, x, y, flags, param):
if self.mode == 0:
# Draw spine_bottom
print "Select midle of a spine_bottom"
if event == cv.CV_EVENT_LBUTTONUP:
self.spine_bottom = Geometry2D.Point(x, y)
self.permanentHighlightPt(self.spine_bottom)
self.mode += 1
else:
pass
elif self.mode == 1:
print "Select bottom point of a collar"
if event == cv.CV_EVENT_LBUTTONUP:
self.spine_top = Geometry2D.Point(x, y)
self.permanentHighlightPt(self.spine_top)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.mode += 1
else:
self.highlightSegment(Geometry2D.LineSegment(self.spine_bottom, Geometry2D.Point(x, y)))
elif self.mode == 2:
# Draw collar
print "Select left upper point of a collar"
if event == cv.CV_EVENT_LBUTTONUP:
self.collar = Geometry2D.Point(x, y)
self.virtual_collar = Geometry2D.mirrorPt(
self.collar, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.permanentHighlightPt(self.collar)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.spine_top, self.collar))
self.permanentHighlightPt(self.virtual_collar)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.spine_top, self.virtual_collar))
self.mode += 1
else:
temp_collar = Geometry2D.Point(x, y)
virtual_collar = Geometry2D.mirrorPt(
temp_collar, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.highlightPt(temp_collar)
self.highlightSegment(Geometry2D.LineSegment(self.spine_top, temp_collar))
self.highlightPt(virtual_collar)
self.highlightSegment(Geometry2D.LineSegment(self.spine_top, virtual_collar))
elif self.mode == 3:
# Draw shoulder
print "Select point between a left shoulder and a left armpit"
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.shoulder_joint = new_pt
self.virtual_shoulder_joint = Geometry2D.mirrorPt(
self.shoulder_joint, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.permanentHighlightPt(self.shoulder_joint)
self.permanentHighlightPt(self.virtual_shoulder_joint)
seg = Geometry2D.LineSegment(self.shoulder_joint, self.virtual_shoulder_joint)
self.permanentHighlightSegment(seg)
self.bottom_left = Geometry2D.Point(
self.spine_bottom.x() - 0.5 * seg.dx(), self.spine_bottom.y() - 0.5 * seg.dy()
)
self.bottom_right = Geometry2D.Point(
self.spine_bottom.x() + 0.5 * seg.dx(), self.spine_bottom.y() + 0.5 * seg.dy()
)
bottom_seg = Geometry2D.LineSegment(self.bottom_left, self.bottom_right)
self.permanentHighlightPt(self.bottom_left)
self.permanentHighlightPt(self.bottom_right)
self.permanentHighlightSegment(bottom_seg)
self.mode += 1
else:
temp_shoulder_joint = new_pt
virtual_shoulder_joint = Geometry2D.mirrorPt(
temp_shoulder_joint, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.highlightPt(temp_shoulder_joint)
self.highlightPt(virtual_shoulder_joint)
seg = Geometry2D.LineSegment(temp_shoulder_joint, virtual_shoulder_joint)
self.highlightSegment(seg)
bottom_left = Geometry2D.Point(
self.spine_bottom.x() - 0.5 * seg.dx(), self.spine_bottom.y() - 0.5 * seg.dy()
)
bottom_right = Geometry2D.Point(
self.spine_bottom.x() + 0.5 * seg.dx(), self.spine_bottom.y() + 0.5 * seg.dy()
)
bottom_seg = Geometry2D.LineSegment(bottom_left, bottom_right)
self.highlightPt(bottom_left)
self.highlightPt(bottom_right)
self.highlightSegment(bottom_seg)
elif self.mode == 4:
# Draw shoulder
print "Select point in a left shoulder"
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.shoulder_top = new_pt
seg = Geometry2D.LineSegment(self.shoulder_top, self.shoulder_joint)
self.armpit = seg.extrapolate(2.0)
self.virtual_shoulder_top = Geometry2D.mirrorPt(
self.shoulder_top, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.virtual_armpit = Geometry2D.mirrorPt(
self.armpit, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.permanentHighlightPt(self.shoulder_top)
self.permanentHighlightPt(self.armpit)
self.permanentHighlightPt(self.virtual_shoulder_top)
self.permanentHighlightPt(self.virtual_armpit)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.armpit, self.shoulder_top))
self.permanentHighlightSegment(Geometry2D.LineSegment(self.virtual_armpit, self.virtual_shoulder_top))
self.mode += 1
else:
shoulder_top = new_pt
seg = Geometry2D.LineSegment(shoulder_top, self.shoulder_joint)
armpit = seg.extrapolate(2.0)
virtual_shoulder_top = Geometry2D.mirrorPt(
shoulder_top, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
virtual_armpit = Geometry2D.mirrorPt(armpit, Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.highlightPt(shoulder_top)
self.highlightPt(armpit)
self.highlightPt(virtual_shoulder_top)
self.highlightPt(virtual_armpit)
self.highlightSegment(Geometry2D.LineSegment(armpit, shoulder_top))
self.highlightSegment(Geometry2D.LineSegment(virtual_armpit, virtual_shoulder_top))
elif self.mode == 5:
# Draw sleeve
print "Select midle point at the end of a left sleeve"
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.sleeve_node = new_pt
self.virtual_sleeve_node = Geometry2D.mirrorPt(
self.sleeve_node, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.permanentHighlightPt(self.sleeve_node)
self.permanentHighlightPt(self.virtual_sleeve_node)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.shoulder_joint, self.sleeve_node))
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.virtual_shoulder_joint, self.virtual_sleeve_node)
)
self.mode += 1
else:
temp_sleeve_node = new_pt
virtual_sleeve_node = Geometry2D.mirrorPt(
temp_sleeve_node, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.highlightPt(temp_sleeve_node)
self.highlightPt(virtual_sleeve_node)
self.highlightSegment(Geometry2D.LineSegment(self.shoulder_joint, temp_sleeve_node))
self.highlightSegment(Geometry2D.LineSegment(self.virtual_shoulder_joint, virtual_sleeve_node))
elif self.mode == 6:
# Draw shoulder
print "Select upper end point at the end of a left sleeve"
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.sleeve_top = new_pt
seg = Geometry2D.LineSegment(self.sleeve_top, self.sleeve_node)
self.sleeve_bottom = seg.extrapolate(2.0)
self.virtual_sleeve_top = Geometry2D.mirrorPt(
self.sleeve_top, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.virtual_sleeve_bottom = Geometry2D.mirrorPt(
self.sleeve_bottom, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.permanentHighlightPt(self.sleeve_top)
self.permanentHighlightPt(self.sleeve_bottom)
self.permanentHighlightPt(self.virtual_sleeve_top)
self.permanentHighlightPt(self.virtual_sleeve_bottom)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.sleeve_bottom, self.sleeve_top))
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.virtual_sleeve_bottom, self.virtual_sleeve_top)
)
# self.addCVShape(CVPolygon(cv.CV_RGB(0,0,0),self.front(),
# Geometry2D.Polygon( self.bottom_left,self.armpit,self.sleeve_bottom,self.sleeve_top,self.shoulder_top,self.collar,
# self.virtual_collar,self.virtual_shoulder_top,self.virtual_sleeve_top,self.virtual_sleeve_bottom,self.virtual_armpit,self.bottom_right)))
self.mode += 1
else:
sleeve_top = new_pt
seg = Geometry2D.LineSegment(sleeve_top, self.sleeve_node)
sleeve_bottom = seg.extrapolate(2.0)
virtual_sleeve_top = Geometry2D.mirrorPt(
sleeve_top, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
virtual_sleeve_bottom = Geometry2D.mirrorPt(
sleeve_bottom, Geometry2D.LineSegment(self.spine_bottom, self.spine_top)
)
self.highlightPt(sleeve_top)
self.highlightPt(sleeve_bottom)
self.highlightPt(virtual_sleeve_top)
self.highlightPt(virtual_sleeve_bottom)
self.highlightSegment(Geometry2D.LineSegment(sleeve_bottom, sleeve_top))
self.highlightSegment(Geometry2D.LineSegment(virtual_sleeve_bottom, virtual_sleeve_top))
def pantsSkelDrawer(self, event, x, y, flags, param):
if self.mode == 0:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.mid_center = new_pt
self.permanentHighlightPt(self.mid_center)
self.mode += 1
elif self.mode == 1:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.top_center = new_pt
self.crotch = Geometry2D.LineSegment(self.mid_center, self.top_center).extrapolate(-1)
self.permanentHighlightPt(self.top_center)
self.permanentHighlightPt(self.crotch)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.crotch, self.top_center))
self.mode += 1
else:
top_center = new_pt
crotch = Geometry2D.LineSegment(self.mid_center, top_center).extrapolate(-1)
self.highlightPt(top_center)
self.highlightPt(crotch)
self.highlightSegment(Geometry2D.LineSegment(crotch, top_center))
elif self.mode == 2:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.mid_left = new_pt
self.mid_right = Geometry2D.mirrorPt(
self.mid_left, Geometry2D.LineSegment(self.mid_center, self.top_center)
)
self.permanentHighlightPt(self.mid_left)
self.permanentHighlightPt(self.mid_right)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.mid_left, self.mid_right))
self.mode += 1
else:
mid_left = new_pt
mid_right = Geometry2D.mirrorPt(mid_left, Geometry2D.LineSegment(self.mid_center, self.top_center))
self.highlightPt(mid_left)
self.highlightPt(mid_right)
self.highlightSegment(Geometry2D.LineSegment(mid_left, mid_right))
elif self.mode == 3:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.left_leg_center = new_pt
self.right_leg_center = Geometry2D.mirrorPt(
self.left_leg_center, Geometry2D.LineSegment(self.mid_center, self.top_center)
)
self.permanentHighlightPt(self.left_leg_center)
self.permanentHighlightPt(self.right_leg_center)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.left_leg_center, self.mid_left))
self.permanentHighlightSegment(Geometry2D.LineSegment(self.right_leg_center, self.mid_right))
self.mode += 1
else:
left_leg_center = new_pt
right_leg_center = Geometry2D.mirrorPt(
left_leg_center, Geometry2D.LineSegment(self.mid_center, self.top_center)
)
self.highlightPt(left_leg_center)
self.highlightPt(right_leg_center)
self.highlightSegment(Geometry2D.LineSegment(left_leg_center, self.mid_left))
self.highlightSegment(Geometry2D.LineSegment(right_leg_center, self.mid_right))
elif self.mode == 4:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.left_leg_left = new_pt
self.left_leg_right = Geometry2D.LineSegment(self.left_leg_center, self.left_leg_left).extrapolate(-1)
self.right_leg_left = Geometry2D.mirrorPt(
self.left_leg_right, Geometry2D.LineSegment(self.mid_center, self.top_center)
)
self.right_leg_right = Geometry2D.mirrorPt(
self.left_leg_left, Geometry2D.LineSegment(self.mid_center, self.top_center)
)
self.permanentHighlightPt(self.left_leg_left)
self.permanentHighlightPt(self.left_leg_right)
self.permanentHighlightPt(self.right_leg_left)
self.permanentHighlightPt(self.right_leg_right)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.left_leg_left, self.left_leg_right))
self.permanentHighlightSegment(Geometry2D.LineSegment(self.right_leg_left, self.right_leg_right))
top_displ = Geometry2D.ptDiff(self.top_center, self.mid_center)
left_displ = Geometry2D.ptDiff(self.left_leg_left, self.left_leg_center)
self.top_left = Geometry2D.ptSum(Geometry2D.ptSum(self.mid_left, top_displ), left_displ)
self.top_right = Geometry2D.mirrorPt(
self.top_left, Geometry2D.LineSegment(self.mid_center, self.top_center)
)
self.permanentHighlightPt(self.top_left)
self.permanentHighlightPt(self.top_right)
self.mode += 1
else:
left_leg_left = new_pt
left_leg_right = Geometry2D.LineSegment(self.left_leg_center, left_leg_left).extrapolate(-1)
right_leg_left = Geometry2D.mirrorPt(
left_leg_right, Geometry2D.LineSegment(self.mid_center, self.top_center)
)
right_leg_right = Geometry2D.mirrorPt(
left_leg_left, Geometry2D.LineSegment(self.mid_center, self.top_center)
)
self.highlightPt(left_leg_left)
self.highlightPt(left_leg_right)
self.highlightPt(right_leg_left)
self.highlightPt(right_leg_right)
self.highlightSegment(Geometry2D.LineSegment(left_leg_left, left_leg_right))
self.highlightSegment(Geometry2D.LineSegment(right_leg_left, right_leg_right))
top_displ = Geometry2D.ptDiff(self.top_center, self.mid_center)
left_displ = Geometry2D.ptDiff(left_leg_left, self.left_leg_center)
top_left = Geometry2D.ptSum(Geometry2D.ptSum(self.mid_left, top_displ), left_displ)
top_right = Geometry2D.mirrorPt(top_left, Geometry2D.LineSegment(self.mid_center, self.top_center))
self.highlightPt(top_left)
self.highlightPt(top_right)
def pantsSkelDrawer(self, event, x, y, flags, param):
if self.mode == 0:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.mid_center = new_pt
self.permanentHighlightPt(self.mid_center)
self.mode += 1
elif self.mode == 1:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.top_center = new_pt
self.crotch = Geometry2D.LineSegment(
self.mid_center, self.top_center).extrapolate(-1)
self.permanentHighlightPt(self.top_center)
self.permanentHighlightPt(self.crotch)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.crotch, self.top_center))
self.mode += 1
else:
top_center = new_pt
crotch = Geometry2D.LineSegment(self.mid_center,
top_center).extrapolate(-1)
self.highlightPt(top_center)
self.highlightPt(crotch)
self.highlightSegment(
Geometry2D.LineSegment(crotch, top_center))
elif self.mode == 2:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.mid_left = new_pt
self.mid_right = Geometry2D.mirrorPt(
self.mid_left,
Geometry2D.LineSegment(self.mid_center, self.top_center))
self.permanentHighlightPt(self.mid_left)
self.permanentHighlightPt(self.mid_right)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.mid_left, self.mid_right))
self.mode += 1
else:
mid_left = new_pt
mid_right = Geometry2D.mirrorPt(
mid_left,
Geometry2D.LineSegment(self.mid_center, self.top_center))
self.highlightPt(mid_left)
self.highlightPt(mid_right)
self.highlightSegment(
Geometry2D.LineSegment(mid_left, mid_right))
elif self.mode == 3:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.left_leg_center = new_pt
self.right_leg_center = Geometry2D.mirrorPt(
self.left_leg_center,
Geometry2D.LineSegment(self.mid_center, self.top_center))
self.permanentHighlightPt(self.left_leg_center)
self.permanentHighlightPt(self.right_leg_center)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.left_leg_center,
self.mid_left))
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.right_leg_center,
self.mid_right))
self.mode += 1
else:
left_leg_center = new_pt
right_leg_center = Geometry2D.mirrorPt(
left_leg_center,
Geometry2D.LineSegment(self.mid_center, self.top_center))
self.highlightPt(left_leg_center)
self.highlightPt(right_leg_center)
self.highlightSegment(
Geometry2D.LineSegment(left_leg_center, self.mid_left))
self.highlightSegment(
Geometry2D.LineSegment(right_leg_center, self.mid_right))
elif self.mode == 4:
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.left_leg_left = new_pt
self.left_leg_right = Geometry2D.LineSegment(
self.left_leg_center, self.left_leg_left).extrapolate(-1)
self.right_leg_left = Geometry2D.mirrorPt(
self.left_leg_right,
Geometry2D.LineSegment(self.mid_center, self.top_center))
self.right_leg_right = Geometry2D.mirrorPt(
self.left_leg_left,
Geometry2D.LineSegment(self.mid_center, self.top_center))
self.permanentHighlightPt(self.left_leg_left)
self.permanentHighlightPt(self.left_leg_right)
self.permanentHighlightPt(self.right_leg_left)
self.permanentHighlightPt(self.right_leg_right)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.left_leg_left,
self.left_leg_right))
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.right_leg_left,
self.right_leg_right))
top_displ = Geometry2D.ptDiff(self.top_center, self.mid_center)
left_displ = Geometry2D.ptDiff(self.left_leg_left,
self.left_leg_center)
self.top_left = Geometry2D.ptSum(
Geometry2D.ptSum(self.mid_left, top_displ), left_displ)
self.top_right = Geometry2D.mirrorPt(
self.top_left,
Geometry2D.LineSegment(self.mid_center, self.top_center))
self.permanentHighlightPt(self.top_left)
self.permanentHighlightPt(self.top_right)
self.mode += 1
else:
left_leg_left = new_pt
left_leg_right = Geometry2D.LineSegment(
self.left_leg_center, left_leg_left).extrapolate(-1)
right_leg_left = Geometry2D.mirrorPt(
left_leg_right,
Geometry2D.LineSegment(self.mid_center, self.top_center))
right_leg_right = Geometry2D.mirrorPt(
left_leg_left,
Geometry2D.LineSegment(self.mid_center, self.top_center))
self.highlightPt(left_leg_left)
self.highlightPt(left_leg_right)
self.highlightPt(right_leg_left)
self.highlightPt(right_leg_right)
self.highlightSegment(
Geometry2D.LineSegment(left_leg_left, left_leg_right))
self.highlightSegment(
Geometry2D.LineSegment(right_leg_left, right_leg_right))
top_displ = Geometry2D.ptDiff(self.top_center, self.mid_center)
left_displ = Geometry2D.ptDiff(left_leg_left,
self.left_leg_center)
top_left = Geometry2D.ptSum(
Geometry2D.ptSum(self.mid_left, top_displ), left_displ)
top_right = Geometry2D.mirrorPt(
top_left,
Geometry2D.LineSegment(self.mid_center, self.top_center))
self.highlightPt(top_left)
self.highlightPt(top_right)
def skelDrawer(self,event,x,y,flags,param):
if self.mode==0:
#Draw spine_bottom
if event==cv.CV_EVENT_LBUTTONUP:
self.permanentHighlightPt(self.toe_center)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.ankle_joint,self.toe_center))
self.mode += 1
else:
self.highlightPoint(self.toe_center)
self.highlightSegment(Geometry2D.LineSegment(self.ankle_joint,self.toe_center))
self.spine_bottom = Geometry2D.Point(x,y)
self.permanentHighlightPt(self.spine_bottom)
self.mode += 1
elif self.mode==1:
if event==cv.CV_EVENT_LBUTTONUP:
self.spine_top = Geometry2D.Point(x,y)
self.permanentHighlightPt(self.spine_top)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.mode += 1
else:
self.highlightSegment(Geometry2D.LineSegment(self.spine_bottom,Geometry2D.Point(x,y)))
elif self.mode==2:
#Draw collar
if event==cv.CV_EVENT_LBUTTONUP:
self.collar = Geometry2D.Point(x,y)
self.virtual_collar = Geometry2D.mirrorPt(self.collar,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.permanentHighlightPt(self.collar)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.spine_top,self.collar))
self.permanentHighlightPt(self.virtual_collar)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.spine_top,self.virtual_collar))
self.mode += 1
else:
temp_collar = Geometry2D.Point(x,y)
virtual_collar = Geometry2D.mirrorPt(temp_collar,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.highlightPt(temp_collar)
self.highlightSegment(Geometry2D.LineSegment(self.spine_top,temp_collar))
self.highlightPt(virtual_collar)
self.highlightSegment(Geometry2D.LineSegment(self.spine_top,virtual_collar))
elif self.mode==3:
#Draw shoulder
new_pt = Geometry2D.Point(x,y)
if event==cv.CV_EVENT_LBUTTONUP:
self.shoulder_joint = new_pt
self.virtual_shoulder_joint = Geometry2D.mirrorPt(self.shoulder_joint,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.permanentHighlightPt(self.shoulder_joint)
self.permanentHighlightPt(self.virtual_shoulder_joint)
seg = Geometry2D.LineSegment(self.shoulder_joint,self.virtual_shoulder_joint)
self.permanentHighlightSegment(seg)
self.bottom_left = Geometry2D.Point(self.spine_bottom.x()-0.5*seg.dx(),self.spine_bottom.y()-0.5*seg.dy())
self.bottom_right = Geometry2D.Point(self.spine_bottom.x()+0.5*seg.dx(),self.spine_bottom.y()+0.5*seg.dy())
bottom_seg = Geometry2D.LineSegment(self.bottom_left,self.bottom_right)
self.permanentHighlightPt(self.bottom_left)
self.permanentHighlightPt(self.bottom_right)
self.permanentHighlightSegment(bottom_seg)
self.mode += 1
else:
temp_shoulder_joint = new_pt
virtual_shoulder_joint = Geometry2D.mirrorPt(temp_shoulder_joint,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.highlightPt(temp_shoulder_joint)
self.highlightPt(virtual_shoulder_joint)
seg = Geometry2D.LineSegment(temp_shoulder_joint,virtual_shoulder_joint)
self.highlightSegment(seg)
bottom_left = Geometry2D.Point(self.spine_bottom.x()-0.5*seg.dx(),self.spine_bottom.y()-0.5*seg.dy())
bottom_right = Geometry2D.Point(self.spine_bottom.x()+0.5*seg.dx(),self.spine_bottom.y()+0.5*seg.dy())
bottom_seg = Geometry2D.LineSegment(bottom_left,bottom_right)
self.highlightPt(bottom_left)
self.highlightPt(bottom_right)
self.highlightSegment(bottom_seg)
elif self.mode==4:
new_pt = Geometry2D.Point(x,y)
if event==cv.CV_EVENT_LBUTTONUP:
self.shoulder_top = new_pt
seg = Geometry2D.LineSegment(self.shoulder_top,self.shoulder_joint)
self.armpit = seg.extrapolate(2.0)
self.virtual_shoulder_top = Geometry2D.mirrorPt(self.shoulder_top,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.virtual_armpit = Geometry2D.mirrorPt(self.armpit,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.permanentHighlightPt(self.shoulder_top)
self.permanentHighlightPt(self.armpit)
self.permanentHighlightPt(self.virtual_shoulder_top)
self.permanentHighlightPt(self.virtual_armpit)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.armpit,self.shoulder_top))
self.permanentHighlightSegment(Geometry2D.LineSegment(self.virtual_armpit,self.virtual_shoulder_top))
self.mode+=1
else:
shoulder_top = new_pt
seg = Geometry2D.LineSegment(shoulder_top,self.shoulder_joint)
armpit = seg.extrapolate(2.0)
virtual_shoulder_top = Geometry2D.mirrorPt(shoulder_top,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
virtual_armpit = Geometry2D.mirrorPt(armpit,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.highlightPt(shoulder_top)
self.highlightPt(armpit)
self.highlightPt(virtual_shoulder_top)
self.highlightPt(virtual_armpit)
self.highlightSegment(Geometry2D.LineSegment(armpit,shoulder_top))
self.highlightSegment(Geometry2D.LineSegment(virtual_armpit,virtual_shoulder_top))
elif self.mode==5:
new_pt = Geometry2D.Point(x,y)
if event==cv.CV_EVENT_LBUTTONUP:
self.sleeve_node = new_pt
self.virtual_sleeve_node = Geometry2D.mirrorPt(self.sleeve_node,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.permanentHighlightPt(self.sleeve_node)
self.permanentHighlightPt(self.virtual_sleeve_node)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.shoulder_joint,self.sleeve_node))
self.permanentHighlightSegment(Geometry2D.LineSegment(self.virtual_shoulder_joint,self.virtual_sleeve_node))
self.mode += 1
else:
temp_sleeve_node = new_pt
virtual_sleeve_node = Geometry2D.mirrorPt(temp_sleeve_node,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.highlightPt(temp_sleeve_node)
self.highlightPt(virtual_sleeve_node)
self.highlightSegment(Geometry2D.LineSegment(self.shoulder_joint,temp_sleeve_node))
self.highlightSegment(Geometry2D.LineSegment(self.virtual_shoulder_joint,virtual_sleeve_node))
elif self.mode==6:
new_pt = Geometry2D.Point(x,y)
if event==cv.CV_EVENT_LBUTTONUP:
self.sleeve_top = new_pt
seg = Geometry2D.LineSegment(self.sleeve_top,self.sleeve_node)
self.sleeve_bottom = seg.extrapolate(2.0)
self.virtual_sleeve_top = Geometry2D.mirrorPt(self.sleeve_top,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.virtual_sleeve_bottom = Geometry2D.mirrorPt(self.sleeve_bottom,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.permanentHighlightPt(self.sleeve_top)
self.permanentHighlightPt(self.sleeve_bottom)
self.permanentHighlightPt(self.virtual_sleeve_top)
self.permanentHighlightPt(self.virtual_sleeve_bottom)
self.permanentHighlightSegment(Geometry2D.LineSegment(self.sleeve_bottom,self.sleeve_top))
self.permanentHighlightSegment(Geometry2D.LineSegment(self.virtual_sleeve_bottom,self.virtual_sleeve_top))
#self.addCVShape(CVPolygon(cv.CV_RGB(0,0,0),self.front(),
# Geometry2D.Polygon( self.bottom_left,self.armpit,self.sleeve_bottom,self.sleeve_top,self.shoulder_top,self.collar,
# self.virtual_collar,self.virtual_shoulder_top,self.virtual_sleeve_top,self.virtual_sleeve_bottom,self.virtual_armpit,self.bottom_right)))
self.mode+=1
else:
sleeve_top = new_pt
seg = Geometry2D.LineSegment(sleeve_top,self.sleeve_node)
sleeve_bottom = seg.extrapolate(2.0)
virtual_sleeve_top = Geometry2D.mirrorPt(sleeve_top,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
virtual_sleeve_bottom = Geometry2D.mirrorPt(sleeve_bottom,Geometry2D.LineSegment(self.spine_bottom,self.spine_top))
self.highlightPt(sleeve_top)
self.highlightPt(sleeve_bottom)
self.highlightPt(virtual_sleeve_top)
self.highlightPt(virtual_sleeve_bottom)
self.highlightSegment(Geometry2D.LineSegment(sleeve_bottom,sleeve_top))
self.highlightSegment(Geometry2D.LineSegment(virtual_sleeve_bottom,virtual_sleeve_top))
def skelDrawer(self, event, x, y, flags, param):
if self.mode == 0:
#Draw spine_bottom
print "Select midle of a spine_bottom"
if event == cv.CV_EVENT_LBUTTONUP:
self.spine_bottom = Geometry2D.Point(x, y)
self.permanentHighlightPt(self.spine_bottom)
self.mode += 1
else:
pass
elif self.mode == 1:
print "Select bottom point of a collar"
if event == cv.CV_EVENT_LBUTTONUP:
self.spine_top = Geometry2D.Point(x, y)
self.permanentHighlightPt(self.spine_top)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.mode += 1
else:
self.highlightSegment(
Geometry2D.LineSegment(self.spine_bottom,
Geometry2D.Point(x, y)))
elif self.mode == 2:
#Draw collar
print "Select left upper point of a collar"
if event == cv.CV_EVENT_LBUTTONUP:
self.collar = Geometry2D.Point(x, y)
self.virtual_collar = Geometry2D.mirrorPt(
self.collar,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.permanentHighlightPt(self.collar)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.spine_top, self.collar))
self.permanentHighlightPt(self.virtual_collar)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.spine_top,
self.virtual_collar))
self.mode += 1
else:
temp_collar = Geometry2D.Point(x, y)
virtual_collar = Geometry2D.mirrorPt(
temp_collar,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.highlightPt(temp_collar)
self.highlightSegment(
Geometry2D.LineSegment(self.spine_top, temp_collar))
self.highlightPt(virtual_collar)
self.highlightSegment(
Geometry2D.LineSegment(self.spine_top, virtual_collar))
elif self.mode == 3:
#Draw shoulder
print "Select point between a left shoulder and a left armpit"
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.shoulder_joint = new_pt
self.virtual_shoulder_joint = Geometry2D.mirrorPt(
self.shoulder_joint,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.permanentHighlightPt(self.shoulder_joint)
self.permanentHighlightPt(self.virtual_shoulder_joint)
seg = Geometry2D.LineSegment(self.shoulder_joint,
self.virtual_shoulder_joint)
self.permanentHighlightSegment(seg)
self.bottom_left = Geometry2D.Point(
self.spine_bottom.x() - 0.5 * seg.dx(),
self.spine_bottom.y() - 0.5 * seg.dy())
self.bottom_right = Geometry2D.Point(
self.spine_bottom.x() + 0.5 * seg.dx(),
self.spine_bottom.y() + 0.5 * seg.dy())
bottom_seg = Geometry2D.LineSegment(self.bottom_left,
self.bottom_right)
self.permanentHighlightPt(self.bottom_left)
self.permanentHighlightPt(self.bottom_right)
self.permanentHighlightSegment(bottom_seg)
self.mode += 1
else:
temp_shoulder_joint = new_pt
virtual_shoulder_joint = Geometry2D.mirrorPt(
temp_shoulder_joint,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.highlightPt(temp_shoulder_joint)
self.highlightPt(virtual_shoulder_joint)
seg = Geometry2D.LineSegment(temp_shoulder_joint,
virtual_shoulder_joint)
self.highlightSegment(seg)
bottom_left = Geometry2D.Point(
self.spine_bottom.x() - 0.5 * seg.dx(),
self.spine_bottom.y() - 0.5 * seg.dy())
bottom_right = Geometry2D.Point(
self.spine_bottom.x() + 0.5 * seg.dx(),
self.spine_bottom.y() + 0.5 * seg.dy())
bottom_seg = Geometry2D.LineSegment(bottom_left, bottom_right)
self.highlightPt(bottom_left)
self.highlightPt(bottom_right)
self.highlightSegment(bottom_seg)
elif self.mode == 4:
#Draw shoulder
print "Select point in a left shoulder"
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.shoulder_top = new_pt
seg = Geometry2D.LineSegment(self.shoulder_top,
self.shoulder_joint)
self.armpit = seg.extrapolate(2.0)
self.virtual_shoulder_top = Geometry2D.mirrorPt(
self.shoulder_top,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.virtual_armpit = Geometry2D.mirrorPt(
self.armpit,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.permanentHighlightPt(self.shoulder_top)
self.permanentHighlightPt(self.armpit)
self.permanentHighlightPt(self.virtual_shoulder_top)
self.permanentHighlightPt(self.virtual_armpit)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.armpit, self.shoulder_top))
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.virtual_armpit,
self.virtual_shoulder_top))
self.mode += 1
else:
shoulder_top = new_pt
seg = Geometry2D.LineSegment(shoulder_top, self.shoulder_joint)
armpit = seg.extrapolate(2.0)
virtual_shoulder_top = Geometry2D.mirrorPt(
shoulder_top,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
virtual_armpit = Geometry2D.mirrorPt(
armpit,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.highlightPt(shoulder_top)
self.highlightPt(armpit)
self.highlightPt(virtual_shoulder_top)
self.highlightPt(virtual_armpit)
self.highlightSegment(
Geometry2D.LineSegment(armpit, shoulder_top))
self.highlightSegment(
Geometry2D.LineSegment(virtual_armpit,
virtual_shoulder_top))
elif self.mode == 5:
#Draw sleeve
print "Select midle point at the end of a left sleeve"
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.sleeve_node = new_pt
self.virtual_sleeve_node = Geometry2D.mirrorPt(
self.sleeve_node,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.permanentHighlightPt(self.sleeve_node)
self.permanentHighlightPt(self.virtual_sleeve_node)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.shoulder_joint,
self.sleeve_node))
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.virtual_shoulder_joint,
self.virtual_sleeve_node))
self.mode += 1
else:
temp_sleeve_node = new_pt
virtual_sleeve_node = Geometry2D.mirrorPt(
temp_sleeve_node,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.highlightPt(temp_sleeve_node)
self.highlightPt(virtual_sleeve_node)
self.highlightSegment(
Geometry2D.LineSegment(self.shoulder_joint,
temp_sleeve_node))
self.highlightSegment(
Geometry2D.LineSegment(self.virtual_shoulder_joint,
virtual_sleeve_node))
elif self.mode == 6:
#Draw shoulder
print "Select upper end point at the end of a left sleeve"
new_pt = Geometry2D.Point(x, y)
if event == cv.CV_EVENT_LBUTTONUP:
self.sleeve_top = new_pt
seg = Geometry2D.LineSegment(self.sleeve_top, self.sleeve_node)
self.sleeve_bottom = seg.extrapolate(2.0)
self.virtual_sleeve_top = Geometry2D.mirrorPt(
self.sleeve_top,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.virtual_sleeve_bottom = Geometry2D.mirrorPt(
self.sleeve_bottom,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.permanentHighlightPt(self.sleeve_top)
self.permanentHighlightPt(self.sleeve_bottom)
self.permanentHighlightPt(self.virtual_sleeve_top)
self.permanentHighlightPt(self.virtual_sleeve_bottom)
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.sleeve_bottom,
self.sleeve_top))
self.permanentHighlightSegment(
Geometry2D.LineSegment(self.virtual_sleeve_bottom,
self.virtual_sleeve_top))
#self.addCVShape(CVPolygon(cv.CV_RGB(0,0,0),self.front(),
# Geometry2D.Polygon( self.bottom_left,self.armpit,self.sleeve_bottom,self.sleeve_top,self.shoulder_top,self.collar,
# self.virtual_collar,self.virtual_shoulder_top,self.virtual_sleeve_top,self.virtual_sleeve_bottom,self.virtual_armpit,self.bottom_right)))
self.mode += 1
else:
sleeve_top = new_pt
seg = Geometry2D.LineSegment(sleeve_top, self.sleeve_node)
sleeve_bottom = seg.extrapolate(2.0)
virtual_sleeve_top = Geometry2D.mirrorPt(
sleeve_top,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
virtual_sleeve_bottom = Geometry2D.mirrorPt(
sleeve_bottom,
Geometry2D.LineSegment(self.spine_bottom, self.spine_top))
self.highlightPt(sleeve_top)
self.highlightPt(sleeve_bottom)
self.highlightPt(virtual_sleeve_top)
self.highlightPt(virtual_sleeve_bottom)
self.highlightSegment(
Geometry2D.LineSegment(sleeve_bottom, sleeve_top))
self.highlightSegment(
Geometry2D.LineSegment(virtual_sleeve_bottom,
virtual_sleeve_top))
