FlexTpsSource.java

/*
 * Created on Dec 12, 2003
 *
 */
package org.nees.rbnb;

import java.io.DataInputStream;
import java.io.IOException;

import java.net.URL;
import java.net.URLConnection;
import java.net.MalformedURLException;

import com.rbnb.sapi.ChannelMap;
import com.rbnb.sapi.Source;
import com.rbnb.sapi.SAPIException;

import com.rbnb.utility.ArgHandler; //for argument parsing
//import COM.Creare.Utility.ArgHandler; //for argument parsing

public class FlexTpsSource extends RBNBBase {

    private final static String DEFAULT_RBNB_SOURCE_NANE = "SourceName";
    private final static String DEFAULT_RBNB_SOURCE = "TpsVideo";
    private final static String DEFAULT_RBNB_CHANNEL_NANE = "ChannelName";
    private final static String DEFAULT_RBNB_CHANNEL = "video.jpg";
    private final static String CAMERA_HOST_NAME = "TpsCameraHost";
    private final static String CAMERA_HOST = "undefined";

//   value                       notes
//   -----                       -------
//   Camera IP or Host name      required
//   feed                        "public" or "private"
//   stream                      required
//   fps                         optional
//   username                    optional
//   password                    optional

    private final static String CAMERA_URL_FEED_NAME = "URLFeedType";   
    private final static String CAMERA_URL_STREAM_NAME = "URLStream";
    private final static String CAMERA_URL_FPS_NAME = "URLFramesPerSecond";
    private final static String CAMERA_URL_USERNAME_NAME = "URLUsername";
    private final static String CAMERA_URL_PASSWORD_NAME = "URLPassword";
    private final static int DEFAULT_FPS = 10;
    
    private final static long RETRY_INTERVAL = 1000;
    
    private String cameraURLString;
    private String cameraHost = CAMERA_HOST;
    private String rbnbSourceName = DEFAULT_RBNB_SOURCE;
    private String rbnbChannelName = DEFAULT_RBNB_CHANNEL; 

    private static final String CACHE_SIZE_NAME = "CacheSize";  
    private static final int DEFAULT_CACHE_SIZE=30;
    
    private int cacheSize = DEFAULT_CACHE_SIZE;

    private String urlType = null;
    private String urlStreamName = null;
    private String urlUsername = null;
    private String urlPassword = null;
    private int urlFPS = DEFAULT_FPS; 
    
    Source source = null;
    ChannelMap sMap;
    int index;
    boolean connected = false;

    Thread timerThread;
    boolean runit = false;
    
    private String[] parameterNameArray;
    private String[] parameterTypeArray;
    private Object[] parameterDefaultArray;
    
    public static void main(String[] args) {
        // start from command line
        FlexTpsSource a = new FlexTpsSource();
        if (a.setArgs(args))
        {
            a.connect();
            a.startThread();
        }
    }

    public void printUsage()
    {
        System.out.println("Usage for FlexTpsSource...");
        super.printUsage(); // uses h, s, p
        System.out.println("-A TPS host (required)");
        System.out.println("-T Feed type (required public|private)");
        System.out.println("-N User defined stream name (required)");
        System.out.println("[-F Frames per second *" + DEFAULT_FPS + "]");
        System.out.println("[-U Username (optional)]");
        System.out.println("[-P Password (optional)]");
        System.out.println("[-S RBNB Source Name *" + DEFAULT_RBNB_SOURCE + "]");
        System.out.println("[-C RBNB Source Channel Name *" + DEFAULT_RBNB_CHANNEL + "]");
        System.out.println("[-z cache size *" + DEFAULT_CACHE_SIZE + "]");
        System.out.println();       
    }

