#include "d/snd/d_snd_source_group.h"

#include "common.h"
#include "d/snd/d_snd_source.h"
#include "d/snd/d_snd_source_enums.h"
#include "d/snd/d_snd_source_mgr.h"
#include "d/snd/d_snd_state_mgr.h"
#include "nw4r/ut/ut_list.h"

#include <cmath>

void dSndSourceGroup_c::setParam(s32 sourceType, const char *name) {
    resetSoundSourceParam();
    bool assignedParam = false;

    switch (mSourceCategory) {
        case SND_SOURCE_CATEGORY_PLAYER: {
            assignedParam = true;
            mParam.field_0x10 = INFINITY;
            break;
        }
        case SND_SOURCE_CATEGORY_EQUIPMENT:
            if (sourceType == SND_SOURCE_ARROW) {
                assignedParam = true;
                mParam.field_0x00 = 1500.0f;
                mParam.field_0x10 = INFINITY;
            }
            break;
    }

    if (!assignedParam && (sourceType == SND_SOURCE_KENSEI || sourceType == SND_SOURCE_58)) {
        assignedParam = true;
        mParam.field_0x10 = INFINITY;
    }

    if (!assignedParam) {
        assignedParam = setParamFromName(name);
    }

    if (!assignedParam) {
        switch (sourceType) {
            case SND_SOURCE_NPC_51: {
                mParam.field_0x00 = 300.0f;
                mParam.field_0x04 = 800.0;
                mParam.field_0x10 = 2200.0;
                break;
            }
            case SND_SOURCE_NPC_HEAD: {
                if (dSndStateMgr_c::isInStage("F401")) {
                    mParam.field_0x00 = 500.0f;
                    mParam.field_0x04 = 4000.0;
                } else if (dSndStateMgr_c::isInStage("F402")) {
                    mParam.field_0x00 = 400.0f;
                    mParam.field_0x04 = 3500.0;
                } else {
                    mParam.field_0x00 = 300.0f;
                    mParam.field_0x04 = 800.0;
                    mParam.field_0x10 = 2200.0;
                }

                mParam.field_0x10 = INFINITY;
            }
        }
    }
}

void dSndSourceGroup_c::resetSoundSourceParam() {
    mParam.reset();
}

dSndSourceGroup_c::dSndSourceGroup_c()
    : mSubtype(0),
      mIsActive(false),
      field_0x1D(false),
      mName(""),
      mpOrigName(nullptr),
      mpCachedClosestSourceToListener(nullptr),
      mpCachedClosestSourceToPlayer(nullptr),
      mCalculatedClosestToListener(false),
      mCalculatedClosestToPlayer(false) {
    // TODO offsetof
    nw4r::ut::List_Init(&mSubSourceList, 0x138);
    resetSoundSourceParam();
}

dSndSourceGroup_c::dSndSourceGroup_c(s32 sourceType, const char *name, const char *origName, s32 subtype)
    : mSourceCategory(-1),
      mSubtype(subtype),
      mIsActive(false),
      field_0x1D(false),
      mName(name),
      mpOrigName(origName),
      mpCachedClosestSourceToListener(nullptr),
      mpCachedClosestSourceToPlayer(nullptr),
      mCalculatedClosestToListener(false),
      mCalculatedClosestToPlayer(false) {
    nw4r::ut::List_Init(&mSubSourceList, 0x138);
    setParam(sourceType, name);
}

void dSndSourceGroup_c::set(s32 sourceType, const char *name, const char *origName, s32 subtype) {
    mSourceCategory = dSndSourceMgr_c::getSourceCategoryForSourceType(sourceType, name);
    setParam(sourceType, name);
    mSourceType = sourceType;
    mName = name;
    mSubtype = subtype;
    mpCachedClosestSourceToPlayer = nullptr;
    mpCachedClosestSourceToListener = nullptr;
    mCalculatedClosestToPlayer = false;
    mCalculatedClosestToListener = false;
    mpOrigName = origName;
}

void dSndSourceGroup_c::setTemp(s32 sourceType, const char *name, s32 subtype) {
    mSourceType = sourceType;
    mName = name;
    mSubtype = subtype;
}

dSndSourceGroup_c::~dSndSourceGroup_c() {
    clearList();
}

void dSndSourceGroup_c::clear() {
    clearList();
    mIsActive = false;
    field_0x1D = false;
    mName = "";
}

void dSndSourceGroup_c::clearTemp() {
    mIsActive = false;
    field_0x1D = false;
    mName = "";
}

void dSndSourceGroup_c::calc() {
    // Reset freshness flags, computed on-demand
    mCalculatedClosestToPlayer = false;
    mCalculatedClosestToListener = false;
}

s32 dSndSourceGroup_c::getNumSources() const {
    return nw4r::ut::List_GetSize(&mSubSourceList);
}

void dSndSourceGroup_c::registerSource(dSoundSource_c *source) {
    if (source != nullptr) {
        appendSubSource(source);
    }
}

void dSndSourceGroup_c::unregisterSource(dSoundSource_c *source) {
    if (source != nullptr) {
        removeSubSource(source);
    }
}

void dSndSourceGroup_c::clearList() {
    dSoundSource_c *source = getSubSourceFirst();
    while (source != nullptr) {
        removeSubSource(source);
        source = getSubSourceFirst();
    }
}

dSoundSource_c *dSndSourceGroup_c::getSourceClosestToListener() {
    if (mCalculatedClosestToListener) {
        return mpCachedClosestSourceToListener;
    }

    mpCachedClosestSourceToListener = nullptr;
    f32 closest = INFINITY;
    for (dSoundSource_c *source = getSubSourceFirst();
         source != nullptr; source = getSubSourceNext(source)) {
        if (source->getSourceType() != SND_SOURCE_PLAYER_HEAD && source->getSourceType() != SND_SOURCE_NPC_HEAD) {
            f32 dist = source->getDistanceToListener();
            if (dist < closest) {
                closest = dist;
                mpCachedClosestSourceToListener = source;
            }
        }
    }
    mCalculatedClosestToListener = true;
    return mpCachedClosestSourceToListener;
}

dSoundSource_c *dSndSourceGroup_c::getSourceClosestToPlayer() {
    if (mCalculatedClosestToPlayer) {
        return mpCachedClosestSourceToPlayer;
    }

    if ((s32)nw4r::ut::List_GetSize(&mSubSourceList) <= 1) {
        return getSubSourceFirst();
    }

    mpCachedClosestSourceToPlayer = nullptr;
    f32 closest = INFINITY;
    for (dSoundSource_c *source = getSubSourceFirst();
         source != nullptr; source = getSubSourceNext(source)) {
        if (source->getSourceType() != 1 && source->getSourceType() != 48) {
            f32 dist = source->getDistanceToPlayer();
            if (dist < closest) {
                closest = dist;
                mpCachedClosestSourceToPlayer = source;
            }
        }
    }
    mCalculatedClosestToPlayer = true;
    return mpCachedClosestSourceToPlayer;
}

bool dSndSourceGroup_c::setParamFromName(const char *) {
    // TODO
    return false;
}