Computer Shopper 242/Issue 242 - April 2008 - DPCS0408DVD.ISO/Software Money Savers/VirtualDub/Source/VirtualDub-1.7.7-src.7z/src/system/source/cmdline.cpp

C/C++ Source or Header | 2007-08-11 | 4.2 KB | 179 lines

//    VirtualDub - Video processing and capture application
//    System library component
//    Copyright (C) 1998-2005 Avery Lee, All Rights Reserved.
//
//    Beginning with 1.6.0, the VirtualDub system library is licensed
//    differently than the remainder of VirtualDub.  This particular file is
//    thus licensed as follows (the "zlib" license):
//
//    This software is provided 'as-is', without any express or implied
//    warranty.  In no event will the authors be held liable for any
//    damages arising from the use of this software.
//
//    Permission is granted to anyone to use this software for any purpose,
//    including commercial applications, and to alter it and redistribute it
//    freely, subject to the following restrictions:
//
//    1.    The origin of this software must not be misrepresented; you must
//        not claim that you wrote the original software. If you use this
//        software in a product, an acknowledgment in the product
//        documentation would be appreciated but is not required.
//    2.    Altered source versions must be plainly marked as such, and must
//        not be misrepresented as being the original software.
//    3.    This notice may not be removed or altered from any source
//        distribution.

#include "stdafx.h"
#include <vd2/system/cmdline.h>

VDCommandLine::VDCommandLine() {
}

VDCommandLine::VDCommandLine(const wchar_t *s) {
    Init(s);
}

VDCommandLine::~VDCommandLine() {
}

void VDCommandLine::Init(const wchar_t *s) {
    for(;;) {
        while(iswspace(*s))
            ++s;

        if (!*s)
            break;

        Token te = { (int)mLine.size(), *s == L'/', *s == L'"' };

        if (te.mbIsSwitch) {
            mLine.push_back(L'/');
            ++s;
        }

        mTokens.push_back(te);

        // special case for /?
        if (te.mbIsSwitch && *s == L'?') {
            mLine.push_back(L'?');
            ++s;
        }

        while(*s && *s != L' ' && *s != L'/') {
            if (te.mbIsSwitch) {
                if (!isalnum((unsigned char)*s))
                    break;

                mLine.push_back(*s++);
            } else if (*s == L'"') {
                ++s;
                while(*s && *s != L'"')
                    mLine.push_back(*s++);

                if (*s) {
                    ++s;

                    if (*s == ',') {
                        ++s;
                        break;
                    }
                }
            } else
                mLine.push_back(*s++);
        }

        mLine.push_back(0);
    }
}

uint32 VDCommandLine::GetCount() const {
    return mTokens.size();
}

const wchar_t *VDCommandLine::operator[](int index) const {
    return (uint32)index < mTokens.size() ? mLine.data() + mTokens[index].mTokenIndex : NULL;
}

bool VDCommandLine::GetNextArgument(VDCommandLineIterator& it, const wchar_t *& token, bool& isSwitch) const {
    int count = (int)mTokens.size();

    if (it.mIndex >= count)
        return false;

    token = mLine.data() + mTokens[it.mIndex].mTokenIndex;
    isSwitch = mTokens[it.mIndex].mbIsSwitch;

    ++it.mIndex;
    return true;
}

bool VDCommandLine::GetNextNonSwitchArgument(VDCommandLineIterator& it, const wchar_t *& token) const {
    int count = (int)mTokens.size();

    if (it.mIndex >= count)
        return false;

    if (mTokens[it.mIndex].mbIsSwitch)
        return false;

    token = mLine.data() + mTokens[it.mIndex++].mTokenIndex;
    return true;
}

bool VDCommandLine::GetNextSwitchArgument(VDCommandLineIterator& it, const wchar_t *& token) const {
    int count = (int)mTokens.size();

    if (it.mIndex >= count)
        return false;

    if (!mTokens[it.mIndex].mbIsSwitch)
        return false;

    token = mLine.data() + mTokens[it.mIndex++].mTokenIndex;
    return true;
}

bool VDCommandLine::FindAndRemoveSwitch(const wchar_t *name) {
    int count = (int)mTokens.size();

    for(int i=1; i<count; ++i) {
        if (mTokens[i].mbIsSwitch && !_wcsicmp(name, mLine.data() + mTokens[i].mTokenIndex + 1)) {
            mTokens.erase(mTokens.begin() + i);
            return true;
        }
    }

    return false;
}

bool VDCommandLine::FindAndRemoveSwitch(const wchar_t *name, const wchar_t *& token) {
    int count = (int)mTokens.size();
    size_t namelen = wcslen(name);

    for(int i=1; i<count; ++i) {
        if (!mTokens[i].mbIsSwitch)
            continue;

        const wchar_t *s = mLine.data() + mTokens[i].mTokenIndex + 1;

        if (!_wcsnicmp(name, s, namelen)) {
            token = s+namelen;

            switch(*token) {
                case L':':
                    ++token;
                    break;
                case 0:
                    break;
                default:
                    continue;
            }

            mTokens.erase(mTokens.begin() + i);
            return true;
        }
    }

    return false;
}