DBInterface.hh

#ifndef DBInterface_h
#define DBInterface_h


#include "lccd.h"


// forward declarations
class ICondDBMgr ;
class ICondDBDataAccess;
class ICondDBFolderMgr;
class ICondDBTagMgr;
class ICondDBObject;


// LCIO headers
#include "lcio.h"
#include "EVENT/LCCollection.h"


#include <iostream>

namespace lccd {
  
  typedef std::vector< lcio::LCCollection* > ColVec ;
 

  class DBInterface {
    
  public:

    DBInterface( const DBInterface& )  = delete ;
    DBInterface& operator=( const DBInterface& )  = delete ;

    
    DBInterface( const std::string& folder, bool update=false ) ;
    
    DBInterface( const std::string& dbInit, const std::string& folder, bool update=false) ;
    
    virtual ~DBInterface() ;
    
    ICondDBMgr* condDBMgr() { return _condDBmgr ; }
    
    ICondDBDataAccess* condDataAccess() {  return _condDataAccess ; }
    
    ICondDBFolderMgr*  condFolderMgr() { return _condFolderMgr ; }
    
    ICondDBTagMgr * condTagMgr() { return _condTagMgr ; }


    const std::string& folderName() {  return _folder ; } 

    const std::string& dbName() {  return _dbName ; } 


    void storeCollection( LCCDTimeStamp since, LCCDTimeStamp till,
                          lcio::LCCollection* col,
                          const std::string& description ) ;
      

    lcio::LCCollection* findCollection( LCCDTimeStamp timeStamp, 
                                        LCCDTimeStamp& since, LCCDTimeStamp& till,
                                        const std::string& tag="" ) ; 



    lcio::LCCollection* findNextValidCollection( LCCDTimeStamp timeStamp, 
                                                 LCCDTimeStamp& since, LCCDTimeStamp& till,
                                                 const std::string& tag="" ) ; 
    

    lcio::LCCollection* findLastValidCollection( LCCDTimeStamp timeStamp, 
                                                 LCCDTimeStamp& since, LCCDTimeStamp& till,
                                                 const std::string& tag="" ) ; 


    lcio::LCCollection* findCollection( LCCDTimeStamp timeStamp, 
                                        const std::string& tag="" ) ; 
    
    void findCollections( ColVec& colVec, const std::string& tag="" ) ; 

    void findCollections(ColVec &colVec, LCCDTimeStamp since, LCCDTimeStamp till, const std::string &tag="") ; 

    void findCollections( ColVec& colVec, LCCDTimeStamp timeStamp ) ; 

    void findAllCollections( ColVec& colVec ) ; 

    void tagFolder( const std::string& tagName, const std::string& description="", std::string usingTagName = "");


    bool folderBranchContainsTag ( const std::string& tagName ) ;
  

    void createDBFile( const std::string& tag="" ) ;


    void createSimpleFile( LCCDTimeStamp timeStamp, const std::string& tag, bool allLayers=false ) ;

    void dump2File( ) ;


  protected:
    void init() ;

    lcio::LCCollection* collectionFromCondDBObject( ICondDBObject* object, const std::string& tag ) ;
    

    // data members
    ICondDBMgr* _condDBmgr = nullptr ;
    ICondDBDataAccess* _condDataAccess = nullptr ;
    ICondDBFolderMgr* _condFolderMgr = nullptr ;
    ICondDBTagMgr * _condTagMgr = nullptr ; 


  private:

    std::string _dbInit{} ;
    std::string _folder{} ;
    bool _update{} ;
    std::string _dbName{} ;
  };

}  //end namespace

#endif // DBInterface_h