main.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "convertToString.h"
 
#include "OptionParser.h"
 
#include "FrontEnd/HeaderParser.h"
#include "BackEnd/backend.h"
#include "BackEnd/backend_proxy_loader.h"


OptionParser createOptionParser(){
        OptionParser parser(true, __PROGRAM_VERSION__);
        parser.setExampleLongOption("phoenix_hpc_proxy --input=file.ph5 --output=output/dir/");
        parser.setExampleShortOption("phoenix_hpc_proxy -i file1.ph5 file1.ph5 fileN.ph5 -o output/dir/");
        
        parser.addOption("header", "i", OptionType::FILENAME, true, "list of input headers to be used to describe the main library");
        std::string libName("lib_shadok");
        parser.addOption("library", "l", libName, "name of the library to be created");
        
        parser.addOption("libdir", "d", OptionType::DIRECTORY, false, "directory where the sub libraries are installed");
        
        parser.addOption("anyarch", "", OptionType::FILENAME, false, "file name of the library to be used for any architectures (kind of security if the other architectures are not found)");
        parser.addOption("sse", "", OptionType::FILENAME, false, "file name of the library to be used for sse architecture");
        parser.addOption("sse2", "", OptionType::FILENAME, false, "file name of the library to be used for sse2 architecture");
        parser.addOption("ssse3", "", OptionType::FILENAME, false, "file name of the library to be used for ssse3 architecture");
        parser.addOption("sse4.1", "", OptionType::FILENAME, false, "file name of the library to be used for sse4.1 architecture");
        parser.addOption("sse4.2", "", OptionType::FILENAME, false, "file name of the library to be used for sse4.2 architecture");
        parser.addOption("avx", "", OptionType::FILENAME, false, "file name of the library to be used for avx architecture");
        parser.addOption("avx2", "", OptionType::FILENAME, false, "file name of the library to be used for avx2 architecture");
        parser.addOption("avx512f", "", OptionType::FILENAME, false, "file name of the library to be used for avx512f architecture");
        parser.addOption("avx512bw", "", OptionType::FILENAME, false, "file name of the library to be used for avx512wb architecture");
        
        std::string defaultOutputDir(".");
        parser.addOption("output", "o", defaultOutputDir, "Output directory");
        
        return parser;
}


void parseVecArchLib(PVecArchLib & vecArchLib, const OptionMode & defaultMode){
        std::string libAnyArch(""), libSse(""), libSse2(""), libSsse3(""), libSse41(""), libSse42(""), libAvx(""), libAvx2(""), libAvx512f(""), libAvx512wb("");
        defaultMode.getValue(libAnyArch, "anyarch");
        defaultMode.getValue(libSse, "sse");
        defaultMode.getValue(libSse2, "sse2");
        defaultMode.getValue(libSsse3, "ssse3");
        defaultMode.getValue(libSse41, "sse4.1");
        defaultMode.getValue(libSse42, "sse4.2");
        defaultMode.getValue(libAvx, "avx");
        defaultMode.getValue(libAvx2, "avx2");
        defaultMode.getValue(libAvx512f, "avx512f");
        defaultMode.getValue(libAvx512wb, "avx512wb");
        
        addArch(vecArchLib, "anyarch", libAnyArch);
        addArch(vecArchLib, "sse", libSse);
        addArch(vecArchLib, "sse2", libSse2);
        addArch(vecArchLib, "ssse3", libSsse3);
        addArch(vecArchLib, "sse4.1", libSse41);
        addArch(vecArchLib, "sse4.2", libSse42);
        addArch(vecArchLib, "avx", libAvx);
        addArch(vecArchLib, "avx2", libAvx2);
        addArch(vecArchLib, "avx512f", libAvx512f);
        addArch(vecArchLib, "avx512wb", libAvx512wb);
}


bool createSourceFromConfig(const PString & inputFile, const PString & libraryName, const PVecArchLib & vecArchLib,
                                const PString & libDir, const PString & outputDir)
{
        HeaderParser parser;
        if(!parser.load(inputFile)){
                std::cerr << "createSourceFromConfig : cannot parse the input file '"<<inputFile<<"'" << std::endl;
                return false;
        }
        std::string libName(getLibraryFile(vecArchLib));
        PVecSource vecSource = parser.getVecSource();
        if(vecSource.size() == 0lu){
                std::cerr << "createSourceFromConfig : no source to be created" << std::endl;
                return false;
        }
        bool b(cpp_backend(vecSource, libName, outputDir));
        if(!b){
                std::cerr << "createSourceFromConfig : cannot save source/header from input file '"<<inputFile<<"'" << std::endl;
                return b;
        }
        b &= cpp_backend_proxy(vecSource, libraryName, vecArchLib, libDir, outputDir);
        if(!b){
                std::cerr << "createSourceFromConfig : cannot save source/header ProxyLoader class" << std::endl;
                return b;
        }
        return b;
}


int processFiles(const PVecPath & listInputFile, const PString & libName, const PVecArchLib & vecArchLib,
                const PPath & libDir, const PPath & outputDir)
{
        bool b(true);
        for(PVecPath::const_iterator it(listInputFile.begin()); it != listInputFile.end(); ++it){
                b &= createSourceFromConfig(*it, libName, vecArchLib, libDir, outputDir);
        }
        return b - 1;
}
 
int main(int argc, char** argv){
        OptionParser parser = createOptionParser();
        parser.parseArgument(argc, argv);
        
        const OptionMode & defaultMode = parser.getDefaultMode();
        PVecPath listInputFile;
        defaultMode.getValue(listInputFile, "header");
        
        PPath outputDir("."), libDir("");
        PString libName("");
        defaultMode.getValue(outputDir, "output");
        defaultMode.getValue(libName, "library");
        defaultMode.getValue(libDir, "libdir");
        
        PVecArchLib vecArchLib;
        parseVecArchLib(vecArchLib, defaultMode);
        
        return processFiles(listInputFile, libName, vecArchLib, libDir, outputDir);
}