tinydbmain.java

用于传感器网络的节点操作系统 TinyOS 结构设计非常有意思

package net.tinyos.tinydb;

import java.util.*;
import net.tinyos.message.*;
import net.tinyos.sf.*;
import java.io.*;

/** TinyDBMain creates the main UI for the TinyDB java
    classes.
*/
public class TinyDBMain {
    /**
    * @param args the command line arguments
    */
    public static void main(String args[]) {
	
	if (!parseArgs(args)) return;
	try {
	    initMain();
	    if (Config.getParam("show-command-window").equalsIgnoreCase("true"))
		showControlPanel = true;
	    
	    String commandString = Config.getParam("show-command-window");
	    if (commandString != null) {
		if (commandString.toLowerCase().equals("true"))
		    showControlPanel = true;
		else
		    showControlPanel = false;
	    }
	    
	    if (cmdLine) {
		CmdLineOutput cmdLine = new CmdLineOutput(nw, cmdLineQuery);
		while (true) {
		    try {
			Thread.currentThread().sleep(1000);
		    } catch(Exception e) {
		    }
		}
	    } else {
		
		if (showControlPanel) {
		CmdFrame cmf = new CmdFrame(mif);	    	    
		cmf.setLocation(800,30);
		cmf.show();
		}
		
		mf = new MainFrame(nw, startGuiInterface);
		mf.setLocation(5,30);
		mf.show();
	    }
	} catch (Exception e) {
	    if (debug) e.printStackTrace();
	    System.out.println("Open failed -- network won't work.");
	}
	
	
    }
    
    /** Initialize the static variables that TinyDB needs to have set up to run 
        Note that this does not show any of the TinyDB UI, so it can be used
	in apps that have their own user interface

	@param configFile The name of the configuration file to use.
    */
    public static void initMain(String configFileName) throws IOException{
	TinyDBMain.configFile = configFileName;
	initMain();
	
    }
    
    /** Initialize the static variables that TinyDB needs to have set up to run 
        Note that this does not show any of the TinyDB UI, so it can be used
	in apps that have their own user interface
    */
    public static void initMain() throws IOException{
	

	Config.init(configFile);
	Catalog c = new Catalog(Config.getParam("catalog-file"));
	Catalog.curCatalog = c;

	String groupStr = Config.getParam("am-group-id");
	if (groupStr == null) 
	    groupid = GROUP_ID;
	else {
	    try {
		groupid = new Byte(groupStr).byteValue();
		if (debug) System.out.println("GROUPID : " + groupid);
	    } catch (NumberFormatException e) {
		System.out.println("Bad am-group-id entry in config file: " + groupStr);
		groupid = GROUP_ID;
	    }
	}
	
	String startSfString = Config.getParam("start-sf");
	if (startSfString != null)
	    if (startSfString.toLowerCase().equals("false"))
		startSerialForwarder=false;
	


	//open radio comm
	try {
	    Thread t;

	    if (getMoteIF() == null)
		throw new IOException();

	    t = new Thread(mif);
	    t.start();

	    for (int i = 0; i < 3; i++) {
		mif.send((short)-1,CommandMsgs.resetCmd((short)-1));
		try {
		    Thread.currentThread().sleep(200);
		} catch (InterruptedException e) {
		}
	    }
	    
	    nw = new TinyDBNetwork(mif);
	    addQueryListener(nw);
	    

	} catch (IOException e) {
	    if (debug) { e.printStackTrace();
	    System.out.println("Open failed -- network won't work.");
	    } 
	    throw e;
	}

    }

    /** Helper routine to handle the common steps in sending a query
	into the network.
	@param q The query to inject
	@param rl The result listner to receive (processed) results from
	          this query.
    */
    public static void injectQuery(TinyDBQuery q, ResultListener rl) throws IOException{
	//add the query to list of queries we know about
	notifyAddedQuery(q);
	// register the listener for results from this query
	nw.addResultListener(rl, true, q.getId()); 
	//inject the query
	nw.sendQuery(q); 

    }
    
    // The following methods are used to register/remove QueryListeners
    // that are notified whenever a query is started or stopped
    
    
    /** Register the specified QueryListener.  The listener will be notified
	everytime notifyAddedQuery or notifyRemovedQuery is called. 
    */
    public static void addQueryListener(QueryListener ql) {
	qls.addElement(ql);
    }

    /** Deregister the specified QueryListener */
    public static void removeQueryListener(QueryListener ql) {
	qls.removeElement(ql);
    }

