main_split.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "PPath.h"
#include "OptionParser.h"
 
#include "phoenix_clock_mock.h"
 
 
OptionParser createOptionParser(){
        OptionParser parser(true, __PROGRAM_VERSION__);
        parser.setExampleLongOption("phoenix_clock_mock_split --input=file.clockmock --output=split.clockmock");
        parser.setExampleShortOption("phoenix_clock_mock_split -i file2.clockmock file2.clockmock -o split.clockmock");
        
        parser.addOption("input", "i", OptionType::FILENAME, true, "List of input mock to be split");
        
        PPath defaultOutputFile("./split.clockmock");
        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;
}
 
 
PPath phoenix_mockMakeOutputFile(const PPath & baseFileName, size_t indexFile, const PPath & extentionFile){
        PPath extention(extentionFile);
        if(extention == ""){
                extention = PPath("clockmock");
        }
        return PPath(baseFileName + PString("_") + PString::toString(indexFile) + PString(".") + extention);
}
 
 
bool splitMock(const std::vector<PPath> & vecInputFile, const PPath & 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;
        }
        PPath outputExtension(outputFile.getExtension()), baseOutputFile(outputFile.eraseExtension());
        bool b(true);
        for(std::vector<PPath>::const_iterator itFile(vecInputFile.begin()); itFile != vecInputFile.end() && b; ++itFile){
                VecTime vecTmpFile;
                if(data_load(*itFile, vecTmpFile)){
                        size_t nbTimeIn(vecTmpFile.size());
                        if(sizePart == 0lu){
                                if(offsetPart <= nbTimeIn){
                                        sizePart = (nbTimeIn - offsetPart)/nbPart;
                                        if(sizePart == 0lu){
                                                std::cerr << "splitMock : cannot split nbTimeIn = " << nbTimeIn << ", into nbPart = "<<nbPart<<" of size sizePart = "<<sizePart<<" and offset offsetPart = " << offsetPart << std::endl;
                                                b = false;
                                        }else{
                                                for(size_t i(0lu); i < nbPart; ++i){
                                                        VecTime vecMessage;
                                                        splitVecTime(vecMessage, vecTmpFile, offsetPart + i*sizePart, sizePart);
                                                        b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, i, outputExtension), vecMessage);
                                                }
                                        }
                                }else{
                                        VecTime vecMessage;
//                                      splitVecTime(vecMessage, vecTmpFile, offsetPart, sizePart);
                                        b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, 0lu, outputExtension), vecMessage);
                                }
                                
                        }else if(nbPart == 0lu){
                                if((sizePart + offsetPart) <= nbTimeIn){
                                        nbPart = (nbTimeIn-offsetPart)/sizePart;
                                        for(size_t i(0lu); i < nbPart; ++i){
                                                VecTime vecMessage;
                                                splitVecTime(vecMessage, vecTmpFile, offsetPart + i*sizePart, sizePart);
                                                b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, i, outputExtension), vecMessage);
                                        }
                                }else{
                                        VecTime vecMessage;
//                                      splitVecTime(vecMessage, vecTmpFile, offsetPart, sizePart);
                                        b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, 0lu, outputExtension), vecMessage);
                                }
                        }else{
                                if((sizePart*nbPart + offsetPart) <= nbTimeIn){
                                        for(size_t i(0lu); i < nbPart; ++i){
                                                VecTime vecMessage;
                                                splitVecTime(vecMessage, vecTmpFile, offsetPart + i*sizePart, sizePart);
                                                b &= data_save(phoenix_mockMakeOutputFile(baseOutputFile, i, outputExtension), vecMessage);
                                        }
                                }else{
                                        std::cerr << "splitMock : cannot split nbTimeIn = " << nbTimeIn << ", 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<PPath> vecInputFile;
        PPath 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;
}