vrfEmbarkationHierarchy.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2009 MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/
/******************************************************************************
** $RCSfile: DtVrfEmbarkationHierarchy.h,v $ $Revision: 1.3 $ $State: Exp $
******************************************************************************/



#pragma once

#include <vrfVrlSharedSource/vrfDriver.h>

#include <vrvCore/DtElementIDHierarchy.h>
#include <vrvUtil/DtVirtualBaseClass.h>
#include <vrfGuiCore/embarkationHierarchyAttributes.h>

namespace makVrv
{
   class DtConnection;
}

namespace makVrf
{
   namespace VRF_PROTOCOL_NAMESPACE
   {
      class VRF_PROTOCOL_EXPORT DtVrfEmbarkationHierarchy : public makVrv::DtElementIDHierarchy, public makVrv::DtVirtualBaseClass
      {
      public:
         enum EmbarkationFlags
         {
            Environmental = 0x00000001
         };

      public:
         static DtVrfEmbarkationHierarchy& instance(makVrv::DtConnection& sim);

         static void registerInstance(makVrv::DtConnection& sim, DtVrfEmbarkationHierarchy*);

         virtual ~DtVrfEmbarkationHierarchy();

      public:

         makVrv::DtElementID attachedTo(makVrv::DtElementID id) const
         {
            HierarchyNodeMap::const_iterator iter = myFlatHierarchy.find(id);

            if ((iter != myFlatHierarchy.end()) && (iter->second->parent()))
            {
               return iter->second->parent()->elementId();
            }

            return 0;
         }

         bool hasAttachedEnvironmentals(makVrv::DtElementID id) const
         {
            HierarchyNodeMap::const_iterator iter = myFlatHierarchy.find(id);

            if (iter != myFlatHierarchy.end())
            {
               makVrv::DtElementIDHierarchyNode::ChildHierarchyNodes::const_iterator children = iter->second->children().begin();

               while (children != iter->second->children().end())
               {
                  if (((*children)->flags() & Environmental) != 0)
                  {
                     return true;
                  }

                  ++children;
               }
            }

            return false;
         }

         std::vector<makVrv::DtElementID> attachedEnvironmentals(makVrv::DtElementID id) const
         {
            std::vector<makVrv::DtElementID> envs;
            HierarchyNodeMap::const_iterator iter = myFlatHierarchy.find(id);

            if (iter != myFlatHierarchy.end())
            {
               makVrv::DtElementIDHierarchyNode::ChildHierarchyNodes::const_iterator children = iter->second->children().begin();

               while (children != iter->second->children().end())
               {
                  if (((*children)->flags() & Environmental) != 0)
                  {
                     envs.push_back((*children)->elementId());
                  }

                  ++children;
               }
            }

            return envs;
         }


      protected:
         DtVrfEmbarkationHierarchy();
      };
   }
}