cell_index.h

Go to the documentation of this file.

#ifndef CELL_INDEX_H
#define CELL_INDEX_H
#include <map>
#include <string>
#include <vector>
 
class CellIndex
{
  public:
    CellIndex() = default;
    CellIndex(const std::vector<std::string>& atomLabels_in,
              const std::vector<int>& atomCounts_in,
              const std::vector<std::vector<int>>& lnchiCounts_in,
              const int& nspin);
 
  public:
    int get_nat();
    int get_nat(int it);
    int get_ntype();
    int get_nw();
    int get_nw(int iat);
    int get_iwt(int iat, int orbital_index);
    int get_maxL(int iat);
    int get_nchi(int iat, int L);
    std::string get_atom_label(int iat, bool order = false);
    void write_orb_info(std::string out_dir);
 
  private:
    std::vector<int> atomCounts = {};
    std::vector<int> orbitalCounts = {};
    std::vector<std::vector<int>> lnchiCounts = {};
    std::vector<std::string> atomLabels = {};
    int npol_ = 1;
    void cal_orbitalCounts();
    bool check_nspin(int nspin);
    void check_atomCounts();
    int iat2it(int iat);
    int iat2ia(int iat);
    int iw2l(int iat, int iw);
    int iw2z(int iat, int iw);
    int iw2m(int iat, int iw);
};
 
#endif // CELL_INDEX_H