1 /*
2  * Copyright 2023 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <android-base/thread_annotations.h>
20 #include <sys/types.h>
21 #include <optional>
22 #include <unordered_map>
23 #include <vector>
24 
25 namespace android::scheduler {
26 class SmallAreaDetectionAllowMappings {
27     using AppIdThresholdMap = std::unordered_map<int32_t, float>;
28 
29 public:
30     void update(std::vector<std::pair<int32_t, float>>& appIdThresholdMappings);
31     void setThresholdForAppId(int32_t appId, float threshold) EXCLUDES(mLock);
32     std::optional<float> getThresholdForAppId(int32_t uid) EXCLUDES(mLock);
33 
34 private:
isValidThreshold(float threshold)35     static bool isValidThreshold(float threshold) { return threshold >= 0.0f && threshold <= 1.0f; }
36     mutable std::mutex mLock;
37     AppIdThresholdMap mMap GUARDED_BY(mLock);
38 };
39 } // namespace android::scheduler
40