content: /*
   * Copyright (C) 2013 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.launcher3.testing;
  
  import android.app.ActivityManager;
  import android.app.Service;
  import android.content.Context;
  import android.content.Intent;
  import android.os.Binder;
  import android.os.Debug;
  import android.os.Handler;
  import android.os.IBinder;
  import android.os.Message;
  import android.os.SystemClock;
  import android.util.Log;
  import android.util.LongSparseArray;
  
  import com.android.launcher3.util.TestingUtils;
  
  import java.util.ArrayList;
  import java.util.List;
  
  public class MemoryTracker extends Service {
      public static final String TAG = MemoryTracker.class.getSimpleName();
  
      private static final long UPDATE_RATE = 5000;
  
      private static final int MSG_START = 1;
      private static final int MSG_STOP = 2;
      private static final int MSG_UPDATE = 3;
  
      public static class ProcessMemInfo {
          public int pid;
          public String name;
          public long startTime;
          public long currentPss, currentUss;
          public long[] pss = new long[256];
          public long[] uss = new long[256];
              //= new Meminfo[(int) (30 * 60 / (UPDATE_RATE / 1000))]; // 30 minutes
          public long max = 1;
          public int head = 0;
          public ProcessMemInfo(int pid, String name, long start) {
              this.pid = pid;
              this.name = name;
              this.startTime = start;
          }
          public long getUptime() {
              return System.currentTimeMillis() - startTime;
          }
      };
      public final LongSparseArray<ProcessMemInfo> mData = new LongSparseArray<ProcessMemInfo>();
      public final ArrayList<Long> mPids = new ArrayList<Long>();
      private int[] mPidsArray = new int[0];
      private final Object mLock = new Object();
  
      Handler mHandler = new Handler() {
          @Override
          public void handleMessage(Message m) {
              switch (m.what) {
                  case MSG_START:
                      mHandler.removeMessages(MSG_UPDATE);
                      mHandler.sendEmptyMessage(MSG_UPDATE);
                      break;
                  case MSG_STOP:
                      mHandler.removeMessages(MSG_UPDATE);
                      break;
                  case MSG_UPDATE:
                      update();
                      mHandler.removeMessages(MSG_UPDATE);
                      mHandler.sendEmptyMessageDelayed(MSG_UPDATE, UPDATE_RATE);
                      break;
              }
          }
      };
  
      ActivityManager mAm;
  
      public ProcessMemInfo getMemInfo(int pid) {
          return mData.get(pid);
      }
  
      public int[] getTrackedProcesses() {
          return mPidsArray;
      }
  
      public void startTrackingProcess(int pid, String name, long start) {
          synchronized (mLock) {
              final Long lpid = Long.valueOf(pid);
  
              if (mPids.contains(lpid)) return;
  
              mPids.add(lpid);
              updatePidsArrayL();
  
              mData.put(pid, new ProcessMemInfo(pid, name, start));
          }
      }
  
      void updatePidsArrayL() {
          final int N = mPids.size();
          mPidsArray = new int[N];
          StringBuffer sb = new StringBuffer("Now tracking processes: ");
          for (int i=0; i<N; i++) {
              final int p = mPids.get(i).intValue();
              mPidsArray[i] = p;
              sb.append(p); sb.append(" ");
          }
          Log.v(TAG, sb.toString());
      }
  
      void update() {
          synchronized (mLock) {
              Debug.MemoryInfo[] dinfos = mAm.getProcessMemoryInfo(mPidsArray);
              for (int i=0; i<dinfos.length; i++) {
                  Debug.MemoryInfo dinfo = dinfos[i];
                  if (i > mPids.size()) {
                      Log.e(TAG, "update: unknown process info received: " + dinfo);
                      break;
                  }
                  final long pid = mPids.get(i).intValue();
                  final ProcessMemInfo info = mData.get(pid);
                  info.head = (info.head+1) % info.pss.length;
                  info.pss[info.head] = info.currentPss = dinfo.getTotalPss();
                  info.uss[info.head] = info.currentUss = dinfo.getTotalPrivateDirty();
                  if (info.currentPss > info.max) info.max = info.currentPss;
                  if (info.currentUss > info.max) info.max = info.currentUss;
                  // Log.v(TAG, "update: pid " + pid + " pss=" + info.currentPss + " uss=" + info.currentUss);
                  if (info.currentPss == 0) {
                      Log.v(TAG, "update: pid " + pid + " has pss=0, it probably died");
                      mData.remove(pid);
                  }
              }
              for (int i=mPids.size()-1; i>=0; i--) {
                  final long pid = mPids.get(i).intValue();
                  if (mData.get(pid) == null) {
                      mPids.remove(i);
                      updatePidsArrayL();
                  }
              }
          }
      }
  
      @Override
      public void onCreate() {
          mAm = (ActivityManager) getSystemService(Context.ACTIVITY_SERVICE);
  
          // catch up in case we crashed but other processes are still running
          List<ActivityManager.RunningServiceInfo> svcs = mAm.getRunningServices(256);
          for (ActivityManager.RunningServiceInfo svc : svcs) {
              if (svc.service.getPackageName().equals(getPackageName())) {
                  Log.v(TAG, "discovered running service: " + svc.process + " (" + svc.pid + ")");
                  startTrackingProcess(svc.pid, svc.process,
                          System.currentTimeMillis() - (SystemClock.elapsedRealtime() - svc.activeSince));
              }
          }
  
          List<ActivityManager.RunningAppProcessInfo> procs = mAm.getRunningAppProcesses();
          for (ActivityManager.RunningAppProcessInfo proc : procs) {
              final String pname = proc.processName;
              if (pname.startsWith(getPackageName())) {
                  Log.v(TAG, "discovered other running process: " + pname + " (" + proc.pid + ")");
                  startTrackingProcess(proc.pid, pname, System.currentTimeMillis());
              }
          }
      }
  
      @Override
      public void onDestroy() {
          mHandler.sendEmptyMessage(MSG_STOP);
      }
  
      @Override
      public int onStartCommand(Intent intent, int flags, int startId) {
          Log.v(TAG, "Received start id " + startId + ": " + intent);
  
          if (intent != null) {
              if (TestingUtils.ACTION_START_TRACKING.equals(intent.getAction())) {
                  final int pid = intent.getIntExtra("pid", -1);
                  final String name = intent.getStringExtra("name");
                  final long start = intent.getLongExtra("start", System.currentTimeMillis());
                  startTrackingProcess(pid, name, start);
              }
          }
  
          mHandler.sendEmptyMessage(MSG_START);
  
          return START_STICKY;
      }
  
      public class MemoryTrackerInterface extends Binder {
          MemoryTracker getService() {
              return MemoryTracker.this;
          }
      }
  
      private final IBinder mBinder = new MemoryTrackerInterface();
  
      public IBinder onBind(Intent intent) {
          mHandler.sendEmptyMessage(MSG_START);
  
          return mBinder;
      }
  }
  