nnp-select.cpp File Reference

#include "Log.h"
#include "utility.h"
#include <cstddef>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <fstream>

Include dependency graph for nnp-select.cpp:

Go to the source code of this file.

Functions
int	main (int argc, char *argv[])

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 28 of file nnp-select.cpp.

{
    if (argc < 3)
    {
        cout << "USAGE: " << argv[0] << " <mode> <arg1 <arg2>>\n"
             << "       <mode> ... Choose selection mode (random/interval).\n"
             << "       Arguments for mode \"random\":\n"
             << "         <arg1> ... Ratio of selected structures "
             << "(1.0 equals 100 %).\n"
             << "         <arg2> ... Seed for random number generator "
             << "(integer).\n"
             << "       Arguments for mode \"interval\":\n"
             << "         <arg1> ... Select structures in this interval "
             << "(integer).\n"
             << "       Execute in directory with these NNP files present:\n"
             << "       - input.data (structure file)\n";
        return 1;
    }
 
    bool mode = false; // 0 = random, 1 = interval.
    bool writeStructure = false;
    size_t countStructures = 0;
    size_t countSelected = 0;
    size_t interval = 10;
    int seed = 10;
    double ratio = 1.0;
 
    Log log;
    ifstream inputFile;
    ofstream outputFile;
    ofstream rejectFile;
    ofstream logFile;
 
    logFile.open("nnp-select.log");
    log.registerStreamPointer(&logFile);
 
    log << "\n";
    log << "*** NNP-SELECT **************************"
           "**************************************\n";
    log << "\n";
 
    if (strcmp(argv[1], "random") == 0)
    {
        mode = false;
        if (argc != 4)
        {
            throw invalid_argument("ERROR: Wrong number of arguments.\n");
        }
        ratio = atof(argv[2]);
        seed = atoi(argv[3]);
        srand(seed);
        log << strpr("Selecting randomly %.2f %% of all structures.\n",
                     ratio * 100.0);
        log << strpr("Random number generator seed: %d.\n", seed);
    }
    else if (strcmp(argv[1], "interval") == 0)
    {
        mode = true; 
        if (argc != 3)
        {
            throw invalid_argument("ERROR: Wrong number of arguments.\n");
        }
        interval = (size_t)atoi(argv[2]);
        log << strpr("Selecting every %d structure.\n", interval);
    }
    else
    {
        throw invalid_argument("ERROR: Unknown selection mode.\n");
    }
 
    log << "*****************************************"
           "**************************************\n";
 
    inputFile.open("input.data");
    outputFile.open("output.data");
    rejectFile.open("reject.data");
    string line;
    while (getline(inputFile, line))
    {
        if (split(reduce(line)).at(0) == "begin")
        {
            if (mode)
            {
                if (countStructures % interval == 0) writeStructure = true; 
                else writeStructure = false;
            }
            else
            {
                if ((double)rand() / RAND_MAX <= ratio) writeStructure = true;
                else writeStructure = false;
            }
            countStructures++; 
            if (writeStructure)
            {
                log << strpr("Structure %7d selected.\n", countStructures);
                countSelected++;
            }
        }
        if (writeStructure)
        {
            outputFile << line << '\n';
        }
        else
        {
            rejectFile << line << '\n';
        }
    }
    inputFile.close();
    outputFile.close();
    rejectFile.close();
 
    log << "*****************************************"
           "**************************************\n";
    log << strpr("Total    structures           : %7d\n", countStructures);
    log << strpr("Selected structures           : %7d\n", countSelected);
    log << strpr("Selected structures percentage: %.3f %%\n",
                 countSelected / (double)countStructures * 100.0);
    log << "*****************************************"
           "**************************************\n";
    logFile.close();
 
    return 0;
}

References logFile, nnp::reduce(), nnp::Log::registerStreamPointer(), nnp::split(), and nnp::strpr().

Here is the call graph for this function: