#include <M5StickCPlus.h>
#include "AXP192.h"
TFT_eSprite tftSprite = TFT_eSprite(&M5.Lcd); 

void setup() {
    M5.begin();
    M5.Lcd.setRotation(3);
    tftSprite.createSprite(160, 80);
    tftSprite.setRotation(3);
    M5.Axp.EnableCoulombcounter();
}

void loop() {
    tftSprite.fillSprite(BLACK);
    tftSprite.setCursor(0, 0, 1);
    tftSprite.printf("AXP Temp: %.1fC \r\n", M5.Axp.GetTempInAXP192());
    tftSprite.setCursor(0, 10);
    tftSprite.printf("Bat:\r\n  V: %.3fv  I: %.3fma\r\n", M5.Axp.GetBatVoltage(), M5.Axp.GetBatCurrent());
    tftSprite.setCursor(0, 30);  
    tftSprite.printf("USB:\r\n  V: %.3fv  I: %.3fma\r\n", M5.Axp.GetVBusVoltage(), M5.Axp.GetVBusCurrent());
    tftSprite.setCursor(0, 50);    
    tftSprite.printf("5V-In:\r\n  V: %.3fv  I: %.3fma\r\n", M5.Axp.GetVinVoltage(), M5.Axp.GetVinCurrent());
    tftSprite.setCursor(0, 70);    
    tftSprite.printf("Bat power %.3fmw", M5.Axp.GetBatPower());
    tftSprite.pushSprite(40, 20);

    // 0x01 long press(1s), 0x02 press
    if(M5.Axp.GetBtnPress() == 0x02) 
    {
        esp_restart();
    }
    
    if(M5.BtnA.wasPressed())
    {
        // close tft voltage output
        M5.Axp.SetLDO2(false);
    }

    if(M5.BtnB.wasPressed())
    {
        // close tft voltage output
        M5.Axp.SetLDO2(true);
    }
    
    // M5.Axp.SetChargeCurrent(CURRENT_100MA);
    
    M5.update();
    delay(100);
}