/*************************************************************************\
* Copyright (c) 2019 European Spallation Source ERIC
* ecmc is distributed subject to a Software License Agreement found
* in file LICENSE that is included with this distribution. 
*
*  ecmcFFT.cpp
*
*  Created on: Mar 22, 2020
*      Author: anderssandstrom
*      Credits to  https://github.com/sgreg/dynamic-loading 
*
\*************************************************************************/

// Needed to get headers in ecmc right...
#define ECMC_IS_PLUGIN

#include "ecmcFFT.h"
#include "ecmcFFTDefs.h"
#include "ecmcPluginClient.h"
#include "ecmcAsynPortDriver.h"

#define PRINT_IF_DBG_MODE(fmt, ...)        \
   {                                       \
     if(cfgDbgMode_){                         \
       printf(fmt, ## __VA_ARGS__);        \
     }                                     \
   }                                       \


// New data callback from ecmc
static int printMissingObjError = 1;

void f_dataUpdatedCallback(uint8_t* data, size_t size, ecmcEcDataType dt, void* obj) {
  if(!obj) {
    if(printMissingObjError){
      printf("%s/%s:%d: Error: Callback object NULL.. Data will not be added to buffer.\n",
              __FILE__, __FUNCTION__, __LINE__);
      printMissingObjError = 0;
      return;
    }
  }
  ecmcFFT * fftObj = (ecmcFFT*)obj;

  // Call the correct fft object with new data
  fftObj->dataUpdatedCallback(data,size,dt);
}


/** ecmc FFT class
 * This object can throw: 
 *    - bad_alloc
 *    - invalid_argument
 *    - runtime_error
*/
ecmcFFT::ecmcFFT(int   fftIndex,       // index of this object (if several is created)
                 char* configStr) {
  cfgDataSourceStr_ = NULL;
  dataBuffer_       = NULL;
  dataItem_         = NULL;
  fftDouble_        = NULL;
  asynPort_         = NULL;
  elementsInBuffer_ = 0;
  fftCalcDone_      = 0;
  callbackHandle_   = -1;
  objectId_         = fftIndex;
  scale_            = 1.0;
  // Config defaults
  cfgDbgMode_       = 0;
  cfgNfft_          = ECMC_PLUGIN_DEFAULT_NFFT; // samples in fft (must be n^2)
  cfgDcRemove_      = 0;
  cfgApplyScale_    = 1;   // Scale as default to get correct amplitude in fft
  asynPort_         = (ecmcAsynPortDriver*) getEcmcAsynPortDriver();
  if(!asynPort_) {
    throw std::runtime_error("Asyn port NULL");
  }

  parseConfigStr(configStr); // Assigns all configs
  // Check valid nfft
  if(cfgNfft_ <= 0) {
    throw std::out_of_range("NFFT must be > 0 and even N^2.");
  }
  // set scale factor
  scale_ = 1.0 / cfgNfft_; // sqrt((double)cfgNfft_);

  // Allocate buffers
  dataBuffer_      = new double[cfgNfft_];
  fftBuffer_       = new std::complex<double>[cfgNfft_];
  clearBuffers();

  // Allocate KissFFT
  fftDouble_ = new kissfft<double>(cfgNfft_,false);
}

ecmcFFT::~ecmcFFT() {
  if(dataBuffer_) {
    delete[] dataBuffer_;
  }
  // De regeister callback when unload
  if(callbackHandle_ >= 0) {
    dataItem_->deregDataUpdatedCallback(callbackHandle_);
  }
  if(cfgDataSourceStr_) {
    free(cfgDataSourceStr_);
  }
  if(fftDouble_) {
    delete fftDouble_;
  }
}

void ecmcFFT::parseConfigStr(char *configStr) {

  // check config parameters
  if (configStr && configStr[0]) {    
    char *pOptions = strdup(configStr);
    char *pThisOption = pOptions;
    char *pNextOption = pOptions;
    
    while (pNextOption && pNextOption[0]) {
      pNextOption = strchr(pNextOption, ';');
      if (pNextOption) {
        *pNextOption = '\0'; /* Terminate */
        pNextOption++;       /* Jump to (possible) next */
      }
      
      // ECMC_PLUGIN_DBG_OPTION_CMD
      if (!strncmp(pThisOption, ECMC_PLUGIN_DBG_OPTION_CMD, strlen(ECMC_PLUGIN_DBG_OPTION_CMD))) {
        pThisOption += strlen(ECMC_PLUGIN_DBG_OPTION_CMD);
        cfgDbgMode_ = atoi(pThisOption);
      } 
      
      // ECMC_PLUGIN_SOURCE_OPTION_CMD
      else if (!strncmp(pThisOption, ECMC_PLUGIN_SOURCE_OPTION_CMD, strlen(ECMC_PLUGIN_SOURCE_OPTION_CMD))) {
        pThisOption += strlen(ECMC_PLUGIN_SOURCE_OPTION_CMD);
        cfgDataSourceStr_=strdup(pThisOption);
      }

      // ECMC_PLUGIN_NFFT_OPTION_CMD
      else if (!strncmp(pThisOption, ECMC_PLUGIN_NFFT_OPTION_CMD, strlen(ECMC_PLUGIN_NFFT_OPTION_CMD))) {
        pThisOption += strlen(ECMC_PLUGIN_NFFT_OPTION_CMD);
        cfgNfft_ = atoi(pThisOption);
      }

      // ECMC_PLUGIN_APPLY_SCALE_OPTION_CMD
      else if (!strncmp(pThisOption, ECMC_PLUGIN_APPLY_SCALE_OPTION_CMD, strlen(ECMC_PLUGIN_APPLY_SCALE_OPTION_CMD))) {
        pThisOption += strlen(ECMC_PLUGIN_APPLY_SCALE_OPTION_CMD);
        cfgApplyScale_ = atoi(pThisOption);
      }

      // ECMC_PLUGIN_DC_REMOVE_OPTION_CMD
      else if (!strncmp(pThisOption, ECMC_PLUGIN_DC_REMOVE_OPTION_CMD, strlen(ECMC_PLUGIN_DC_REMOVE_OPTION_CMD))) {
        pThisOption += strlen(ECMC_PLUGIN_DC_REMOVE_OPTION_CMD);
        cfgDcRemove_ = atoi(pThisOption);
      }
      pThisOption = pNextOption;
    }    
    free(pOptions);
  }

  // Data source must be defined...
  if(!cfgDataSourceStr_) { 
    throw std::invalid_argument( "Data source not defined.");
  }
}

void ecmcFFT::connectToDataSource() {
  // Get dataItem
  dataItem_        = (ecmcDataItem*) getEcmcDataItem(cfgDataSourceStr_);
  if(!dataItem_) {
    throw std::runtime_error( "Data item NULL." );
  }

  // Register data callback
  callbackHandle_ = dataItem_->regDataUpdatedCallback(f_dataUpdatedCallback, this);
  if (callbackHandle_ < 0) {
    throw std::runtime_error( "Failed to register data source callback.");
  }

  // Check data source
  if( !dataTypeSupported(dataItem_->getEcmcDataType()) ) {
    throw std::invalid_argument( "Data type not supported." );
  }
}

void ecmcFFT::dataUpdatedCallback(uint8_t*       data, 
                                  size_t         size,
                                  ecmcEcDataType dt) {
  // No buffer or full
  if(!dataBuffer_) {
    return;
  }
  
  if(cfgDbgMode_) {
    printEcDataArray(data, size, dt, objectId_);

    if(elementsInBuffer_ == cfgNfft_) {
      printf("Buffer full (%zu elements appended).\n",elementsInBuffer_);
    }
  }
  
  if(elementsInBuffer_ >= cfgNfft_) {
    //Buffer full
    if(!fftCalcDone_){
      calcFFT();
      scaleFFT();
      if(cfgDbgMode_){
        printComplexArray(fftBuffer_,
                    cfgNfft_,
                    objectId_);
        printEcDataArray((uint8_t*)dataBuffer_,
                          cfgNfft_*sizeof(double),
                          ECMC_EC_F64,
                          objectId_);
      }
      // Buffer new data
      clearBuffers();
    }
    return;
  }

  size_t dataElementSize = getEcDataTypeByteSize(dt);

  uint8_t *pData = data;
  for(unsigned int i = 0; i < size / dataElementSize; ++i) {    
    switch(dt) {
      case ECMC_EC_U8:        
        addDataToBuffer((double)getUint8(pData));
        break;
      case ECMC_EC_S8:
        addDataToBuffer((double)getInt8(pData));
        break;
      case ECMC_EC_U16:
        addDataToBuffer((double)getUint16(pData));
        break;
      case ECMC_EC_S16:
        addDataToBuffer((double)getInt16(pData));
        break;
      case ECMC_EC_U32:
        addDataToBuffer((double)getUint32(pData));
        break;
      case ECMC_EC_S32:
        addDataToBuffer((double)getInt32(pData));
        break;
      case ECMC_EC_U64:
        addDataToBuffer((double)getUint64(pData));
        break;
      case ECMC_EC_S64:
        addDataToBuffer((double)getInt64(pData));
        break;
      case ECMC_EC_F32:
        addDataToBuffer((double)getFloat32(pData));
        break;
      case ECMC_EC_F64:
        addDataToBuffer((double)getFloat64(pData));
        break;
      default:
        break;
    }
    
    pData += dataElementSize;
  }
}

void ecmcFFT::addDataToBuffer(double data) {
  if(dataBuffer_ && (elementsInBuffer_ < cfgNfft_) ) {
    //if(cfgApplyScale_) {
    //  dataBuffer_[elementsInBuffer_] = data * scale_;
    //} else {
      dataBuffer_[elementsInBuffer_] = data;
    //}
  }
  elementsInBuffer_ ++;
}

void ecmcFFT::clearBuffers() {
  memset(dataBuffer_, 0, cfgNfft_ * sizeof(double));
  memset(fftBuffer_,  0, cfgNfft_ * sizeof(std::complex<double>));
  elementsInBuffer_ = 0;
  fftCalcDone_      = 0;
}

void ecmcFFT::calcFFT() {
  fftDouble_->transform_real(dataBuffer_, fftBuffer_);
  fftCalcDone_ = 1;
}

void ecmcFFT::scaleFFT() {
  if(!cfgApplyScale_) {
    return;
  }

  for(unsigned int i = 0 ; i < cfgNfft_ ; ++i ) {
    fftBuffer_[i] = fftBuffer_[i] * scale_;
  }
}

void ecmcFFT::printEcDataArray(uint8_t*       data, 
                               size_t         size,
                               ecmcEcDataType dt,
                               int objId) {
  printf("fft id: %d, data: ",objId);

  size_t dataElementSize = getEcDataTypeByteSize(dt);

  uint8_t *pData = data;
  for(unsigned int i = 0; i < size / dataElementSize; ++i) {    
    switch(dt) {
      case ECMC_EC_U8:        
        printf("%hhu\n",getUint8(pData));
        break;
      case ECMC_EC_S8:
        printf("%hhd\n",getInt8(pData));
        break;
      case ECMC_EC_U16:
        printf("%hu\n",getUint16(pData));
        break;
      case ECMC_EC_S16:
        printf("%hd\n",getInt16(pData));
        break;
      case ECMC_EC_U32:
        printf("%u\n",getUint32(pData));
        break;
      case ECMC_EC_S32:
        printf("%d\n",getInt32(pData));
        break;
      case ECMC_EC_U64:
        printf("%" PRIu64 "\n",getInt64(pData));
        break;
      case ECMC_EC_S64:
        printf("%" PRId64 "\n",getInt64(pData));
        break;
      case ECMC_EC_F32:
        printf("%f\n",getFloat32(pData));
        break;
      case ECMC_EC_F64:
        printf("%lf\n",getFloat64(pData));
        break;
      default:
        break;
    }
    
    pData += dataElementSize;
  }
}

void ecmcFFT::printComplexArray(std::complex<double>* fftBuff,
                                size_t elements,
                                int objId) {
  printf("fft id: %d, results: \n",objId);
  for(unsigned int i = 0 ; i < elements ; ++i ) {
    printf("%d: %lf\n", i, std::abs(fftBuff[i]));
  }
}

int ecmcFFT::dataTypeSupported(ecmcEcDataType dt) {

  switch(dt) {
    case ECMC_EC_NONE:      
      return 0;
      break;
    case ECMC_EC_B1:
      return 0;
      break;
    case ECMC_EC_B2:
      return 0;
      break;
    case ECMC_EC_B3:
      return 0;
      break;
    case ECMC_EC_B4:
      return 0;
      break;
    default:
      return 1;
      break;
  }
  return 1;
}

uint8_t   ecmcFFT::getUint8(uint8_t* data) {
  return *data;
}

int8_t    ecmcFFT::getInt8(uint8_t* data) {
  int8_t* p=(int8_t*)data;
  return *p;
}

uint16_t  ecmcFFT::getUint16(uint8_t* data) {
  uint16_t* p=(uint16_t*)data;
  return *p;
}

int16_t   ecmcFFT::getInt16(uint8_t* data) {
  int16_t* p=(int16_t*)data;
  return *p;
}

uint32_t  ecmcFFT::getUint32(uint8_t* data) {
  uint32_t* p=(uint32_t*)data;
  return *p;
}

int32_t   ecmcFFT::getInt32(uint8_t* data) {
  int32_t* p=(int32_t*)data;
  return *p;
}

uint64_t  ecmcFFT::getUint64(uint8_t* data) {
  uint64_t* p=(uint64_t*)data;
  return *p;
}

int64_t   ecmcFFT::getInt64(uint8_t* data) {
  int64_t* p=(int64_t*)data;
  return *p;
}

float     ecmcFFT::getFloat32(uint8_t* data) {
  float* p=(float*)data;
  return *p;
}

double    ecmcFFT::getFloat64(uint8_t* data) {
  double* p=(double*)data;
  return *p;
}

size_t ecmcFFT::getEcDataTypeByteSize(ecmcEcDataType dt){
  switch(dt) {
  case ECMC_EC_NONE:
    return 0;
    break;

  case ECMC_EC_B1:
    return 1;
    break;

  case ECMC_EC_B2:
    return 1;
    break;

  case ECMC_EC_B3:
    return 1;
    break;

  case ECMC_EC_B4:
    return 1;
    break;

  case ECMC_EC_U8:
    return 1;
    break;

  case ECMC_EC_S8:
    return 1;
    break;

  case ECMC_EC_U16:
    return 2;
    break;

  case ECMC_EC_S16:
    return 2;
    break;

  case ECMC_EC_U32:
    return 4;
    break;

  case ECMC_EC_S32:
    return 4;
    break;

  case ECMC_EC_U64:
    return 8;
    break;

  case ECMC_EC_S64:
    return 8;
    break;

  case ECMC_EC_F32:
    return 4;
    break;

  case ECMC_EC_F64:
    return 8;
    break;

  default:
    return 0;
    break;
  }

  return 0;
}