restart_exx_csr.h

Go to the documentation of this file.

#pragma once
#include "source_base/abfs-vector3_order.h"
#include "source_cell/unitcell.h"
#include "source_lcao/module_ri/serialization_cereal.h"
#include <RI/global/Tensor.h>
#include <map>
 
namespace ModuleIO
{
    using TC = std::array<int, 3>;
    using TAC = std::pair<int, TC>;
 
    template<typename Tdata>
    void read_Hexxs_csr(const std::string& file_name, const UnitCell& ucell,
        const int nspin, const int nbasis,
        std::vector<std::map<int, std::map<TAC, RI::Tensor<Tdata>>>>& Hexxs);
 
    template<typename Tdata>
    void read_Hexxs_cereal(const std::string& file_name,
        std::vector<std::map<int, std::map<TAC, RI::Tensor<Tdata>>>>& Hexxs);
 
    template<typename Tdata>
    void write_Hexxs_csr(const std::string& file_name, const UnitCell& ucell,
        const std::map<int, std::map<TAC, RI::Tensor<Tdata>>>& Hexxs);
 
    template<typename Tdata>
    std::map<Abfs::Vector3_Order<int>, std::map<size_t, std::map<size_t, Tdata>>>
        calculate_RI_Tensor_sparse(const double& sparse_threshold,
            const std::vector<std::map<int, std::map<TAC, RI::Tensor<Tdata>>>>& Hexxs,
            const UnitCell& ucell);
}
 
#include "source_io/restart_exx_csr.hpp"