NeuralNetworkToHistoTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
// -*-c++-*-
#ifndef TRKNEURALNETWORKUTILS_NEURALNETWORKTOHISTOTOOL_H
#define TRKNEURALNETWORKUTILS_NEURALNETWORKTOHISTOTOOL_H
 
/******************************************************
    @class NeuralNetworkToHistoTool
 
    Created 19-2-2008
    @author Giacinto Piacquadio (giacinto.piacquadio AT physik.uni-freiburg.de)
********************************************************/
 
#include <TString.h>
#include <vector>
#include <map>
#include <string>
#include "AthenaBaseComps/AthAlgTool.h"
 
class TH1;
class TTrainedNetwork;
 
static const InterfaceID IID_NeuralNetworkToHistoTool("Trk::NeuralNetworkToHistoTool", 1, 0);
 
namespace Trk 
{
  
 
class NeuralNetworkToHistoTool : public AthAlgTool
{
public:
  
  static const InterfaceID& interfaceID() { return IID_NeuralNetworkToHistoTool; };
 
  NeuralNetworkToHistoTool(const std::string& name,
                              const std::string& n, const IInterface* p);
 
  virtual StatusCode initialize();
  virtual StatusCode finalize();
  
  ~NeuralNetworkToHistoTool();
 
  // old style wrappers
  static std::vector<TH1*> fromTrainedNetworkToHisto(TTrainedNetwork*) ;
  static TTrainedNetwork* fromHistoToTrainedNetwork(const std::vector<TH1*> &) ;
  static TTrainedNetwork* fromHistoToTrainedNetwork(const std::vector<const TH1*> &) ;
 
  // wrappers for new NN converter tool
  // note that root's internal hist names are mangled as a workarond for 
  // root's global variable problems. 
  static std::map<std::string,TH1*> histsFromNetwork(TTrainedNetwork*) ;
  static TTrainedNetwork* networkFromHists(const std::map<std::string,const TH1*> &) ;
 
 
private:
  
};
 
}//end namespace Trk
 
 
#endif