AxisSource.java

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

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.rbnb.utility.RBNBProcess; //alternative to System.exit, so
                                       //don't bring down servlet engine
//import COM.Creare.Utility.ArgHandler; //for argument parsing
//import COM.Creare.Utility.RBNBProcess; //alternative to System.exit, so
                                     //don't bring down servlet engine

public class AxisSource {

    private final static String DEFAULT_RBNB_SERVER = "neestpm.mcs.anl.gov";
    private final static String DEFAULT_RBNB_PORT = "3333";
    private final static String DEFAULT_RBNB_SOURCE = "AxisVideo";
    private final static String DEFAULT_RBNB_CHANNEL = "video.jpg";

    private String cameraURLString = null;
    private String cameraHost = null;
    private String rbnbServerName = DEFAULT_RBNB_SERVER;
    private String rbnbServerPort = DEFAULT_RBNB_PORT;
    private String rbnbHostName = rbnbServerName + ":" + rbnbServerPort;
    private String rbnbSourceName = DEFAULT_RBNB_SOURCE;
    private String rbnbChannelName = DEFAULT_RBNB_CHANNEL; 

    private final int requestedFPS = 30; 

    private static final int DEFAULT_CACHE_SIZE=30;
    private int cacheSize = DEFAULT_CACHE_SIZE;

    public AxisSource(String[] args)
    {
        //parse args
        try {
            ArgHandler ah=new ArgHandler(args);
            if (ah.checkFlag('h')) {
                printUsage();
                RBNBProcess.exit(0);                
            }
            if (ah.checkFlag('v')) {
                String a=ah.getOption('v');
                if (a!=null) cameraHost=a;
            }
            if (ah.checkFlag('r')) {
                String a=ah.getOption('r');
                if (a!=null) rbnbServerName=a;
            }
            if (ah.checkFlag('p')) {
                String a=ah.getOption('p');
                if (a!=null) rbnbServerPort=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) {
            System.err.println("AxisSource argument exception "+e.getMessage());
            e.printStackTrace();
            RBNBProcess.exit(0);
        }

        if (cameraHost == null)
        {
            System.err.println("Axis Video Camera Host is required. Use AxisSource -h for help");
            printUsage();
        }

        rbnbHostName = rbnbServerName + ":" + rbnbServerPort;
        cameraURLString = "http://" + cameraHost + "/axis-cgi/mjpg/video.cgi?camera=1&resolution=352x240&showlength=1&compression=25&clock=0&date=0&text=0&req_fps=30&deltatime=1";

    }

    public void start() {
                
        System.out.println("Starting AxisSource with... \n"
            + " camera URL = " + cameraURLString
            + ";\n RBNB Server = " + rbnbHostName
            + "; RBNB Soruce name = " + rbnbSourceName
            + "; RBNB Channel name = " + rbnbChannelName
            + "/nUse AxisSource -h to see optional arguments.");

        
        boolean done = false;
        while (!done) {
            done = !execute();
            try { new java.lang.Thread().wait(1000); } catch (Exception e) {}
        }       
    }

    private static void printUsage()
    {
        System.out.println("Usage for AxisSoruce...");
        System.out.println("  -v Axis camera host (required)");
        System.out.println("  [-r RBNB Server Name *"+ DEFAULT_RBNB_SERVER + "]");
        System.out.println("  [-p RBNB Server Port *" + DEFAULT_RBNB_PORT + "]");
        System.out.println("  [-s RBNB Source Name *" + DEFAULT_RBNB_SOURCE + "]");
        System.out.println("  [-c RBNB Source Channel Name *" + DEFAULT_RBNB_CHANNEL + "]");
        
    }

    private boolean execute() {
        Source mySource = new Source(cacheSize, "none", 0);
        try {
            mySource.OpenRBNBConnection(rbnbHostName, rbnbSourceName);
        } catch (SAPIException e) {
            System.err.println("Failed to connect to RBNB server with host = "
                + rbnbHostName + " and Source = " + rbnbSourceName);
            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);
            mySource.CloseRBNBConnection();
            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);
            mySource.CloseRBNBConnection();
            return false;
        }
        
        URLConnection cameraConnection = null;
        try {
            cameraConnection = cameraURL.openConnection();
        } catch (IOException e) {
            System.err.println("Failed to connect to axis host with " + cameraURLString);
            mySource.CloseRBNBConnection();
            return false;
        }
        
        DataInputStream dis = null;
        try {
            dis = new DataInputStream(cameraConnection.getInputStream());
        } catch (IOException e) {
            System.err.println("Failed to get data stream from axis host.");
            mySource.CloseRBNBConnection();
            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 = requestedFPS;
        
        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 {
                    mySource.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.");
                        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.");
        }
        
        mySource.CloseRBNBConnection();
        
        return !failed;
            
    }
    
    public static void main(String[] args) {
        AxisSource s = new AxisSource(args);
        s.start();
    }
}

---