QtRootController.cxx

Go to the documentation of this file.

#include "QtRootController.h"

#include "RootController.h"
#include "QtRootNTuple.h"

#include "datasrcs/DataSourceController.h"
#include "datasrcs/DataSourceException.h"
#include "python/PyDataSource.h"

#include <qapplication.h>

RootController   * QtRootController::m_instance = 0;
QtRootController * QtRootController::s_instance = 0;

QtRootController::
QtRootController()
{
}

QtRootController *
QtRootController::
instance ()
{
  if ( s_instance == 0 ) {
    s_instance = new QtRootController ();
    m_instance = RootController::instance ();
  }

  return s_instance;
}

DataSource *
QtRootController::
createNTuple ( const std::string & filename, const std::string & treename )
{
  if ( qApp ) qApp -> lock ();

  std::string ds_name = filename;
  ds_name += ": ";
  ds_name += treename;

  DataSourceController * controller = DataSourceController::instance();
  DataSource * ntuple = controller -> getDataSource ( ds_name );

  if ( ntuple == 0 ) {
    TTree * tree = m_instance -> getTree ( filename, treename );
    QtRootNTuple * rtuple = new QtRootNTuple ( tree );
    ntuple = rtuple;
    ntuple -> setTitle ( treename );

    ntuple -> setName ( ds_name );
    controller -> registerNTuple ( ntuple );
    controller -> registerDataSourceFile ( ntuple );
  }

  if ( qApp ) qApp -> unlock ();

  return ntuple;
}

#ifdef HAVE_NUMARRAY
PyDataSource *
QtRootController::
createDataArray ( const std::string & filename, const std::string & treename )
{
  DataSource * rtuple = createNTuple ( filename, treename );
  PyDataSource * ds = new PyDataSource ( "RootNTuple", rtuple );

  return ds;
}
#endif

const std::vector < std::string > &
QtRootController::
getNTupleNames ( const std::string & filename )
{
  if ( qApp ) qApp -> lock ();
  try {
    const std::vector< std::string > & names 
      =  m_instance -> getNTupleNames ( filename );
    if ( qApp ) qApp -> unlock ();

    return names;
  }
  catch ( const DataSourceException & e ) {
    if ( qApp ) qApp -> unlock ();
    throw e;
  }
  static const std::vector < std::string > names;

  return names;
}

---