main_split.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "string_utils.h"
#include "convertToString.h"
#include "OptionParser.h"
 
#include "phoenix_mock.h"
 
 
OptionParser createOptionParser(){
        OptionParser parser(true, __PROGRAM_VERSION__);
        parser.setExampleLongOption("phoenix_mock_split --input=file.pmock --output=split.pmock");
        parser.setExampleShortOption("phoenix_mock_split -i file2.mock file2.pmock -o split.pmock");
        
        parser.addOption("input", "i", OptionType::FILENAME, true, "List of input mock to be split");
        
        std::string defaultOutputFile("./split.pmock");
        parser.addOption("output", "o", defaultOutputFile, "Name of the output split mock file");
        
        size_t defaultOffset(0lu), defaultSizePart(1lu), defaultNbPart(0lu);
        parser.addOption("offset", "f", defaultOffset, "Offset of the first message to be extracted in a split mock");
        parser.addOption("sizepart", "s", defaultSizePart, "Size of each split part (number of messages in each part to be split)");
        parser.addOption("nbpart", "n", defaultNbPart, "Number of output mock to be created (0 means automatic by respect to --offset and --sizepart)");
        return parser;
}
 
 
std::string phoenix_mockMakeOutputFile(const std::string & baseFileName, size_t indexFile, const std::string & extentionFile){
        std::string extention(extentionFile);
        if(extention == ""){
                extention = "pmock";
        }
        return baseFileName + "_" + convertToString(indexFile) + "." + extention;
}
 
 
bool splitMock(const std::vector<std::string> & vecInputFile, const std::string & outputFile,
                size_t offsetPart, size_t sizePart, size_t nbPart)
{
        if(sizePart == 0lu && nbPart == 0lu){
                std::cerr << "Error sizePart cannot be 0 id nbPart == 0, aborting split" << std::endl;
                return false;
        }
        std::string outputExtension(getExtention(outputFile)), baseOutputFile(eraseExtension(outputFile));
        bool b(true);
        for(std::vector<std::string>::const_iterator itFile(vecInputFile.begin()); itFile != vecInputFile.end() && b; ++itFile){
                std::vector<std::vector<char> > vecTmpFile;
                if(data_load(*itFile, vecTmpFile)){
                        size_t nbMessageIn(vecTmpFile.size());
                        if(sizePart == 0lu){
                                if(offsetPart <= nbMessageIn){
                                        sizePart = (nbMessageIn - offsetPart)/nbPart;
                                        if(sizePart == 0lu){
                                                std::cerr << "splitMock : cannot split nbMessageIn = " << nbMessageIn << ", into nbPart = "<<nbPart<<" of size sizePart = "<<sizePart<<" and offset offsetPart = " << offsetPart << std::endl;
                                                b = false;
                                        }else{
                                                for(size_t i(0lu); i < nbPart; ++i){
                                                        std::vector<std::vector<char> > vecMessage;
                                                        splitVecMessage(vecMessage, vecTmpFile, offsetPart + i*sizePart, sizePart);
                                                        b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, i, outputExtension), vecMessage);
                                                }
                                        }
                                }else{
                                        std::vector<std::vector<char> > vecMessage;
//                                      splitVecMessage(vecMessage, vecTmpFile, offsetPart, sizePart);
                                        b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, 0lu, outputExtension), vecMessage);
                                }
                                
                        }else if(nbPart == 0lu){
                                if((sizePart + offsetPart) <= nbMessageIn){
                                        nbPart = (nbMessageIn-offsetPart)/sizePart;
                                        for(size_t i(0lu); i < nbPart; ++i){
                                                std::vector<std::vector<char> > vecMessage;
                                                splitVecMessage(vecMessage, vecTmpFile, offsetPart + i*sizePart, sizePart);
                                                b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, i, outputExtension), vecMessage);
                                        }
                                }else{
                                        std::vector<std::vector<char> > vecMessage;
//                                      splitVecMessage(vecMessage, vecTmpFile, offsetPart, sizePart);
                                        b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, 0lu, outputExtension), vecMessage);
                                }
                        }else{
                                if((sizePart*nbPart + offsetPart) <= nbMessageIn){
                                        for(size_t i(0lu); i < nbPart; ++i){
                                                std::vector<std::vector<char> > vecMessage;
                                                splitVecMessage(vecMessage, vecTmpFile, offsetPart + i*sizePart, sizePart);
                                                b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, i, outputExtension), vecMessage);
                                        }
                                }else{
                                        std::cerr << "splitMock : cannot split nbMessageIn = " << nbMessageIn << ", into nbPart = "<<nbPart<<" of size sizePart = "<<sizePart<<" and offset offsetPart = " << offsetPart << std::endl;
                                        b = false;
                                }
                        }
                }else{
                        b = false;
                }
        }
        return b;
}
 
int main(int argc, char** argv){
        OptionParser parser = createOptionParser();
        parser.parseArgument(argc, argv);
        
        const OptionMode & defaultMode = parser.getDefaultMode();
        std::vector<std::string> vecInputFile;
        std::string outputFile("");
        defaultMode.getValue(vecInputFile, "input");
        defaultMode.getValue(outputFile, "output");
        
        size_t offsetPart(0lu), sizePart(1lu), nbPart(0lu);
        defaultMode.getValue(offsetPart, "offset");
        defaultMode.getValue(sizePart, "sizepart");
        defaultMode.getValue(nbPart, "nbpart");
        
        return splitMock(vecInputFile, outputFile, offsetPart, sizePart, nbPart) - 1;
}