Gaudi::RootEventExtractor Struct Reference

Collaboration diagram for Gaudi::RootEventExtractor:

Public Member Functions
	RootEventExtractor ()=default
	Default constructor. More...
	RootEventExtractor (const char *input, const char *output, const char *output_option)
	Initializing constructor directly opening the input file and the output file. More...
virtual	~RootEventExtractor ()
	Default destructor. More...
ExtractStatus	openInput (const char *name)
	Open input file. More...
ExtractStatus	closeInput ()
	Close input file. More...
ExtractStatus	openOutput (const char *name, const char *option)
	Open output file. More...
ExtractStatus	closeOutput ()
	Close output file. More...
ExtractStatus	cancel ()
	Clear the list of event numbers to be extracted from the file. More...
ExtractStatus	select (int evt_num)
	Add a given entry number to the list of events to be selected to the output file. More...
ExtractStatus	extract ()
	Extract all previously selected events from the input file and write it to the output. More...

Protected Attributes
std::unique_ptr< TFile >	m_in
TTree *	m_evt_in = nullptr
TTree *	m_ref_in = nullptr
TFile *	m_out = nullptr
TTree *	m_evt_out = nullptr
TTree *	m_ref_out = nullptr
int	m_localDB_id = 0
std::vector< int >	m_eventList

Detailed Description

Definition at line 116 of file extractEvt.C.

Constructor & Destructor Documentation

RootEventExtractor() [1/2]

Gaudi::RootEventExtractor::RootEventExtractor ( )

default

Default constructor.

RootEventExtractor() [2/2]

RootEventExtractor::RootEventExtractor	(	const char *	input,
		const char *	output,
		const char *	output_option
	)

Initializing constructor directly opening the input file and the output file.

Definition at line 169 of file extractEvt.C.

                                                                                                         {
  if ( EXTRACT_SUCCESS != openInput( input ) ) {
    throw std::runtime_error( "Failed to open input file:" + std::string( input ) );
  }
  if ( EXTRACT_SUCCESS != openOutput( output, output_option ) ) {
    throw std::runtime_error( "Failed to open output file:" + std::string( output ) );
  }
}

~RootEventExtractor()

RootEventExtractor::~RootEventExtractor ( )

virtual

Default destructor.

Definition at line 179 of file extractEvt.C.

                                        {
  closeInput();
  closeOutput();
}

Member Function Documentation

cancel()

ExtractStatus RootEventExtractor::cancel

(

)

Clear the list of event numbers to be extracted from the file.

Definition at line 185 of file extractEvt.C.

                                         {
  m_eventList.clear();
  ::printf( "+++ Event list cleared.\n" );
  return EXTRACT_SUCCESS;
}

closeInput()

ExtractStatus RootEventExtractor::closeInput

(

)

Close input file.

Definition at line 218 of file extractEvt.C.

                                             {
  if ( m_in ) {
    ::printf( "+++ Closing input  file:%s\n", m_in->GetName() );
    m_in->Close();
  }
  m_in.reset();
  m_evt_in = nullptr;
  m_ref_in = nullptr;
  return EXTRACT_SUCCESS;
}

closeOutput()

ExtractStatus RootEventExtractor::closeOutput

(

)

Close output file.

Definition at line 230 of file extractEvt.C.

                                              {
  if ( m_out ) {
    ::printf( "+++ Closing output file:%s\n", m_out->GetName() );
    if ( m_evt_out ) m_evt_out->Write();
    if ( m_ref_out ) m_ref_out->Write();
    m_out->Close();
    delete m_out;
  }
  m_out     = nullptr;
  m_evt_out = nullptr;
  m_ref_out = nullptr;
  return EXTRACT_SUCCESS;
}

extract()

ExtractStatus RootEventExtractor::extract

(

)

Extract all previously selected events from the input file and write it to the output.

