Kool Diggs automatic dog bed

/* 
 * Project DogBed
 * Author: David Barbour
 * Date: 3/1/24
 */

#include "Particle.h"
#include "IoTClassroom_CNM.h"
#include "neopixel.h"
#include "Colors.h"
#include "math.h"
#include "IoTTimer.h"
#include "Adafruit_GFX.h"
#include "Adafruit_SSD1306.h"
#include "Adafruit_BME280.h"
#include "Graphic.h"
#include "Button.h"
#include "wemo.h"

//CONSTANTS
    //joystick 
    const int joyHorz = A1;
    const int joyVert = A2;
    const int joySwitch = D6;

    //neo pixel
    const int PIXELCOUNT = 20;

    //temperature reading
    const int SENSORWAITTINE=10;

    //Motion detector
    const int DETECTPIN=D9;

    //hue light bulb
    const int BULB=3;

//Variables
    //joystick
    int newVer;int newHor;int oldVer;int oldHor;
    bool setVertDown = false;bool setVertUp = false;
    bool setHorRight = false;bool setHorLeft = false;
    Button joyButton(joySwitch,true);

    //Neo pixels
    Adafruit_NeoPixel pixel ( PIXELCOUNT , SPI1 , WS2812B );

    //temperature reading
    bool forceHeat = false;bool forceCool = false;
    float coolingTemp = 74;float heatingTemp = 73;
    float currentTemp;
    bool status;
    Adafruit_BME280 bme;

    //display setup
    Adafruit_SSD1306 display(-1);
    bool showDisplay=false;

    //timers
    IoTTimer waitTimer;
    int waitedTime=0;

    //Motion detector
    bool motionDetected=false;

    //wemo device numbers
    int wemoCool=4;int wemoHeat=2; 

    //hue light bulb
    bool sendCmdToHue = true;

    //debugging button
    Button debugButton(D4,false);

    //application and setup states
    int applicationState=0;int prevApplicationState;
    int setupState=0;int subSetupState=0;

    //functions
    void PixelFill(int startPixel, int endPixel, int theColor);
    void setPixelDisplay(int theState);
    void programLogic();

    
    SYSTEM_MODE(MANUAL);

void setup() {

    //start serial monitor
    Serial.begin(9600);
    waitFor (Serial.isConnected,10000);

    //turn on wi fi
    WiFi.on();
    WiFi.clearCredentials();
    WiFi.setCredentials("IoTNetwork");
    WiFi.connect();
    while(WiFi.connecting()) {
        delay(50);
        Serial.printf(".");}

    //start the display
    display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
    display.clearDisplay();
    display.display();
    showDisplay=true;

    //start the neo pixels
    pixel.begin();
    pixel.setBrightness(22);
    pixel.clear();
    pixel.show();

    //set up motion detector
    pinMode(DETECTPIN,INPUT);

    //start bme temp guage
    status=bme.begin (0x76);
    if (status==false ) {
        Serial.printf (" BME280 at address %c failed to start ", 0x76 );}

}

void loop() 
{
    programLogic();

 }

