RootDirFcn.cpp

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2024 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#define ROOTHISTCNV_ROOTDIRFCN_CPP
 
#include "RootDirFcn.h"
#include <GaudiKernel/Kernel.h>
 
#include <TDirectory.h>
#include <TFile.h>
#include <list>
#include <string>
 
namespace RootHistCnv {
 
  //-----------------------------------------------------------------------------
  bool RootCd( const std::string& full )
  //-----------------------------------------------------------------------------
  {
    int p, i = 1;
 
    gDirectory->cd( "/" );
    while ( ( p = full.find( "/", i ) ) != -1 ) {
      auto sdir = full.substr( i, p - i );
      if ( !gDirectory->GetKey( sdir.c_str() ) ) { return false; }
      gDirectory->cd( sdir.c_str() );
      i = p + 1;
    }
    gDirectory->cd( full.substr( i ).c_str() );
    return true;
  }
 
  //-----------------------------------------------------------------------------
  bool RootMkdir( const std::string& full )
  //-----------------------------------------------------------------------------
  {
 
    TDirectory* gDir = gDirectory;
 
    int  i = 1;
    auto p = full.find( ":", 0 );
    if ( p != std::string::npos ) {
      auto fil = full.substr( 0, p );
      i        = p + 1;
      fil += ":/";
      gDirectory->cd( fil.c_str() );
    }
 
    std::vector<std::string> lpath;
    while ( ( p = full.find( "/", i ) ) != std::string::npos ) {
      lpath.push_back( full.substr( i, p - i ) );
      i = p + 1;
    }
    lpath.push_back( full.substr( i ) );
 
    if ( full.compare( 0, 1, "/" ) == 0 ) gDirectory->cd( "/" );
 
    for ( const auto& lp : lpath ) {
      if ( !gDirectory->GetKey( lp.c_str() ) ) { gDirectory->mkdir( lp.c_str() ); }
      gDirectory->cd( lp.c_str() );
    }
    gDirectory = gDir;
 
    return true;
  }
 
  //-----------------------------------------------------------------------------
  std::string RootPwd()
  //-----------------------------------------------------------------------------
  {
    std::string dir = gDirectory->GetPath();
 
    return ( dir );
  }
 
  //-----------------------------------------------------------------------------
  bool RootTrimLeadingDir( std::string& full, std::string dir )
  //-----------------------------------------------------------------------------
  {
 
    if ( dir.compare( 0, 1, "/" ) != 0 ) { dir.insert( 0, "/" ); }
 
    if ( dir.compare( dir.length() - 1, 1, "/" ) != 0 ) { dir += "/"; }
 
    long ll = full.find( dir );
    if ( ll != 0 ) { return false; }
 
    full.erase( 0, dir.length() - 1 );
 
    return true;
  }
 
} // namespace RootHistCnv