#Pan Tilt application for ONVIF devices


from time import sleep

from onvif import ONVIFCamera

XMAX = 1
XMIN = -1
YMAX = 1
YMIN = -1

def perform_move(ptz, request, timeout):
    # Start continuous move
    ptz.ContinuousMove(request)
    # Wait a certain time
    sleep(timeout)
    # Stop continuous move
    ptz.Stop({'ProfileToken': request.ProfileToken})

def move_up(ptz, request, timeout=1):
    print 'move up...'
    request.Velocity.PanTilt._x = 0
    request.Velocity.PanTilt._y = YMAX
    perform_move(ptz, request, timeout)

def move_down(ptz, request, timeout=1):
    print 'move down...'
    request.Velocity.PanTilt._x = 0
    request.Velocity.PanTilt._y = YMIN
    perform_move(ptz, request, timeout)

def move_right(ptz, request, timeout=1):
    print 'move right...'
    request.Velocity.PanTilt._x = XMAX
    request.Velocity.PanTilt._y = 0
    perform_move(ptz, request, timeout)

def move_left(ptz, request, timeout=1):
    print 'move left...'
    request.Velocity.PanTilt._x = XMIN
    request.Velocity.PanTilt._y = 0
    perform_move(ptz, request, timeout)

def continuous_move():
    mycam = ONVIFCamera('192.168.1.11', 80, 'username_provided_in_webinterface', 'password')
    # Create media service object
    media = mycam.create_media_service()
    # Create ptz service object
    ptz = mycam.create_ptz_service()

    # Get target profile
    media_profile = media.GetProfiles()[0];

    # Get PTZ configuration options for getting continuous move range
    request = ptz.create_type('GetConfigurationOptions')
    request.ConfigurationToken = media_profile.PTZConfiguration._token
    ptz_configuration_options = ptz.GetConfigurationOptions(request)

    request = ptz.create_type('ContinuousMove')
    request.ProfileToken = media_profile._token

    ptz.Stop({'ProfileToken': media_profile._token})

    # Get range of pan and tilt
    # NOTE: X and Y are velocity vector
    global XMAX, XMIN, YMAX, YMIN
    XMAX = ptz_configuration_options.Spaces.ContinuousPanTiltVelocitySpace[0].XRange.Max
    XMIN = ptz_configuration_options.Spaces.ContinuousPanTiltVelocitySpace[0].XRange.Min
    YMAX = ptz_configuration_options.Spaces.ContinuousPanTiltVelocitySpace[0].YRange.Max
    YMIN = ptz_configuration_options.Spaces.ContinuousPanTiltVelocitySpace[0].YRange.Min

    #PT control
    var=1
    while var==1: #Create infinite loop
        direction=raw_input('Type the direction that you want to move the camera, Use the arrows with Numlock On: 6 for Right, 4 for Left, 8 for Up, 2 for Down. Use duble or tripple digits for larger moves: ')
        print direction
        if direction=='6':
            move_right(ptz, request)
        if direction=='4':
            move_left(ptz, request)
        if direction=='8':
            move_up(ptz, request)
        if direction=='2':
            move_down(ptz, request)
        #Double Moves
        if direction=='66':
            move_right(ptz, request)
            move_right(ptz, request)
        if direction=='44':
            move_left(ptz, request)
            move_left(ptz, request)
        if direction=='88':
            move_up(ptz, request)
            move_up(ptz, request)
        if direction=='22':
            move_down(ptz, request)
            move_down(ptz, request)

        #Tripple Moves
        if direction=='666':
            move_right(ptz, request)
            move_right(ptz, request)
            move_right(ptz, request)
        if direction=='444':
            move_left(ptz, request)
            move_left(ptz, request)
            move_left(ptz, request)
        if direction=='888':
            move_up(ptz, request)
            move_up(ptz, request)
            move_up(ptz, request)
        if direction=='222':
            move_down(ptz, request)
            move_down(ptz, request)
            move_down(ptz, request)
            
if __name__ == '__main__':
    continuous_move()