root/OpenSceneGraph-TrainingMaterials/trunk/Sources/Exercises/utils/runapp/runapp.cpp @ 10042

#include <osg/ArgumentParser>
#include <osgDB/FileNameUtils>

#include <iostream>
#include <fstream>
#include <set>
#include <vector>
#include <stdlib.h>

// #define DEBUG

int main(int argc, char** argv)
{
    osg::ArgumentParser arguments(&argc, argv);
    
#ifdef DEBUG    
    std::ofstream fout("output.txt");
#endif
    
    if (arguments.argc()<=1)
    {
        std::cout<<"Nothing to run"<<std::endl;
        return 1;
    }
    
    typedef std::vector<std::string> ArgumentList;
    ArgumentList argumentList;
    for(int i=1; i<arguments.argc(); ++i)
    {
        std::string argument = arguments[i];
        std::cout<<"argument = ["<<argument<<"]"<<std::endl;
        std::string::size_type pos = 0;
        std::string::size_type prev_start = 0;
        for(pos = 0; pos<argument.size();)
        {
            if (argument[pos]=='"')
            {
                if (pos-prev_start>0)
                {
                    argumentList.push_back(argument.substr(prev_start, pos-prev_start));
                }
                
                prev_start = pos;
                ++pos;

                while (pos<argument.size() && argument[pos]!='"') { ++pos; }

                if (pos<argument.size())
                {
                    // skip over trailing "
                    ++pos;
                }

                if (pos-prev_start>0)
                {
                    argumentList.push_back(argument.substr(prev_start, pos-prev_start));
                }
                
                prev_start = pos;                

            }
            if (argument[pos]=='\'')
            {
                if (pos-prev_start>0)
                {
                    argumentList.push_back(argument.substr(prev_start, pos-prev_start));
                }
                
                prev_start = pos;
                ++pos;

                while (pos<argument.size() && argument[pos]!='\'') { ++pos; }

                if (pos<argument.size())
                {
                    // skip over trailing "
                    ++pos;
                }

                if (pos-prev_start>0)
                {
                    argumentList.push_back(argument.substr(prev_start, pos-prev_start));
                }
                
                prev_start = pos;                

            }
            else if (argument[pos]==' ')
            {
                if (pos-prev_start>0)
                {
                    argumentList.push_back(argument.substr(prev_start, pos-prev_start));
                }
                
                while (pos<argument.size() && argument[pos]==' ') { ++pos; }
                prev_start = pos;                
            }
            else
            {
                ++pos;
            }            
        }
        
        if (pos-prev_start>0)
        {
            argumentList.push_back(argument.substr(prev_start, pos-prev_start));
        }
    }
    
    
    for(ArgumentList::iterator itr = argumentList.begin();
        itr != argumentList.end();
        ++itr)
    {
        std::cout<<"result ["<<*itr<<"]"<<std::endl;
    }
    
    if (argumentList.empty()) 
    {
        std::cout<<"Nothing to run"<<std::endl;
        return 1;
    }

    for(int i=0; i<argumentList.size(); ++i)
    {
        std::string& argument = argumentList[i];
        std::string::size_type start_pos = argument.find("${");
        if (start_pos != std::string::npos)
        {        
            std::string::size_type end_pos = argument.find("}",start_pos);
            if (start_pos != std::string::npos)
            {
                std::string var = argument.substr(start_pos+2, end_pos-start_pos-2);
                const char* str = getenv(var.c_str());
#ifdef DEBUG    
                fout<<"for argument = ["<<argument<<"]"<<std::endl;
                fout<<"Found ["<<var<<"] = "<<str<<std::endl;
#endif
                argument.erase(start_pos, end_pos-start_pos+1);
                argument.insert(start_pos, str);

#ifdef DEBUG    
                fout<<"new argument = ["<<argument<<"]"<<std::endl;
#endif
            }
        }
    }



    if (argumentList[0]=="run")
    {
        std::string runstring;
        for(int i=1; i<argumentList.size(); ++i)
        {
            runstring += argumentList[i];
            runstring += ' ';
        }
        
        std::cout<<"Running : "<<runstring<<std::endl;
#ifdef DEBUG    
        fout<<"Running : "<<runstring<<std::endl;
#endif
        
        return system(runstring.c_str());
    }

    if (argumentList[0]=="file")
    {
        std::string runstring("dolphin");

        for(int i=1; i<argumentList.size(); ++i)
        {
            runstring += ' ';
            runstring += argumentList[i];
        }
        
        std::cout<<"File : "<<runstring<<std::endl;

#ifdef DEBUG    
        fout<<"File : "<<runstring<<std::endl;
#endif
        
        return system(runstring.c_str());
    }

    typedef std::set<std::string> StringSet;
    StringSet fileEndings;
    
    for(int i=0; i<argumentList.size(); ++i)
    {
        fileEndings.insert(osgDB::getFileExtension(argumentList[i]));
    }
    
    for(StringSet::iterator itr = fileEndings.begin();
        itr != fileEndings.end();
        ++itr)
    {
        std::cout<<"File ending : "<<*itr<<std::endl;
    }

    std::string application;
    if (fileEndings.count("p3d"))
    {
        application = "present3D";
    }
    else if (fileEndings.count("osg") ||
             fileEndings.count("ive") ||
             fileEndings.count("osga") ||
             fileEndings.count("3ds") ||
             fileEndings.count("obj") ||
             fileEndings.count("flt"))
    {
        application = "osgviewer";
    }
    else
    {
        application = "firefox";
    }
    
    std::string runstring = application;
    for(int i=0; i<argumentList.size(); ++i)
    {
        runstring += ' ';
        runstring += argumentList[i];
    }

    std::cout<<"Running : "<<runstring<<std::endl;

#ifdef DEBUG    
    fout<<"Running : "<<runstring<<std::endl;
#endif

    return system(runstring.c_str());
}