#!/usr/bin/env python3

##########################################################################
# basf2 (Belle II Analysis Software Framework)                           #
# Author: The Belle II Collaboration                                     #
#                                                                        #
# See git log for contributors and copyright holders.                    #
# This file is licensed under LGPL-3.0, see LICENSE.md.                  #
##########################################################################

######################################################
# shows how to use the logging object exposed by the framework
######################################################

from basf2 import Path, process, logging, set_log_level, set_debug_level, \
                  LogConfig, LogLevel, LogInfo, B2DEBUG, B2INFO, B2WARNING

# show default loglevel
print('Default loglevel:', logging.log_level)

# change loglevel to show also debug messages
logging.log_level = LogLevel.DEBUG
# or alternatively:
set_log_level(LogLevel.DEBUG)
print('New loglevel:    ', logging.log_level)

# change debug level
logging.debug_level = 123
# or alternatively:
set_debug_level(123)

# set specific LogConfig for pxd package:
# log level WARNING, debug level 0, abort on errors
pxd_logging = LogConfig(LogLevel.WARNING, 0)
pxd_logging.abort_level = LogLevel.ERROR
logging.set_package('pxd', pxd_logging)

# set specific log options for svd package:
# debug level 10, print level, message, and file name for info messages
logging.package('svd').debug_level = 10
logging.package('svd').set_info(LogLevel.INFO, LogInfo.LEVEL | LogInfo.MESSAGE |
                                LogInfo.FILE)

# show default loginfo for all levels
for (name, value) in LogLevel.names.items():
    if name != 'default':
        print('LogInfo for level %8s: %3d' % (name, logging.get_info(value)))

# reset logging connections (default log connection to the console is deleted)
logging.reset()
# add logfile for output
logging.add_file('test.log')
# add colored console output
logging.add_console(True)

# It is possible to send log messages from python to have consistent messages
# print some log messages
B2DEBUG(100, 'Debug Message')
B2INFO('Info Message')
B2WARNING('Warning Message')
# B2ERROR('Error Message')

# Repeated messages (provided they originate from the same line) can be
# filtered. Note that this is not the default, you can enable this using
# a logging.reset() and logging.add_console(True), as done above.
for i in range(0, 42):
    B2DEBUG(100, 'A repeated message!')

B2INFO('Some other message')

# next line will bail since the default abort-level is FATAL,
# so leaving it commented.
# B2FATAL('Fatal Message')

# show number of log_messages per level
for (level, num) in logging.log_stats.items():
    print('Messages for level %8s: %2d' % (level.name, num))

main = Path()
eventinfosetter = main.add_module('EventInfoSetter')
# configure logging for the EventInfoSetter module:
# loglevel ERROR and info for ERROR message is file name and line number
eventinfosetter.logging.log_level = LogLevel.ERROR
eventinfosetter.logging.set_info(LogLevel.ERROR, LogInfo.FILE | LogInfo.LINE)
# or alternatively:
eventinfosetter.set_log_level(LogLevel.ERROR)
eventinfosetter.set_log_info(LogLevel.ERROR, LogInfo.FILE | LogInfo.LINE)

# add time stamp to all INFO messages
currentInfo = logging.get_info(LogLevel.INFO)
logging.set_info(LogLevel.INFO, currentInfo | LogInfo.TIMESTAMP)

# run some events to see time stamp
set_log_level(LogLevel.INFO)
eventinfosetter.param('evtNumList', [30])
main.add_module('EventInfoPrinter', logLevel=LogLevel.DEBUG)
process(main)