Definition at line 245 of file extractEvt.C.

                                          {
  char     text[1024];
  bool     new_output = false;
  TBranch *br_in, *br_out;
  if ( !m_in || m_in->IsZombie() ) {
    throw std::runtime_error( "Input file missing - cannot extract events." );
  } else if ( !m_out || m_out->IsZombie() ) {
    throw std::runtime_error( "Output file missing - cannot extract events." );
  } else {
    m_evt_out = (TTree*)m_out->Get( "Event" );
    if ( !m_evt_out ) { m_evt_out = m_evt_in->CloneTree( 0 ); }
    m_ref_out = (TTree*)m_out->Get( "Refs" );
    if ( !m_ref_out ) {
      m_ref_out  = m_ref_in->CloneTree( 0 );
      new_output = true;
    }
  }
  m_localDB_id = -1;
  if ( new_output ) {
    ::printf( "+++ Copy refs table to new output file:%s\n", m_out->GetName() );
    br_in = m_ref_in->GetBranch( "Databases" );
    br_in->SetAddress( text );
    br_out = m_ref_out->GetBranch( "Databases" );
    br_out->SetAddress( text );
    for ( int i = 0; i < br_in->GetEntries(); ++i ) {
      br_in->GetEntry( i );
      br_out->Fill();
      if ( m_localDB_id < 0 && strcmp( text, "<localDB>" ) == 0 ) { m_localDB_id = i; }
    }

    br_in = m_ref_in->GetBranch( "Containers" );
    br_in->SetAddress( text );
    br_out = m_ref_out->GetBranch( "Containers" );
    br_out->SetAddress( text );
    for ( int i = 0; i < br_in->GetEntries(); ++i ) {
      br_in->GetEntry( i );
      br_out->Fill();
    }

    br_in = m_ref_in->GetBranch( "Links" );
    br_in->SetAddress( text );
    br_out = m_ref_out->GetBranch( "Links" );
    br_out->SetAddress( text );
    for ( int i = 0; i < br_in->GetEntries(); ++i ) {
      br_in->GetEntry( i );
      br_out->Fill();
    }

    br_in = m_ref_in->GetBranch( "Params" );
    br_in->SetAddress( text );
    br_out = m_ref_out->GetBranch( "Params" );
    br_out->SetAddress( text );
    for ( int i = 0; i < br_in->GetEntries(); ++i ) {
      br_in->GetEntry( i );
      if ( strncmp( text, "PFN=", 4 ) == 0 ) { // Update PFN entry
        ::snprintf( text, sizeof( text ), "PFN=%s", br_out->GetFile()->GetName() );
        ::printf( "+++ PFN of the created output file is:%s\n", text );
      } else if ( strncmp( text, "FID=", 4 ) == 0 ) { // Create new FID for new file
        static const char* fmt = "FID=%08lX-%04hX-%04hX-%02hhX%02hhX-%02hhX%02hhX%02hhX%02hhX%02hhX%02hhX";
        union uuid_data {
          unsigned char  buf[16];
          unsigned short sbuf[8];
          unsigned int   ibuf[4];
        } d;
        TUUID uuid;
        uuid.GetUUID( d.buf );
        ::snprintf( text, sizeof( text ), fmt, d.ibuf[0], d.sbuf[2], d.sbuf[3], d.buf[8], d.buf[9], d.buf[10],
                    d.buf[11], d.buf[12], d.buf[13], d.buf[14], d.buf[15] );
        ::printf( "+++ FID of the created output file is:%s\n", text );
      }
      br_out->Fill();
    }
    m_ref_out->Write();
  }

  TIter next( m_evt_in->GetListOfBranches() );
  while ( ( br_in = (TBranch*)next() ) ) {
    TString name     = br_in->GetName();
    TClass* br_class = gROOT->GetClass( br_in->GetClassName(), kTRUE );
    br_out           = m_evt_out->GetBranch( name );
    if ( !br_out ) {
      ::printf( "+++ ERROR: Input and output event trees are incompatible. Selection not possible.\n" );
      return EXTRACT_ERROR;
    }
    int out_num_entries = br_out->GetEntries();
    for ( const auto& i : m_eventList ) {
      int         num_evt = i;
      const char* br_type = "DataObject";
      void*       pObject = br_class->New();
      br_in->SetAddress( &pObject );
      br_out->SetAddress( &pObject );
      int num_rd = br_in->GetEntry( num_evt );
      if ( num_rd < 0 ) { // Definitive error
        ::printf( "+++ ERROR: Failed to read data from branch:%s\n", name.Data() );
        return EXTRACT_ERROR;
      }
      if ( name.EndsWith( "_R." ) ) {
        RootObjectRefs* refs = (RootObjectRefs*)pObject;
        for ( auto& ir : refs->refs ) {
          RootRef& r = ir;
          if ( r.dbase == m_localDB_id ) r.entry = out_num_entries;
        }
        br_type = "*Index*";
      }
      int num_wr = br_out->Fill();
      if ( num_wr < 0 ) {
        ::printf( "+++ ERROR: Failed to write data to extraction branch:%s "
                  "read:%d wrote:%d bytes [Length-mismatch]\n",
                  name.Data(), num_rd, num_wr );
        return EXTRACT_ERROR;
      }
      ::printf( "+++ Copied %8d bytes to %-10s branch %s(%d)\n", num_rd, br_type, name.Data(), out_num_entries );
      ++out_num_entries;
    }
    m_evt_out->SetEntries( br_out->GetEntries() + 1 );
  }
  m_evt_out->Write();
  return cancel();
}

