registerExtendedAttributeClassesDIS.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2017 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/



#pragma once

#ifdef DtDIS

#include <vrfExtObjects/stateRepositoryWithExtendedAttributesDIS.h>
#include <vrfExtObjects/extendedAttributeVariableRecordDIS.h>
#include <vrfExtObjects/vrfUnknownVariableRecordDIS.h>
#include <vl/aggregatePublisher.h>

#define DtALMOST(x, y, e) (((x)<=((y)+(e))) && ((x)>=((y)-(e))))

namespace makVrfVrlinkExtToolkit
{
   template <typename StateRepType, typename PublisherType, typename ReflectedObjType>
   void registerExtendedAttributeClasses(DtExerciseConn* exConn)
   {
      // Do not reregister if already done. Need to check if this is the same exConn as last time
      // to handle the case of disconnect/reconnect. Unfortunately sometimes the memory address is
      // reused, so we can't simply check that. Instead use the exConn's clock to check the time
      // the exConn was initialized.
      static DtTime prevExConnInitTime = -1.0;
      DtTime exConnInitTime = exConn->clock()->absRealTime() - exConn->clock()->elapsedRealTime();
      if (DtALMOST(prevExConnInitTime, exConnInitTime, 0.0001))
      {
         // Already registered
         return;
      }

      // Set the state repository creators
      PublisherType::setStateRepCreator((typename PublisherType::DtStateRepCreator)
         StateRepType::create);
      ReflectedObjType::setStateRepCreator((typename ReflectedObjType::DtStateRepCreator)
         StateRepType::create);
      
      // Set the variable record factory
      if (!dynamic_cast<DtVrfVariableRecordFactory*>(exConn->pduFactory()->standardVariableRecordFactory()))
      {
         // Already registered
         DtStandardVariableRecordFactory* oldFactory =
            exConn->pduFactory()->setStandardVariableRecordFactory(new DtVrfVariableRecordFactory);
         delete oldFactory;
      }

      prevExConnInitTime = exConnInitTime;
   }
}

#endif