/*******************************************************************************
 * The Things Network - ABP Feather
 *
 * Example of using an Adafruit Feather M0 and DHT22 with a
 * single-channel TheThingsNetwork gateway.
 *
 * This uses ABP (Activation by Personalization), where session keys for
 * communication would be assigned/generated by TTN and hard-coded on the device.
 *
 * Learn Guide: https://learn.adafruit.com/lora-pi
 *
 * Copyright (c) 2015 Thomas Telkamp and Matthijs Kooijman
 * Copyright (c) 2018 Terry Moore, MCCI
 * Copyright (c) 2018 Brent Rubell, Adafruit Industries
 *
 * Permission is hereby granted, free of charge, to anyone
 * obtaining a copy of this document and accompanying files,
 * to do whatever they want with them without any restriction,
 * including, but not limited to, copying, modification and redistribution.
 * NO WARRANTY OF ANY KIND IS PROVIDED.
 *******************************************************************************/
#include <lmic.h>
#include <hal/hal.h>
#include <SPI.h>
#include <TinyGPS++.h>
#include <SoftwareSerial.h>

#include "Base64.h"
#include "gateway/message.pb.h"

#include "pb_common.h"
#include "pb.h"
#include "pb_encode.h"
#include "pb_decode.h"

#include "lmic_project_config.h"
#include "gateway/message.pb.h"

// DHT digital pin and sensor type
#define DHTPIN 60
#define DHTTYPE DHT22

//
// For normal use, we require that you edit the sketch to replace FILLMEIN
// with values assigned by the TTN console. However, for regression tests,
// we want to be able to compile these scripts. The regression tests define
// COMPILE_REGRESSION_TEST, and in that case we define FILLMEIN to a non-
// working but innocuous value.
//
/*
#ifdef COMPILE_REGRESSION_TEST
# define FILLMEIN 0
#else
# warning "You must replace the values marked FILLMEIN with real values from the TTN control panel!"
# define FILLMEIN (#dont edit this, edit the lines that use FILLMEIN)
#endif
*/
// LoRaWAN NwkSKey, network session key
static const PROGMEM u1_t NWKSKEY[16] = { 0x52, 0x92, 0xC0, 0x72, 0x2D, 0x3C, 0x55, 0x5E, 0xE4, 0xB9, 0x9E, 0x9B, 0x88, 0x66, 0x47, 0xF1 };

// LoRaWAN AppSKey, application session key
static const u1_t PROGMEM APPSKEY[16] = { 0xC4, 0x30, 0xEF, 0x56, 0x4F, 0x6D, 0xA2, 0x56, 0x1F, 0x15, 0x2F, 0xB8, 0x62, 0xC7, 0xCA, 0xC2 };

// LoRaWAN end-device address (DevAddr)
// See http://thethingsnetwork.org/wiki/AddressSpace
// The library converts the address to network byte order as needed.
#ifndef COMPILE_REGRESSION_TEST
static const u4_t DEVADDR = 0x260212B6;
#else
static const u4_t DEVADDR = 0;
#endif

// These callbacks are only used in over-the-air activation, so they are
// left empty here (we cannot leave them out completely unless
// DISABLE_JOIN is set in arduino-lmic/project_config/lmic_project_config.h,
// otherwise the linker will complain).
void os_getArtEui (u1_t* buf) { }
void os_getDevEui (u1_t* buf) { }
void os_getDevKey (u1_t* buf) { }

// payload to send to TTN gateway
//static uint8_t payload[] = "Hello, world!";

// Data Packet to Send to TTN
u1_t loraData[Fenceless_CollarResponse_size+1] = {0};
static osjob_t sendjob;

// Schedule TX every this many seconds (might become longer due to duty
// cycle limitations).
const unsigned TX_INTERVAL = 10;

// Pin mapping for Adafruit Feather M0 LoRa
const lmic_pinmap lmic_pins = {
  .nss = 10,
  .rxtx = LMIC_UNUSED_PIN,
  .rst = 9,
  .dio = {2, 3, LMIC_UNUSED_PIN},
  .rxtx_rx_active = 0,
  .rssi_cal = 8,              // LBT cal for the Adafruit Feather M0 LoRa, in dB
  .spi_freq = 8000000,
};

// init. DHT
//DHT dht(DHTPIN, DHTTYPE);

/****************************************************
 * Arduino drivers
 * - LoRaWAN
 * - GPS
 * - Software Serial
 ***************************************************/
TinyGPSPlus gps;
SoftwareSerial ss(6, 7);

/****************************************************
 * Track each pair of X and Y coordinates
 * - arrays are used by the pnpoly function
 ***************************************************/
const uint8_t N_POLY_MAX=10;
float polyx[N_POLY_MAX];
float polyy[N_POLY_MAX];
int n_poly=0;

