SFBAlarm.h

Go to the documentation of this file.

/*                                              -*- mode:C++; fill-column:100 -*-
  SFBAlarm.h - Support for time-based events
  Copyright (C) 2009 The Regents of the University of New Mexico.  All rights reserved.

  This library is free software; you can redistribute it and/or
  modify it under the terms of the GNU Lesser General Public
  License as published by the Free Software Foundation; either
  version 2.1 of the License, or (at your option) any later version.

  This library is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  Lesser General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this library; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
  USA
*/

#ifndef SFBALARM_H_
#define SFBALARM_H_

#include "SFBTypes.h"
#include "SFBWiring.h"  /* for millis() */

typedef void (*SFBAlarmHandler)(u32 when);

typedef void (*SFBAlarmArgHandler)(u32 when, void *arg);

#ifndef SFBALARM_QUEUE_SIZE
#define SFBALARM_QUEUE_SIZE 51  

#endif

#if SFBALARM_QUEUE_SIZE < 256
typedef u8 SFBAlarmIndexType;   
#else
#if SFBALARM_QUEUE_SIZE < 65536
typedef u16 SFBAlarmIndexType;
#else
typedef u32 SFBAlarmIndexType;
#endif
#endif

class SFBAlarm {
  enum { MAX_EVENTS = SFBALARM_QUEUE_SIZE };
 public: 

  SFBAlarm() ;

  u32 create(SFBAlarmHandler handler) ;  

  u32 create(SFBAlarmArgHandler handler, void * arg) ; /* Create a new alarm object, return alarm number > 0 or 0 if full */ 

  void update(u32 alarmNumber, SFBAlarmHandler handler) { 
    update(alarmNumber,(SFBAlarmArgHandler) handler, (void*) 0);
  }

  void update(u32 alarmNumber, SFBAlarmArgHandler handler, void * arg); /* Alter the handler and/or arg of an existing alarm */ 

  u32 get(u32 alarmNumber) ;

  bool set(u32 alarmNumber, u32 when) ;

  bool cancel(u32 alarmNumber) ;   

  u32 currentAlarmNumber() { return currentAlarmIndex; }

  int runDueEvents(int maxEvents, u32 deadline) ; 

  int runPriorityEvents(int maxEvents, u32 priorityAtLeast) ; 

  bool canCreate() ;

  u32 created() { 
    return alarms; 
  }

  bool canAdd() ; 
  bool canRemove() { return used > 0; } 
  u32 count() { return used; } 
  bool inUse(u32 alarmIndex) {     
    return alarmIndex <= alarms
      && idxs[alarmIndex] > 0;
  }

  u32 getTopTime() {
    return times[slots[1]];
  }

 private:

  int getUsed() {
    return used;
  }

  void removeAndRunTop() ;

  void deleteElt(int eltIdx);              /* eltIdx must currently be in pq! */

  u32 times[MAX_EVENTS];                   /* when this event goes off */
  SFBAlarmArgHandler handlers[MAX_EVENTS]; /* who to call when it does */
  void * args[MAX_EVENTS];                 /* argument to pass back to them */
  SFBAlarmIndexType slots[MAX_EVENTS];     /* priority queue map slot idx->elt idx */
  SFBAlarmIndexType idxs[MAX_EVENTS];      /* reverse map elt idx->slot idx */
  SFBAlarmIndexType used;
  SFBAlarmIndexType alarms;
  SFBAlarmIndexType currentAlarmIndex;

  void bubbleUp(int fromIndex);
  void bubbleDown(int fromIndex);

#ifdef TEST_SFBALARM
  friend void heapcheck(int) ;
  friend void runRandoms() ;
#endif

};

extern SFBAlarm Alarms;

#endif /*SFBALARM_H_*/