# IUserDefinedSolutionHandle API implementation.    
# Calculates output values and sets them using IUDSInputData/IUDSOutputData API.
def Compute(self, inData, outData, propList, progMon):

  # Get the sweeps associated with the probe and validate
  # use the probe name that we had defined earlier
  sweeps = inData.GetSweepNamesForProbe("probe1")
  if (sweeps == None or sweeps.Count > 1):
    AddErrorMessage(self.GetName() + "Unexpected sweep count 0 or > 1 in Compute")
    return False
                                                           
  # Get the data associated with our probe
  probeData = inData.GetDoubleProbeData("probe1")
  sweepData = inData.GetSweepsDataForProbe("probe1", sweeps[0])

  # Get the user specified properties.
  # Note that ideally, these "X Min" etc names should be 
  # written as constant members and referred to in both 
  # the GetInputUDSParams and in Compute to reduce the 
  # change of typos.
  useXRangeProp = propList.GetMenuProperty("Activate X Limits").SelectedMenuChoice
  xRangeStart = propList.GetNumberProperty("X Min").ValueSI
  xRangeEnd = propList.GetNumberProperty("X Max").ValueSI

  # At this stage, one can look at the 
  # RequestedQuantities and create a dictionary to later 
  # check against. However, I am simply computing all
  # the quantities.
  minVal = 0
  maxVal = 0
  avgVal = 0

  # Check if we need to perform range computation
  if useXRangeProp == "Yes":
    seenAny = False
    avgSum = 0
    count = 0

  # zip is used since we also need to pull in sweep data
  # an index and the array notation could also have been
  # used
  for probeVal, sweepVal in zip(probeData, sweepData):
    if sweepVal < xRangeStart or sweepVal > xRangeEnd:
      pass

  # Note that in a better written script, this code 
  # would be refactored into its own function to
  # avoid code duplication
  if not seenAny:
    minVal = probeVal
    maxVal = probeVal
    avgSum = probeVal
    seenAny = True
    count = 1
  else:
    if probeVal < minVal:
      minVal = probeVal

    if probeVal > maxVal:
      maxVal = probeVal

    avgSum += probeVal
    count += 1

  if seenAny:
    avgVal = avgSum/count

  else:
    seenAny = False
    avgSum = 0
    for probeVal in probeData:
      if not seenAny:
        minVal = probeVal
        maxVal = probeVal
        avgSum = probeVal
        seenAny = True
      else:
        if probeVal < minVal:
          minVal = probeVal

  if probeVal > maxVal:
    maxVal = probeVal

  avgSum += probeVal

  if seenAny:
    avgVal = avgSum/probeData.Count

  # Finally set the output values. Note that these are 
  # always set as lists even if we have just one item.
  outData.SetDoubleQuantityData("min_val", [minVal])
  outData.SetDoubleQuantityData("max_val", [maxVal])
  outData.SetDoubleQuantityData("avg_val", [avgVal])

  # And we are done.
  return True