Added ability to manually specify PID to be traced

Author: alxhoff

EnergyDebugger.py

@@ -362,6 +362,7 @@ def __init__(self, parent=None):
         self.graph = False
         self.subgraph = False
         self.skip_tracing = False
+        self.pid = None
 
         # Governor control
         self.governorRadioButtons = []
@@ -649,6 +650,9 @@ def buttonrun(self):
         try:
             self.checkrun()
 
+            if self.lineEditAppPID.text() != "":
+                self.pid = self.lineEditAppPID.text()
+
             self.application_name = self.lineEditApplicationName.text()
             self.duration = self.doubleSpinBoxDuration.value()
             self.events = []
@@ -682,21 +686,20 @@ def buttonrun(self):
                 for x in range(test_start_value,
                                test_start_value + no_of_tests):
                     try:
-                        buttonrunprocess(
-                            self.application_name,
-                            self.governor,
-                            self.duration,
-                            self.events,
-                            self.events_to_process,
-                            self.preamble,
-                            self.subgraph,
-                            self.graph,
-                            self.skip_tracing,
-                            progress_signal=self.changed_progress,
-                            open_func=self.openallresults,
-                            subdir=self.application_name + "/" +
-                            self.governor + "/" + str(x) + "/",
-                        )
+                        buttonrunprocess(self.application_name,
+                                         self.governor,
+                                         self.duration,
+                                         self.events,
+                                         self.events_to_process,
+                                         self.preamble,
+                                         self.subgraph,
+                                         self.graph,
+                                         self.skip_tracing,
+                                         progress_signal=self.changed_progress,
+                                         open_func=self.openallresults,
+                                         subdir=self.application_name + "/" +
+                                         self.governor + "/" + str(x) + "/",
+                                         pid=self.pid)
                         if self.checkBoxTestAutomationPrompt.isChecked():
                             QMessageBox.information(
                                 self,
@@ -811,22 +814,22 @@ def buttonrunprocess(
         progress_signal=None,
         open_func=None,
         subdir=None,
+        pid=None,
 ):
 
     try:
-        current_debugger = EnergyDebugger(
-            application=application_name,
-            governor=governor,
-            duration=duration,
-            events=events,
-            event_count=events_to_process,
-            preamble=preamble,
-            graph=graph,
-            subgraph=subgraph,
-            skip_tracing=skip_tracing,
-            progress_signal=progress_signal,
-            results_subdir=subdir,
-        )
+        current_debugger = EnergyDebugger(application=application_name,
+                                          governor=governor,
+                                          duration=duration,
+                                          events=events,
+                                          event_count=events_to_process,
+                                          preamble=preamble,
+                                          graph=graph,
+                                          subgraph=subgraph,
+                                          skip_tracing=skip_tracing,
+                                          progress_signal=progress_signal,
+                                          results_subdir=subdir,
+                                          pid=pid)
         current_debugger.run()
         if open_func is not None:
             open_func(subdir)
@@ -895,20 +898,19 @@ def run(self):
 
 
 class EnergyDebugger:
-    def __init__(
-            self,
-            application,
-            governor,
-            duration,
-            events,
-            event_count,
-            preamble,
-            graph,
-            subgraph,
-            skip_tracing,
-            progress_signal,
-            results_subdir,
-    ):
+    def __init__(self,
+                 application,
+                 governor,
+                 duration,
+                 events,
+                 event_count,
+                 preamble,
+                 graph,
+                 subgraph,
+                 skip_tracing,
+                 progress_signal,
+                 results_subdir,
+                 pid=None):
         self.adb = ADBInterface()
         self.application = application
         self.governor = governor
@@ -922,13 +924,14 @@ def __init__(
         self.skip_tracing = skip_tracing
         self.progress_signal = progress_signal
         self.results_subdir = results_subdir
+        self.pid = pid
         """ Required objects for tracking system metrics and interfacing with a target system, connected
         via an ADB connection.
         """
 
         start_time = time.time()
         try:
-            self.pid_tool = PIDTool(self.adb, self.application)
+            self.pid_tool = PIDTool(self.adb, self.application, self.pid)
         except Exception, e:
             raise Exception("Trace failed: {}".format(e))
         print("PIDs gathered --- %s Sec" % (time.time() - start_time))

MainInterface.py

@@ -39,7 +39,7 @@ def setupUi(self, MainWindow):
         self.scrollArea.setWidgetResizable(True)
         self.scrollArea.setObjectName("scrollArea")
         self.scrollAreaWidgetContents = QtWidgets.QWidget()
-        self.scrollAreaWidgetContents.setGeometry(QtCore.QRect(0, 0, 283, 762))
+        self.scrollAreaWidgetContents.setGeometry(QtCore.QRect(0, 0, 283, 792))
         self.scrollAreaWidgetContents.setObjectName("scrollAreaWidgetContents")
         self.verticalLayout_2 = QtWidgets.QVBoxLayout(
             self.scrollAreaWidgetContents)
@@ -79,6 +79,15 @@ def setupUi(self, MainWindow):
         self.lineEditApplicationName.setObjectName("lineEditApplicationName")
         self.horizontalLayout_3.addWidget(self.lineEditApplicationName)
         self.verticalLayout_5.addLayout(self.horizontalLayout_3)
+        self.horizontalLayout_14 = QtWidgets.QHBoxLayout()
+        self.horizontalLayout_14.setObjectName("horizontalLayout_14")
+        self.label_35 = QtWidgets.QLabel(self.frameTraceOptions)
+        self.label_35.setObjectName("label_35")
+        self.horizontalLayout_14.addWidget(self.label_35)
+        self.lineEditAppPID = QtWidgets.QLineEdit(self.frameTraceOptions)
+        self.lineEditAppPID.setObjectName("lineEditAppPID")
+        self.horizontalLayout_14.addWidget(self.lineEditAppPID)
+        self.verticalLayout_5.addLayout(self.horizontalLayout_14)
         self.horizontalLayout_4 = QtWidgets.QHBoxLayout()
         self.horizontalLayout_4.setSizeConstraint(
             QtWidgets.QLayout.SetNoConstraint)
@@ -832,6 +841,9 @@ def retranslateUi(self, MainWindow):
             _translate("MainWindow", "Android Energy Debugger"))
         self.label_6.setText(_translate("MainWindow", "<b>Trace</b>"))
         self.label.setText(_translate("MainWindow", "Application:"))
+        self.label_35.setText(_translate("MainWindow", "Manual PID"))
+        self.lineEditAppPID.setPlaceholderText(
+            _translate("MainWindow", "Input PID for hard-to-find apps"))
         self.label_3.setText(_translate("MainWindow", "Trace Duration:"))
         self.checkBoxDrawGraph.setText(
             _translate("MainWindow", "Draw Graph File (.dot)"))

PIDTool.py

@@ -28,12 +28,14 @@ class PIDTool:
     """ Probes the target system using ps and grep to extract all relevant threads to bother the target
     application, system services and binder threads.
     """
-    def __init__(self, adb_device, name):
+    def __init__(self, adb_device, name, pid=None):
 
         try:
             self.adb_device = adb_device
             self.name = name
-            main_pid = self._find_main_pid()
+
+            main_pid = self._find_main_pid(pid)
+
             if main_pid is None:
                 print(
                     "Failed to find main PID for given application: {}".format(
@@ -68,13 +70,15 @@ def _find_all_pid(self):
         self._find_binder_pids()
         print("---- Found Binder PIDs --- %s Sec" % (time.time() - start_time))
 
-    def _find_main_pid(self):
+    def _find_main_pid(self, pid=None):
         """ Will find the parent PID of the target application.
 
         :return: PID object of the main process for the target application
         """
-
-        res = self.adb_device.command("ps | grep " + self.name)
+        if pid is None:
+            res = self.adb_device.command("ps | grep " + self.name)
+        else:
+            res = self.adb_device.command("ps | grep " + pid)
         if res == "":
             return None
 

UI/MainInterface.ui

@@ -51,7 +51,7 @@
              <x>0</x>
              <y>0</y>
              <width>283</width>
-             <height>762</height>
+             <height>792</height>
             </rect>
            </property>
            <layout class="QVBoxLayout" name="verticalLayout_2">
@@ -100,6 +100,24 @@
                  </item>
                 </layout>
                </item>
+               <item>
+                <layout class="QHBoxLayout" name="horizontalLayout_14">
+                 <item>
+                  <widget class="QLabel" name="label_35">
+                   <property name="text">
+                    <string>Manual PID</string>
+                   </property>
+                  </widget>
+                 </item>
+                 <item>
+                  <widget class="QLineEdit" name="lineEditAppPID">
+                   <property name="placeholderText">
+                    <string>Input PID for hard-to-find apps</string>
+                   </property>
+                  </widget>
+                 </item>
+                </layout>
+               </item>
                <item>
                 <layout class="QHBoxLayout" name="horizontalLayout_4">
                  <property name="sizeConstraint">