TupleTools.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# =============================================================================
## @file TupleTools.py 
#
#  Demonstration how to use TupleTools with Bender
#
#  This file is a part of 
#  <a href="http://cern.ch/lhcb-comp/Analysis/Bender/index.html">Bender project</a>
#  <b>``Python-based Interactive Environment for Smart and Friendly 
#   Physics Analysis''</b>
#
#  The package has been designed with the kind help from
#  Pere MATO and Andrey TSAREGORODTSEV. 
#  And it is based on the 
#  <a href="http://cern.ch/lhcb-comp/Analysis/LoKi/index.html">LoKi project:</a>
#  ``C++ ToolKit for Smart and Friendly Physics Analysis''
#
#  By usage of this code one clearly states the disagreement 
#  with the smear campaign of Dr.O.Callot et al.: 
#  ``No Vanya's lines are allowed in LHCb/Gaudi software.''
#
#  @author Vanya BELYAEV Ivan.Belyaev@itep.ru
#  @date 2015-10-31
#
# =============================================================================
"""Demonstration how to use TupleTools with Bender

oooooooooo.                              .o8                     
`888'   `Y8b                            \"888                     
 888     888  .ooooo.  ooo. .oo.    .oooo888   .ooooo.  oooo d8b 
 888oooo888' d88' `88b `888P\"Y88b  d88' `888  d88' `88b `888\"\"8P 
 888    `88b 888ooo888  888   888  888   888  888ooo888  888     
 888    .88P 888    .o  888   888  888   888  888    .o  888     
o888bood8P'  `Y8bod8P' o888o o888o `Y8bod88P\" `Y8bod8P' d888b    

This file is a part of BENDER project:
   ``Python-based Interactive Environment for Smart and Friendly Physics Analysis''

The project has been designed with the kind help from
Pere MATO and Andrey TSAREGORODTSEV. 

And it is based on the LoKi project:
   ``C++ ToolKit for Smart and Friendly Physics Analysis''

By usage of this code one clearly states the disagreement 
with the smear campaign of Dr.O.Callot et al.: 
    ``No Vanya's lines are allowed in LHCb/Gaudi software.''

"""
# =============================================================================
__author__  = " Vanya BELYAEV Ivan.Belyaev@itep.ru "
__date__    = " 2014-10-31" 
__version__ = " $Revision$"
# ============================================================================= 
## import everything from bender 
from   Bender.Main import *
# =============================================================================
## optional logging
# =============================================================================
from Bender.Logger import getLogger 
if '__main__' == __name__ : logger = getLogger ( 'BenderTutor.TupleTools' )
else                      : logger = getLogger ( __name__ )
# =============================================================================
## @class TupleTools
#  Demonstration how to use TupleTools with Bender
#  @author Vanya BELYAEV Ivan.Belyaev@itep.ru
#  @date 2015-10-31
class TupleTools(Algo):
    """
    Demonstration how to use TupleTools with Bender
    """
    def initialize ( self ) :

        sc = Algo.initialize ( self ) ## initialize the base class
        if sc.isFailure() : return sc
        
        IPTT = cpp.IParticleTupleTool
        IETT = cpp.IEventTupleTool
        
        self.t1 = self.tool( IPTT,'TupleToolPid'      , parent=self )
        if not self.t1 : return FAILURE 
        
        self.t2 = self.tool( IPTT,'TupleToolGeometry' , parent=self )
        if not self.t2 : return FAILURE

        self.e1 = self.tool( IETT,'TupleToolBeamSpot' , parent=self )
        if not self.e1 : return FAILURE 
        
        return SUCCESS

    ## the main 'analysis' method 
    def analyse( self ) :   ## IMPORTANT! 
        """
        The main 'analysis' method
        """
        
        ## get particles from the input locations 
        particles = self.select ( 'bmesons', 'Beauty -> J/psi(1S) K+ K-'  )
        if particles.empty() : return self.Warning('No input particles', SUCCESS )
        
        tup = self.nTuple('MyTuple')
        
        for b in particles :
            
            psi = b ( 1 ) ## the first daughter: J/psi 
            
            ## Particle Tuple Tool
            sc = self.t1.fill ( b , psi , 'psi_' , tup )
            if sc.isFailure() : return sc
            
            ## Particle Tuple Tool 
            sc = self.t2.fill ( b , b   , 'b_'   , tup )
            if sc.isFailure() : return sc            
            
            ## Event Tuple Tool 
            sc = self.e1.fill ( tup )
            if sc.isFailure() : return sc            
            
            tup.write() 
            
        ## 
        return SUCCESS      ## IMPORTANT!!! 
