testing/performance/talos/cmanager_win32.py

   1 # ***** BEGIN LICENSE BLOCK *****
   2 # Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3 #
   4 # The contents of this file are subject to the Mozilla Public License Version
   5 # 1.1 (the "License"); you may not use this file except in compliance with
   6 # the License. You may obtain a copy of the License at
   7 # http://www.mozilla.org/MPL/
   8 #
   9 # Software distributed under the License is distributed on an "AS IS" basis,
  10 # WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11 # for the specific language governing rights and limitations under the
  12 # License.
  13 #
  14 # The Original Code is standalone Firefox Windows performance test.
  15 #
  16 # The Initial Developer of the Original Code is Google Inc.
  17 # Portions created by the Initial Developer are Copyright (C) 2006
  18 # the Initial Developer. All Rights Reserved.
  19 #
  20 # Contributor(s):
  21 #   Annie Sullivan <annie.sullivan@gmail.com> (original author)
  22 #   Ben Hearsum    <bhearsum@wittydomain.com> (OS independence)
  23 #
  24 # Alternatively, the contents of this file may be used under the terms of
  25 # either the GNU General Public License Version 2 or later (the "GPL"), or
  26 # the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27 # in which case the provisions of the GPL or the LGPL are applicable instead
  28 # of those above. If you wish to allow use of your version of this file only
  29 # under the terms of either the GPL or the LGPL, and not to allow others to
  30 # use your version of this file under the terms of the MPL, indicate your
  31 # decision by deleting the provisions above and replace them with the notice
  32 # and other provisions required by the GPL or the LGPL. If you do not delete
  33 # the provisions above, a recipient may use your version of this file under
  34 # the terms of any one of the MPL, the GPL or the LGPL.
  35 #
  36 # ***** END LICENSE BLOCK *****
  37
  38
  39 import win32pdh
  40 import win32pdhutil
  41
  42
  43 class CounterManager:
  44
  45   def __init__(self, process, counters=None):
  46     self.process = process
  47     self.registeredCounters = {}
  48     self.registerCounters(counters)
  49     # PDH might need to be "refreshed" if it has been queried while Firefox
  50     # is closed
  51     win32pdh.EnumObjects(None, None, 0, 1)
  52
  53   def registerCounters(self, counters):
  54     for counter in counters:
  55       self.registeredCounters[counter] = []
  56
  57   def unregisterCounters(self, counters):
  58     for counter in counters:
  59       if counter in self.registeredCounters:
  60         del self.registeredCounters[counter]
  61
  62   def getRegisteredCounters(self):
  63     return keys(self.registeredCounters)
  64
  65   def getCounterValue(self, counter):
  66     hq = self.registeredCounters[counter][0]
  67     hc = self.registeredCounters[counter][1]
  68     try:
  69       win32pdh.CollectQueryData(hq)
  70       type, val = win32pdh.GetFormattedCounterValue(hc, win32pdh.PDH_FMT_LONG)
  71       return val
  72     except:
  73       return None
  74
  75   def getProcess(self):
  76     return self.process
  77
  78   def startMonitor(self):
  79     # PDH might need to be "refreshed" if it has been queried while Firefox
  80     # is closed
  81     win32pdh.EnumObjects(None, None, 0, 1)
  82
  83     for counter in self.registeredCounters:
  84       path = win32pdh.MakeCounterPath((None, 'process', self.process,
  85                                        None, -1, counter))
  86       hq = win32pdh.OpenQuery()
  87       try:
  88         hc = win32pdh.AddCounter(hq, path)
  89       except:
  90         win32pdh.CloseQuery(hq)
  91
  92       self.registeredCounters[counter] = [hq, hc]
  93
  94   def stopMonitor(self):
  95     for counter in self.registeredCounters:
  96       win32pdh.RemoveCounter(self.registeredCounters[counter][1])
  97       win32pdh.CloseQuery(self.registeredCounters[counter][0])
  98     self.registeredCounters.clear()