/****************************************************
 * Add a coordinate to the arrays
 * - stores a total of N_POLY_MAX pairs
 ***************************************************/
int push_vert(float x, float y) {
  if(n_poly>N_POLY_MAX)
    return 0;
  polyx[n_poly]=x;
  polyy[n_poly]=y;
  n_poly++;
  return 1;
}
/****************************************************
 * 'Clear' pairs of coordinates
 ***************************************************/
void clear_verts() {
  n_poly=0;
}

/****************************************************
 * Check a pair of coordinates against two lists
 * of vertices
 * - https://wrf.ecse.rpi.edu//Research/Short_Notes/pnpoly.html
 ***************************************************/
int pnpoly(int nvert, float *vertx, float *verty, float testx, float testy)
{ 
  int i, j, c = 0;
  for (i = 0, j = nvert-1; i < nvert; j = i++) {
    if ( ((verty[i]>testy) != (verty[j]>testy)) &&
        (testx < (vertx[j]-vertx[i]) * (testy-verty[i]) / (verty[j]-verty[i]) + vertx[i]) )
      c = !c;
  }
  return c;
}
/****************************************************
 * Test a coordinate against all vertices
 ***************************************************/
int check_bounds(float x, float y) {
  return pnpoly(n_poly, polyx, polyy, x, y);
}
/****************************************************
 * Load coordinates from protobuff stream
 * - currently a maximum of 10 coordinates
 * - loading arrays in nanopb does not appear
 *   to work.
 ***************************************************/
typedef struct {
  float x,y;
} coord;
void import_protobuf(uint8_t *protobuffer, uint32_t size) {

#define TYPE_STRING 0x0A
#define TYPE_VARIANT 0x10
#define PROTO_LEN 0x0A
#define FIELD_ONE_FLOAT 0x0D
#define FIELD_TWO_FLOAT 0x15
#define FIELD_TWO_VARIANT 0x10
#define FIELD_ONE_VARIANT 0x08
#define FIELD_TWO_STRING 0x12
#define FIELD_THREE_STRING 0x1A
#define FIELD_FOUR_STRING 0x22
#define FIELD_FIVE_STRING 0x2A
#define FIELD_SIX_STRING 0x32
#define FIELD_SEVEN_STRING 0x3A
#define FIELD_EIGHT_STRING 0x42
#define FIELD_NINE_STRING 0x4A
#define FIELD_TEN_STRING 0x52
#define FIELD_ELEVEN_STRING 0x5A

  /*uint8_t buffer0[] {
    FIELD_ONE_VARIANT, 0x01,
    FIELD_TWO_STRING, 0x0A, 
    FIELD_ONE_FLOAT, 0x1B, 0x91, 0xF6, 0xC2,
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40, 
    FIELD_THREE_STRING, 0x0A, 
    FIELD_ONE_FLOAT, 0xB5, 0x3B, 0x32, 0x42,
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40, 
    FIELD_FOUR_STRING, 0x0A,
    FIELD_ONE_FLOAT, 0x00, 0x00, 0x80, 0x3F, 
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40, 
    FIELD_FIVE_STRING, 0x0A,
    FIELD_ONE_FLOAT, 0x00, 0x00, 0x80, 0x3F, 
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40, 
    FIELD_SIX_STRING, 0x0A,
    FIELD_ONE_FLOAT, 0x00, 0x00, 0x80, 0x3F, 
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40, 
    FIELD_SEVEN_STRING, 0x0A,
    FIELD_ONE_FLOAT, 0x00, 0x00, 0x80, 0x3F, 
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40, 
    FIELD_EIGHT_STRING, 0x0A,
    FIELD_ONE_FLOAT, 0x00, 0x00, 0x80, 0x3F, 
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40, 
    FIELD_NINE_STRING, 0x0A,
    FIELD_ONE_FLOAT, 0x00, 0x00, 0x80, 0x3F, 
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40, 
    FIELD_TEN_STRING, 0x0A,
    FIELD_ONE_FLOAT, 0x00, 0x00, 0x80, 0x3F, 
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40, 
    FIELD_ELEVEN_STRING, 0x0A,
    FIELD_ONE_FLOAT, 0x00, 0x00, 0x80, 0x3F, 
    FIELD_TWO_FLOAT, 0x00, 0x00, 0x00, 0x40
    };*/

  Serial.println("DECODE FUNCTION");
  if(size != 122) {
    Serial.println("Failed to decode");
    Serial.print("Size:");
    Serial.println(size);
    return;
  }
  /* 
   * this stuff does not work yet
   */

  /*coord coordinates[10]; 
    int32_t isr;

    isr = 0;
    isr = protobuffer[1];

    uint8_t *ptr = protobuffer+5;
    for(int i=0;i<isr;i++)
    {
    memcpy(&coordinates[i].x, ptr + i*12, 4);
    memcpy(&coordinates[i].y, ptr + i*12+5, 4);
    }

    Serial.println("Recieved valid protobuf data?");
    clear_verts();
    for(int i=0;i<isr;i++) {
    push_vert(coordinates[i].x, coordinates[i].y);
    }*/
}