    /** Notify all currently registered QueryListeners with a addQuery message */
    public static void notifyAddedQuery(TinyDBQuery q) {
	Enumeration e;
	e = qls.elements();
	while (e.hasMoreElements()) {
	    QueryListener ql = (QueryListener)e.nextElement();
	    ql.addQuery(q);
	}
    }

    /** Notify all currently register QueryListeners with a removeQuery message */
    public static void notifyRemovedQuery(TinyDBQuery q) {
	Enumeration e;
	e = qls.elements();
	while (e.hasMoreElements()) {
	    QueryListener ql = (QueryListener)e.nextElement();
	    ql.removeQuery(q);
	}
    }

    static boolean parseArgs(String[] args) {
	boolean setInterface = false;
	try {
	    for (int i = 0; i < args.length; i++) {
		if (args[i].equals("-text")) {
		    startGuiInterface = false;
		    if (setInterface) {
			System.out.println("-text option conflicts with previous input window option. \n" + usage);
			return false;
		    }
		    setInterface = true;
		}
		else if (args[i].equals("-cmdwindow"))
		    showControlPanel = true;
		else if (args[i].equals("-gui")) {
		    startGuiInterface = true;
		    if (setInterface) {
			System.out.println("-gui option conflicts with previous input window option. \n" + usage);
			return false;
		    }
		    setInterface = true;
		} else if (args[i].equals("-configfile")) {
		    configFile = args[++i];
		} else if (args[i].equals("-run")) {
		    cmdLine = true;
		    cmdLineQuery = args[++i];
		} else if (args[i].equals("-debug")) {
		    debug = true;
		} else {
		    System.out.println("Unknown argument: " + args[i]);
		    System.out.println(usage);
		    return false;
		}
	    } 
	} catch (ArrayIndexOutOfBoundsException e) {
	    System.out.println("Missing expected command line argument! \n" + usage);
	    return false;
	}

	return true;
    }


    /** RootId is the id of the mote that is the root of the query */
    static int getRootId() {
	if (mf != null) {
	    return mf.getRootId();
	} else return 0;
    }

    static MoteIF getMoteIF() throws IOException{
	String host = "localhost";
	short port = 9000;
	String commPort = Config.getParam("sf-comm-port");
	
	if (commPort == null)
	    commPort = "COM1";

	if (mif == null) {
	    if (startSerialForwarder)
		SerialForward.run(false, commPort, MSG_SIZE);
	    try {
		Thread.currentThread().sleep(100);
	    } catch (Exception e) {
		e.printStackTrace();
	    }
	    if (!startSerialForwarder) {
		String portString;

		host = Config.getParam("sf-host");
		if (host == null) {
		    System.out.println("sf-host config file option is not set, but start-sf is!");
		    return null;
		}
		portString = Config.getParam("sf-port");
		if (portString == null) {
		    System.out.println("sf-port config file option is not set, but start-sf is!");
		    return null;
		}
		try {
		    port = new Integer(portString).shortValue();
		} catch (NumberFormatException e) {
		    System.out.println("couldn't parse sf-port config file option (" + portString + " is not a valid port number.");
		    return null; //failed
		}
		
	    }
		try
		{
			mif = new MoteIF(host, port ,groupid,DATA_SIZE, false);
		}
		catch (Exception e)
		{
			e.printStackTrace();
		}
	} 
	return mif;
    }
    
    public static MoteIF mif = null;

    static byte groupid;


    public static TinyDBNetwork nw;

    static boolean debug = false;
    static boolean startSerialForwarder = true;
    static boolean startGuiInterface = true;
    static boolean showControlPanel = false;
    private static boolean cmdLine = false;
    private static String cmdLineQuery;

    //private static String configFile = "tinydb.conf";
    private static String configFile = "net/tinyos/tinydb/tinydb.conf";

    static final String usage = "Usage: java TinyDBMain [-text|-gui] [-cmdwindow] [-configfile file] [-run query] [-debug] \n\t-text: Use text input window \n\t-gui: Use gui input window (default) \n\t-cmdwindow: Display the command window. \n\t-configfile: The name of the configuration file to use (default: " + configFile + ")\n\t-run: Run the specified query on the command line\n\t-debug: Enable debugging messages";

    private static Vector qls = new Vector(); //vector of query listeners


    public static final int DATA_SIZE = QueryMsg.DEFAULT_MESSAGE_SIZE;
    public static final int MSG_SIZE = DATA_SIZE + 7;
    private static final int GROUP_ID = 105;

    private static MainFrame mf = null;



}