TestPrefixTbl.C

Go to the documentation of this file.

// Copyright (C) 2001, Compaq Computer Corporation
// 
// This file is part of Vesta.
// 
// Vesta is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// 
// Vesta is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
// 
// You should have received a copy of the GNU Lesser General Public
// License along with Vesta; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

// Created on Sat Mar 28 23:02:19 PST 1998 by heydon
// Last modified on Thu Aug  5 10:12:42 EDT 2004 by ken@xorian.net
//      modified on Mon Mar 30 18:34:09 PST 1998 by heydon

/* A test of the "PrefixTbl" class. */

#include <Basics.H>
#include <FS.H>
#include "PrefixTbl.H"

using std::ofstream;
using std::ifstream;
using std::cout;
using std::cin;
using std::cerr;
using std::endl;

int main(int argc, char *argv[]) {
    // check for correct # of arguments
    if (argc != 2) {
        cerr << "Incorrect number of arguments." << endl;
        cerr << "Syntax: TestPrefixTbl filename" << endl;
        return 1;
    }

    // read from stdin and create the table
    const int MaxLineLen = 80;
    char line[MaxLineLen + 1];
    PrefixTbl tbl(/*sizeHint=*/ 10);
    TextIntTbl tempTbl;
    cout << "Adding paths:" << endl;
    while (cin.peek() != EOF) {
        cin.getline(line, MaxLineLen, '\n');
        FV2::T seq(line);
        int idx = tbl.Put(seq, /*INOUT*/ tempTbl);
        cout << "  " << line << " -> " << idx << endl;
    }
    cout << endl;

    // print the table
    cout << "Here is the resulting table:" << endl << endl;
    tbl.Print(cout);
    cout << endl;

    Text fname(argv[1]);
    try {
        // write the table out to disk
        cout << "Pickling table to " << fname << "..." << endl << endl;
        ofstream ofs;
        FS::OpenForWriting(fname, /*OUT*/ ofs);
        tbl.Write(ofs);
        FS::Close(ofs);

        // read it back in and print it
        ifstream ifs;
        FS::OpenReadOnly(fname, /*OUT*/ ifs);
        PrefixTbl newTbl(ifs);
        if (!FS::AtEOF(ifs)) {
            cerr << "Error: stream not at end-of-file as expected!" << endl;
        }
        FS::Close(ifs);
        cout << "Here is the unpickled version of the table:" << endl << endl;
        newTbl.Print(cout);
    } catch (const FS::Failure &f) {
        cerr << "Error pickling table: " << f << "; exiting..." << endl;
        return 1;
    }
    return 0;
}

---