1 // Copyright (C) 2009 - 2012 Mathias Froehlich - Mathias.Froehlich@web.de
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Library General Public
5 // License as published by the Free Software Foundation; either
6 // version 2 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Library General Public License for more details.
13 // You should have received a copy of the GNU General Public License
14 // along with this program; if not, write to the Free Software
15 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19 # include <simgear_config.h>
22 #include <simgear/compiler.h>
24 #include "HLAObjectClass.hxx"
26 #include "simgear/debug/logstream.hxx"
27 #include "RTIFederate.hxx"
28 #include "RTIObjectClass.hxx"
29 #include "RTIObjectInstance.hxx"
30 #include "HLADataType.hxx"
31 #include "HLADataTypeVisitor.hxx"
32 #include "HLAFederate.hxx"
33 #include "HLAObjectInstance.hxx"
37 HLAObjectClass::InstanceCallback::~InstanceCallback()
42 HLAObjectClass::InstanceCallback::discoverInstance(const HLAObjectClass&, HLAObjectInstance& objectInstance, const RTIData& tag)
47 HLAObjectClass::InstanceCallback::removeInstance(const HLAObjectClass&, HLAObjectInstance& objectInstance, const RTIData& tag)
52 HLAObjectClass::InstanceCallback::registerInstance(const HLAObjectClass&, HLAObjectInstance& objectInstance)
57 HLAObjectClass::InstanceCallback::deleteInstance(const HLAObjectClass&, HLAObjectInstance& objectInstance)
61 HLAObjectClass::RegistrationCallback::~RegistrationCallback()
65 HLAObjectClass::HLAObjectClass(const std::string& name, HLAFederate* federate) :
70 SG_LOG(SG_NETWORK, SG_ALERT, "HLAObjectClass::HLAObjectClass(): "
71 "No parent federate given for object class \"" << getName() << "\"!");
74 federate->_insertObjectClass(this);
77 HLAObjectClass::~HLAObjectClass()
79 // HLAObjectClass objects only get deleted when the parent federate
80 // dies. So we do not need to deregister there.
82 _clearRTIObjectClass();
86 HLAObjectClass::getName() const
91 const SGWeakPtr<HLAFederate>&
92 HLAObjectClass::getFederate() const
98 HLAObjectClass::getNumAttributes() const
100 return _attributeVector.size();
104 HLAObjectClass::addAttribute(const std::string& name)
106 unsigned index = _attributeVector.size();
107 _nameIndexMap[name] = index;
108 _attributeVector.push_back(Attribute(name));
109 _resolveAttributeIndex(name, index);
114 HLAObjectClass::getAttributeIndex(const std::string& name) const
116 NameIndexMap::const_iterator i = _nameIndexMap.find(name);
117 if (i == _nameIndexMap.end())
123 HLAObjectClass::getAttributeName(unsigned index) const
125 if (_attributeVector.size() <= index)
126 return std::string();
127 return _attributeVector[index]._name;
131 HLAObjectClass::getAttributeDataType(unsigned index) const
133 if (_attributeVector.size() <= index)
135 return _attributeVector[index]._dataType.get();
139 HLAObjectClass::setAttributeDataType(unsigned index, const HLADataType* dataType)
141 if (_attributeVector.size() <= index)
143 _attributeVector[index]._dataType = dataType;
147 HLAObjectClass::getAttributeUpdateType(unsigned index) const
149 if (_attributeVector.size() <= index)
150 return HLAUndefinedUpdate;
151 return _attributeVector[index]._updateType;
155 HLAObjectClass::setAttributeUpdateType(unsigned index, HLAUpdateType updateType)
157 if (_attributeVector.size() <= index)
159 _attributeVector[index]._updateType = updateType;
163 HLAObjectClass::getAttributeSubscriptionType(unsigned index) const
165 if (_attributeVector.size() <= index)
166 return HLAUnsubscribed;
167 return _attributeVector[index]._subscriptionType;
171 HLAObjectClass::setAttributeSubscriptionType(unsigned index, HLASubscriptionType subscriptionType)
173 if (_attributeVector.size() <= index)
175 _attributeVector[index]._subscriptionType = subscriptionType;
179 HLAObjectClass::getAttributePublicationType(unsigned index) const
181 if (_attributeVector.size() <= index)
182 return HLAUnpublished;
183 return _attributeVector[index]._publicationType;
187 HLAObjectClass::setAttributePublicationType(unsigned index, HLAPublicationType publicationType)
189 if (_attributeVector.size() <= index)
191 _attributeVector[index]._publicationType = publicationType;
194 HLADataElement::IndexPathPair
195 HLAObjectClass::getIndexPathPair(const HLADataElement::StringPathPair& stringPathPair) const
197 unsigned index = getAttributeIndex(stringPathPair.first);
198 if (getNumAttributes() <= index) {
199 SG_LOG(SG_NETWORK, SG_ALERT, "HLAObjectClass::getIndexPathPair(\""
200 << HLADataElement::toString(stringPathPair)
201 << "\"): Could not resolve attribute \"" << stringPathPair.first
202 << "\" for object class \"" << getName() << "\"!");
204 return HLADataElement::IndexPathPair(index, stringPathPair.second);
207 HLADataElement::IndexPathPair
208 HLAObjectClass::getIndexPathPair(const std::string& path) const
210 return getIndexPathPair(HLADataElement::toStringPathPair(path));
214 HLAObjectClass::subscribe()
216 if (!_rtiObjectClass.valid()) {
217 SG_LOG(SG_NETWORK, SG_WARN, "HLAObjectClass::subscribe(): "
218 "No RTIObject class for object class \"" << getName() << "\"!");
222 HLAIndexList indexList;
223 for (unsigned i = 1; i < getNumAttributes(); ++i) {
224 if (_attributeVector[i]._subscriptionType != HLASubscribedActive)
226 indexList.push_back(i);
228 if (!indexList.empty()) {
229 if (!_rtiObjectClass->subscribe(indexList, true))
234 for (unsigned i = 1; i < getNumAttributes(); ++i) {
235 if (_attributeVector[i]._subscriptionType != HLASubscribedPassive)
237 indexList.push_back(i);
239 if (!indexList.empty()) {
240 if (!_rtiObjectClass->subscribe(indexList, false))
247 HLAObjectClass::unsubscribe()
249 if (!_rtiObjectClass.valid()) {
250 SG_LOG(SG_NETWORK, SG_WARN, "HLAObjectClass::unsubscribe(): "
251 "No RTIObject class for object class \"" << getName() << "\"!");
254 return _rtiObjectClass->unsubscribe();
258 HLAObjectClass::publish()
260 if (!_rtiObjectClass.valid()) {
261 SG_LOG(SG_NETWORK, SG_WARN, "HLAObjectClass::publish(): "
262 "No RTIObject class for object class \"" << getName() << "\"!");
266 HLAIndexList indexList;
267 for (unsigned i = 1; i < getNumAttributes(); ++i) {
268 if (_attributeVector[i]._publicationType == HLAUnpublished)
270 indexList.push_back(i);
272 if (indexList.empty())
274 if (!_rtiObjectClass->publish(indexList))
280 HLAObjectClass::unpublish()
282 if (!_rtiObjectClass.valid()) {
283 SG_LOG(SG_NETWORK, SG_WARN, "HLAObjectClass::unpublish(): "
284 "No RTIObject class for object class \"" << getName() << "\"!");
287 return _rtiObjectClass->unpublish();
291 HLAObjectClass::discoverInstance(HLAObjectInstance& objectInstance, const RTIData& tag)
293 if (_instanceCallback.valid())
294 _instanceCallback->discoverInstance(*this, objectInstance, tag);
298 HLAObjectClass::removeInstance(HLAObjectInstance& objectInstance, const RTIData& tag)
300 if (_instanceCallback.valid())
301 _instanceCallback->removeInstance(*this, objectInstance, tag);
305 HLAObjectClass::registerInstance(HLAObjectInstance& objectInstance)
307 if (_instanceCallback.valid())
308 _instanceCallback->registerInstance(*this, objectInstance);
312 HLAObjectClass::deleteInstance(HLAObjectInstance& objectInstance)
314 if (_instanceCallback.valid())
315 _instanceCallback->deleteInstance(*this, objectInstance);
319 HLAObjectClass::startRegistration() const
324 HLAObjectClass::stopRegistration() const
329 HLAObjectClass::createObjectInstance(const std::string& name)
331 SGSharedPtr<HLAFederate> federate = _federate.lock();
332 if (!federate.valid())
334 return federate->createObjectInstance(this, name);
338 HLAObjectClass::createAttributeDataElements(HLAObjectInstance& objectInstance)
340 unsigned numAttributes = getNumAttributes();
341 for (unsigned i = 0; i < numAttributes; ++i)
342 objectInstance.createAndSetAttributeDataElement(i);
346 HLAObjectClass::createAttributeDataElement(HLAObjectInstance& objectInstance, unsigned index)
348 // FIXME here we want to have a vector of factories and if this fails do the following
349 const HLADataType* dataType = getAttributeDataType(index);
352 HLADataElementFactoryVisitor dataElementFactoryVisitor;
353 dataType->accept(dataElementFactoryVisitor);
354 return dataElementFactoryVisitor.getDataElement();
358 HLAObjectClass::_setRTIObjectClass(RTIObjectClass* objectClass)
360 if (_rtiObjectClass) {
361 SG_LOG(SG_NETWORK, SG_ALERT, "HLAObjectClass: Setting RTIObjectClass twice for object class \"" << getName() << "\"!");
364 _rtiObjectClass = objectClass;
365 if (_rtiObjectClass->_objectClass != this) {
366 SG_LOG(SG_NETWORK, SG_ALERT, "HLAObjectClass: backward reference does not match!");
369 for (unsigned i = 0; i < _attributeVector.size(); ++i)
370 _resolveAttributeIndex(_attributeVector[i]._name, i);
374 HLAObjectClass::_resolveAttributeIndex(const std::string& name, unsigned index)
376 if (!_rtiObjectClass)
378 if (!_rtiObjectClass->resolveAttributeIndex(name, index))
379 SG_LOG(SG_NETWORK, SG_ALERT, "HLAObjectClass: Could not resolve attribute \""
380 << name << "\" for object class \"" << getName() << "\"!");
384 HLAObjectClass::_clearRTIObjectClass()
386 if (!_rtiObjectClass.valid())
388 _rtiObjectClass->_objectClass = 0;
393 HLAObjectClass::_discoverInstance(RTIObjectInstance* rtiObjectInstance, const RTIData& tag)
395 SGSharedPtr<HLAFederate> federate = _federate.lock();
396 if (!federate.valid()) {
397 SG_LOG(SG_NETWORK, SG_ALERT, "RTI: could not find parent federate while discovering object instance");
401 SGSharedPtr<HLAObjectInstance> objectInstance = createObjectInstance(rtiObjectInstance->getName());
402 if (!objectInstance.valid()) {
403 SG_LOG(SG_NETWORK, SG_INFO, "RTI: could not create new object instance for discovered \""
404 << rtiObjectInstance->getName() << "\" object");
407 SG_LOG(SG_NETWORK, SG_INFO, "RTI: create new object instance for discovered \""
408 << rtiObjectInstance->getName() << "\" object");
409 objectInstance->_setRTIObjectInstance(rtiObjectInstance);
410 if (!federate->_insertObjectInstance(objectInstance)) {
411 SG_LOG(SG_NETWORK, SG_ALERT, "RTI: could not insert new object instance for discovered \""
412 << rtiObjectInstance->getName() << "\" object");
415 objectInstance->discoverInstance(tag);
416 objectInstance->createAttributeDataElements();
420 HLAObjectClass::_removeInstance(HLAObjectInstance& objectInstance, const RTIData& tag)
422 SGSharedPtr<HLAFederate> federate = _federate.lock();
423 if (!federate.valid()) {
424 SG_LOG(SG_NETWORK, SG_ALERT, "RTI: could not find parent federate while removing object instance");
427 SG_LOG(SG_NETWORK, SG_INFO, "RTI: remove object instance \"" << objectInstance.getName() << "\"");
428 objectInstance.removeInstance(tag);
429 federate->_eraseObjectInstance(objectInstance.getName());
433 HLAObjectClass::_registerInstance(HLAObjectInstance* objectInstance)
435 SGSharedPtr<HLAFederate> federate = _federate.lock();
436 if (!federate.valid()) {
437 SG_LOG(SG_NETWORK, SG_ALERT, "RTI: could not find parent federate while registering object instance");
442 // We can only register object instances with a valid name at the rti.
443 // So, we cannot do that at HLAObjectInstance creation time.
444 if (!federate->_insertObjectInstance(objectInstance)) {
445 SG_LOG(SG_NETWORK, SG_ALERT, "RTI: could not insert new object instance \""
446 << objectInstance->getName() << "\" object");
449 registerInstance(*objectInstance);
450 objectInstance->createAttributeDataElements();
454 HLAObjectClass::_deleteInstance(HLAObjectInstance& objectInstance)
456 SGSharedPtr<HLAFederate> federate = _federate.lock();
457 if (!federate.valid()) {
458 SG_LOG(SG_NETWORK, SG_ALERT, "RTI: could not find parent federate while deleting object instance");
461 deleteInstance(objectInstance);
462 federate->_eraseObjectInstance(objectInstance.getName());
466 HLAObjectClass::_startRegistration()
468 if (_registrationCallback.valid())
469 _registrationCallback->startRegistration(*this);
475 HLAObjectClass::_stopRegistration()
477 if (_registrationCallback.valid())
478 _registrationCallback->stopRegistration(*this);
483 } // namespace simgear