testtxt.cpp

来自「把html转成txt 把html转成txt」· C++ 代码 · 共 84 行

#include <iostream>
#include <fstream>
#include <stdlib.h>

#include "WriterFactory.h"
#include "ReaderFactory.h"
#include "ReaderInterface.h"
#include "WriterInterface.h"

int main(int argc, char* argv[])
{
  using namespace DoxEngine;
  using namespace std;
  
  DebugLog log;

  try
  {
    if (argc != 4)
    {
      std::cout << "test <infile> <outfile> <repititions>\n";
      return 0;
    }

    int i;
    for (i=0;i<atoi(argv[3]);i++)
    {


      ifstream input(argv[1], fstream::in);
      ofstream output(argv[2], fstream::out);


	  DoxEngine::WriterFactories &writerFactories = DoxEngine::WriterFactoriesSingleton::GetWriterFactories();

	  DoxEngine::WriterFactories::iterator writerIterator = writerFactories.find(FORMAT_RTF);

      DoxEngine::WriterInterface *writer;
	  
	  if (writerIterator == writerFactories.end())
	  {
	    std::cerr << "Internal error initialising writer";
		break;
	  }
	  else
	  {
		// First type is the key (file format)
		// Second type is the value (factory instance)
		writer = writerIterator->second->Create(output, log);
	  }

 	  DoxEngine::ReaderFactories &readerFactories = DoxEngine::ReaderFactoriesSingleton::GetReaderFactories();

	  DoxEngine::ReaderFactories::iterator readerIterator = readerFactories.find(FORMAT_TEXT);
      DoxEngine::ReadInterface *reader;
	  if (readerIterator == readerFactories.end())
	  {
		std::cerr << "Internal error initialising reader";
		break;
	  }
	  else
	  {
		// First type is the key (file format)
		// Second type is the value (factory instance)
		reader = readerIterator->second->Create(input, *writer, log);
	  }


      while(reader->processData());

      delete reader;
      delete writer;
    }


  }
  catch(...)
  {
    cout<<"Exception occured";
  }

}