diff --git a/src/CashFlows.py b/src/CashFlows.py index 098e7179b..e07ca3e70 100644 --- a/src/CashFlows.py +++ b/src/CashFlows.py @@ -1101,11 +1101,19 @@ def calculateCashflow(self, variables, lifetimeCashflows, lifetime, verbosity): ## FIXME what if I have set the values already? # get variable values, if needed need = {'alpha': self._alpha, 'driver': self._driver} + # load alpha, driver from variables if need be need = self.loadFromVariables(need, variables, lifetimeCashflows, lifetime) # for Capex, use m * alpha * (D/D')^X alpha = need['alpha'] driver = need['driver'] + # check if not-zero alpha > 1 + maskAlpha = np.where(alpha != 0.0, )[0] + if len(maskAlpha) > 1: + if len(np.where(driver != 0.0, )[0]) == 1: + val = driver[np.where(driver != 0.0, )] + driver[maskAlpha] = val + reference = self.getParam('reference') if reference is None: reference = 1.0