#include "Runtime/Particle/CDecalDataFactory.hpp"

#include "Runtime/CRandom16.hpp"
#include "Runtime/CSimplePool.hpp"
#include "Runtime/Graphics/CModel.hpp"
#include "Runtime/Particle/CGenDescription.hpp"
#include "Runtime/Particle/CElectricDescription.hpp"
#include "Runtime/Particle/CParticleDataFactory.hpp"
#include "Runtime/Particle/CSwooshDescription.hpp"

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

using CPF = CParticleDataFactory;
std::unique_ptr<CDecalDescription> CDecalDataFactory::GetGeneratorDesc(CInputStream& in, CSimplePool* resPool) {
  return CreateGeneratorDescription(in, resPool);
}

std::unique_ptr<CDecalDescription> CDecalDataFactory::CreateGeneratorDescription(CInputStream& in,
                                                                                 CSimplePool* resPool) {
  const FourCC clsId = CPF::GetClassID(in);

  if (clsId == FOURCC('DPSM')) {
    auto desc = std::make_unique<CDecalDescription>();
    if (CreateDPSM(desc.get(), in, resPool)) {
      return desc;
    }
  }

  return nullptr;
}

bool CDecalDataFactory::CreateDPSM(CDecalDescription* desc, CInputStream& in, CSimplePool* resPool) {
  CRandom16 rand;
  CGlobalRandom gr{rand};
  FourCC clsId = CPF::GetClassID(in);

  while (clsId != SBIG('_END')) {
    bool loadFirstDesc = false;
    switch (clsId.toUint32()) {
    case SBIG('1SZE'):
    case SBIG('1LFT'):
    case SBIG('1ROT'):
    case SBIG('1OFF'):
    case SBIG('1CLR'):
    case SBIG('1TEX'):
    case SBIG('1ADD'):
      loadFirstDesc = true;
      [[fallthrough]];
    case SBIG('2LFT'):
    case SBIG('2SZE'):
    case SBIG('2ROT'):
    case SBIG('2OFF'):
    case SBIG('2CLR'):
    case SBIG('2TEX'):
    case SBIG('2ADD'):
      if (loadFirstDesc)
        GetQuadDecalInfo(in, resPool, clsId, desc->x0_Quads[0]);
      else
        GetQuadDecalInfo(in, resPool, clsId, desc->x0_Quads[1]);
      break;

    case SBIG('DMDL'):
      desc->x38_DMDL = CPF::GetModel(in, resPool);
      break;
    case SBIG('DLFT'):
      desc->x48_DLFT = CPF::GetIntElement(in);
      break;
    case SBIG('DMOP'):
      desc->x4c_DMOP = CPF::GetVectorElement(in);
      break;
    case SBIG('DMRT'):
      desc->x50_DMRT = CPF::GetVectorElement(in);
      break;
    case SBIG('DMSC'):
      desc->x54_DMSC = CPF::GetVectorElement(in);
      break;
    case SBIG('DMCL'):
      desc->x58_DMCL = CPF::GetColorElement(in);
      break;
    case SBIG('DMAB'):
      desc->x5c_24_DMAB = CPF::GetBool(in);
      break;
    case SBIG('DMOO'):
      desc->x5c_25_DMOO = CPF::GetBool(in);
      break;
    default: {
      Log.report(logvisor::Fatal, FMT_STRING("Unknown DPSC class {} @{}"), clsId, in.position());
      return false;
    }
    }

    clsId = CPF::GetClassID(in);
  }
  return true;
}

void CDecalDataFactory::GetQuadDecalInfo(CInputStream& in, CSimplePool* resPool, FourCC clsId, SQuadDescr& quad) {
  switch (clsId.toUint32()) {
  case SBIG('1LFT'):
  case SBIG('2LFT'):
    quad.x0_LFT = CPF::GetIntElement(in);
    break;
  case SBIG('1SZE'):
  case SBIG('2SZE'):
    quad.x4_SZE = CPF::GetRealElement(in);
    break;
  case SBIG('1ROT'):
  case SBIG('2ROT'):
    quad.x8_ROT = CPF::GetRealElement(in);
    break;
  case SBIG('1OFF'):
  case SBIG('2OFF'):
    quad.xc_OFF = CPF::GetVectorElement(in);
    break;
  case SBIG('1CLR'):
  case SBIG('2CLR'):
    quad.x10_CLR = CPF::GetColorElement(in);
    break;
  case SBIG('1TEX'):
  case SBIG('2TEX'):
    quad.x14_TEX = CPF::GetTextureElement(in, resPool);
    break;
  case SBIG('1ADD'):
  case SBIG('2ADD'):
    quad.x18_ADD = CPF::GetBool(in);
    break;
  }
}

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