atomic_basis.h

/*
 * @file basis.h
 */
#pragma once
#include <map>
#include <utility>
#include <vector>
 
namespace LIBRPA {
 
class AtomicBasis
{
private:
    std::vector<std::size_t> nbs_;
    std::vector<std::size_t> part_range;
    void initialize();
public:
    std::size_t n_atoms;
    std::size_t nb_total;
    AtomicBasis(): nbs_(), part_range(), n_atoms(0), nb_total(0) {};
    AtomicBasis(const std::vector<std::size_t>& nbs);
    AtomicBasis(const std::vector<int>& atoms,
                const std::map<int, std::size_t>& map_atom_nb);
    AtomicBasis(const std::map<std::size_t, std::size_t>& iatom_nbs);
    void set(const std::vector<std::size_t>& nbs);
    void set(const std::map<std::size_t, std::size_t>& iatom_nbs);
    std::size_t get_global_index(const int& i_atom, const std::size_t& i_loc_b) const;
    int get_i_atom(const std::size_t& i_glo_b) const;
    void get_local_index(const std::size_t& i_glo_b, int& i_atom, int& i_loc_b) const;
    int get_local_index(const std::size_t& i_glo_b, const int& i_atom) const;
    std::pair<int, int> get_local_index(const std::size_t& i_glo_b) const;
    std::size_t get_atom_nb(const int& i_atom) const { return nbs_[i_atom]; }
    const std::vector<std::size_t>& get_part_range() const { return part_range; }
};
 
extern AtomicBasis atomic_basis_wfc;
extern AtomicBasis atomic_basis_abf;
 
} // namespace LIBRPA