openInput()

ExtractStatus RootEventExtractor::openInput

(

const char *

name

)

Open input file.

Close input file.

Definition at line 198 of file extractEvt.C.

                                                              {
  if ( m_in ) closeInput();
  m_in.reset( TFile::Open( name ) );
  if ( m_in && !m_in->IsZombie() ) {
    m_evt_in = (TTree*)m_in->Get( "Event" );
    m_ref_in = (TTree*)m_in->Get( "Refs" );
    return EXTRACT_SUCCESS;
  }
  return EXTRACT_ERROR;
}

openOutput()

ExtractStatus RootEventExtractor::openOutput	(	const char *	name,
		const char *	option
	)

Open output file.

Close input file.

Definition at line 210 of file extractEvt.C.

                                                                                   {
  if ( m_out ) closeOutput();
  m_out     = TFile::Open( name, option );
  m_evt_out = m_ref_out = nullptr;
  return EXTRACT_SUCCESS;
}

select()

ExtractStatus RootEventExtractor::select

(

int

evt_num

)

Add a given entry number to the list of events to be selected to the output file.

Definition at line 192 of file extractEvt.C.

                                                      {
  m_eventList.push_back( evt_num );
  return EXTRACT_SUCCESS;
}

Member Data Documentation

m_eventList

std::vector<int> Gaudi::RootEventExtractor::m_eventList

protected

Definition at line 127 of file extractEvt.C.

m_evt_in

TTree* Gaudi::RootEventExtractor::m_evt_in = nullptr

protected

Definition at line 119 of file extractEvt.C.

m_evt_out

TTree* Gaudi::RootEventExtractor::m_evt_out = nullptr

protected

Definition at line 123 of file extractEvt.C.

m_in

std::unique_ptr<TFile> Gaudi::RootEventExtractor::m_in

protected

Definition at line 118 of file extractEvt.C.

m_localDB_id

int Gaudi::RootEventExtractor::m_localDB_id = 0

protected

Definition at line 126 of file extractEvt.C.

m_out

TFile* Gaudi::RootEventExtractor::m_out = nullptr

protected

Definition at line 122 of file extractEvt.C.

m_ref_in

TTree* Gaudi::RootEventExtractor::m_ref_in = nullptr

protected

Definition at line 120 of file extractEvt.C.

m_ref_out

TTree* Gaudi::RootEventExtractor::m_ref_out = nullptr

protected

Definition at line 124 of file extractEvt.C.

---

The documentation for this struct was generated from the following file:

• RootCnv/merge/extractEvt.C