TB_Parameter_Id.h

#ifndef QNANO_NEW_PARAMETER_ID_DEFINED_H
#define QNANO_NEW_PARAMETER_ID_DEFINED_H

#include "tightbinding/TB_Parameter_List.h"
#include "tightbinding/TB_Hopping_Parameter_List.h"

class TB_Parameter_Id{
public:
  std::string str;
  int element[2];
  
  std::string get_name()const{
    std::stringstream ss;
    ss<<str<<" "<<element[0]<<" "<<element[1];
    return ss.str();
  }
  bool operator==(const TB_Parameter_Id &other)const{
    return str==other.str && element[0]==other.element[0] && element[1]==other.element[1];
  }
  TB_Parameter_Id(const std::string &str_=0, int e0_=0, int e1_=0){
    str=str_;
    element[0]=e0_;
    element[1]=e1_;
  }
  TB_Parameter_Id(const std::vector<std::string> &strvec){
    if(strvec.size()<3){
      std::cerr<<"Initializing TB_Parameter_Id from string vector ";
      for(size_t i=0; i<strvec.size(); i++){
        std::cerr<<"'"<<strvec[i]<<"', ";
      }
      std::cerr<<std::endl;
      exit(1);
    }
    str=strvec[0];
    element[0]=Reader::readDouble(strvec[1], "TB_Parameter_Id, element[0]");
    element[1]=Reader::readDouble(strvec[2], "TB_Parameter_Id, element[1]");
  }
};


class TB_Parameter_Id_List: public List_Class_NoFind<TB_Parameter_Id>{
public:

  void print(std::ostream &os=std::cout)const{
    for(size_t i=0; i<list.size(); i++){
      os<<list[i].get_name()<<std::endl;
    }
  }
  int find(const TB_Parameter_Id & other) const{
    for(size_t i=0;i<list.size();i++){
      if(list[i]==other)return i;
    }
    return -1;
  }
  int find_check(const TB_Parameter_Id & element) const{
    int i=find(element);
    if(i<0){
      std::cerr<<"TB_Parameter_Id_List:: '"<<element.get_name()<<"' not found!"<<std::endl;
      std::cerr<<"Registered parameters:"<<std::endl;
      print(std::cerr);
      exit(1);
    }
    return i;
  }
  
  bool add(const std::string &str, int e0, int e1){
    TB_Parameter_Id id(str,e0,e1);
    if(find(id)>=0){
      std::cout<<"WARNING: TB_Parameter_Id_List: Parameter '"<<id.get_name()<<"' already exists!"<<std::endl;
      return false;
    }
    list.push_back(id);
    return true;
  }

};

#endif