ParamList.h

Go to the documentation of this file.

/*-------------------------------------------------------------------
Copyright 2011 Ravishankar Sundararaman

This file is part of JDFTx.

JDFTx is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

JDFTx is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with JDFTx.  If not, see <http://www.gnu.org/licenses/>.
-------------------------------------------------------------------*/

#ifndef JDFTX_COMMAND_PARAMLIST_H
#define JDFTX_COMMAND_PARAMLIST_H

#include <core/Util.h>

class ParamList
{
        istringstream iss;
public:
        ParamList(string params) : iss(params) {}

        void rewind() { iss.seekg(0, std::ios_base::beg); }
        
        template<typename T>
        void get(T& t, T tDefault, string paramName, bool required=false)
        {       iss.clear(); //clear previous errors
                iss >> t;
                if(iss.bad()) throw string("I/O error while reading parameter <"+paramName+">.");
                if(iss.eof())
                {       t = tDefault;
                        if(required) throw string("Parameter <"+paramName+"> must be specified.");
                        else return;
                }
                if(iss.fail()) { t = tDefault; throw string("Conversion of parameter <"+paramName+"> failed."); }
        }

        template<typename T>
        void get(T& t, T tDefault, const EnumStringMap<T>& tMap, string paramName, bool required=false)
        {       iss.clear(); //clear previous errors
                string key;
                iss >> key;
                if(iss.bad()) throw string("I/O error while reading parameter <"+paramName+">.");
                if(iss.eof())
                {       t = tDefault;
                        if(required) throw string("Parameter <"+paramName+"> must be specified.");
                        else return;
                }
                if(!tMap.getEnum(key.c_str(), t))
                {       t = tDefault;
                        throw string("Parameter <"+paramName+"> must be one of "+tMap.optionList());
                }
        }
        
        string getRemainder()
        {       if(iss.eof()) return string();
                size_t curPos = iss.tellg();
                iss.seekg(0, std::ios_base::end);
                size_t endPos = iss.tellg();
                if(endPos > curPos)
                {       string buf(endPos-curPos, 0);
                        iss.seekg(curPos);
                        iss.read(&buf.at(0), buf.length());
                        trim(buf);
                        return buf;
                }
                else return string();
        }
};

#endif //JDFTX_COMMAND_PARAMLIST_H