15f9996aaSopenharmony_ci#!/usr/bin/env python3 25f9996aaSopenharmony_ci# -*- coding: utf-8 -*- 35f9996aaSopenharmony_ci 45f9996aaSopenharmony_ci# 55f9996aaSopenharmony_ci# Copyright (c) 2024 Huawei Device Co., Ltd. 65f9996aaSopenharmony_ci# Licensed under the Apache License, Version 2.0 (the "License"); 75f9996aaSopenharmony_ci# you may not use this file except in compliance with the License. 85f9996aaSopenharmony_ci# You may obtain a copy of the License at 95f9996aaSopenharmony_ci# 105f9996aaSopenharmony_ci# http://www.apache.org/licenses/LICENSE-2.0 115f9996aaSopenharmony_ci# 125f9996aaSopenharmony_ci# Unless required by applicable law or agreed to in writing, software 135f9996aaSopenharmony_ci# distributed under the License is distributed on an "AS IS" BASIS, 145f9996aaSopenharmony_ci# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 155f9996aaSopenharmony_ci# See the License for the specific language governing permissions and 165f9996aaSopenharmony_ci# limitations under the License. 175f9996aaSopenharmony_ci# 185f9996aaSopenharmony_ciimport time 195f9996aaSopenharmony_cifrom util.log_util import LogUtil 205f9996aaSopenharmony_ci 215f9996aaSopenharmony_ci 225f9996aaSopenharmony_ciclass TimerUtil(): 235f9996aaSopenharmony_ci def __init__(self) -> None: 245f9996aaSopenharmony_ci pass 255f9996aaSopenharmony_ci 265f9996aaSopenharmony_ci @staticmethod 275f9996aaSopenharmony_ci def cost_time(func): 285f9996aaSopenharmony_ci def inner(*arg, **kwarg): 295f9996aaSopenharmony_ci s_time = time.monotonic() 305f9996aaSopenharmony_ci res = func(*arg, **kwarg) 315f9996aaSopenharmony_ci e_time = time.monotonic() 325f9996aaSopenharmony_ci LogUtil.hb_info("The run time for {} is {} s".format(func.__name__, round(e_time - s_time, 2))) 335f9996aaSopenharmony_ci return res 345f9996aaSopenharmony_ci 355f9996aaSopenharmony_ci return inner 36