/*
 * Copyright (c) 2023-2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef HISTREAMER_STEADY_CLOCK_H
#define HISTREAMER_STEADY_CLOCK_H

#include <chrono>

namespace OHOS {
namespace Media {
class SteadyClock {
public:
    static int64_t GetCurrentTimeMs();
    static int64_t GetCurrentTimeNanoSec();
    SteadyClock();
    ~SteadyClock() = default;
    void Reset();
    int64_t ElapsedMilliseconds();
    int64_t ElapsedMicroseconds();
    int64_t ElapsedNanoseconds();
    int64_t ElapsedSeconds();

private:
    std::chrono::time_point<std::chrono::high_resolution_clock> begin_;
};
} // namespace Media
} // namespace OHOS

#if defined(PROFILE)
#include "foundation/log.h"
#define PROFILE_BEGIN(message, ...)                                                                                    \
    do {                                                                                                               \
        MEDIA_LOG_I(message ", timestamp(ms): " PUBLIC_LOG_D64, ##__VA_ARGS__, SteadyClock::GetCurrentTimeMs());      \
    } while (0);                                                                                                       \
    OHOS::Media::SteadyClock _steadyClock;
#define PROFILE_RESET() _steadyClock.Reset()

#define PROFILE_END(message, ...)                                                                                      \
    do {                                                                                                               \
        MEDIA_LOG_I(message ", timestamp(ms): " PUBLIC_LOG_D64 ", duration(ms): " PUBLIC_LOG_D64,                    \
                    ##__VA_ARGS__, SteadyClock::GetCurrentTimeMs(), _steadyClock.ElapsedMilliseconds());               \
    } while (0)

#else
#define PROFILE_BEGIN(message, ...)
#define PROFILE_RESET()
#define PROFILE_END(message, ...)
#endif

#endif // HISTREAMER_STEADY_CLOCK_H