void programLogic()
{
    
    //There are multiple states the application can be in
    // 0 - off
    // 1 - setup mode
    // 2 - waiting to cool
    // 3 - cooling
    // 4 - waiting to heat
    // 5 - heating

    //read the temperature
    currentTemp = (bme.readTemperature ()*9/5)+32.0; // deg F

    //this is for debugging only
    if (debugButton.isClicked()==true){
        //Serial.printf("Start %i, Temp %0.1f%cF\n\n",0,currentTemp,248);
        Serial.printf("Application %i, SetupState %i \n",applicationState,setupState);
        Serial.printf("Forcecool %i, forceheat %i motion %i\n",forceCool,forceHeat,motionDetected);
        Serial.printf("Currenttemp %f, cooltemp %f heatingtemp %f\n",currentTemp,coolingTemp,heatingTemp);
        Serial.printf("\n");
        }

    //determine what state you should be in
    prevApplicationState = applicationState;
    if (forceCool==true){
        forceHeat=false;
        applicationState=3;
        setupState=0;
        }

    if (forceHeat==true){
        forceCool = false;
        applicationState = 5;
        setupState=0;
        }

    //when you're in running mode
    if(forceCool==false && forceHeat==false && applicationState>1){  
        //go throught the temperature and motion settings
        if(currentTemp>=coolingTemp){   
            if (motionDetected==true){
                applicationState=3;
                }
            else{
                applicationState=2;
            }
        }
        
        //if the temperature goes below the heating temp, turn on 
        if(currentTemp<heatingTemp)
        {
            if (motionDetected==true){
                applicationState=5;
            }
            else{
                applicationState=4;
            }
        }
    }


    //only update the display if something changes
    if(applicationState!=prevApplicationState){showDisplay=true;}

    switch (applicationState){
    case 0: 
        //this is off

        //turn off the neopixels
        setPixelDisplay(0);

        //turn off the display
        if(showDisplay){
            display.clearDisplay();
            display.display();
            showDisplay=false;

            //turn off the heater
            wemoWrite(wemoHeat,LOW);

            //turn off the fan
            wemoWrite(wemoCool,LOW);
        }
          
        //if user clicks on button, start  running process
        if (joyButton.isPressed()==true){
            Serial.print("Push joystick button");
            applicationState=1;
            showDisplay=true;
        }

        //set these back to default off
        forceHeat = false;
        forceCool = false;
        break;
    
    case 1:  
        //setup mode

        setPixelDisplay(1);
        newVer = analogRead(joyVert);
        newHor = analogRead(joyHorz);
        switch (setupState){
        case 0: //on off screen
            
            if(showDisplay){
                display.clearDisplay();
                display.drawBitmap(20, 0,graphic_onoff, 80,68, 1);
                display.display();
                showDisplay=false;

                //turn off the heater
                wemoWrite(wemoHeat,LOW);

                //turn off the fan
                wemoWrite(wemoCool,LOW);

            }

            //DOWN set the application state back to off
            if(newVer>3500){setVertDown=true;}
            if(newVer<3500 && setVertDown==true){
                applicationState = 0;
                setVertDown = false;
                showDisplay=true;
            }

            //UP turn it on
            if(newVer<1500){setVertUp=true;}
            if(newVer>1500 && setVertUp==true){
                applicationState = 2;
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                showDisplay=true;
            }


            //RIGHT, go to the cooling teperature screen
            if(newHor>3500){setHorRight=true;}
            if(newHor<3500 && setHorRight==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                setupState = 1;
                showDisplay=true;
            }
            
            //set these back to default off
            forceHeat = false;
            forceCool = false;

            break;

        case 1:
            //set cooling termperature

            //up down arrow
            if (showDisplay){
                display.clearDisplay();
                display.drawBitmap(0, 9,graphic_updown,16,46, 1);
                display.setTextColor(WHITE);
                display.setTextSize(1);
                display.setCursor(20,0);
                display.printf("Set cooling temp");

                display.setTextSize(2);
                display.setCursor(60,25);
                display.printf("%.0f",coolingTemp);
                display.display();
                showDisplay=false;
            }
            
            //RIGHT, go to the heating teperature screen
            if(newHor>3500){setHorRight=true;}
            if(newHor<3500 && setHorRight==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                setupState = 2;
                showDisplay=true;
            }

            //LEFT, go to on off screen
            if(newHor<1000){setHorLeft=true;}
            if(newHor>1000 && setHorLeft==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                setupState = 0;
                showDisplay=true;
            }

            //UP, increase the cooling termperature
            if(newVer>3000){setVertUp=true;}
            if(newVer<3000 && setVertUp==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                coolingTemp--;

                //keep the temps from overlapping
                if(heatingTemp>=coolingTemp){heatingTemp=coolingTemp-1.0;}

                showDisplay=true;
            }

            //DOWN, decrease the cooling termperature
            if(newVer<1000){setVertDown=true;}
            if(newVer>1000 && setVertDown==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                coolingTemp++;

                //keep the temps from overlapping
                if(heatingTemp>=coolingTemp){heatingTemp=coolingTemp-1.0;}

                showDisplay=true;
            }

            break;
        
        case 2:
            //set heting termperature
            if(showDisplay){
                display.clearDisplay();
                display.drawBitmap(0, 9,graphic_updown,16,46, 1);
                display.setTextColor(WHITE);
                display.setTextSize(1);
                display.setCursor(20,0);
                display.printf("Set heating temp");

                display.setTextSize(2);
                display.setCursor(60,25);
                display.printf("%.0f",heatingTemp);
                display.display();
                showDisplay=false;
            }

            //LEFT, go to on off screen
            if(newHor<1000){setHorLeft=true;}
            if(newHor>1000 && setHorLeft==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                setupState = 1;
                showDisplay=true;
            }

            //UP, increase the heating termperature
            if(newVer>3000){setVertUp=true;}
            if(newVer<3000 && setVertUp==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                heatingTemp--;

                //keep the temps from overlapping
                if(heatingTemp>=coolingTemp){coolingTemp=heatingTemp+1.0;}

                showDisplay=true;
            }

            //DOWN, increase the heating termperature
            if(newVer<1000){setVertDown=true;}
            if(newVer>1000 && setVertDown==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                heatingTemp++;

                //keep the temps from overlapping
                if(heatingTemp>=coolingTemp){coolingTemp=heatingTemp+1.0;}

                showDisplay=true;
            }

            //Right, go to manual screen
            if(newHor>3000){setHorRight=true;}
            if(newHor<3000 && setHorRight==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                setupState = 3;
                showDisplay=true;
            }

            break;
        
        case 3:
            //manual heat or cool
            if(showDisplay){
                display.clearDisplay();
                display.drawBitmap(9, 5,graphic_hotcold,110,51, 1);
                display.setTextColor(WHITE);
                display.display();
                showDisplay=false;
            }

            //UP, start heat
            if(newVer<1000){setVertDown=true;}
            if(newVer>1000 && setVertDown==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                forceHeat = true;
                showDisplay=true;
            }

            //DOWN, start cooling
            if(newVer>3000){setVertUp=true;}
            if(newVer<3000 && setVertUp==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                forceCool = true;
                showDisplay=true;
            }

            //LEFT, to the heat temp screen
            if(newHor<1000){setHorLeft=true;}
            if(newHor>1000 && setHorLeft==true){
                setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
                setupState = 2;
                showDisplay=true;
            }
            break;
        }
        break;
    
    case 2:  //wating to cool
        setPixelDisplay(2);
        
        //tell the user, it's waiting to cool
        if(showDisplay){
            waitedTime = 0;
            waitTimer.startTimer(1000);   
            motionDetected=0;

            display.clearDisplay();
            display.setTextColor(WHITE);
            display.setTextSize(1);
            display.setCursor(20,0);
            display.printf("Waiting to cool");
            display.display();
            showDisplay=false;
        }

        //do a countdown, so the motion sensor doesn't start reading
        //for 10 seconds
        if (waitedTime < SENSORWAITTINE){
            if (waitTimer.isTimerReady()==true){
                display.fillRect(50,20, 70,30,BLACK);
                display.setTextColor(WHITE);
                display.setTextSize(2);
                display.setCursor(50,20);
                display.printf("%i",SENSORWAITTINE-waitedTime);
                display.display();
                waitedTime++;
                waitTimer.startTimer(1000); 
                forceCool=false;forceHeat=false;
            }
        }
        else{
            //read the motion sensor
            display.fillRect(50,20, 70,30,BLACK);
            display.display();
            motionDetected = digitalRead(DETECTPIN);
        }

        //LEFT, go back to setup screen
        if(newHor<1000){setHorLeft=true;}
        if(newHor>1000 && setHorLeft==true){

            //Serial.print("Cooling left button");
            setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
            applicationState = 1;
            setupState=0;
            waitedTime = 0;
            motionDetected=false;
            showDisplay=true;
            forceCool=false;forceHeat=false;
        }
        break;

    case 3:  //Cooling
        setPixelDisplay(3);
        newHor = analogRead(joyHorz);

        //tell the user, it's cooling
        if(showDisplay){
            display.clearDisplay();
            display.setTextColor(WHITE);
            display.setTextSize(2);
            display.setCursor(20,0);
            display.printf("Cooling");
            display.display();
            showDisplay=false;

            //turn on fan here
            wemoWrite(wemoCool,HIGH);
        }

        //LEFT, go back to setup screen
        if(newHor<1000){setHorLeft=true;}
        if(newHor>1000 && setHorLeft==true){
            setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
            applicationState = 1;
            setupState=0;
            waitedTime = 0;
            motionDetected=false;
            showDisplay=true;
            forceCool=false;forceHeat=false;
        }

        break;

    case 4:  //Wating to heat
        setPixelDisplay(4);

        //tell the user, it's cooling
        if(showDisplay){   
            waitedTime = 0;
            waitTimer.startTimer(1000); 
            motionDetected=0;

            display.clearDisplay();
            display.setTextColor(WHITE);
            display.setTextSize(1);
            display.setCursor(20,0);
            display.printf("Wating to heat");
            display.display();
            showDisplay=false;
        }

        //do a countdown, so the motion sensor doesn't 
        //start reading for 10 seconds
        if (waitedTime < SENSORWAITTINE){
            if (waitTimer.isTimerReady()==true){
                display.fillRect(50,20, 70,30,BLACK);
                display.setTextColor(WHITE);
                display.setTextSize(2);
                display.setCursor(50,20);
                display.printf("%i",SENSORWAITTINE-waitedTime);
                display.display();
                waitedTime++;
                waitTimer.startTimer(1000); 
            }
        }
        else
        {
            //read the motion sensor
            display.fillRect(50,20, 70,30,BLACK);
            display.display();
            motionDetected = digitalRead(DETECTPIN);
        }

        //LEFT, go back to setup screen
        if(newHor<1000){setHorLeft=true;}
        if(newHor>1000 && setHorLeft==true){
            setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
            applicationState = 1;
            setupState=0;
            waitedTime = 0;
            motionDetected=false;
            showDisplay=true;
            forceCool=false;forceHeat=false;
        }

        break;

    case 5:  //heating
        setPixelDisplay(5);
        newHor = analogRead(joyHorz);
        
        //tell the user, it's cooling
        if(showDisplay){
            display.clearDisplay();
            display.setTextColor(WHITE);
            display.setTextSize(2);
            display.setCursor(20,0);
            display.printf("Heating");
            display.display();
            showDisplay=false;

            //turn on heater here
            wemoWrite(wemoHeat,HIGH);
        }

        //LEFT, go back to setup screen
        if(newHor<1000){setHorLeft=true;}
        if(newHor>1000 && setHorLeft==true){
            Serial.print("Heating left button");
            setHorRight = false;setHorLeft= false;setVertDown=false;setVertUp=false;
            applicationState = 1;
            setupState=0;
            waitedTime = 0;
            motionDetected=false;
            forceHeat=false;
            showDisplay=true;
        }

        break;

    default:
        break;
    }


}


