src/motecontrol/client/SimpleMoteManager.java

   1 package diku.distlab.motecontrolclientlib.client;
   2 import java.util.ArrayList;
   3
   4 import diku.distlab.motecontrolclientlib.protocols.MsgPayload;
   5 import diku.distlab.motecontrolclientlib.protocols.motecontrol.*;
   6
   7 public class SimpleMoteManager extends MoteManager {
   8
   9         public SimpleMoteManager() {
  10                 super();
  11         }
  12
  13         protected void grantMoteControl(long mote_id) {
  14                 SimpleMote mote;
  15                 mote = (SimpleMote)moteList.get(mote_id);
  16                 if (mote == null)
  17                 {
  18                         mote = new SimpleMote(mote_id,this);
  19                         moteList.add(mote);
  20                 }
  21                 mote.getControlStatus().grantControl();
  22                 // automatically detect new mote status
  23                 status(toArray(mote.getId()));
  24         }
  25
  26         protected void revokeMoteControl(long mote_id) {
  27                 SimpleMote mote;
  28                 mote = (SimpleMote) moteList.remove(mote_id);
  29                 if (mote != null)
  30                 {
  31                         mote.getControlStatus().revokeControl();
  32                         mote.getStatus().setStatus(SimpleMoteStatus.UNAVAILABLE);
  33                 }
  34
  35         }
  36
  37         protected void revokeAllMoteControls()
  38         {
  39                 Object[] motes = moteList.getCollection().toArray();
  40
  41                 for (int i=0;i<motes.length;i++)
  42                 {
  43                         revokeMoteControl(((Mote)motes[i]).getId());
  44                 }
  45         }
  46
  47         protected void confirm(long mote_id, MsgPayload moteMsg)
  48         {
  49                 MoteMsg msg = new MoteMsg();
  50                 try
  51                 {
  52                         msg.setBytes(moteMsg.getData());
  53                 }
  54                 catch (Exception e)
  55                 {
  56                         e.printStackTrace();
  57                 }
  58                 SimpleMote mote;
  59                 mote = (SimpleMote)moteList.get(mote_id);
  60                 if (mote != null)
  61                 {
  62                         switch (msg.getType().getValue())
  63                         {
  64                                 case MoteMsgType.CONFIRM:
  65                                         mote.getStatus().update(msg.getConfirm());
  66                                         break;
  67                                 case MoteMsgType.DATA:
  68                                         mote.dataOut(msg.getData());
  69                                         break;
  70                                 default:
  71                                         // TODO:take corrective action here
  72                                         //throw new Exception("Invalid mote message!");
  73                         }
  74                 }
  75         }
  76
  77         public void requestMotes(long[] moteids)
  78         {
  79                 session.requestControl(moteids);
  80         }
  81
  82         public void releaseMotes(long[] moteids) {
  83                 Mote mote;
  84                 for (int i = 0; i < moteids.length; i++)
  85                 {
  86                         mote = moteList.remove(moteids[i]);
  87                         if (mote != null) mote.getControlStatus().releaseControl();
  88                 }
  89                 session.releaseControl(moteids);
  90         }
  91
  92         public void releaseAllMotes()
  93         {
  94                 // TODO: release all mote controls
  95         }
  96
  97         public void cancelProgramming(long[] moteids) {
  98                 requestCommand(MsgCommand.CANCELPROGRAMMING, moteids);
  99         }
 100
 101         public void status(long[] moteids) {
 102                 requestCommand(MsgCommand.STATUS, moteids);
 103         }
 104
 105         public void start(long[] moteids) {
 106                 requestCommand(MsgCommand.START, moteids);
 107         }
 108
 109         public void stop(long[] moteids) {
 110                 requestCommand(MsgCommand.STOP, moteids);
 111         }
 112
 113         public void reset(long[] moteids) {
 114                 requestCommand(MsgCommand.RESET, moteids);
 115         }
 116
 117         public void program(long[] moteids, byte[] flashImage) {
 118                 MoteMsg msg = new MoteMsg();
 119                 msg.getType().setValue(MoteMsgType.REQUEST);
 120                 msg.getRequest().getCommand().setValue(MsgCommand.PROGRAM);
 121                 msg.getRequest().getFlashImage().setData(flashImage);
 122                 session.sendMoteMsg(msg,filterReady(moteids,msg));
 123         }
 124
 125         protected void sendData(long moteid, byte[] data) {
 126                 MoteMsg msg = new MoteMsg();
 127                 msg.getType().setValue(MoteMsgType.DATA);
 128                 msg.getData().setData(data);
 129                 session.sendMoteMsg(msg,toArray(moteid));
 130         }
 131
 132         private long[] filterReady(long[] moteids,MoteMsg msg)
 133         {
 134                 SimpleMote mote;
 135                 int i;
 136                 ArrayList list = new ArrayList();
 137                 for (i=0; i < moteids.length; i++)
 138                 {
 139                         mote = (SimpleMote)moteList.get(moteids[i]);
 140                         if (mote !=null && mote.getStatus().isReady())
 141                         {
 142                                 list.add(new Long(moteids[i]));
 143                                 mote.getStatus().update(msg.getRequest());
 144                         }
 145                 }
 146                 long[] res = new long[list.size()];
 147                 for (i=0; i < list.size(); i++)
 148                 {
 149                         res[i] = ((Long)list.get(i)).longValue();
 150                 }
 151                 return res;
 152         }
 153
 154         private void requestCommand(short command,long[] moteids)
 155         {
 156                 MoteMsg msg = new MoteMsg();
 157                 msg.getType().setValue(MoteMsgType.REQUEST);
 158                 msg.getRequest().getCommand().setValue(command);
 159                 session.sendMoteMsg(msg,filterReady(moteids,msg));
 160         }
 161
 162         private long[] toArray(long value)
 163         {
 164                 long[] res = new long[1];
 165                 res[0] = value;
 166                 return res;
 167         }
 168 }