---

CmdLineParser.h

Go to the documentation of this file.

#ifndef __CMD_LINE_PARSER_H__
#define  __CMD_LINE_PARSER_H__

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream.h>

#ifdef WIN32
        #include <UnixFuncPrototypes.h>
#else
        #include <getopt.h>
#endif

#define MAX_OPT_ARGS 500

class CmdLineParser;

typedef bool (CmdLineParser::*CLP_ACTION)();
typedef const char* LPCSTR;

LPCSTR asciitoascii(LPCSTR sz);

class CmdLineArgT
{
        char m_cShortName;
        LPCSTR m_szLongName;
        LPCSTR m_szDesc;
        CLP_ACTION m_pfAction;
        bool m_bSelected;

public:
        CmdLineArgT(char cShortName, LPCSTR szLongName, LPCSTR szDesc, CLP_ACTION pfAction = NULL)
        {
                m_bSelected = false;
                m_cShortName = cShortName;
                m_szLongName = szLongName;
                m_szDesc = szDesc;
                m_pfAction = pfAction;
        }

        virtual ~CmdLineArgT() {}
        virtual void Select(LPCSTR szOptArg) { m_bSelected = true; }
        virtual bool HasOptArg() const { return false; }
        virtual void Print(ostream& os = cout) const { }

        bool HasShortName() const { return m_cShortName != 0; }
        bool HasLongName() const { return m_szLongName != 0 && *m_szLongName != '\0'; }
        bool IsSelected() const { return m_bSelected; }
        bool HasAction() const { return m_pfAction != 0; }
        CLP_ACTION GetAction() const { return m_pfAction; }
        char GetShortName() const { return m_cShortName; }
        LPCSTR GetLongName() const { return m_szLongName; }
        LPCSTR GetDesc() const { return m_szDesc; }
};

template<class T> class CmdLineArg : public CmdLineArgT
{
        T* m_pOptArg;
        T (*m_pfAtoT)(LPCSTR);
        
public:
        CmdLineArg(char cShortName, LPCSTR szLongName, LPCSTR szDesc, 
                T* pOptArg, const T& defVal, CLP_ACTION pfAction, T (*pfAtoT)(LPCSTR)) 
                : CmdLineArgT(cShortName, szLongName, szDesc, pfAction)
        { 
                m_pOptArg = pOptArg; 
                *m_pOptArg = defVal; 
                m_pfAtoT = pfAtoT;
        }

        void Select(LPCSTR szOptArg)
        {
                CmdLineArgT::Select(szOptArg);
                *m_pOptArg = (*m_pfAtoT)(szOptArg); 
        }

        void Print(ostream& os = cout) const 
        { 
                os << GetLongName() << " = " << *m_pOptArg << endl; 
        }

        bool HasOptArg() const { return true; }
};

struct OPTION : public option
{
        void operator=(const CmdLineArgT& arg)
        {
                name = arg.GetLongName();
                has_arg = arg.HasOptArg();
                flag = 0;
                val = arg.GetShortName();
        }

        void SetToZero()
        {
                name = 0;
                has_arg = 0;
                flag = 0;
                val = 0;
        }
};

class CmdLineParser
{
protected:
        CmdLineArgT* m_pOptArgs[MAX_OPT_ARGS];
        char** m_pNonOptArgs;
        int m_nOptArgs;
        int m_nNonOptArgs;
        const char* m_szProgramName;
        const char* m_szOptArgHelp;
        
public: 
        CmdLineParser()
        {
                m_pNonOptArgs = NULL;
                m_nOptArgs = 0;
                m_nNonOptArgs = 0;
                m_szProgramName = "prgname";
                m_szOptArgHelp = NULL;
        }

        ~CmdLineParser()
        {
                while (m_nOptArgs) 
                        delete m_pOptArgs[--m_nOptArgs];
        }

        int GetArgCount() const { return m_nNonOptArgs; }
        const char* GetArg(int i) const { return m_pNonOptArgs[i]; }
        
        void Add(char cShortName, LPCSTR szLongName, LPCSTR szDesc, CLP_ACTION pfAction = NULL)
        {
                m_pOptArgs[m_nOptArgs++] = new CmdLineArgT(cShortName, szLongName, 
                        szDesc, pfAction);
        }

        void Add(char cShortName, LPCSTR szLongName, LPCSTR szDesc,
                int* pOptArg, int defVal, CLP_ACTION pfAction = NULL)
        { 
                m_pOptArgs[m_nOptArgs++] = new CmdLineArg<int>(cShortName, szLongName, 
                        szDesc, pOptArg, defVal, pfAction, atoi);
        }

        void Add(char cShortName, LPCSTR szLongName, LPCSTR szDesc,
                double* pOptArg, double defVal, CLP_ACTION pfAction = NULL)
        { 
                m_pOptArgs[m_nOptArgs++] = new CmdLineArg<double>(cShortName, szLongName, 
                        szDesc, pOptArg, defVal, pfAction, atof);
        }

        void Add(char cShortName, LPCSTR szLongName, LPCSTR szDesc,
                LPCSTR* pOptArg, LPCSTR defVal, CLP_ACTION pfAction = NULL)
        { 
                m_pOptArgs[m_nOptArgs++] = new CmdLineArg<LPCSTR>(cShortName, szLongName, szDesc, 
                        pOptArg, defVal, pfAction, asciitoascii);
        }

        void SetOptArgHelp(const char* szOptArgHelp) 
        { 
                m_szOptArgHelp = szOptArgHelp; 
        }

        void SetPrgName(const char* szProgramName);

        void Print(ostream& os = cout);

        void ShowUsageMsg();
        void ShowErrorMsg(LPCSTR szPgmName);
        bool Parse(int argc, char** argv);
        void ExecActions();
        int GetArgIndex(char c);
        int GetArgIndex(const char* szArgName);
        bool IsSelected(char c);
        bool IsSelected(const char* szArgName);

        virtual void InitOptions() = 0;
        virtual bool ExecAction(CLP_ACTION pAction) = 0;
        virtual void ShowUsageExamples() {}
};

#endif //__CMD_LINE_PARSER_H__

---