restart_exx_csr.h

Go to the documentation of this file.

#pragma once
#include "source_cell/unitcell.h"
#include "source_lcao/module_ri/RI_Util.h"
#include "source_lcao/module_ri/abfs-vector3_order.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);
} // namespace ModuleIO
 
#include "restart_exx_csr.hpp"