TreeFactory.h

// -*- C++ -*-
//
// TreeFactory.h is a part of ThePEG - Toolkit for HEP Event Generation
// Copyright (C) 1999-2019 Leif Lonnblad
//
// ThePEG is licenced under version 3 of the GPL, see COPYING for details.
// Please respect the MCnet academic guidelines, see GUIDELINES for details.
//
#ifndef LWH_TreeFactory_H
#define LWH_TreeFactory_H
//
// This is the declaration of the TreeFactory class.
//

#include "AITreeFactory.h"
#include <string>
#include <stdexcept>
#include "Tree.h"

namespace LWH {

using namespace AIDA;

class TreeFactory: public ITreeFactory {

public:

  virtual ~TreeFactory() {
    clear();
  }

  ITree * create() {
    Tree * tree = new Tree;
    trees.insert(tree);
    return tree;
  }

  Tree * createTree(const std::string & storeName) {
    return new Tree(storeName);
  }

  ITree * create(const std::string & storeName,
                 const std::string & storeType = "",
                 bool readOnly = false, bool createNew = false,
                 const std::string & = "") {
    if ( storeType != "xml" && storeType != "" && storeType != "flat" )
      throw std::runtime_error("Can only store trees in xml or flat format.");
    if ( readOnly || !createNew )
      throw std::runtime_error("Cannot read in trees.");
    return new Tree(storeName, storeType != "flat");
  }

private:

  void clear() {
    for ( std::set<Tree *>::iterator it = trees.begin();
          it != trees.end(); ++it ) delete *it;
    trees.clear();
  }

  std::set<Tree *> trees;

};

}

#endif /* LWH_TreeFactory_H */