/**
 * Furnace Tracker - multi-system chiptune tracker
 * Copyright (C) 2021-2025 tildearrow and contributors
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <string.h>
#include "asio.h"
#include "../ta-log.h"

static TAAudioASIO* callbackInstance=NULL;
extern AsioDrivers* asioDrivers;
bool loadAsioDriver(char *name);

static void _onBufferSwitch(long index, ASIOBool isDirect) {
  if (callbackInstance==NULL) return;
  callbackInstance->onProcess(index);
}

static void _onSampleRate(ASIOSampleRate rate) {
  if (callbackInstance==NULL) return;
  callbackInstance->onSampleRate(*(double*)(&rate));
}

static long _onMessage(long type, long value, void* msg, double* opt) {
  if (callbackInstance==NULL) return 0;
  switch (type) {
    case kAsioSelectorSupported:
      switch (value) {
        case kAsioSelectorSupported:
        case kAsioEngineVersion:
        case kAsioResetRequest:
        case kAsioBufferSizeChange:
        case kAsioResyncRequest:
          return 1;
        default:
          return 0;
      }
      break;
    case kAsioEngineVersion:
      return 2;
      break;
    case kAsioResetRequest:
      callbackInstance->requestDeviceChange();
      return 1;
      break;
    case kAsioBufferSizeChange:
      callbackInstance->onBufferSize(value);
      return 1;
      break;
    case kAsioResyncRequest:
      // ignore
      return 1;
      break;
  }
  return 0;
}

void TAAudioASIO::onSampleRate(double rate) {
  sampleRateChanged(SampleRateChangeEvent(rate));
}

void TAAudioASIO::onBufferSize(int bufsize) {
  bufferSizeChanged(BufferSizeChangeEvent(bufsize));
  desc.bufsize=bufsize;
}

void TAAudioASIO::onProcess(int index) {
  if (audioProcCallback!=NULL) {
    if (midiIn!=NULL) midiIn->gather();
    audioProcCallback(audioProcCallbackUser,inBufs,outBufs,desc.inChans,desc.outChans,desc.bufsize);
  }

  // upload here...
  for (int i=0; i<totalChans; i++) {
    if (chanInfo[i].isInput==ASIOTrue) continue;
    int ch=chanInfo[i].channel;
    if (ch>=desc.outChans) continue;
    float* srcBuf=outBufs[ch];

    switch (chanInfo[i].type) {
      // little-endian
      case ASIOSTInt16LSB: {
        short* buf=(short*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          buf[j]=CLAMP(srcBuf[j],-1.0,1.0)*32767.0f;
        }
        break;
      }
      case ASIOSTInt24LSB: {
        unsigned char* buf=(unsigned char*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          int val=CLAMP(srcBuf[j],-1.0,1.0)*8388608.0f;
          if (val<-8388608) val=-8388608;
          if (val>8388607) val=-8388607;
          *(buf++)=(val)&0xff;
          *(buf++)=(val>>8)&0xff;
          *(buf++)=(val>>16)&0xff;
        }
        break;
      }
      case ASIOSTInt32LSB:
      case ASIOSTInt32LSB16:
      case ASIOSTInt32LSB18:
      case ASIOSTInt32LSB20:
      case ASIOSTInt32LSB24: {
        int* buf=(int*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          int val=CLAMP(srcBuf[j],-1.0,1.0)*8388608.0f;
          if (val<-8388608) val=-8388608;
          if (val>8388607) val=-8388607;
          val<<=8;
          buf[j]=val;
        }
        break;
      }
      case ASIOSTFloat32LSB: {
        float* buf=(float*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          buf[j]=srcBuf[j];
        }
        break;
      }
      case ASIOSTFloat64LSB: {
        double* buf=(double*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          buf[j]=srcBuf[j];
        }
        break;
      }

      // big-endian
      case ASIOSTInt16MSB: {
        unsigned short* buf=(unsigned short*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          unsigned short val=(unsigned short)((short)(CLAMP(srcBuf[j],-1.0,1.0)*32767.0f));
          buf[j]=(val>>8)|(val<<8);
        }
        break;
      }
      case ASIOSTInt24MSB: {
        unsigned char* buf=(unsigned char*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          int val=CLAMP(srcBuf[j],-1.0,1.0)*8388608.0f;
          if (val<-8388608) val=-8388608;
          if (val>8388607) val=-8388607;
          *(buf++)=(val>>16)&0xff;
          *(buf++)=(val>>8)&0xff;
          *(buf++)=(val)&0xff;
        }
        break;
      }
      case ASIOSTInt32MSB:
      case ASIOSTInt32MSB16:
      case ASIOSTInt32MSB18:
      case ASIOSTInt32MSB20:
      case ASIOSTInt32MSB24: {
        unsigned int* buf=(unsigned int*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          int val=CLAMP(srcBuf[j],-1.0,1.0)*8388608.0f;
          if (val<-8388608) val=-8388608;
          if (val>8388607) val=-8388607;
          val<<=8;
          unsigned char* uVal=(unsigned char*)&val;
          buf[j]=(uVal[0]<<24)|(uVal[1]<<16)|(uVal[2]<<8)|(uVal[3]);
        }
        break;
      }
      case ASIOSTFloat32MSB: {
        unsigned int* buf=(unsigned int*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          float val=srcBuf[j];
          unsigned char* uVal=(unsigned char*)&val;
          buf[j]=(uVal[0]<<24)|(uVal[1]<<16)|(uVal[2]<<8)|(uVal[3]);
        }
        break;
      }
      case ASIOSTFloat64MSB: {
        unsigned char* buf=(unsigned char*)bufInfo[i].buffers[index];
        for (unsigned int j=0; j<desc.bufsize; j++) {
          double val=srcBuf[j];
          unsigned char* uVal=(unsigned char*)&val;
          *(buf++)=uVal[7];
          *(buf++)=uVal[6];
          *(buf++)=uVal[5];
          *(buf++)=uVal[4];
          *(buf++)=uVal[3];
          *(buf++)=uVal[2];
          *(buf++)=uVal[1];
          *(buf++)=uVal[0];
        }
        break;
      }

      default: // unsupported
        break;
    }
  }
}

String TAAudioASIO::getErrorStr(ASIOError which) {
  switch (which) {
    case ASE_OK:
      return "OK";
      break;
    case ASE_SUCCESS:
      return "Success";
      break;
    case ASE_NotPresent:
      return "Not present";
      break;
    case ASE_HWMalfunction:
      return "Hardware error";
      break;
    case ASE_InvalidParameter:
      return "Invalid parameter";
      break;
    case ASE_InvalidMode:
      return "Invalid mode";
      break;
    case ASE_SPNotAdvancing:
      return "Sample position not advancing";
      break;
    case ASE_NoClock:
      return "Clock not initialized";
      break;
    case ASE_NoMemory:
      return "Out of memory";
      break;
    default:
      break;
  }
  return "Unknown error";
}

String TAAudioASIO::getFormatName(ASIOSampleType which) {
  switch (which) {
    case ASIOSTInt16LSB:
      return "16-bit LSB";
      break;
    case ASIOSTInt24LSB:
      return "24-bit packed LSB";
      break;
    case ASIOSTInt32LSB:
      return "32-bit LSB";
      break;
    case ASIOSTFloat32LSB:
      return "32-bit float LSB";
      break;
    case ASIOSTFloat64LSB:
      return "64-bit float LSB";
      break;
    case ASIOSTInt16MSB:
      return "16-bit MSB";
      break;
    case ASIOSTInt24MSB:
      return "24-bit packed MSB";
      break;
    case ASIOSTInt32MSB:
      return "32-bit MSB";
      break;
    case ASIOSTFloat32MSB:
      return "32-bit float MSB";
      break;
    case ASIOSTFloat64MSB:
      return "64-bit float MSB";
      break;
    case ASIOSTInt32LSB16:
      return "16-bit padded LSB";
      break;
    case ASIOSTInt32LSB18:
      return "18-bit padded LSB";
      break;
    case ASIOSTInt32LSB20:
      return "20-bit padded LSB";
      break;
    case ASIOSTInt32LSB24:
      return "24-bit padded LSB";
      break;
    case ASIOSTInt32MSB16:
      return "16-bit padded MSB";
      break;
    case ASIOSTInt32MSB18:
      return "18-bit padded MSB";
      break;
    case ASIOSTInt32MSB20:
      return "20-bit padded MSB";
      break;
    case ASIOSTInt32MSB24:
      return "24-bit padded MSB";
      break;
    case ASIOSTDSDInt8LSB1:
      return "1-bit LSB";
      break;
    case ASIOSTDSDInt8MSB1:
      return "1-bit MSB";
      break;
    case ASIOSTDSDInt8NER8:
      return "1-bit padded";
      break;
  }
  return "Unknown";
}

void* TAAudioASIO::getContext() {
  return (void*)&driverInfo;
}

int TAAudioASIO::specialCommand(TAAudioCommand which) {
  switch (which) {
    case TA_AUDIO_CMD_SETUP:
      if (ASIOControlPanel()==ASE_NotPresent) return 0;
      return 1;
      break;
  }
  return -1;
}

bool TAAudioASIO::quit() {
  if (!initialized) return false;

  ASIOError result;

  if (running) {
    logV("CRASH: STOPPING NOW (QUIT)......");
    result=ASIOStop();
    if (result!=ASE_OK) {
      logE("could not stop on quit! (%s)",getErrorStr(result));
    }
    running=false;
  }

  logV("CRASH: ASIODisposeBuffers()");
  result=ASIODisposeBuffers();
  if (result!=ASE_OK) {
    logE("could not destroy buffers! (%s)",getErrorStr(result));
  }
  
  logV("CRASH: erase inBufs");
  for (int i=0; i<desc.inChans; i++) {
    delete[] inBufs[i];
    inBufs[i]=NULL;
  }
  logV("CRASH: erase outBufs");
  for (int i=0; i<desc.outChans; i++) {
    delete[] outBufs[i];
    outBufs[i]=NULL;
  }

  logV("CRASH: erase arrays");
  if (inBufs!=NULL) {
    delete[] inBufs;
    inBufs=NULL;
  }
  if (outBufs!=NULL) {
    delete[] outBufs;
    outBufs=NULL;
  }

  logV("CRASH: ASIOExit()");
  result=ASIOExit();
  if (result!=ASE_OK) {
    logE("could not exit!",getErrorStr(result));
  }
  logV("CRASH: removeCurrentDriver()");
  asioDrivers->removeCurrentDriver();

  logV("CRASH: reset callback instance");
  callbackInstance=NULL;
  initialized=false;

  return true;
}

bool TAAudioASIO::setRun(bool run) {
  if (!initialized) return false;
  if (running==run) {
    return running;
  }

  ASIOError result;

  if (run) {
    result=ASIOStart();
    if (result!=ASE_OK) {
      logE("could not start running! (%s)",getErrorStr(result));
      return false;
    }
    running=true;
  } else {
    // does it matter whether stop was successful?
    logV("CRASH: STOPPING NOW......");
    result=ASIOStop();
    if (result!=ASE_OK) {
      logE("could not stop running! (%s)",getErrorStr(result));
    }
    running=false;
  }
  return running;
}

void TAAudioASIO::requestDeviceChange() {
  deviceStatus=TA_AUDIO_DEVICE_RESET;
}

bool TAAudioASIO::init(TAAudioDesc& request, TAAudioDesc& response) {
  if (initialized) return false;
  if (callbackInstance) {
    logE("can't initialize more than one output!");
    return false;
  }

  desc=request;
  desc.outFormat=TA_AUDIO_FORMAT_F32;

  if (desc.deviceName.empty()) {
    // load first driver if not specified
    logV("getting driver names...");
    if (!driverNamesInit) {
      for (int i=0; i<ASIO_DRIVER_MAX; i++) {
        // 64 just in case
        driverNames[i]=new char[64];
      }
      driverNamesInit=true;
    }
    driverCount=asioDrivers->getDriverNames(driverNames,ASIO_DRIVER_MAX);

    // quit if we couldn't find any drivers
    if (driverCount<1) {
      logE("no ASIO drivers available");
      return false;
    }

    desc.deviceName=driverNames[0];
  }

  // load driver
  logV("loading ASIO driver... (%s)",desc.deviceName);
  strncpy(deviceNameCopy,desc.deviceName.c_str(),63);
  if (!loadAsioDriver(deviceNameCopy)) {
    logE("failed to load ASIO driver!");
    return false;
  }

  // init
  memset(&driverInfo,0,sizeof(driverInfo));
  memset(bufInfo,0,sizeof(ASIOBufferInfo)*ASIO_CHANNEL_MAX*2);
  memset(chanInfo,0,sizeof(ASIOChannelInfo)*ASIO_CHANNEL_MAX*2);
  driverInfo.asioVersion=2;
  driverInfo.sysRef=NULL;
  ASIOError result=ASIOInit(&driverInfo);

  if (result!=ASE_OK) {
    logE("could not init device! (%s)",getErrorStr(result));
    asioDrivers->removeCurrentDriver();
    return false;
  }

  // setup callbacks
  // unfortunately, only one TAAudio instance may exist at a time when using ASIO...
  callbackInstance=this;
  memset(&callbacks,0,sizeof(ASIOCallbacks));
  callbacks.bufferSwitch=_onBufferSwitch;
  callbacks.sampleRateDidChange=_onSampleRate;
  callbacks.asioMessage=_onMessage;

  // get driver information
  long maxInChans=0;
  long maxOutChans=0;

  result=ASIOGetChannels(&maxInChans,&maxOutChans);
  if (result!=ASE_OK) {
    logE("could not get channel count! (%s)",getErrorStr(result));
    ASIOExit();
    asioDrivers->removeCurrentDriver();
    return false;
  }

  if (maxInChans>ASIO_CHANNEL_MAX) maxInChans=ASIO_CHANNEL_MAX;
  if (maxOutChans>ASIO_CHANNEL_MAX) maxOutChans=ASIO_CHANNEL_MAX;

  if (desc.inChans>maxInChans) desc.inChans=maxInChans;
  if (desc.outChans>maxOutChans) desc.outChans=maxOutChans;

  long minBufSize=0;
  long maxBufSize=0;
  long actualBufSize=0;
  long bufSizeGranularity=0;
  result=ASIOGetBufferSize(&minBufSize,&maxBufSize,&actualBufSize,&bufSizeGranularity);
  if (result!=ASE_OK) {
    logE("could not get buffer size! (%s)",getErrorStr(result));
    ASIOExit();
    asioDrivers->removeCurrentDriver();
    return false;
  }

  ASIOSampleRate outRate;
  result=ASIOGetSampleRate(&outRate);
  if (result!=ASE_OK) {
    logE("could not get sample rate! (%s)",getErrorStr(result));
    ASIOExit();
    asioDrivers->removeCurrentDriver();
    return false;
  }
  desc.rate=*(double*)(&outRate);

  totalChans=0;

  if (desc.inChans>0) {
    inBufs=new float*[desc.inChans];
    for (int i=0; i<desc.inChans; i++) {
      chanInfo[totalChans].channel=i;
      chanInfo[totalChans].isInput=ASIOTrue;
      result=ASIOGetChannelInfo(&chanInfo[totalChans]);
      if (result!=ASE_OK) {
        logW("failed to get channel info for input channel %d! (%s)",i,getErrorStr(result));
      }
      bufInfo[totalChans].channelNum=i;
      bufInfo[totalChans++].isInput=ASIOTrue;
      inBufs[i]=new float[actualBufSize];
    }
  }
  if (desc.outChans>0) {
    outBufs=new float*[desc.outChans];
    for (int i=0; i<desc.outChans; i++) {
      chanInfo[totalChans].channel=i;
      chanInfo[totalChans].isInput=ASIOFalse;
      result=ASIOGetChannelInfo(&chanInfo[totalChans]);
      if (result!=ASE_OK) {
        logW("failed to get channel info for output channel %d! (%s)",i,getErrorStr(result));
      }
      bufInfo[totalChans].channelNum=i;
      bufInfo[totalChans++].isInput=ASIOFalse;
      outBufs[i]=new float[actualBufSize];
    }
  }

  for (int i=0; i<totalChans; i++) {
    logV("channel %d info: (index %d)",chanInfo[i].channel,i);
    logV("- name: %s",chanInfo[i].name);
    logV("- sample type: %s",getFormatName(chanInfo[i].type));
    logV("- group: %d",chanInfo[i].channelGroup);
    logV("- is input: %s",(chanInfo[i].isInput==ASIOTrue)?"yes":"no");
    logV("- is active: %s",(chanInfo[i].isActive==ASIOTrue)?"yes":"no");
  }

  result=ASIOCreateBuffers(bufInfo,totalChans,actualBufSize,&callbacks);
  if (result!=ASE_OK) {
    logE("could not create buffers! (%s)",getErrorStr(result));
    if (inBufs!=NULL) {
      for (int i=0; i<desc.inChans; i++) {
        if (inBufs[i]!=NULL) {
          delete[] inBufs[i];
          inBufs[i]=NULL;
        }
      }
      delete[] inBufs;
      inBufs=NULL;
    }
    if (outBufs!=NULL) {
      for (int i=0; i<desc.outChans; i++) {
        if (outBufs[i]!=NULL) {
          delete[] outBufs[i];
          outBufs[i]=NULL;
        }
      }
      delete[] outBufs;
      outBufs=NULL;
    }
    result=ASIOExit();
    if (result!=ASE_OK) {
      logE("could not exit either! (%s)",getErrorStr(result));
    }
    asioDrivers->removeCurrentDriver();
    return false;
  }

  desc.bufsize=actualBufSize;
  desc.fragments=2;

  response=desc;
  initialized=true;
  return true;
}

std::vector<String> TAAudioASIO::listAudioDevices() {
  std::vector<String> ret;

  if (!asioDrivers) asioDrivers=new AsioDrivers;

  if (!driverNamesInit) {
    for (int i=0; i<ASIO_DRIVER_MAX; i++) {
      // 64 just in case
      driverNames[i]=new char[64];
    }
    driverNamesInit=true;
  }
  driverCount=asioDrivers->getDriverNames(driverNames,ASIO_DRIVER_MAX);
  for (int i=0; i<driverCount; i++) {
    ret.push_back(driverNames[i]);
  }

  return ret;
}