btrtest.cpp

Go to the documentation of this file.

#include <gmp.h>
#include <gmpxx.h>
#include <fstream>
#include <cstring>

#include "verifierutils.hpp"

using namespace verifierUtils;

#define PATH_TO_INPUT_FILE  "./input.txt"
#define PATH_TO_OUTPUT_FILE "./actual_output.txt"


void test_ByteTree_parseString(std::ifstream &input,
                               std::ofstream &output)
{
        ByteTree * bt;
        std::string line;
        while (input>>line)
        {
                bt = ByteTree::parseString(line);
                output<<bt->toString()<<std::endl;
        }
}


void test_ByteTree_parseFile(std::ifstream &input,
                             std::ofstream &output)
{
        ByteTree * bt;
        std::string line;
        while (input>>line)
        {
                bt = ByteTree::parseFile(line);
                output<<bt->toString()<<std::endl;
        }
}


void test_Leaf_toInteger(std::ifstream &input, std::ofstream &output)
{
        ByteTree * bt;
        std::string line;
        while (input>>line)
        {
                mpz_class integer(line,10);
                bt = new Leaf(integer);
                output<<bt->toInteger()<<std::endl;
        }
}


void test_Leaf_toBoolArray(std::ifstream &input, std::ofstream &output)
{
        ByteTree * bt;
        unsigned int number;
        while (input>>number)
        {
                std::vector<uint8_t> content(number);
                for (unsigned int i=0; i<number; i++)
                {
                        bool b;
                        input>>b;
                        if (b)
                                content[i] = 1;
                        else
                                content[i] = 0;
                }
                bt = new Leaf(content);
                std::vector<bool> array = bt->toBoolArray();
                for (unsigned int i=0; i<array.size(); i++)
                        if (array[i])
                                output<<"1";
                        else
                                output<<"0";
                output<<std::endl;
        }
}


void printHelp()
{
        std::cout<<"Run ./bt_test <function to test> and it will parse the"
                 <<PATH_TO_INPUT_FILE<< "file to produce the "
                 <<PATH_TO_OUTPUT_FILE<<" file.  If this is giberish to you,"
                 <<"you\nshould read the documentation ^^"<<"\n\n"
                 <<"Available functions are:"<<"\n"
                 <<" - ByteTree::parseString\n"
                 <<" - ByteTree::parseFile\n"
                 <<" - Leaf.toInteger\n"
                 <<" - Leaf.toBoolArray\n"
                 <<"\n"<<std::endl;
}



int main(int argc, char ** argv)
{
        if (argc<2)
                printHelp();
        else
        {
                std::ifstream input(PATH_TO_INPUT_FILE,
                                    std::ifstream::in);
                std::ofstream output(PATH_TO_OUTPUT_FILE);
                if (strcmp(argv[1],"ByteTree::parseString") == 0)
                        test_ByteTree_parseString(input,output);
                else if (strcmp(argv[1],"ByteTree::parseFile") == 0)
                        test_ByteTree_parseFile(input,output);
                else if (strcmp(argv[1],"Leaf.toInteger") == 0)
                        test_Leaf_toInteger(input,output);
                else if (strcmp(argv[1],"Leaf.toBoolArray") == 0)
                        test_Leaf_toBoolArray(input,output);
                else
                        printHelp();
                output.close();
        }        
        
     return 0;
}