image_object.py

import numpy as np
import cv2
import sys
from enum import Enum
from var import mode

cropImgSize = (500, 500)
class img_obj:

	def __init__(self, master, newImg):
		self.imgLive = newImg.copy()
		self.imgCopy = newImg.copy()
		self.master = master

	def new(self, newImg):
		self.imgLive = newImg.copy()
		self.imgCopy = newImg.copy()

	def reset(self):
		self.imgLive = self.imgCopy.copy()

	def get_image_live(self):
		return self.imgLive

	def crop_roi(self):
		from target_image_object import targ_img_obj

		for i in range(0,len(self.master.pts),2):
			roi = self.get_roi(i)

			imgObj = targ_img_obj(self.master, roi)
			self.master.cropList.append(imgObj)

		self.master.image_loop()

	def get_roi(self, index):
		i = index

		pts = self.master.pts
		# uses this formula
		#     clone[y0:y1, x0:x1]
		roi = self.imgCopy[pts[i][1]:pts[i+1][1], pts[i][0]:pts[i+1][0]]

		# set size of new image
		roi = cv2.resize(roi, cropImgSize)
		return roi

		# ensures positive area and draws the rectangle
	def make_rectangle(self):
		# ensures the set of two pts is in order from lowest to highest
		# so that "high-low >= 0" is true
		# always ensure the area of the pts will not be negative
		pt0 = self.master.pt0
		pt1 = self.master.pt1

		h, w, c = (self.imgLive).shape
		x0 = min(pt0[0],pt1[0], w)
		y0 = min(pt0[1],pt1[1], h)
		x1 = max(pt0[0],pt1[0], 0)
		y1 = max(pt0[1],pt1[1], 0)

		self.master.pts.append((x0,y0))
		self.master.pts.append((x1,y1))
		# draws rectangle at two pts in color red (BGR) with width 2
		cv2.rectangle(self.imgLive, pt0, pt1, (0, 0, 255), 2)
		self.master.imageModified = True