def test_valuefederate_test_vectord(): broker = createBroker() vFed, fedinfo = createValueFederate() defaultValue = [0.0, 1.0, 2.0] testValue = [3.0, 4.0, 5.0] pubid = h.helicsFederateRegisterGlobalPublication(vFed, "pub1", h.HELICS_DATA_TYPE_VECTOR, "") subid = h.helicsFederateRegisterSubscription(vFed, "pub1", "") h.helicsInputSetDefaultVector(subid, defaultValue) h.helicsFederateEnterExecutingMode(vFed) h.helicsPublicationPublishVector(pubid, testValue) assert h.helicsInputGetVector(subid) == defaultValue grantedtime = h.helicsFederateRequestTime(vFed, 1.0) assert grantedtime == 0.01 value = h.helicsInputGetVector(subid) assert value == testValue destroyFederate(vFed, fedinfo) destroyBroker(broker)
def _get_input_based_on_type(self, in_put): """ Get input based on type :param in_put: :return: """ val = None sub_id = in_put['sub_id'] try: if in_put['type'] == 'integer': val = h.helicsInputGetInteger(sub_id) elif in_put['type'] == 'double': val = h.helicsInputGetDouble(sub_id) elif in_put['type'] == 'string': val = h.helicsInputGetString(sub_id) elif in_put['type'] == 'complex': real, imag = h.helicsInputGetComplex(sub_id) val = [real, imag] elif in_put['type'] == 'vector': val = h.helicsInputGetVector(sub_id) elif in_put['type'] == 'boolean': val = h.helicsInputGetBoolean(sub_id) else: _log.error("Unknown datatype: {}".format(in_put['type'])) except h._helics.HelicsException as e: _log.exception("Error getting input from HELICS {}".format(e)) return val
def test_value_federate_runFederateTestVectorD(vFed): defaultValue = [0, 1, 2] testValue = [3, 4, 5] pubid = h.helicsFederateRegisterGlobalPublication(vFed, "pub1", h.helics_data_type_vector, "") subid = h.helicsFederateRegisterSubscription(vFed, "pub1", "") h.helicsInputSetDefaultVector(subid, defaultValue) h.helicsFederateEnterExecutingMode(vFed) h.helicsPublicationPublishVector(pubid, testValue) value = h.helicsInputGetVector(subid) assert value == [0, 1, 2] grantedtime = h.helicsFederateRequestTime(vFed, 1.0) assert grantedtime == 0.01 value = h.helicsInputGetVector(subid) assert value == [3, 4, 5]
def updateHelicsSubscriptions(self): for element_name, sub_info in self._subscriptions.items(): if 'Subscription' in sub_info: value = None if sub_info['Data type'].lower() == 'double': value = helics.helicsInputGetDouble( sub_info['Subscription']) elif sub_info['Data type'].lower() == 'vector': value = helics.helicsInputGetVector( sub_info['Subscription']) elif sub_info['Data type'].lower() == 'string': value = helics.helicsInputGetString( sub_info['Subscription']) elif sub_info['Data type'].lower() == 'boolean': value = helics.helicsInputGetBoolean( sub_info['Subscription']) elif sub_info['Data type'].lower() == 'integer': value = helics.helicsInputGetInteger( sub_info['Subscription']) if value and value != 0: value = value * sub_info['Multiplier'] dssElement = self._objects_by_element[element_name] dssElement.SetParameter(sub_info['Property'], value) self._logger.info( 'Value for "{}.{}" changed to "{}"'.format( element_name, sub_info['Property'], value * sub_info['Multiplier'])) if self._settings.helics.iterative_mode: if self.c_seconds != self.c_seconds_old: self._subscription_dState[element_name] = [ self.init_state ] * self.n_states else: self._subscription_dState[element_name].insert( 0, self._subscription_dState[element_name].pop()) self._subscription_dState[element_name][0] = value else: self._logger.warning( '{} will not be used to update element for "{}.{}" '. format(value, element_name, sub_info['Property'])) self.c_seconds_old = self.c_seconds
h.helicsFederateEnterExecutingMode(vfed) basevolt = 4.16 for t in range(1, 1440): time_requested = t * 60 #while time_requested < r_seconds: currenttime = h.helicsFederateRequestTime(vfed, time_requested) iteration_state = h.helics_iteration_result_iterating for i in range(20): currenttime, iteration_state = h.helicsFederateRequestTimeIterative( vfed, time_requested, h.helics_iteration_request_iterate_if_needed) print(iteration_state) h.helicsPublicationPublishDouble(pubA, basevolt + random.random() / 30) h.helicsPublicationPublishDouble(pubB, basevolt + random.random() / 30) h.helicsPublicationPublishDouble(pubC, basevolt + random.random() / 30) value = h.helicsInputGetVector(sub1) print("PyDSS.Circuit.heco19021.TotalPower: {} kW @ time: {}".format( value, currenttime)) #i+=1 h.helicsFederateFinalize(vfed) print("PI SENDER: Federate finalized") while h.helicsBrokerIsConnected(broker): time.sleep(1) h.helicsFederateFree(vfed) h.helicsCloseLibrary() print("PI SENDER: Broker disconnected")
def test_valuefederate_default_value_tests(): broker = createBroker() vFed1, fedinfo = createValueFederate(1, "fed0") inp_raw1 = h.helicsFederateRegisterInput(vFed1, "key1", h.HELICS_DATA_TYPE_RAW, "raw") inp_raw2 = h.helicsFederateRegisterInput(vFed1, "key2", h.HELICS_DATA_TYPE_RAW, "raw") inp_bool = h.helicsFederateRegisterInput(vFed1, "key3", h.HELICS_DATA_TYPE_BOOLEAN, "") inp_time = h.helicsFederateRegisterInput(vFed1, "key4", h.HELICS_DATA_TYPE_TIME, "") inp_char = h.helicsFederateRegisterInput(vFed1, "key5", h.HELICS_DATA_TYPE_STRING, "") inp_vect = h.helicsFederateRegisterInput(vFed1, "key6", h.HELICS_DATA_TYPE_VECTOR, "V") inp_double = h.helicsFederateRegisterInput(vFed1, "key7", h.HELICS_DATA_TYPE_DOUBLE, "kW") inp_double2 = h.helicsFederateRegisterInput(vFed1, "key8", h.HELICS_DATA_TYPE_DOUBLE, "") inp_np = h.helicsFederateRegisterInput(vFed1, "key9", h.HELICS_DATA_TYPE_NAMED_POINT, "") h.helicsInputSetMinimumChange(inp_double, 1100.0) h.helicsInputSetDefaultDouble(inp_double, 10000.0) h.helicsInputSetOption(inp_double2, h.HELICS_HANDLE_OPTION_CONNECTION_REQUIRED, True) pub = h.helicsFederateRegisterPublication(vFed1, "", h.HELICS_DATA_TYPE_INT, "MW") h.helicsPublicationSetOption(pub, h.HELICS_HANDLE_OPTION_CONNECTION_REQUIRED, True) h.helicsPublicationAddTarget(pub, "Testfed0/key7") h.helicsPublicationAddTarget(pub, "Testfed0/key8") h.helicsInputSetDefaultRaw(inp_raw1, "") data = "this is a string" h.helicsInputSetDefaultRaw(inp_raw2, data) h.helicsInputSetDefaultBoolean(inp_bool, True) h.helicsInputSetDefaultTime(inp_time, 12.3) h.helicsInputSetDefaultChar(inp_char, "q") h.helicsInputSetDefaultVector(inp_vect, []) h.helicsInputSetDefaultNamedPoint(inp_np, data, 15.7) h.helicsFederateEnterExecutingMode(vFed1) assert h.helicsInputGetInjectionUnits(inp_double) == "MW" assert h.helicsInputGetInjectionUnits(inp_double2) == "MW" assert h.helicsInputGetType(inp_double) == "double" assert h.helicsInputGetPublicationType(inp_double) == "int64" c2 = h.helicsInputGetChar(inp_char) assert c2 == "q" h.helicsInputGetVector(inp_vect) optset = h.helicsInputGetOption(inp_double2, h.HELICS_HANDLE_OPTION_CONNECTION_REQUIRED) assert optset == 1 optset = h.helicsPublicationGetOption(pub, h.HELICS_HANDLE_OPTION_CONNECTION_REQUIRED) assert optset == 1 h.helicsPublicationPublishInteger(pub, 12) h.helicsFederateRequestNextStep(vFed1) assert h.helicsInputGetDouble(inp_double) == 12000.0 assert h.helicsInputGetDouble(inp_double2) == 12.0 h.helicsPublicationPublishInteger(pub, 13) h.helicsFederateRequestNextStep(vFed1) assert h.helicsInputIsUpdated(inp_double) is False assert h.helicsInputIsUpdated(inp_double2) is True assert h.helicsInputGetDouble(inp_double) == 12000.0 assert h.helicsInputGetDouble(inp_double2) == 13.0 h.helicsPublicationPublishInteger(pub, 15) h.helicsFederateRequestNextStep(vFed1) assert h.helicsInputIsUpdated(inp_double) is True assert h.helicsInputIsUpdated(inp_double2) is True h.helicsInputClearUpdate(inp_double) h.helicsInputClearUpdate(inp_double2) assert h.helicsInputIsUpdated(inp_double) is False assert h.helicsInputIsUpdated(inp_double2) is False _, rval = h.helicsInputGetNamedPoint(inp_np) assert rval == 15.7 out, rval = h.helicsInputGetNamedPoint(inp_np) assert out == "this is a string" assert rval == 15.7 h.helicsFederateFinalize(vFed1) destroyFederate(vFed1, fedinfo) destroyBroker(broker)