    protected boolean setInstanceArgs(String[] args) throws IllegalArgumentException
    {
        //parse args
        try {
            ArgHandler ah=new ArgHandler(args);
            if (ah.checkFlag('A')) {
                String a=ah.getOption('A');
                if (a!=null) cameraHost=a;
            }
            if (ah.checkFlag('T')) {
                String a=ah.getOption('T');
                if (a!=null) urlType=a;
            }
            if (ah.checkFlag('N')) {
                String a=ah.getOption('N');
                if (a!=null) urlStreamName=a;
            }
            if (ah.checkFlag('F')) {
                String a=ah.getOption('F');
                if (a!=null)
                try
                {
                    Integer i =  new Integer(a);
                    int value = i.intValue();
                    urlFPS = value;
                }
                catch (Exception ignore) {} 
            }
            if (ah.checkFlag('U')) {
                String a=ah.getOption('U');
                if (a!=null) urlUsername=a;
            }
            if (ah.checkFlag('P')) {
                String a=ah.getOption('P');
                if (a!=null) urlPassword=a;
            }
            if (ah.checkFlag('S')) {
                String a=ah.getOption('S');
                if (a!=null) rbnbSourceName=a;
            }
            if (ah.checkFlag('C')) {
                String a=ah.getOption('C');
                if (a!=null) rbnbChannelName=a;
            }
            if (ah.checkFlag('z')) {
                String a=ah.getOption('z');
                if (a!=null)
                try
                {
                    Integer i =  new Integer(a);
                    int value = i.intValue();
                    cacheSize = value;
                }
                catch (Exception ignore) {} 
            }
        } catch (Exception e) {
            throw new IllegalArgumentException("Argument Exception: " + e);
        }

        if (cameraHost == null)
        {
            System.err.println("TPS camera hostname is required. Use FlexTpsSource -h for help");
            printUsage();
            return false;
        }
        
        if (urlType == null)
        {
            System.err.println("TPS camera stream type is required. Use FlexTpsSource -h for help");
            printUsage();
            return false;
        }
        
        if (!(urlType.equals("public") || urlType.equals("high-speed")))
        {
            System.err.println("TPS camera stream type must be either \"public\" or \"high-speed\".");
            System.err.println("Use FlexTpsSource -h for help");
            printUsage();
            return false;
        }

        if (urlStreamName == null)
        {
            System.err.println("TPS camera stream name is required. Use FlexTpsSource -h for help");
            printUsage();
            return false;
        }

        cameraURLString = "http://" + cameraHost + "/"
            + urlType + "/nph-mjpeg_stream.pl?stream="+ urlStreamName 
            + "&fps=" + urlFPS;
        
        System.out.println("Starting FlexTpsSource with... \n"
            + " camera URL = " + cameraURLString + "\n"
            + " RBNB Server = " + getServer() + "\n"
            + " RBNB Source name = " + rbnbSourceName + "\n"
            + " RBNB Channel name = " + rbnbChannelName + "\n"
            + "\nUse FlexTpsSource -h to all arguments.");

        return true;
    }

    public FlexTpsSource()
    {
        setParamterArrays();
        initialize();
    }

    protected void setParamterArrays() {

        String [] pNameArray = super.getBaseParameterNameArray();
        String [] pTypeArray = super.getBaseParameterTypeArray();
        Object [] pDefaultArray = super.getBaseParameterDefaultArray();
        
        int base = pNameArray.length;
        int numberOfparameters = 4;
        parameterNameArray = new String[base + numberOfparameters];
        parameterTypeArray = new String[base + numberOfparameters];
        parameterDefaultArray = new Object[base + numberOfparameters];
        
        for (int i = 0; i < base; i++)
        {
            parameterNameArray[i] = pNameArray[i];
            parameterTypeArray[i] = pTypeArray[i];
            parameterDefaultArray[i] = pDefaultArray[i];
        }

        parameterNameArray[base + 0] = DEFAULT_RBNB_SOURCE_NANE;
        parameterTypeArray[base + 0] = RBNBBaseParameterHolder.STRING_TYPE;
        parameterDefaultArray[base + 0] = new String(DEFAULT_RBNB_SOURCE);      

        parameterNameArray[base + 1] = DEFAULT_RBNB_CHANNEL_NANE;
        parameterTypeArray[base + 1] = RBNBBaseParameterHolder.STRING_TYPE;
        parameterDefaultArray[base + 1] = new String(DEFAULT_RBNB_CHANNEL);     

        parameterNameArray[base + 2] = CAMERA_HOST_NAME;
        parameterTypeArray[base + 2] = RBNBBaseParameterHolder.STRING_TYPE;
        parameterDefaultArray[base + 2] = new String(CAMERA_HOST);      

        parameterNameArray[base + 3] = CACHE_SIZE_NAME;
        parameterTypeArray[base + 3] = RBNBBaseParameterHolder.INTEGER_TYPE;
        parameterDefaultArray[base + 3] = new Integer(DEFAULT_CACHE_SIZE);      

    }
    
