SoComputeShaderScheduler.h

/*==============================================================================
**    THE CONTENT OF THIS WORK IS PROPRIETARY TO FEI S.A.S, (FEI S.A.S.),     **
**              AND IS DISTRIBUTED UNDER A LICENSE AGREEMENT.                 **
**                                                                            **
**  REPRODUCTION, DISCLOSURE,  OR USE,  IN WHOLE OR IN PART,  OTHER THAN AS   **
**  SPECIFIED  IN THE LICENSE ARE  NOT TO BE  UNDERTAKEN  EXCEPT WITH PRIOR   **
**  WRITTEN AUTHORIZATION OF FEI S.A.S.                                       **
**                                                                            **
**                        RESTRICTED RIGHTS LEGEND                            **
**  USE, DUPLICATION, OR DISCLOSURE BY THE GOVERNMENT OF THE CONTENT OF THIS  **
**  WORK OR RELATED DOCUMENTATION IS SUBJECT TO RESTRICTIONS AS SET FORTH IN  **
**  SUBPARAGRAPH (C)(1) OF THE COMMERCIAL COMPUTER SOFTWARE RESTRICTED RIGHT  **
**  CLAUSE  AT FAR 52.227-19  OR SUBPARAGRAPH  (C)(1)(II)  OF  THE RIGHTS IN  **
**  TECHNICAL DATA AND COMPUTER SOFTWARE CLAUSE AT DFARS 52.227-7013.         **
**                                                                            **
**                   COPYRIGHT (C) 1996-2025 BY FEI S.A.S,                    **
**                              BORDEAUX, FRANCE                              **
**                            ALL RIGHTS RESERVED                             **
==============================================================================*/
 
#pragma once
 
#include <Inventor/nodes/SoNode.h>
#include <Inventor/fields/SoSFEnum.h>
#include <Inventor/fields/SoSFVec3i32.h>
#include <Inventor/fields/SoSFTrigger.h>
#include <Inventor/renderer/RendererResourceMacro.h>
 
class SoGLRenderAction;
class SoNotList;

class  SoComputeShaderScheduler : public SoNode
{
  SO_NODE_HEADER(SoComputeShaderScheduler);
  RENDERER_RESOURCE(SoComputeShaderScheduler);
 
public:
  enum Policy
  {
    EACH_FRAME,

    ON_TRIGGER,
  };

  SoSFVec3i32 numWorkGroups;

  SoSFEnum policy;

  SoSFTrigger trigger;
 
  SoComputeShaderScheduler();

  virtual SbBool affectsState() const
  {
    return FALSE;
  }
 
private:
  virtual void GLRender(SoGLRenderAction* action);
 
private:
  static void initClass();
  static void exitClass();

  SoSFVec3i32 workGroupSize;
 
  virtual void fieldHasChanged(SoField* field);
  virtual void notify(SoNotList* list);
 
  virtual bool shouldCompute(SoGLRenderAction* action);

  static void computeWorkGroups(SoState* state, const SbVec3i32& totalComputeSize, SbVec3i32& numWorkGroups, SbVec3i32& workGroupSize);
 
private:
  virtual ~SoComputeShaderScheduler();
 
private:
  bool m_triggerCalled;
};