#include <Arduino_FreeRTOS.h>

TaskHandle_t TaskHandle_2;
// TaskHandle_t TaskHandle_3;
// TaskHandle_t TaskHandle_4;

void setup()
{  
    Serial.begin(9600);
    Serial.println(F("In Setup function"));

    /* Use INT0(pin2) falling edge interrupt for resuming tasks */
    attachInterrupt(digitalPinToInterrupt(2), ExternalInterrupt, FALLING); 

    /* Create Task 2 with priority 2. Capture the Task details to the respective handler */
    xTaskCreate(MyTask2, "Task2", 100, NULL, 2, &TaskHandle_2);  
    // xTaskCreate(MyTask3, "Task3", 100, NULL, 3, &TaskHandle_3);
    // xTaskCreate(MyTask4, "Task4", 100, NULL, 4, &TaskHandle_4);  
}

void loop()
{
    // Hooked to Idle task, it will run whenever CPU is idle
    Serial.println(F("Loop function"));
    delay(1000);
}

static void ExternalInterrupt()
{
    static int count = 0;
    BaseType_t taskYieldRequired = 0;

    if(count < 1) // Modified condition to handle only one task
    {
        count++;
    }

    switch(count) // Resume one task depending on count value
    {    
    case 1:
        Serial.println(F("ISR Resuming Task2"));
        taskYieldRequired = xTaskResumeFromISR(TaskHandle_2);
        Serial.println(F("Leaving ISR"));
        break;

    // case 2:
    //     Serial.println(F("ISR Resuming Task3"));
    //     taskYieldRequired = xTaskResumeFromISR(TaskHandle_3);
    //     Serial.println(F("Leaving ISR"));
    //     break;

    // case 3:
    //     Serial.println(F("ISR Resuming Task4"));
    //     taskYieldRequired = xTaskResumeFromISR(TaskHandle_4);
    //     Serial.println(F("Leaving ISR"));
    //     break;

    default:
        //DO nothing
        break;
    }

    if(taskYieldRequired == 1) // If the taskYield is required then trigger the same.
    {
        taskYIELD();
    }
}

/* Task2 with priority 2 */
static void MyTask2(void* pvParameters)
{        
    Serial.println(F("Task2, Deleting itself"));
    vTaskDelete(NULL);     // Delete own task by passing NULL(TaskHandle_2 can also be used)
}

/*
// Task3 with priority 3
static void MyTask3(void* pvParameters)
{
    Serial.println(F("Task3, Deleting Itself"));
    vTaskDelete(NULL);     // Delete own task by passing NULL(TaskHandle_3 can also be used)  
}

// Task4 with priority 4
static void MyTask4(void* pvParameters)
{
    Serial.println(F("Task4 Running, Suspending all tasks"));
    vTaskSuspend(TaskHandle_2); // Suspend Task2/3
    vTaskSuspend(TaskHandle_3);
    vTaskSuspend(NULL);         // Suspend Own Task

    Serial.println(F("Back in Task4, Deleting Itself"));
    vTaskDelete(TaskHandle_4);       
}
*/