src/com/android/loganalysis/item/CompactMemInfoItem.java

/*
 * Copyright (C) 2014 The Android Open Source Project
 *
 * 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.
 */
package com.android.loganalysis.item;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * Contains a list of processes and how much memory they are using. Generated from parsing
 * compact mem info file. Refer to CompactMemInfoParser for more details.
 */
public class CompactMemInfoItem implements IItem {
    /** Constants for JSON output */
    public static final String PID_JSON_KEY = "pid";
    public static final String NAME_JSON_KEY = "name";
    public static final String PSS_JSON_KEY = "pss";
    public static final String SWAP_JSON_KEY = "swap";
    public static final String TYPE_JSON_KEY = "type";
    public static final String ACTIVITIES_JSON_KEY = "activities";
    public static final String PROCESSES_JSON_KEY = "processes";
    public static final String LOST_RAM_JSON_KEY = "lostRam";
    /** Constants for attributes HashMap */
    private static final String NAME_ATTR_KEY = "name";
    private static final String PSS_ATTR_KEY = "pss";
    private static final String SWAP_ATTR_KEY = "swap";
    private static final String TYPE_ATTR_KEY = "type";
    private static final String ACTIVITIES_ATTR_KEY = "activities";

    private Map<Integer, Map<String, Object>> mPids = new HashMap<Integer, Map<String, Object>>();
    private long mLostRam;

    @Override
    public IItem merge(IItem other) throws ConflictingItemException {
        throw new ConflictingItemException("Compact meminfo items cannot be merged");
    }

    @Override
    public boolean isConsistent(IItem other) {
       return false;
    }

    @Override
    public JSONObject toJson() {
        JSONObject object = new JSONObject();
        JSONArray processes = new JSONArray();
        // Add processes and their attributes
        for(int pid : getPids()) {
            JSONObject proc = new JSONObject();
            try {
                proc.put(PID_JSON_KEY, pid);
                proc.put(NAME_JSON_KEY, getName(pid));
                proc.put(PSS_JSON_KEY, getPss(pid));
                proc.put(SWAP_JSON_KEY, getSwap(pid));
                proc.put(TYPE_JSON_KEY, getType(pid));
                proc.put(ACTIVITIES_JSON_KEY, hasActivities(pid));
                processes.put(proc);
            } catch (JSONException e) {
                // ignore
            }
        }
        try {
            object.put(PROCESSES_JSON_KEY, processes);
            // Add the lost RAM field
            object.put(LOST_RAM_JSON_KEY, getLostRam());
        } catch (JSONException e) {
            // ignore
        }

        return object;
    }

    /**
     * Get the list of pids of the processes that were added so far.
     * @return
     */
    public Set<Integer> getPids() {
        return mPids.keySet();
    }

    private Map<String, Object> get(int pid) {
        return mPids.get(pid);
    }

    /**
     * Adds a process to the list stored in this item.
     */
    public void addPid(int pid, String name, String type, long pss, long swap, boolean activities) {
        Map<String, Object> attributes = new HashMap<String, Object>();
        attributes.put(NAME_ATTR_KEY, name);
        attributes.put(TYPE_ATTR_KEY, type);
        attributes.put(PSS_ATTR_KEY, pss);
        attributes.put(SWAP_ATTR_KEY, swap);
        attributes.put(ACTIVITIES_ATTR_KEY, activities);
        mPids.put(pid, attributes);
    }

    /**
     * Returns the name of the process with a given pid.
     */
    public String getName(int pid) {
        return (String)get(pid).get(NAME_ATTR_KEY);
    }

    /**
     * Return pss of the process with a given name.
     */
    public long getPss(int pid) {
        return (Long)get(pid).get(PSS_ATTR_KEY);
    }

    /**
     * Return swap memory of the process with a given name.
     */
    public long getSwap(int pid) {
        return (Long)get(pid).get(SWAP_ATTR_KEY);
    }

    /**
     * Returns the type of the process with a given pid. Some possible types are native, cached,
     * foreground and etc.
     */
    public String getType(int pid) {
        return (String)get(pid).get(TYPE_ATTR_KEY);
    }

    /**
     * Returns true if a process has any activities assosiated with it. False otherwise.
     */
    public boolean hasActivities(int pid) {
        return (Boolean)get(pid).get(ACTIVITIES_ATTR_KEY);
    }

    /**
     * Sets the lost RAM field
     */
    public void setLostRam(long lostRam) {
        mLostRam = lostRam;
    }

    /**
     * Returns the lost RAM field
     */
    public long getLostRam() {
        return mLostRam;
    }
}