void setPixelDisplay(int theState)
{
    static int lastSwitch;
    static bool onOff;
    int brightNess;
    bool useHue = true;

    switch (theState)
    {
        case 0:
            //bed is off
            pixel.clear();
            pixel.show();
            if (useHue) {setHue(BULB,false,0,0,0);}
            break;

        case 1:  
            //setup mode  (blinking white every second)
            if(millis()-lastSwitch>500)
            {
                pixel.setBrightness(10);
                onOff = !onOff;
                if (onOff==true)
                {
                    PixelFill(0,PIXELCOUNT-1,white);
                    if (useHue) {setHue(BULB,true,HueOrange,75,10);}
                }
                else
                {
                    pixel.clear();
                    if (useHue) {setHue(BULB,false,0,0,0);}
                }
                pixel.show();
                lastSwitch = millis();
            }
            break;

        case 2: 
            // bed is ready to be cold (breath blue)
            PixelFill(0,PIXELCOUNT-1,blue);
            brightNess = 7 * sin(2.0*M_PI*(2.0/5.0)*millis()/1000.0)+ 10;
            pixel.setBrightness(brightNess);
            pixel.show();
            if (useHue) {setHue(BULB,true,HueBlue,brightNess,255);}
            break;

        case 3:
            //bed is in cold mode (steady blue)
            PixelFill(0,PIXELCOUNT-1,blue);
            pixel.setBrightness(40);
            pixel.show();
            if (useHue) {setHue(BULB,true,HueBlue,100,255);}
            break;

        case 4:
            // bed is ready to be hot (breathing yellow)
            PixelFill(0,PIXELCOUNT-1,yellow);
            brightNess = 7 * sin(2.0*M_PI*(2.0/5.0)*millis()/1000.0)+ 10;
            pixel.setBrightness(brightNess);
            pixel.show();
            if (useHue) {setHue(BULB,true,HueYellow,brightNess,255);}
            break;

        case 5:
            //bed is in heat mode (steady yellow)
            PixelFill(0,PIXELCOUNT-1,yellow);
            pixel.setBrightness(40);
            pixel.show();
            if (useHue) {setHue(BULB,true,HueYellow,100,255);}
            break;

        default:
            //bed is off
            pixel.clear();
            pixel.show();
            if (useHue) {setHue(BULB,false,0,0,0);}
            break;
    }

}

void PixelFill(int startPixel, int endPixel, int theColor)
{
  int theLoop;
  for (theLoop=startPixel;theLoop<=endPixel;theLoop++){
    pixel.setPixelColor(theLoop,theColor);
  }
}