# =============================================================================

# =============================================================================
## The configuration of the job
def configure ( inputdata        ,    ## the list of input files  
                catalogs = []    ,    ## xml-catalogs (filled by GRID)
                castor   = False ,    ## use the direct access to castor/EOS ? 
                params   = {}    ) :
    
    ## import DaVinci 
    from Configurables import DaVinci
    ## delegate the actual configuration to DaVinci
    rootInTES = '/Event/PSIX'
    the_line  = 'Phys/SelPsi2KForPsiX/Particles'
    
    from PhysConf.Filters import LoKi_Filters
    fltrs = LoKi_Filters (
        VOID_Code = """
        0 < CONTAINS('%s/%s')
        """ % ( rootInTES , the_line ) 
        )
    dv = DaVinci ( DataType        = '2012'              ,
                   InputType       = 'MDST'              ,
                   RootInTES       = rootInTES           ,
                   EventPreFilters = fltrs.filters('WG') ,   
                   TupleFile       = 'TupleTools.root' ## IMPORTANT 
                   )

    from PhysConf.Selections import AutomaticData
    input = AutomaticData( the_line )
    
    ## (1) create the algorithm with given name 
    sel   = BenderSelection ( 
        'TupleTools' ,
        [ input ]    , 
        )
    dv.UserAlgorithms.append ( sel ) 
    
    ## define the input data
    setData  ( inputdata , catalogs , castor )
    
    ## get/create application manager
    gaudi = appMgr() 
    
    ## (1) create the algorithm with given name 
    alg   = TupleTools ( sel )
             
    return SUCCESS 
# =============================================================================

# =============================================================================
## Job steering 
if __name__ == '__main__' :

    logger.info ( 80*'*'  ) 
    logger.info ( __doc__ ) 
    logger.info ( ' Author  : %s ' %  __author__  ) 
    logger.info ( ' Version : %s ' %  __version__ ) 
    logger.info ( ' Date    : %s ' %  __date__    ) 
    logger.info ( 80*'*'  ) 
    
    ## job configuration
    ## BKQuery ( '/LHCb/Collision12/Beam4000GeV-VeloClosed-MagDown/Real Data/Reco14/Stripping20/WGBandQSelection7/90000000/PSIX.MDST'   )
    inputdata = [
        '/lhcb/LHCb/Collision12/PSIX.MDST/00035290/0000/00035290_00000221_1.psix.mdst',
        '/lhcb/LHCb/Collision12/PSIX.MDST/00035290/0000/00035290_00000282_1.psix.mdst',
        '/lhcb/LHCb/Collision12/PSIX.MDST/00035290/0000/00035290_00000238_1.psix.mdst',
        '/lhcb/LHCb/Collision12/PSIX.MDST/00035290/0000/00035290_00000185_1.psix.mdst',
        '/lhcb/LHCb/Collision12/PSIX.MDST/00035290/0000/00035290_00000278_1.psix.mdst',
        ]
    
    configure( inputdata , castor = True )
    
    ## event loop 
    run(10000)
    
    gaudi = appMgr()
    alg   = gaudi.algorithm('TupleTools')
    alg.NTuplePrint = True 
                          
# =============================================================================
# The END
# =============================================================================