ipcam_test.py

# -*- coding: utf-8 -*-
"""
Created on Wed May  6 14:26:27 2015

@author: chuong
"""
# Python 3.x behavior
from __future__ import absolute_import, division, print_function

import urllib2
import base64
import os
import io
from datetime import datetime
import numpy as np
import csv
import PIL.Image
from scipy import misc
import time
import yaml

# This is for Axis camera
IPVAL = '10.132.11.32:4442'
USERVAL = 'root'
PASSVAL = 'admin'
ImageSize = [1280, 720]
URL_Capture = 'IPVAL/axis-cgi/bitmap/image.bmp?resolution=WIDTHVALxHEIGHTVAL&compression=0'
URL_SetPanTilt = 'IPVAL/axis-cgi/com/ptz.cgi?pan=PANVAL&tilt=TILTVAL'
URL_SetZoom = 'IPVAL/axis-cgi/com/ptz.cgi?zoom=ZOOMVAL'
URL_SetFocusMode = 'IPVAL/axis-cgi/com/ptz.cgi?autofocus=FOCUSMODE'
URL_GetZoom = 'IPVAL/axis-cgi/com/ptz.cgi?query=position'
RET_GetZoom = '*zoom={}*'
URL_GetFocusMode = 'IPVAL/axis-cgi/com/ptz.cgi?query=position'
RET_GetFocusMode = '*autofocus={}*'


def callURL(URL, IPVAL, USERVAL, PASSVAL):
    URL_Str = 'http://' + URL
    URL_Str = URL_Str.replace("IPVAL", IPVAL)
    print(URL_Str),
    request = urllib2.Request(URL_Str)
    base64string = base64.encodestring('%s:%s' % (USERVAL, PASSVAL)).replace('\n', '')
    request.add_header("Authorization", "Basic %s" % base64string)
    try:
        stream = urllib2.urlopen(request, timeout=60)
    except urllib2.URLError, e:
        raise Exception("Time out error: %r" % e)
    return stream.read()


def captureImage():
    URL_Str = URL_Capture.replace("WIDTHVAL", str(ImageSize[0])).replace("HEIGHTVAL", str(ImageSize[1]))
    output = callURL(URL_Str, IPVAL, USERVAL, PASSVAL)
    byte_array = io.BytesIO(output)
    print(' Read successfull')
    Image = np.array(PIL.Image.open(byte_array))
    return Image


def captureImage2File2(OutputFileName):
    try:
        Image = captureImage()
    except Exception as e:
        print('Error when capturing an image: {}'.format(e))
        return False
    try:
        print('Save to ' + OutputFileName)
        misc.imsave(OutputFileName, Image)
        return True
    except Exception as e:
        print('Error when saving an image: {}'.format(e))
        return False


def setPanTilt(PANVAL, TILTVAL):
    URL_Str = URL_SetPanTilt.replace("PANVAL", str(PANVAL)).replace("TILTVAL", str(TILTVAL))
    try:
        callURL(URL_Str, IPVAL, USERVAL, PASSVAL)
        return True
    except Exception as e:
        print('Error when setting pan/tilt: {}'.format(e))
        return False


def setZoom(ZOOMVAL):
    URL_Str = URL_SetZoom.replace("ZOOMVAL", str(ZOOMVAL))
    try:
        callURL(URL_Str, IPVAL, USERVAL, PASSVAL)
        return True
    except Exception as e:
        print('Error when setting zoom: {}'.format(e))
        return False


def getZoom():
    Output = callURL(URL_GetZoom, IPVAL, USERVAL, PASSVAL).strip()
    ZOOMVAL = extractInfo(Output, RET_GetZoom)
    return ZOOMVAL


def setAutoFocusMode(FOCUSMODE):
    URL_Str = URL_SetFocusMode.replace("FOCUSMODE", str(FOCUSMODE))
    print(URL_Str)
    try:
        callURL(URL_Str, IPVAL, USERVAL, PASSVAL)
        return True
    except Exception as e:
        print('Error when setting autofocus mode: {}'.format(e))
        return False


def isCameraAvailable():
    try:
        getZoom()
        return True
    except:
        return False


def extractInfo(Text, RET_Str):
        StrList = RET_Str.split("*")
        StrList = [Str for Str in StrList if len(Str) > 0]
        Vals = []
        for Str in StrList:
            WordList = Str.split("{}")
            WordList = [Word for Word in WordList if len(Word) > 0]
            if len(WordList) == 1:
                Pos = Text.find(WordList[0])
                if Pos >= 0:
                    Val = Text[Pos + len(WordList[0]):]
                    ValList = Val.split("\n")
                    Vals.append(ValList[0].strip())
            elif len(WordList) == 2:
                Pos1 = Text.find(WordList[0])
                Pos2 = Text.find(WordList[1], Pos1 + len(WordList[0]))
                if Pos1 >= 0 and Pos2 >= Pos1:
                    Vals.append(Text[Pos1 + len(WordList[0]):Pos2])
            else:
                print("Unhandled case {}". format(Str))
        return Vals[0]


def readRunInfo(FileName):
    with open(FileName, 'r') as File:
        csvread = csv.DictReader(File)
        RunConfig = {"Index": [], "Col": [], "Row": [],
                     "PanDeg": [], "TiltDeg": [],
                     "Zoom": [], "Focus": []}
        for row in csvread:
            RunConfig["Index"].append(int(row["Index"]))
            RunConfig["Col"].append(int(row["Col"]))
            RunConfig["Row"].append(int(row["Row"]))
            RunConfig["PanDeg"].append(float(row["PanDeg"]))
            RunConfig["TiltDeg"].append(float(row["TiltDeg"]))
            RunConfig["Zoom"].append(int(row["Zoom"]))
            RunConfig["Focus"].append(row["Focus"])
        return RunConfig
    return None


