Commit 7132e1d3 authored by Charlles Abreu's avatar Charlles Abreu
Browse files

BiasVariable allows arguments with units of measurement

parent 1f4451ec
...@@ -440,15 +440,20 @@ class BiasVariable(object): ...@@ -440,15 +440,20 @@ class BiasVariable(object):
a reasonable value is chosen automatically. a reasonable value is chosen automatically.
""" """
self.force = force self.force = force
self.minValue = minValue self.minValue = self._standardize(minValue)
self.maxValue = maxValue self.maxValue = self._standardize(maxValue)
self.biasWidth = biasWidth self.biasWidth = self._standardize(biasWidth)
self.periodic = periodic self.periodic = periodic
if gridWidth is None: if gridWidth is None:
self.gridWidth = int(np.ceil(5*(maxValue-minValue)/biasWidth)) self.gridWidth = int(np.ceil(5*(maxValue-minValue)/biasWidth))
else: else:
self.gridWidth = gridWidth self.gridWidth = gridWidth
self._scaledVariance = (biasWidth/(maxValue-minValue))**2 self._scaledVariance = (self.biasWidth/(self.maxValue-self.minValue))**2
def _standardize(self, quantity):
if unit.is_quantity(quantity):
return quantity.value_in_unit_system(unit.md_unit_system)
else:
return quantity
_LoadedBias = namedtuple('LoadedBias', ['id', 'index', 'bias']) _LoadedBias = namedtuple('LoadedBias', ['id', 'index', 'bias'])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment