add time base scheduler

Author: shaoziyang

tmScheduler/README.md

@@ -0,0 +1,36 @@
+# Simple Time Base task scheduler
+
+simple Time Base task scheduler for micropython.
+
+
+
+## usage
+
+1. define task, set function, paramters, interval, task state etc.
+2. define scheduler, recommend use **machine.Timer(-1)** (soft timer) as time base.
+3. add tasks to scheduler.
+4. run scheduler. 
+
+## demo
+
+```
+from scheduler import Scheduler, Task
+import machine
+
+def LED():
+    pyb.LED(1).toggle()
+
+def pn(n):
+    print(n)
+
+task1 = Task(LED, None, 1000)
+task2 = Task(pn, 1, 1000)
+task3 = Task(pn, 2, 1500)
+
+sc = Scheduler(machine.Timer(-1))
+sc.add(task1)
+sc.add(task2)
+sc.add(task3)
+sc.scheduler()
+```
+

tmScheduler/demo1.py

@@ -0,0 +1,18 @@
+from scheduler import Scheduler, Task
+import machine
+
+def LED():
+    pyb.LED(1).toggle()
+
+def pn(n):
+    print(n)
+
+task1 = Task(LED, None, 1000)
+task2 = Task(pn, 1, 1000)
+task3 = Task(pn, 2, 1500)
+
+sc = Scheduler(machine.Timer(-1))
+sc.add(task1)
+sc.add(task2)
+sc.add(task3)
+sc.scheduler()

tmScheduler/scheduler.py

@@ -0,0 +1,95 @@
+'''
+    simple Time base Task Scheduler
+'''
+
+from micropython import const
+
+TASK_STOP = const(0)
+TASK_RUN = const(1)
+
+class Task():
+
+    def __init__(self, callback, param = None, interval = 1000, state = TASK_RUN):
+        self._callback = callback
+        self._param = param
+        self._cnt = 10
+        self._intv = interval
+        self._state = state
+        self._rt = 0
+
+    def pause(self):
+        self._state = TASK_STOP
+
+    def resume(self):
+        self._state = TASK_RUN
+
+    def run(self):
+        if self._param == None:
+            self._callback()
+        else:
+            self._callback(self._param)
+
+class Scheduler():
+    def __init__(self, tm, interval = 100, task_idle = None, task_err = None):
+        self._tasks = []
+        self._task_idle = task_idle
+        self._task_err = task_err
+        self._interval = interval
+        self._tmr = tm
+        self._tmr.init(period = interval, callback=self._tmrirq)
+
+    def _tmrirq(self, t):
+         for i in range(len(self._tasks)):
+            if self._tasks[i]._state == TASK_RUN:
+                self._tasks[i]._rt += 1
+
+    def _run(self, task):
+        if task._state == TASK_RUN:
+            try:
+                if task._rt >= task._cnt:
+                    task._rt = 0
+                    task.run()
+            except:
+                if self._task_err:
+                    self._task_err()
+
+    def scheduler(self):
+        while True:
+            for i in range(len(self._tasks)):
+                task = self._tasks[i]
+                self._run(task)
+            if self._task_idle:
+                self._task_idle()
+
+    def find(self, task):
+        try:
+            return self._tasks.index(task)
+        except:
+            return None
+
+    def clear(self):
+        self._tasks.clear()
+
+    def add(self, task):
+        if self.find(task) == None:
+            self._tasks.append(task)
+            task._cnt = task._intv // self._interval
+
+    def delete(self, task):
+        try:
+            self._tasks.remove(task)
+        except:
+            print('del task <', task, '> error')
+
+    def pause(self, task):
+        if self.find(task) != None:
+            self._tasks[self.find(task)].pause()
+
+    def resume(self, task):
+        if self.find(task) != None:
+            self._tasks[self.find(task)].resume()
+
+    def run(self, task):
+        if self.find(task) != None:
+            task._rt = task._cnt
+            self._run(task)