AnalysisFactory.h

// -*- C++ -*-
//
// AnalysisFactory.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_AnalysisFactory_H
#define LWH_AnalysisFactory_H
//
// This is the declaration of the AnalysisFactory class.
//

#include "AIAnalysisFactory.h"
#include "TreeFactory.h"
#include "HistogramFactory.h"
#include "DataPointSetFactory.h"
#include <set>

namespace LWH {

using namespace AIDA;

class AnalysisFactory: public IAnalysisFactory {

public: 
  virtual ~AnalysisFactory() {
    clear();
  }

  ITreeFactory * createTreeFactory() {
    return new TreeFactory;
  }

  IHistogramFactory * createHistogramFactory(ITree & tree) {
    Tree & tr = dynamic_cast<Tree &>(tree);
    HistogramFactory * hf = new HistogramFactory(tr);
    histfacs.insert(hf);
    return hf;
  }

  IDataPointSetFactory * createDataPointSetFactory(ITree & tree) {
    Tree & tr = dynamic_cast<Tree &>(tree);
    DataPointSetFactory * df = new DataPointSetFactory(tr);
    datafacs.insert(df);
    return df;
  }

  ITupleFactory * createTupleFactory(ITree &) {
    return 0;
  }

  IFunctionFactory * createFunctionFactory(ITree &) {
    return 0;
  }

  IPlotterFactory * createPlotterFactory(int = 0, char * * = 0,
                                         const std::string & = "",
                                         const std::string & = "") {
    return 0;
  }

  IFitFactory * createFitFactory() {
    return 0;
  }

private:

  void clear() {
    for ( std::set<HistogramFactory *>::iterator it = histfacs.begin();
          it != histfacs.end(); ++it ) delete *it;
    for ( std::set<DataPointSetFactory *>::iterator it = datafacs.begin();
          it != datafacs.end(); ++it ) delete *it;
    for ( std::set<TreeFactory *>::iterator it = treefacs.begin();
          it != treefacs.end(); ++it ) delete *it;
    histfacs.clear();
    datafacs.clear();
    treefacs.clear();
  }

  std::set<HistogramFactory *> histfacs;

  std::set<DataPointSetFactory *> datafacs;

  std::set<TreeFactory *> treefacs;

};

}

#endif /* LWH_AnalysisFactory_H */