    /* (non-Javadoc)
     * @see org.nees.rbnb.RBNBBase#setVariablesFromParameters()
     */
    public void setVariablesFromParameters() {
        super.setBaseVarialbesFromParametes();

        rbnbSourceName = DEFAULT_RBNB_SOURCE;
        try{
            Object obj = getValueWithDefault(DEFAULT_RBNB_SOURCE_NANE, DEFAULT_RBNB_SOURCE);
            rbnbSourceName = (String)obj;
        } catch (Throwable ignore){}

        rbnbChannelName = DEFAULT_RBNB_CHANNEL; 
        try{
            Object obj = getValueWithDefault(DEFAULT_RBNB_CHANNEL_NANE, DEFAULT_RBNB_CHANNEL);
            rbnbChannelName = (String)obj;
        } catch (Throwable ignore){}

        cameraHost = CAMERA_HOST;
        try{
            Object obj = getValueWithDefault(CAMERA_HOST_NAME, CAMERA_HOST);
            cameraHost = (String)obj;
        } catch (Throwable ignore){}

        cacheSize = DEFAULT_CACHE_SIZE;
        try{
            Integer i = new Integer(DEFAULT_CACHE_SIZE);
            Object obj = getValueWithDefault(CACHE_SIZE_NAME, i);
            cacheSize = ((Integer)obj).intValue();
        } catch (Throwable ignore){}
        
    }

    public void connect()
    {
        if (connected) return;
        try {
            // Create a source and connect:
            source=new Source(cacheSize, "none", 0);
            source.OpenRBNBConnection(getServer(),rbnbSourceName);
            connected = true;
            System.out.println("Connecting to FlexTpsSource with... \n"
                + " camera URL = " + cameraURLString
                + ";\n RBNB Server = " + getServer()
                + "; RBNB Source name = " + rbnbSourceName
                + "; RBNB Channel name = " + rbnbChannelName
                + "\n\nUse FlexTpsSource -h to see optional arguments.");
        } catch (SAPIException se) { se.printStackTrace(); }
    }

    private void disconnect() {
        source.CloseRBNBConnection();
        connected = false;
        source = null;
    }

    public void startThread()
    {
        
        if (!connected) return;
        
        // Use this inner class to hide the public run method
        Runnable r = new Runnable() {
            public void run() {
              runWork();
            }
        };
        runit = true;
        timerThread = new Thread(r, "Timer");
        timerThread.start();
        System.out.println("TspSource: Started thread.");
    }

    public void stopThread()
    {
        if (!connected) return;
        
        runit = false;
        timerThread.interrupt();
        System.out.println("TspSource: Stopped thread.");
    }

    public void runWork() {
                
        boolean done = false;
        while (!done) {
            done = !execute();
            try {Thread.sleep(RETRY_INTERVAL); } catch (Exception e) {}
        }
        stop();     
    }

