1/*
2 * Copyright (c) 2021 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#ifndef CPU_ENTITY_H
17#define CPU_ENTITY_H
18
19#include <map>
20
21#include "cpu_time_reader.h"
22#include "entities/battery_stats_entity.h"
23#include "stats_helper.h"
24
25namespace OHOS {
26namespace PowerMgr {
27class CpuEntity : public BatteryStatsEntity {
28public:
29    CpuEntity();
30    ~CpuEntity() = default;
31    void Calculate(int32_t uid = StatsUtils::INVALID_VALUE) override;
32    double GetEntityPowerMah(int32_t uidOrUserId = StatsUtils::INVALID_VALUE) override;
33    double GetStatsPowerMah(StatsUtils::StatsType statsType, int32_t uid = StatsUtils::INVALID_VALUE) override;
34    int64_t GetCpuTimeMs(int32_t uid) override;
35    void Reset() override;
36    void DumpInfo(std::string& result, int32_t uid = StatsUtils::INVALID_VALUE) override;
37    void UpdateCpuTime() override;
38private:
39    std::shared_ptr<CpuTimeReader> cpuReader_;
40    std::map<int32_t, int64_t> cpuTimeMap_;
41    std::map<int32_t, double> cpuTotalPowerMap_;
42    std::map<int32_t, double> cpuActivePowerMap_;
43    std::map<int32_t, double> cpuClusterPowerMap_;
44    std::map<int32_t, double> cpuSpeedPowerMap_;
45    double CalculateCpuActivePower(int32_t uid);
46    double CalculateCpuClusterPower(int32_t uid);
47    double CalculateCpuSpeedPower(int32_t uid);
48};
49} // namespace PowerMgr
50} // namespace OHOS
51#endif // CPU_ENTITY_H