1/*
2 * Copyright (C) 2006 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
17package com.android.internal.app;
18
19import android.util.ArrayMap;
20import android.util.SparseArray;
21
22public class ProcessMap<E> {
23    final ArrayMap<String, SparseArray<E>> mMap
24            = new ArrayMap<String, SparseArray<E>>();
25
26    public E get(String name, int uid) {
27        SparseArray<E> uids = mMap.get(name);
28        if (uids == null) return null;
29        return uids.get(uid);
30    }
31
32    public E put(String name, int uid, E value) {
33        SparseArray<E> uids = mMap.get(name);
34        if (uids == null) {
35            uids = new SparseArray<E>(2);
36            mMap.put(name, uids);
37        }
38        uids.put(uid, value);
39        return value;
40    }
41
42    public E remove(String name, int uid) {
43        SparseArray<E> uids = mMap.get(name);
44        if (uids != null) {
45            final E old = uids.removeReturnOld(uid);
46            if (uids.size() == 0) {
47                mMap.remove(name);
48            }
49            return old;
50        }
51        return null;
52    }
53
54    public ArrayMap<String, SparseArray<E>> getMap() {
55        return mMap;
56    }
57
58    public int size() {
59        return mMap.size();
60    }
61}
62