void onEvent (ev_t ev) {
  Serial.print(os_getTime());
  Serial.print(": ");
  switch(ev) {
    case EV_SCAN_TIMEOUT:
      Serial.println(F("EV_SCAN_TIMEOUT"));
      break;
    case EV_BEACON_FOUND:
      Serial.println(F("EV_BEACON_FOUND"));
      break;
    case EV_BEACON_MISSED:
      Serial.println(F("EV_BEACON_MISSED"));
      break;
    case EV_BEACON_TRACKED:
      Serial.println(F("EV_BEACON_TRACKED"));
      break;
    case EV_JOINING:
      Serial.println(F("EV_JOINING"));
      break;
    case EV_JOINED:
      Serial.println(F("EV_JOINED"));
      break;
      /*
         || This event is defined but not used in the code. No
         || point in wasting codespace on it.
         ||
         || case EV_RFU1:
         ||     Serial.println(F("EV_RFU1"));
         ||     break;
         */
    case EV_JOIN_FAILED:
      Serial.println(F("EV_JOIN_FAILED"));
      break;
    case EV_REJOIN_FAILED:
      Serial.println(F("EV_REJOIN_FAILED"));
      break;
    case EV_TXCOMPLETE:
      Serial.println(F("EV_TXCOMPLETE (includes waiting for RX windows)"));

      // if (LMIC.dataLen) {
        // import_protobuf(LMIC.frame, LMIC.dataLen);
      // }

       if (LMIC.txrxFlags & TXRX_ACK)
        Serial.println(F("Received ack"));
       if (LMIC.dataLen) {
         Serial.println(F("Received "));
         Serial.println(LMIC.dataLen);
         Serial.println(F(" bytes of payload"));
       }

      break;
    case EV_LOST_TSYNC:
      Serial.println(F("EV_LOST_TSYNC"));
      break;
    case EV_RESET:
      Serial.println(F("EV_RESET"));
      break;
    case EV_RXCOMPLETE:
      // data received in ping slot
      Serial.println(F("EV_RXCOMPLETE"));
      break;
    case EV_LINK_DEAD:
      Serial.println(F("EV_LINK_DEAD"));
      break;
    case EV_LINK_ALIVE:
      Serial.println(F("EV_LINK_ALIVE"));
      break;
      /*
         || This event is defined but not used in the code. No
         || point in wasting codespace on it.
         ||
         || case EV_SCAN_FOUND:
         ||    Serial.println(F("EV_SCAN_FOUND"));
         ||    break;
         */
    case EV_TXSTART:
      Serial.println(F("EV_TXSTART"));
      break;
    case EV_TXCANCELED:
      Serial.println(F("EV_TXCANCELED"));
      break;
    case EV_RXSTART:
      /* do not print anything -- it wrecks timing */
      break;
    case EV_JOIN_TXCOMPLETE:
      Serial.println(F("EV_JOIN_TXCOMPLETE: no JoinAccept"));
      break;
    default:
      Serial.print(F("Unknown event: "));
      Serial.println((unsigned) ev);
      break;
  }
}

#define TYPE_STRING 0x0A
#define TYPE_VARIANT 0x10
#define PROTO_LEN 0x0A
#define FIELD_ONE_FLOAT 0x0D
#define FIELD_TWO_FLOAT 0x15
#define FIELD_TWO_VARIANT 0x10


void do_send(osjob_t* j){
  uint8_t buffer[15] = {
    TYPE_STRING, 
    PROTO_LEN, 
    FIELD_ONE_FLOAT, 0x00, 0x00, 0x48, 0x43,
    FIELD_TWO_FLOAT, 0x00, 0x00, 0xc8, 0x42,
    FIELD_TWO_VARIANT, 0, 0};

  // Check if there is not a current TX/RX job running
  if (LMIC.opmode & OP_TXRXPEND) {
    Serial.println(F("OP_TXRXPEND, not sending"));
  } else {
    // prepare upstream data transmission at the next possible time.
    // transmit on port 1 (the first parameter); you can use any value from 1 to 223 (others are reserved).
    // don't request an ack (the last parameter, if not zero, requests an ack from the network).
    // Remember, acks consume a lot of network resources; don't ask for an ack unless you really need it.

    float latitude = 123;
    float longitude = 456;

    latitude = 44.558308; 
    longitude = -123.28341;

    memcpy(buffer+3, (void*)&latitude, 4);
    memcpy(buffer+8, (void*)&longitude, 4);

    LMIC_setTxData2(1, buffer, 14, 0);
  }
  // Next TX is scheduled after TX_COMPLETE event.
}

