From: AceVest <zhaoyanbai@126.com>
Date: Mon, 18 Jun 2018 07:11:54 +0000 (+0800)
Subject: ...
X-Git-Url: http://zhaoyanbai.com/repos/%22http:/www.isc.org/icons/zlib_tech.html?a=commitdiff_plain;h=5e896bdb6f31a57907640a35221b6b0e04f90a44;p=acecode.git

...
---

diff --git a/arduino/hardware/ace/avr/cores/avr/AceAvr.h b/arduino/hardware/ace/avr/cores/avr/AceAvr.h
index 38fb062..c2a676d 100644
--- a/arduino/hardware/ace/avr/cores/avr/AceAvr.h
+++ b/arduino/hardware/ace/avr/cores/avr/AceAvr.h
@@ -59,3 +59,5 @@ volatile uint8_t* get_digital_pin_port_output_register(uint8_t pin);
 
 void set_digital_pin_mode(uint8_t pin, uint8_t mode);
 void digital_write(uint8_t pin, uint8_t val);
+void yield(void);
+
diff --git a/arduino/hardware/ace/avr/cores/avr/Arduino.h b/arduino/hardware/ace/avr/cores/avr/Arduino.h
index f90c621..fa3dd62 100644
--- a/arduino/hardware/ace/avr/cores/avr/Arduino.h
+++ b/arduino/hardware/ace/avr/cores/avr/Arduino.h
@@ -8,6 +8,4 @@
  */
 
 #pragma once
-
 #include <AceAvr.h>
-
diff --git a/arduino/hardware/ace/avr/cores/avr/kernel.cpp b/arduino/hardware/ace/avr/cores/avr/kernel.cpp
index aa5cf9a..31875da 100644
--- a/arduino/hardware/ace/avr/cores/avr/kernel.cpp
+++ b/arduino/hardware/ace/avr/cores/avr/kernel.cpp
@@ -11,23 +11,24 @@
 
 void delay(unsigned long ms);
 
+const uint8_t task_cnt = MAX_TASK_CNT + 1;
+const uint8_t idle_task_priority = task_cnt - 1;
 struct task *current_task = 0;
-const uint8_t max_task_cnt = 8;
-struct task tasks[max_task_cnt];
-uint8_t idle_task_stack[TASK_STACK_SIZE];
+struct task tasks[task_cnt];
 uint32_t ticks = 0;
-void task_scheduler();
+int8_t kernel_initialized = 0;
 
-#define IDLE_TASK (tasks+max_task_cnt-1)
+#define IDLE_TASK (tasks+task_cnt-1)
 #define LED_TASK (tasks+0);
 #define DEBUG_TASK (tasks+1);
 