    private boolean execute()
    {
        if (!connected) return false;
        
        ChannelMap cmap = new ChannelMap();
        int channelId = -1;
        try {
            channelId = cmap.Add(rbnbChannelName);
        } catch (SAPIException e) {
            System.err.println("Failed to add video channel to channel map; name = "
                + rbnbChannelName);
            disconnect();
            return false;
        }
        cmap.PutTimeAuto("timeofday");
        cmap.PutMime(channelId, "image/jpeg");

        URL cameraURL = null;
        try {
            cameraURL = new URL(cameraURLString);
        } catch (MalformedURLException e) {
            System.err.println("URL is malformed; URL = " + cameraURLString);
            disconnect();
            return false;
        }
        
        URLConnection cameraConnection = null;
        try {
            cameraConnection = cameraURL.openConnection();
        } catch (IOException e) {
            System.err.println("Failed to connect to axis host with " + cameraURLString);
            disconnect();
            return false;
        }
        
        if (urlUsername != null) {
            if (urlPassword == null) urlPassword = "";
            String authString = urlUsername + ":" + urlPassword;
            String encodedAuthString = new sun.misc.BASE64Encoder().encode(authString.getBytes());
            cameraConnection.setRequestProperty("Authorization", "Basic " + encodedAuthString);
        }
        
        DataInputStream dis = null;
        try {
            dis = new DataInputStream(cameraConnection.getInputStream());
        } catch (IOException e) {
            System.err.println("Failed to get data stream from axis host.");
            System.err.println(e.getMessage() + ".");
            disconnect();
            return false;
        }

        String delimiter = "";
        String contentType = "";
        int contentLength = 0;
        int deltaTime = 0;
        byte[] imageData;
        
        StringBuffer inputLine = new StringBuffer();
        int currImage = 0;
        boolean readingData = false;
        boolean gotHeader = false;

        double lastTime = 0;
        double time = 0;
        double frameRate = 0;
        double averageFrameRate = urlFPS;
        
        boolean failed = false;         

        while(currImage < 1000000) {
                        
            if (readingData) {
                                                                    
                if (contentLength > 0) {                    
                    imageData = new byte[contentLength];
                    try {
                        dis.readFully(imageData);
                    } catch (IOException e) {
                        System.err.println("Failed to read JPEG image data from data stream.");
                        failed = true;
                        break;
                    }                                       
                } else { //this works but is could be implemented better
                    int index = 0;
                    imageData = new byte[100000];

                    while (index < imageData.length) {
                        if (index >= 4 && (char)imageData[index-4] == '\r' && (char)imageData[index-3] == '\n' && (char)imageData[index-2] == '\r' && (char)imageData[index-1] == '\n') {
                            index = index - 4;
                            break;
                        }
                        
                        try {
                            imageData[index++] = dis.readByte();
                        } catch (IOException e) {
                            System.err.println("Failed to read JPEG image data from data stream.");
                            failed = true;
                            break;
                        }
                    }
                    
                    if (failed) break;
                    
                    if (index == imageData.length) {
                        System.err.println("Ran out of space in data buffer.");
                        continue;
                    }

                }
                
                if (currImage > 0) {
                    frameRate = 1000/(double)deltaTime;
                    averageFrameRate = 0.995*averageFrameRate + 0.005*frameRate;
                }               

                try {
                    cmap.PutDataAsByteArray(channelId, imageData);
                } catch (SAPIException e) {
                    System.err.println("Failed to put image data into channel map.");
                    failed = true;
                    break;
                }               
                                
                try {
                    source.Flush(cmap, true);
                } catch (SAPIException e) {
                    System.err.println("Failed to flush output data to server.");
                    failed = true;
                    break;
                }               
                                                         
                if (currImage % 30 == 0) System.out.print(((long)(averageFrameRate*10))/((double)10) + " fps  \r");
                
                readingData = false;
                gotHeader = false;

                delimiter = "";
                contentType = "";
                contentLength = 0;
                deltaTime = 0;
                        
                currImage++;

            } else {
                char c;
                try {
                    c = (char) dis.readByte();
                } catch (IOException e) {
                    System.err.println("Failed to read header data.");
                    failed = true;
                    break;
                }

                if (c  == '\r') {
                    try {
                        dis.readByte(); // get \n also
                    } catch (IOException e) {
                        System.err.println("Failed to read header data (looking for new line).");
                        failed = true;
                        break;
                    } 
                    
                    if (inputLine.toString().startsWith("--")) {
                        delimiter = inputLine.substring(2);
                        gotHeader = true;
                    } else if (inputLine.toString().toLowerCase().startsWith("content-type")) {
                        contentType = inputLine.substring(14);                  
                    } else if (inputLine.toString().toLowerCase().startsWith("content-length")) {
                        contentLength = Integer.parseInt(inputLine.substring(16));
                    } else if (inputLine.toString().toLowerCase().startsWith("delta-time")) {
                        deltaTime = Integer.parseInt(inputLine.substring(12));
                    } else if (gotHeader && inputLine.toString().trim().length() == 0) {
                        readingData = true;
                    } else if (inputLine.toString().trim().length() != 0){
                        System.out.println("Received unexpected data.");
                    }

                    inputLine = new StringBuffer();         
                } else {
                    inputLine.append(c);
                }
            }               
        }
    
        try {
            dis.close();
        } catch (IOException e) {
            System.err.println("Failed to close connect to axis host.");
        }
        
        return !failed;         
    }

    public boolean isRunning()
    {
        return (connected && runit);
    }

    /* (non-Javadoc)
     * @see org.nees.rbnb.RBNBBase#getParameterNameArray()
     */
    public String[] getParameterNameArray() {
        return parameterNameArray;
    }

    /* (non-Javadoc)
     * @see org.nees.rbnb.RBNBBase#getParameterTypeArray()
     */
    public String[] getParameterTypeArray() {
        return parameterTypeArray;
    }

    /* (non-Javadoc)
     * @see org.nees.rbnb.RBNBBase#getParameterDefaultsArray()
     */
    public Object[] getParameterDefaultArray() {
        return parameterDefaultArray;
    }

    /* (non-Javadoc)
     * @see org.nees.rbnb.RBNBBase#start()
     */
    public boolean start() {
        if (isRunning()) return false;
        if (connected) disconnect();
        setVariablesFromParameters();
        connect();
        if (!connected) return false;
        startThread();
        return true;
    }

    /* (non-Javadoc)
     * @see org.nees.rbnb.RBNBBase#stop()
     */
    public boolean stop() {
        if (!isRunning()) return false;
        stopThread();
        disconnect();
        return true;
    }

}

---