void setup() {
  delay(1000);
  Serial.begin(9600);
  ss.begin(4800);

  delay(100);
  Serial.println(F("Starting"));

  pinMode(LED_BUILTIN, OUTPUT);

  // LMIC init
  os_init();
  // Reset the MAC state. Session and pending data transfers will be discarded.
  LMIC_reset();

  // Set static session parameters. Instead of dynamically establishing a session
  // by joining the network, precomputed session parameters are be provided.
  // On AVR, these values are stored in flash and only copied to RAM
  // once. Copy them to a temporary buffer here, LMIC_setSession will
  // copy them into a buffer of its own again.
  uint8_t appskey[sizeof(APPSKEY)];
  uint8_t nwkskey[sizeof(NWKSKEY)];
  memcpy_P(appskey, APPSKEY, sizeof(APPSKEY));
  memcpy_P(nwkskey, NWKSKEY, sizeof(NWKSKEY));
  LMIC_setSession (0x13, DEVADDR, nwkskey, appskey);

  /*
  // We'll disable all 72 channels used by TTN
  for (int c = 0; c < 72; c++){
  LMIC_disableChannel(c);
  }

  // We'll only enable Channel 16 (905.5Mhz) since we're transmitting on a single-channel
  LMIC_enableChannel(16);
  */

  LMIC_selectSubBand(1);

  // Disable link check validation
  LMIC_setLinkCheckMode(0);

  // TTN uses SF9 for its RX2 window.
  LMIC.dn2Dr = DR_SF9;

  // Set data rate and transmit power for uplink
  LMIC_setDrTxpow(DR_SF7,14);

  delay(100);
  // Start job
  // do_send(&sendjob);
}

/****************************************************
 * Read a byte from GPS over software serial
 ***************************************************/
int read_gps() {
  int ret = 0;
  // while(ss.available()>0) {
  //   gps.encode(ss.read());
  //   ret = 1;
  // }
  return ret;
}
/****************************************************
 * Set cursor to beginning of line and clear it
 ***************************************************/
const int16_t PROGRESS_BAR_COUNT = 50;
const int16_t START_OF_LINE = 13;

void clear_line() {
  Serial.write(START_OF_LINE);
  for(int i=0;i<PROGRESS_BAR_COUNT;i++)
    Serial.write(' ');
  Serial.write(START_OF_LINE);
}
/****************************************************
 * State variables
 * - track events of main loop
 ***************************************************/
enum STATE_ {
  START_GPS,
  WAITING_GPS,
  VERIFYING_GPS,
  SENDING_LORA,
  WAITING_LORA,
  LORA_DONE
};
// int state = START_GPS;
int state = SENDING_LORA;
int loopCounter = 0;
int startTime = 0;

void loop() {

  if(state == START_GPS) {
    Serial.println("Waiting for GPS");
    state = WAITING_GPS;
  }
  else if(state == WAITING_GPS) {
    int got_data = 
      read_gps();
    /****************************************************
     * loading bar animation
     ***************************************************/
    if(got_data) {
      //if(loopCounter%100==0)
      //  Serial.write('.');
      if(loopCounter>PROGRESS_BAR_COUNT*100) {
      //  clear_line();
        loopCounter=0;

        state = VERIFYING_GPS;
      }
      loopCounter++;
    }
  }
  else if(state == VERIFYING_GPS) {
    /****************************************************
     * if no data has been received from the gps in 5 seconds
     * then the GPS is probably not connected properly
     ***************************************************/
    if (millis() > 5000 && gps.charsProcessed() < 10)
    {
      Serial.println(F("No GPS detected: check wiring."));
      while(true);
    }
    /****************************************************
     * only send to LoRaWAN if valid GPS coordinates are
     * available
     ***************************************************/
    if(gps.location.isValid())
      state = SENDING_LORA;
    else
      state = WAITING_GPS;
  }
  else if(state == SENDING_LORA) {
    /****************************************************
     * send encoded buffer over LoRaWAN
     ***************************************************/
    Serial.println("Sending LoRa Data...");
    do_send(&sendjob);

    /****************************************************
     * set reference time for LoRaWAN transmission delay
     ***************************************************/
    startTime = millis() / 1000;
    state = WAITING_LORA;
  }
  else if(state == WAITING_LORA) {
    /****************************************************
     * don't block the GPS from reading here
     ***************************************************/
    // read_gps();

    /****************************************************
     * if enough seconds have been delayed then move to
     * next state
     ***************************************************/
    if((millis()/1000 - startTime) >= TX_INTERVAL) {
      Serial.println("Lora has finished waiting");
      state = LORA_DONE;
    }
  }
  else if(state == LORA_DONE) {
    state = VERIFYING_GPS;
    state = SENDING_LORA;
  }
  os_runloop_once();
}