#include "CParticleElectricDataFactory.hpp"
#include "CSwooshDescription.hpp"
#include "CElectricDescription.hpp"
#include "CGenDescription.hpp"
#include "Graphics/CModel.hpp"
#include "CToken.hpp"
#include "CSimplePool.hpp"
#include "CRandom16.hpp"

namespace urde {
static logvisor::Module Log("urde::CParticleElectricDataFactory");

using CPF = CParticleDataFactory;

CElectricDescription* CParticleElectricDataFactory::GetGeneratorDesc(CInputStream& in, CSimplePool* resPool) {
  return CreateElectricDescription(in, resPool);
}

CElectricDescription* CParticleElectricDataFactory::CreateElectricDescription(CInputStream& in, CSimplePool* resPool) {
  FourCC cid = CPF::GetClassID(in);
  if (cid == FOURCC('ELSM')) {
    CElectricDescription* desc = new CElectricDescription;
    CreateELSM(desc, in, resPool);
    LoadELSMTokens(desc);
    return desc;
  }

  return nullptr;
}

bool CParticleElectricDataFactory::CreateELSM(CElectricDescription* desc, CInputStream& in, CSimplePool* resPool) {
  CRandom16 rand;
  CGlobalRandom gr{rand};

  FourCC clsId = CPF::GetClassID(in);
  while (clsId != SBIG('_END')) {
    switch (clsId.toUint32()) {
    case SBIG('LIFE'):
      desc->x0_LIFE = CPF::GetIntElement(in);
      break;
    case SBIG('SLIF'):
      desc->x4_SLIF = CPF::GetIntElement(in);
      break;
    case SBIG('GRAT'):
      desc->x8_GRAT = CPF::GetRealElement(in);
      break;
    case SBIG('SCNT'):
      desc->xc_SCNT = CPF::GetIntElement(in);
      break;
    case SBIG('SSEG'):
      desc->x10_SSEG = CPF::GetIntElement(in);
      break;
    case SBIG('COLR'):
      desc->x14_COLR = CPF::GetColorElement(in);
      break;
    case SBIG('IEMT'):
      desc->x18_IEMT = CPF::GetEmitterElement(in);
      break;
    case SBIG('FEMT'):
      desc->x1c_FEMT = CPF::GetEmitterElement(in);
      break;
    case SBIG('AMPL'):
      desc->x20_AMPL = CPF::GetRealElement(in);
      break;
    case SBIG('AMPD'):
      desc->x24_AMPD = CPF::GetRealElement(in);
      break;
    case SBIG('LWD1'):
      desc->x28_LWD1 = CPF::GetRealElement(in);
      break;
    case SBIG('LWD2'):
      desc->x2c_LWD2 = CPF::GetRealElement(in);
      break;
    case SBIG('LWD3'):
      desc->x30_LWD3 = CPF::GetRealElement(in);
      break;
    case SBIG('LCL1'):
      desc->x34_LCL1 = CPF::GetColorElement(in);
      break;
    case SBIG('LCL2'):
      desc->x38_LCL2 = CPF::GetColorElement(in);
      break;
    case SBIG('LCL3'):
      desc->x3c_LCL3 = CPF::GetColorElement(in);
      break;
    case SBIG('SSWH'):
      desc->x40_SSWH = CPF::GetSwooshGeneratorDesc(in, resPool);
      break;
    case SBIG('GPSM'): {
      std::vector<CAssetId> tracker;
      tracker.reserve(8);
      desc->x50_GPSM = CPF::GetChildGeneratorDesc(in, resPool, tracker);
      break;
    }
    case SBIG('EPSM'): {
      std::vector<CAssetId> tracker;
      tracker.reserve(8);
      desc->x60_EPSM = CPF::GetChildGeneratorDesc(in, resPool, tracker);
      break;
    }
    case SBIG('ZERY'):
      desc->x70_ZERY = CPF::GetBool(in);
      break;
    default: {
      Log.report(logvisor::Fatal, fmt("Unknown ELSM class {} @{}"), clsId, in.position());
      return false;
    }
    }
    clsId = CPF::GetClassID(in);
  }
  return true;
}

void CParticleElectricDataFactory::LoadELSMTokens(CElectricDescription* desc) {
  if (desc->x40_SSWH.m_found)
    desc->x40_SSWH.m_swoosh = desc->x40_SSWH.m_token.GetObj();
  if (desc->x50_GPSM.m_found)
    desc->x50_GPSM.m_gen = desc->x50_GPSM.m_token.GetObj();
  if (desc->x60_EPSM.m_found)
    desc->x60_EPSM.m_gen = desc->x60_EPSM.m_token.GetObj();
}

CFactoryFnReturn FParticleElectricDataFactory(const SObjectTag& tag, CInputStream& in, const CVParamTransfer& vparms,
                                              CObjectReference*) {
  CSimplePool* sp = vparms.GetOwnedObj<CSimplePool*>();
  return TToken<CElectricDescription>::GetIObjObjectFor(
      std::unique_ptr<CElectricDescription>(CParticleElectricDataFactory::GetGeneratorDesc(in, sp)));
}
} // namespace urde