+void task_scheduler();
+
 uint8_t debug_task_stack[TASK_STACK_SIZE];
 void debug_task() {
     while(1) task_delay(1000);
     uint8_t pin = 12;
     set_digital_pin_mode(pin, OUTPUT);
-
     while(1) {
         digital_write(pin, HIGH);
         task_delay(20);
@@ -40,7 +41,6 @@ uint8_t led_task_stack[TASK_STACK_SIZE];
 void led_task() {
     uint8_t pin = 13;
     set_digital_pin_mode(pin, OUTPUT);
-
     while(1) {
         digital_write(pin, HIGH);
         task_delay(10);
@@ -52,9 +52,11 @@ void led_task() {
 
 // idle_task 在没有进程READY的情况下都会调度运行
 // 所以task_delay不能在此进程生效
+uint8_t idle_task_stack[TASK_STACK_SIZE];
+uint32_t idle_cnt = 0;
 void idle_task() {
     sei();
-    static uint32_t idle_cnt = 0;
+    kernel_initialized = 1;
     uint8_t pin = 12;
     set_digital_pin_mode(pin, OUTPUT);
     uint8_t state = LOW;
@@ -66,6 +68,7 @@ void idle_task() {
     }
 }
 
+
 void task_delay(uint16_t ticks)
 {
     cli();
@@ -77,7 +80,11 @@ void task_delay(uint16_t ticks)
 
 void create_task(void (*handler)(), uint8_t *stack, uint8_t priority)
 {
-    if(priority >= max_task_cnt) {
+    if(priority == idle_task_priority && handler != idle_task) {
+        return;
+    }
+
+    if(priority > task_cnt) {
         return;
     }
 
@@ -210,30 +217,9 @@ void task_switch(struct task *prev, struct task *next) {
     );
 }
 
-void init_tasks() {
-    cli();
-    for(uint8_t i=0; i<max_task_cnt; i++) {
-        struct task *t = tasks+i;
-        t->pid          = i;
-        t->handler      = 0;
-        t->stack        = 0;
-        t->state        = TASK_STATE_EMPTY;
-        t->delay_ticks  = 0;
-    }
-
-    create_task(led_task, led_task_stack, 0);
-    create_task(debug_task, debug_task_stack, 1);
-    create_task(idle_task, idle_task_stack, max_task_cnt-1);
-    current_task = IDLE_TASK;
-    IDLE_TASK->stack = idle_task_stack+TASK_STACK_SIZE - 3;
-
-    SP = (uint16_t)(IDLE_TASK->stack);
-    asm("ret;");
-}
-
 void task_scheduler() {
     struct task *next = IDLE_TASK;
-    for(uint8_t i=0; i<max_task_cnt; i++) {
+    for(uint8_t i=0; i<task_cnt; i++) {
         struct task *t = tasks+i;
         if(TASK_STATE_READY == t->state) {
             next = t;
@@ -244,13 +230,21 @@ void task_scheduler() {
     task_switch(current_task, next);
 }
 
+void yield(void)
+{
+    if(kernel_initialized == 0) {
+        return ;
+    }
+
+    task_scheduler();
+}
 
 extern "C" void TIMER1_COMPA_vect() __attribute__ ((signal,used, externally_visible));
 void TIMER1_COMPA_vect()
 {
     ticks++;
 
-    for(uint8_t i=0; i<max_task_cnt; i++) {
+    for(uint8_t i=0; i<task_cnt; i++) {
         struct task *t = tasks+i;
         if(TASK_STATE_SLEEP != t->state) {
             continue;
@@ -282,3 +276,26 @@ void init_timer1() {
     // enable timer compare interrupt
     TIMSK1 |= (1 << OCIE1A);
 }
+
+void init_tasks() {
+    cli();
+    for(uint8_t i=0; i<task_cnt; i++) {
+        struct task *t = tasks+i;
+        t->pid          = i;
+        t->handler      = 0;
+        t->stack        = 0;
+        t->state        = TASK_STATE_EMPTY;
+        t->delay_ticks  = 0;
+    }
+
+    create_task(led_task, led_task_stack, 0);
+    create_task(debug_task, debug_task_stack, 1);
+    create_task(idle_task, idle_task_stack, idle_task_priority);
+    current_task = IDLE_TASK;
+    IDLE_TASK->stack = idle_task_stack+TASK_STACK_SIZE - 3;
+
+    SP = (uint16_t)(IDLE_TASK->stack);
+    asm("ret;");
+}
+
+
diff --git a/arduino/hardware/ace/avr/cores/avr/kernel.h b/arduino/hardware/ace/avr/cores/avr/kernel.h
index 63606e4..6cf0863 100644
--- a/arduino/hardware/ace/avr/cores/avr/kernel.h
+++ b/arduino/hardware/ace/avr/cores/avr/kernel.h
@@ -10,8 +10,13 @@
 #pragma once
 #include <stdint.h>
 
+#define MAX_TASK_CNT    8
 #define TASK_STACK_SIZE 64
 
+#if MAX_TASK_CNT > 32
+#error "too much tasks"
+#endif
+
 enum task_state {
     TASK_STATE_EMPTY  = 0,
     TASK_STATE_READY  = 1,
@@ -30,3 +35,4 @@ void init_tasks();
 void create_task(void (*handler)(), char *stack, uint8_t priority);
 void task_switch(struct task *prev, struct task *next);
 void task_delay(uint16_t ticks);
+//void yield(void);
diff --git a/arduino/hardware/ace/avr/cores/avr/main.cpp b/arduino/hardware/ace/avr/cores/avr/main.cpp
index 8e4ee2a..3881bbd 100644
--- a/arduino/hardware/ace/avr/cores/avr/main.cpp
+++ b/arduino/hardware/ace/avr/cores/avr/main.cpp
@@ -6,7 +6,7 @@
  * Description: none
  * ------------------------------------------------------------------------
  */
-#include <Arduino.h>
+#include <AceAvr.h>
 
 void init_timer0();
 void init_timer1();
diff --git a/arduino/hardware/ace/avr/cores/avr/timer.cpp b/arduino/hardware/ace/avr/cores/avr/timer.cpp
index ea3dde6..ac02f3a 100644
--- a/arduino/hardware/ace/avr/cores/avr/timer.cpp
+++ b/arduino/hardware/ace/avr/cores/avr/timer.cpp
@@ -7,19 +7,10 @@
  * ------------------------------------------------------------------------
  */
 
-#include<Arduino.h>
+#include<AceAvr.h>
 
 
 volatile unsigned long timer0_overflow_count = 0;
-#if 0
-static void __empty() {
-    // Empty
-}
-
-void yield(void) __attribute__ ((weak, alias("__empty")));
-#else
-void yield(void) { }
-#endif
 
 unsigned long micros() {
     unsigned long m;