DtEntityEmitterVolumeVisualizer.h

Go to the documentation of this file.

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


#pragma once

#include <vrvCore/DtStateVisualizer.h>

#include <vrvCore/DtUniqueID.h>

#include <matrix/vlVector.h>
#include <vlutil/vlExceptions.h>

#include <unordered_map>

namespace makVrv
{

   class DtModelAgent;
   class DtEntityStateListener;
   class DtEmitterBeamStateListener;

   class DT_DLL_VRVCORE DtEntityEmitterVolumeVisualizer : 
      public DtStateVisualizer
   {
   public:

      enum ColoringMode
      {
         PulseFrequency,
         Frequency,
         Fixed
      };

      DtEntityEmitterVolumeVisualizer( DtBaseConnection& simulation, DtStateListener& listener,
         DtSceneObjectAgent* sceneObject, DtModelSetId modelSet );

      virtual ~DtEntityEmitterVolumeVisualizer();

      DtEntityEmitterVolumeVisualizer() = delete;

      DtEntityEmitterVolumeVisualizer( const DtEntityEmitterVolumeVisualizer& ) = delete;

      DtEntityEmitterVolumeVisualizer& operator=( const DtEntityEmitterVolumeVisualizer& ) = delete; 

   public:

      virtual void initialize() override;

      virtual void createModelAgents() override;

      virtual void destroyModelAgents() override;

      virtual void setModelDefinitionForAgents() override;

      virtual void slot_emitterBeamDiscovered( const DtUniqueID& id );

      virtual void slot_emitterBeamToBeRemoved( const DtUniqueID& id );

      virtual void setColoringMode( ColoringMode method );

      virtual const ColoringMode coloringMode() const;

   protected:

      virtual DtEntityStateListener& entityStateListener();
      virtual const DtEntityStateListener& entityStateListener() const;

      virtual DtModelAgent* createBeamModel();

      virtual DtModelAgent* findBeamModel( const DtUniqueID& beamModelId );

      virtual bool beamIsVisible( float elevationSweep, float azimuthSweep, float emittedPower);

      virtual void setBeamModelVisible( DtModelAgent* beamModel, bool visibility );

      virtual void slot_entityEmitterVolumesVisibleChanged(bool visible, bool instanceOnly);

      virtual void slot_setBeamlRelativePosition( const DtUniqueID& beamId, DtVector32 offset );
      virtual void slot_setBeamAzimuthRange( const DtUniqueID& beamId, float center, float sweep );

      virtual void slot_setBeamElevationRange( const DtUniqueID& beamId, float center, float sweep );
      virtual void slot_setBeamERP( const DtUniqueID& beamId, float erp );
      virtual void slot_setBeamFrequency( const DtUniqueID& beamId, float freq );
      virtual void slot_setBeamPulseWidth( const DtUniqueID& beamId, float width );


      virtual void setBeamModelRelativePosition( DtModelAgent* beamModel, DtVector32 offset) = 0;
      virtual void setBeamModelAzimuthRange( DtModelAgent* beamModel, float center, float sweep ) = 0;
      virtual void setBeamModelElevationRange( DtModelAgent* beamModel, float center, float sweep ) = 0;
      virtual void setBeamModelScale( DtModelAgent* beamModel, double scale) = 0;
      virtual void updateBeamModelColors( DtModelAgent* beamModel, float freq, float prf);
      virtual void setBeamInsideColor( DtModelAgent* beamModel, float r, float g, float b, float a ) = 0;
      virtual void setBeamOutsideColor( DtModelAgent* beamModel, float r, float g, float b, float a ) = 0;

   protected:
      virtual void setInitialValuesFromVisualizerDefinition() override;

      virtual void connectToListenerSignals() override;

      virtual double calculateBeamScale( float elevationRangeSweep, float azimuthRangeSweep, float erPower );

      virtual void computeColorForFrequency( double hi, double low,
         double act, float innerColor[4], float outerColor[4] );

      static void hsv2rgb( float hue, float sat, float val, float color[3] );

      virtual void destroyEmitterBeam( const DtUniqueID& );
      virtual void destroyEmitterBeamModel( DtModelAgent* );
   protected:
      ColoringMode myColoringMode;
      DtEntityStateListener* myEntityStateListener;

      using EmitterBeamModelMap = std::unordered_map<DtUniqueID, DtModelAgent*>;
      EmitterBeamModelMap myEmitterBeamModelMap;

      double myLinearScale;
      double myExponentialScale;

      double myHighPulseFrequency;
      double myLowPulseFrequency;
      double myHighFrequency;
      double myLowFrequency;

      float myNearColor[4];
      float myFarColor[4];

      bool myUseCorrectedBeamScaleFormula;
   };
}