import cv2
import numpy as np

def draw_bbox(image,box,color=(255,0,0),thickness=1):
  if thickness==0:
    return
  
  left = int(box[0])
  top = int(box[1])
  right = int(box[0]+box[2])
  bottom = int(box[1]+box[3])
  box_points =[(left,top),(right,top),(right,bottom),(left,bottom)]
  
  cv2.polylines(image, [np.array(box_points)], isClosed=True, color=color, thickness=thickness)


def to_int_points(points):
  int_points=[]
  for point in points:
    int_points.append([int(point[0]),int(point[1])])
  return int_points

def draw_text(img, text, point, font_scale=0.5, color=(200, 200, 200), thickness=1):
  font = cv2.FONT_HERSHEY_SIMPLEX
  cv2.putText(img, str(text), point, font, font_scale, color, thickness, cv2.LINE_AA)

plot_text_color = (200, 200, 200)
plot_text_font_scale = 0.5
plot_index = 1
plot_text = True

def set_plot_text(is_plot,text_font_scale,text_color):
  global plot_index,plot_text,plot_text_font_scale,plot_text_color
  plot_text = is_plot
  plot_index = 1
  plot_text_font_scale = text_font_scale
  plot_text_color = text_color

def plot_points(image,points,isClosed=False,circle_size=3,circle_color=(255,0,0),line_size=1,line_color=(0,0,255)):
    global plot_index,plot_text
    int_points = to_int_points(points)
    if circle_size>0:
      for point in int_points:
        cv2.circle(image,point,circle_size,circle_color,-1)
        if plot_text:
          draw_text(image,plot_index,point,plot_text_font_scale,plot_text_color)
        plot_index+=1
    if line_size>0:
      cv2.polylines(image, [np.array(int_points)], isClosed=isClosed, color=line_color, thickness=line_size)