def writeRunInfo(FileName, RunConfig):
    with open(FileName, 'w') as File:
        FieldNames = ["Index", "Col", "Row", "PanDeg", "TiltDeg", "Zoom",
                      "Focus"]
        File.write(','.join(FieldNames))
        for i in range(len(RunConfig["Index"])):
            row = [str(RunConfig[key][i]) for key in FieldNames]
            File.write('\n' + ','.join(row))
        return True
    return False


def getPanoFolder(RootFolder, CameraName, NoPanoInSameHour):
    Start = datetime.now()
    PanoFolder = os.path.join(
        RootFolder,
        CameraName,
        Start.strftime("%Y"),
        Start.strftime("%Y_%m"),
        Start.strftime("%Y_%m_%d"),
        Start.strftime("%Y_%m_%d_%H"),
        "{}_{}_{:02}".format(CameraName,
                             Start.strftime("%Y_%m_%d_%H"),
                             NoPanoInSameHour))
    if not os.path.exists(PanoFolder):
        os.makedirs(PanoFolder)
        return PanoFolder
    return None


def getFileName(PanoFolder, CameraName, PanoImageNo, FileExtension='jpg'):
    Now = datetime.now()
    FileName = os.path.join(PanoFolder,
                            "{}_{}_00_00_{:04}.{}".format(
                                CameraName,
                                Now.strftime("%Y_%m_%d_%H_%M"),
                                PanoImageNo, FileExtension))
    return FileName


def setFocusAt(PanDeg, TiltDeg):
    # set focus at the middle of field of view
    # this may work only with Axis camera
    setAutoFocusMode('on')
    setZoom(RunConfig["Zoom"][0])
    setPanTilt(PanDeg, TiltDeg)
    time.sleep(3)
    captureImage()
    setAutoFocusMode('off')

if __name__ == '__main__':
    # settings information
    RootFolder = '/home/chuong/data/PanoFallback'
#    RootFolder = '/home/chuong/data/gigavision'
    CameraName = 'ARB-GV-HILL-1'
    StartHour = 7
    EndHour = 17
    LoopIntervalMinute = 60  # take panoram every 1 hour
    PanoWaitMin = 15
    RunInfoFileName = '/home/chuong/data/PanoFallback/test/RunInfo.cvs'
    CamConfigFile = '/home/chuong/workspace/Gigavision/AxisCamera.yml'

    RunConfig = readRunInfo(RunInfoFileName)
    CamConfig = yaml.load(open(CamConfigFile, 'r'))

    while True:
        Start = datetime.now()
        WithinHourRange = (Start.hour >= StartHour and
                           Start.hour <= EndHour)
        if WithinHourRange:
            # wait until camera is available
            while not isCameraAvailable():
                print('Camera is not available. Check again in 15 min.')
                time.sleep(15*60)  # sleep 15 minute

            # set focus at the middle of field of view
            i_mid = int(len(RunConfig["Index"])/2)
            setFocusAt(RunConfig["PanDeg"][i_mid], RunConfig["TiltDeg"][i_mid])

            for h in range(10):
                PanoFolder = getPanoFolder(RootFolder, CameraName, h)
                if PanoFolder is not None:
                    break

            # make sure zoom is set at begining
            setZoom(RunConfig["Zoom"][0])
            setPanTilt(RunConfig["PanDeg"][0], RunConfig["TiltDeg"][0])
            time.sleep(3)
            for i in RunConfig["Index"]:
                if not setPanTilt(RunConfig["PanDeg"][i], RunConfig["TiltDeg"][i]):
                    print('Failed to set pan/tilt. Skip this panorama')
                    break
                if i > 0 and RunConfig["Col"][i-1] != RunConfig["Col"][i]:
                    # move to next column need more time
                    time.sleep(3)
                    print('Sleep 3 secs')
                else:
                    time.sleep(0.5)
#                ImageFileName = getFileName(PanoFolder, CameraName, i, 'bmp')
#                captureImage2File(ImageFileName)
                ImageFileName = getFileName(PanoFolder, CameraName, i, 'jpg')
                if not captureImage2File2(ImageFileName):
                    print('Error in capture panorama. Skip this panorama')
                    break

            # write panoram config file
            os.makedirs(os.path.join(PanoFolder, '_data'))
            RunConfigFile = os.path.join(PanoFolder, '_data', 'RunInfo.csv')
            writeRunInfo(RunConfigFile, RunConfig)

            print('Finished one panorama')

            # wait until next hour
            while True:
                End = datetime.now()
                Quotient, Remainder = divmod((End.hour*60 + End.minute),
                                             LoopIntervalMinute)
                if Remainder <= PanoWaitMin:
                    break
                DueTime = (Quotient+1)*LoopIntervalMinute
                WaitMin = DueTime - (End.hour*60 + End.minute)
                print("Wait for {} minutes before start.".format(WaitMin))
                time.sleep(WaitMin*60)
        else:
            # sleep until start of hour range
            Now = datetime.now()
            DueTime = (24 + StartHour)*60
            WaitMin = DueTime - (Now.hour*60 + Now.minute)
            Hours, Mins = divmod(WaitMin, 60)
            print("Wait {} hours and {} minutes".format(Hours, Mins))
            time.sleep(WaitMin*60)