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