#include "output_sk.h"

#include "source_base/tool_quit.h"

namespace ModuleIO
{

template <typename TK>
Output_Sk<TK>::Output_Sk(hamilt::Hamilt<TK>* p_hamilt, Parallel_Orbitals* ParaV, int nspin, int nks)
    : p_hamilt_(p_hamilt), ParaV_(ParaV), nspin_(nspin), nks_(nks)
{
}

template <>
double* Output_Sk<double>::get_Sk(int ik)
{
    if (ik < 0 || ik >= this->nks_)
    {
        ModuleBase::WARNING_QUIT("Output_Sk::get_sk", "ik out of range");
    }
    return dynamic_cast<hamilt::HamiltLCAO<double, double>*>(this->p_hamilt_)->getSk();
}

template <>
std::complex<double>* Output_Sk<std::complex<double>>::get_Sk(int ik)
{
    if (ik < 0 || ik >= this->nks_)
    {
        ModuleBase::WARNING_QUIT("Output_Sk::get_sk", "ik out of range");
    }
    if (this->nspin_ == 4)
    {
        dynamic_cast<hamilt::HamiltLCAO<std::complex<double>, std::complex<double>>*>(this->p_hamilt_)
            ->updateSk(ik, 1);
        return dynamic_cast<hamilt::HamiltLCAO<std::complex<double>, std::complex<double>>*>(this->p_hamilt_)->getSk();
    }
    else
    {
        dynamic_cast<hamilt::HamiltLCAO<std::complex<double>, double>*>(this->p_hamilt_)->updateSk(ik, 1);
        return dynamic_cast<hamilt::HamiltLCAO<std::complex<double>, double>*>(this->p_hamilt_)->getSk();
    }
    return nullptr;
}

template class Output_Sk<double>;
template class Output_Sk<std::complex<double>>;

} // namespace ModuleIO