Selected_Manybody_States_with_Maps.h

#ifndef QNANO_NEW_SELECTED_MANYBODY_STATES_WITH_MAPS_DEFINED_H
#define QNANO_NEW_SELECTED_MANYBODY_STATES_WITH_MAPS_DEFINED_H

#include <map>
#include "manybody/Selected_Manybody_States.h"


class Selected_Manybody_States_with_Maps: public Selected_Manybody_States{
public:
  typedef std::map<Fermion_Double_Index, int>::iterator map_iter;
  typedef std::map<Fermion_Double_Index, int>::const_iterator map_const_iter;
  typedef std::map<int, Fermion_Double_Index>::iterator rev_map_iter;
  typedef std::map<int, Fermion_Double_Index>::const_iterator rev_map_const_iter;

  std::map<Fermion_Double_Index, int> map;
  std::map<int, Fermion_Double_Index> rev_map;
  

  virtual size_t size()const{return map.size();}
  void clear(){map.clear();}
  map_const_iter begin()const{return map.begin();}
  map_const_iter end()const{return map.end();}
  bool is_contained(const Fermion_Double_Index &fi)const;


  int find(const Fermion_Double_Index &fi)const;
  int find_check(const Fermion_Double_Index &fi)const;
  Fermion_Double_Index find_reverse(int index)const;
  
  void check_consistency()const;

  void add_if_not_contained(const Fermion_Double_Index &fi);
  void add_from_file(const std::string &fname, int rank_e, int rank_h, int max_nr_e, int max_nr_h);

  void print(std::ostream &os=std::cout)const;
  void print(const std::string &fname)const{
    std::ofstream ofs(fname.c_str());
    print(ofs);
  }

  void add_all(int rank_e, int rank_h, int nr_e, int nr_h, int start_e=0, int start_h=0);
  void add_singles(int nr_e, int nr_h);

  void swap(Selected_Manybody_States_with_Maps &other);
};
#endif