DtVrlinkEllipseVisualizers.h

Go to the documentation of this file.

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


#pragma once

#include <vrvVrl/vrvVrl.h>
#include <vrvCore/Dt2DEllipseVisualizer.h>
#include <vrvCore/Dt3DEllipseVisualizer.h>

#include <vrvUtil/DtCoordinateSystem.h>
#include <vrvVrl/DtVrlinkConnection.h>

#include <matrix/topoCoord.h>

namespace makVrv
{
   namespace DT_PROTOCOL_NAMESPACE
   {
      template <typename T>
      class DtVrlinkEllipseVisualizer : public T
      {
      public:
         DtVrlinkEllipseVisualizer(DtBaseConnection& simulation, DtStateListener& listener,
            int modelSet)
         : T(simulation, listener, modelSet)
         , myVrlinkConnection((dynamic_cast<DtVrlinkConnection&>(simulation)))
         {
         }

         virtual ~DtVrlinkEllipseVisualizer()
         {
         }

         void setPosition(int index, const DtVector& v)
         {
            DtVector local;

            myVrlinkConnection.coordinateSystem().netToLocalPos(v, local);

            T::setPosition(index, local);
         }

         void setColor(float r, float g, float b, float a)
         {
            if ((r == 0) && (g == 0) && (b == 0))
            {
               this->useDefaultColor();
            }
            else
            {
               T::setColor(r, g, b, a);
            }
         }

      protected:
         DtVrlinkConnection& myVrlinkConnection;
      };

      // @{
      typedef DtStateVisualizerCreatorTemplate<DtVrlinkEllipseVisualizer<Dt3DEllipseVisualizer> >
         DtVrlink3DEllipseVisualizerCreator;
      typedef DtStateVisualizerCreatorTemplate<DtVrlinkEllipseVisualizer<Dt2DEllipseVisualizer> >
         DtVrlink2DEllipseVisualizerCreator;
      // @}
   }
}