import cv2 
import numpy as np
import jovian

def ask_for_tracker():
    print("Select tracker API")
    print("Enter 0 for BOOSTING")
    print("Enter 1 for MIL")
    print("Enter 2 for KCF")
    print("Enter 3 for TLD")
    print("Enter 4 for MEDIANFLOW")
    print("Enter 5 for GOTURN")
    print("Enter 6 for MOSSE")
    print("Enter 7 for CSRT")
    
    choice = input("Please Select tracker...")
    
    if choice == '0':
        tracker = cv2.TrackerBoosting_create()
    if choice == '1':
        tracker = cv2.TrackerMIL_create()
    if choice == '2':
        tracker = cv2.TrackerKCF_create()
    if choice == '3':
        tracker = cv2.TrackerTLD_create()
    if choice == '4':
        tracker = cv2.TrackerMedianFlow_create()
    if choice == '5':
        tracker = cv2.TrackerGOTURN_create()
    if choice == '6':
        tracker = cv2.TrackerMOSSE_create()
    if choice == '7':
        tracker = cv2.TrackerCSRT_create()
        
    return tracker
        

# Tracker
tracker = ask_for_tracker()
tracker_name = str(tracker).split()[0][1:]

# capture video
cap = cv2.VideoCapture('carvid1.mp4')
#First frame
ret, frame = cap.read()
#select roi
roi = cv2.selectROI(frame,False)

# initialize tracker
ret = tracker.init(frame,roi)

# loop
while True:
    ret,frame = cap.read()
    
    success,roi = tracker.update(frame)
    
    (x,y,w,h) = tuple(map(int,roi))
    
    if success:
        
        pt1 = (x,y)
        pt2 = (x+w,y+h)
        
        cv2.rectangle(frame,
                     pt1,pt2,
                     (255,100,12),
                     2)
    else:
        cv2.putText(frame,
                   "Fail to track",
                   (59,69),
                   cv2.FONT_HERSHEY_SIMPLEX,1,
                   (0,0,250),2)
     
    cv2.putText(frame,
               tracker_name,
               (20,400),
               cv2.FONT_HERSHEY_SIMPLEX,1,
                (25,125,255),2)
    cv2.imshow(tracker_name,frame)
    
    if cv2.waitKey(3) & 0xFF == 27 or ret == False:
        break
        
    
cap.release()
cv2.destroyAllWindows()

jovian.commit()