Functions
def	get_spills (day, deltaDay, beam)

def	num_spills_between (t0, t1, spills)

def	main ()

Variables
	view = sys.argv[1]

	beam = str("numi")

int	ndays = -1

int	interval = -1

string	BASE_DIR = "/nusoft/app/web/htdoc/nova/datacheck/nearline/data/"

	beamStruct = BeamStruct()

	f = TFile(BASE_DIR + 'POTSpillRateDay.root','RECREATE')

	tree = TTree('BeamMetrics','POTtree')

Detailed Description

  Compute average delivered POT during a given time range
  Justin Vasel (jvasel@fnal.gov)

Function Documentation

def POTSpillRate.get_spills	(	day,
		deltaDay,
		beam
	)

Get info about spills during a particular day

  Args:
  day : A datetime object
  deltaDay : Number of days to consider

  Returns:
  spills : A list of spills (time in sec, pot number in 1E+12)

Definition at line 113 of file POTSpillRate.py.

References check_time_usage.float, novadaq::HexUtils.format(), and runNovaSAM.str.

Referenced by main().

 def get_spills(day, deltaDay, beam):
   """ Get info about spills during a particular day
 
     Args:
     day : A datetime object
     deltaDay : Number of days to consider
 
     Returns:
     spills : A list of spills (time in sec, pot number in 1E+12)
     """
 
   # Initialize list of spills
   spills = []
 
   # Convert start and end of day to UNIX timestamp. We must explicitly specify
   # the UTC timezone (+00:00) because the ifb data server uses CST (-06:00) as
   # the default
   t0  = str(day.isoformat('T')) + "+00:00"
   t1  = str((day + timedelta(days = deltaDay)).isoformat('T')) + "+00:00"
 
   print "Getting spills for the period between {} and {}".format(t0,t1)
 
   # URL where the spill data live
   if beam == str("numi"):
     url = 'http://ifb-data.fnal.gov:8099/ifbeam/data/data?'\
       'v=E:TRTGTD&e=e,a9&'\
       't0={}&t1={}&&f=csv'.format(t0, t1)
   elif beam == str("bnb"):
     url = 'http://ifb-data.fnal.gov:8099/ifbeam/data/data?'\
       'v=E:TOR860&e=e,1d&'\
       't0={}&t1={}&&f=csv'.format(t0, t1)
   
   print "URL: {}".format(url)
 
   # Store the HTTP GET response here
   response = None
 
   print "Trying for a response...",
 
   # Try to retrieve the URL contents. If this fails three times, then
   # give up and return an empty array.
   try:
     response = urllib2.urlopen(url, None, 200)
   except:
     print "No response. Trying again..."
     try:
       response = urllib2.urlopen(url, None, 200)
     except:
       print "No response. Trying again..."
       try:
         response = urllib2.urlopen(url, None, 200)
       except:
         print "No response. Will not try again."
         sys.exit()
 
   if (response == None):
     return spills
 
   # Use csv module to parse the response as CSV
   csvdata = csv.reader(response)
   print "SUCCESS"
 
   # The response will look like this:
   # "e,a9", E:TRTGTD, 1406851201047, E12, 22.0598346211
   # We are interested in the third column (unix timestamp in ms)
   # and the fifth column (value). The list will be populated with these
   # values.
   print "Importing Spills..."
   if beam == str("numi"):
     for row in csvdata:
       if (row.count('E:TRTGTD') > 0):
         if (row[4] != 'null'):
           spills.append( (float(row[2]) / 1000., float(row[4])) )
 
   elif beam == str("bnb"):
     for row in csvdata:
       if (row.count('E:TOR860') > 0):
         if (row[4] != 'null'):
           spills.append( (float(row[2]) / 1000., float(row[4])) )
   print "DONE"
 
   # Return spill info
   return spills
 
 #.......................................................................

def POTSpillRate.main ( )

1D plots hits vs w //

Definition at line 231 of file POTSpillRate.py.

References check_time_usage.float, novadaq::HexUtils.format(), get_spills(), makeTrainCVSamples.int, num_spills_between(), and PandAna.Demos.demo1.range.

 def main():
 
   # Set startTime and currentTime to now
   # (startTime is a constant; currenTime is variable).
   startTime   = datetime.utcnow()
   currentTime = startTime
 
   # Loop over days
   for i in range(1, int(ndays) + 1):
       print "DAY {} of {}".format(i, ndays)
       print "====================="
 
       # Get two days worth of spills (we'll only use a subset of this)
       spills = get_spills(startTime - timedelta(days = i), 2, beam)
 
       # Get number of time steps
       nsteps = int(math.floor(ndays * 24 * 60 / interval))
 
       print "Begin looping over {} intervals of {} minutes each".format(nsteps, interval)
 
       # Loop over steps in day
       for j in range(0, int(nsteps / ndays)):
           # Find total number of spills in this timestep
           spillTotal = num_spills_between(currentTime - timedelta(minutes = interval), currentTime, spills)
 
           # Calculate spill rate (spills per second)
           spillRate = spillTotal * 1.0 / (interval * 60)
 
           # Pring spill rate to screen
           print spillRate
 
           # Fill ROOT branches/tree
           beamStruct.Time = int((currentTime - datetime(1970,1,1)).total_seconds())
           beamStruct.Beam = float(spillRate)
           tree.Fill()
 
           # Advance to next timestep
           currentTime = currentTime - timedelta(minutes = interval)
 
       print "\n"
 
   f.Write()
   f.Close()
 
 
 
 # Prevent execution on import

def POTSpillRate.num_spills_between	(	t0,
		t1,
		spills
	)

Determines the number of POT in a given time interval

  Args:
  t0     : A datetime object
  t1     : A datetime object
  spills : A list of spills

  Returns:
  potTotal : int

Definition at line 198 of file POTSpillRate.py.

References DCS_db_parser.database_connect(), novadaq::HexUtils.format(), runNovaSAM.str, and POTGetData.unix_timestamp().

Referenced by main().

 def num_spills_between(t0, t1, spills):
   """ Determines the number of POT in a given time interval
 
     Args:
     t0     : A datetime object
     t1     : A datetime object
     spills : A list of spills
 
     Returns:
     potTotal : int
     """
 
   # Initiate connection to psql database
   database_connect(beam)
 
   print "Couting spills between {} and {}: ".format(str(t0.isoformat('T')),str(t1.isoformat('T'))),
 
   # Convert to unix timestamps
   t0 = unix_timestamp(t0)
   t1 = unix_timestamp(t1)
 
   # initialize counter
   counter = 0
 
   # Loop over spills, but only count the ones that fall into the
   # appropriate time window.
   for (time, value) in spills:
     if ((time >= t0) & (time <= t1)):
       counter = counter + 1
 
   return counter
 
 #.......................................................................