DaqToRbnb.java

Go to the documentation of this file.

package org.nees.daq;

import java.io.IOException;
import java.net.UnknownHostException;
import java.net.Socket;
import java.util.Hashtable;
import com.rbnb.sapi.*;

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 DaqToRbnb {

    private ControlPort controlPort = null;
    private DataThread dataThread = null;
    private ChannelEntry[] channels = new ChannelEntry[0];
    
    private static final String DEFAULT_DAQ_SERVER = "localhost";
    private static final int DEFAULT_DAQ_CONTROL_PORT = 55055;
    private static final int DEFAULT_DAQ_DATA_PORT = 55056;
    
    private String daqServerName = DEFAULT_DAQ_SERVER;
    private int daqControlPort = DEFAULT_DAQ_CONTROL_PORT;
    private int daqDataPort = DEFAULT_DAQ_DATA_PORT;
    
    private static final String DEFAULT_RBNB_SERVER = "localhost";
    private static final int DEFAULT_RBNB_PORT = 3333;
    private static final String DEFAULT_RBNB_SOURCE_NAME = "FromDAQ";

    private String rbnbServerName = DEFAULT_RBNB_SERVER;
    private int rbnbServerPort = DEFAULT_RBNB_PORT;
    private String rbnbSourceName = DEFAULT_RBNB_SOURCE_NAME;
    
    private Socket controlSocket = null;
    private Socket dataSocket = null;
    
    private Source source;

    public static final void main(String[] args)
    {
        try
        {
            DaqToRbnb control = new DaqToRbnb(args);
            control.startDaqConnections(); // note both connects must be made first
            control.connectToDaqControl();
            control.buildChannels();
            control.start();    
        }
        catch (Exception e)
        {
            e.printStackTrace();
        }
    }

    public DaqToRbnb(String[] args)
    {
        parseArgs(args);
    }
    
    private void printUsage() {
        System.out.println("DaqToRbnb: usage is...");       
        System.out.print("DaqToRbnb ");
        System.out.print("[-q DAQ Server *" + DEFAULT_DAQ_SERVER + "] ");
        System.out.print("[-c DAQ Control Port *" + DEFAULT_DAQ_CONTROL_PORT + "] ");
        System.out.print("[-d DAQ Control Port *" + DEFAULT_DAQ_DATA_PORT + "] ");
        System.out.print("[-s RBNB Server *" + DEFAULT_RBNB_SERVER + "] ");
        System.out.print("[-p RBNB Port *" + DEFAULT_RBNB_PORT + "] ");
        System.out.print("[-n RBNB Source Name *" + DEFAULT_RBNB_SOURCE_NAME + "] ");
        System.out.println();
    }

    private void parseArgs (String[] args)
    {
        String a;
        //parse args
        try {
            ArgHandler ah=new ArgHandler(args);
            if (ah.checkFlag('h')) {
                printUsage();
                RBNBProcess.exit(0);                
            }
            if (ah.checkFlag('q')) {
                a=ah.getOption('q');
                if (a!=null) daqServerName=a;
            }
            if (ah.checkFlag('c')) {
                a=ah.getOption('c');
                if (a!=null) daqControlPort=Integer.parseInt(a);
            }
            if (ah.checkFlag('d')) {
                a=ah.getOption('d');
                if (a!=null) daqDataPort=Integer.parseInt(a);
            }
            if (ah.checkFlag('s')) {
                a=ah.getOption('s');
                if (a!=null) rbnbServerName=a;
            }
            if (ah.checkFlag('p')) {
                a=ah.getOption('p');
                if (a!=null) rbnbServerPort=Integer.parseInt(a);
            }
            if (ah.checkFlag('n')) {
                a=ah.getOption('n');
                if (a!=null) rbnbSourceName=a;
            }   
        } catch (Exception e) {
            System.err.println("DaqToRbnb argument exception "+e.getMessage());
            e.printStackTrace();
            RBNBProcess.exit(0);                
        }
        System.out.println("Starting DaqToRbnb with");
        System.out.println("  DAQ: server = " + daqServerName + 
            "; control port = "+ daqControlPort + "; data port = " + daqDataPort );
        System.out.println("  RBNB: server = " + rbnbServerName + 
            "; port = "+ rbnbServerPort + "; source name = " + rbnbSourceName );
        System.out.println("  Use DaqToRbnb -h to see optional parameters");
    }
    
    
    private void startDaqConnections()
        throws UnknownHostException, IOException
    {
        controlSocket = new Socket(daqServerName,daqControlPort);

        try
        {
            Thread.sleep(1000);
        } catch (Exception ignore) {}
        System.out.println("...Continued");
        
        dataSocket = new Socket(daqServerName,daqDataPort);
    }

    private void connectToDaqControl()
        throws UnknownHostException, IOException
    {
        controlPort = null;
        
        ControlPort port = null;

        // use a local vaiable in case there is an exception
        port = new ControlPort(controlSocket);

        System.out.println("Pause for one sec after ControlPort...");
        try
        {
            Thread.sleep(1000);
        } catch (Exception ignore) {}
        System.out.println("...Continued");

        controlPort = port;
    }

    private void buildChannels()
        throws IOException, SAPIException
    {
        if (controlPort == null) return;

        dataThread = new DataThread(dataSocket);
    
        source = new Source();
        source.OpenRBNBConnection(rbnbServerName + ":" + rbnbServerPort, rbnbSourceName);
    
        String[] channelList = controlPort.getChannels();
        
        channels = new ChannelEntry[channelList.length];

        for (int i = 0; i < channelList.length; i++)
        {
            String name = channelList[i];
            channels[i] = new ChannelEntry(name);
            dataThread.addListener(name, channels[i]);
        }
        
    }

    private void start() throws IOException
    {
        if (controlPort == null) return;

        if (dataThread == null) return;

        for (int i = 0; i < channels.length; i++)
        {
            channels[i].subscribe();
        }

        dataThread.start();
        
    }
    
    private class ChannelEntry implements DaqListener
    {
            
        String name;
        ChannelMap map;
        boolean subscribed = false;
        int index;

        ChannelEntry(String name) throws SAPIException
        {
            this.name = name;
            map = new ChannelMap();
            index = map.Add(name);
            System.out.println("RBNB: Connection made to sever = "
                + rbnbServerName + ":" + rbnbServerPort + 
                " as source = " + rbnbSourceName + 
                " with channel = " + name + ".");
        }

        void subscribe() throws IOException
        {
            controlPort.subscribe(name);
            subscribed = true;
        }
        
        void unsubscribe() throws IOException
        {
            controlPort.unsubscribe(name);
            subscribed = false;
        }
    
        /*
            @brief Put DAQ data into turbine
            @param name Channel name
            @param time timestamp, RBNB format
            @param data DAQ datum
        */
        public void postData(String name, double  time, double data)
            throws SAPIException
        {
            if (!name.equals(this.name)) return;
            
            map.PutTime(time, 0.0);
            //map.PutTimeAuto("timeofday");
            double dataArray[] = new double[1];
            dataArray[0] = data;
            map.PutDataAsFloat64(index,dataArray);
            // System.out.println("Put Data, " + name + ": " + data);   
            source.Flush(map);
        }
    }
}

---