TwinCadModelSystem.h

#pragma once
 
#include "Rendering/Common.h"
 
#include "Resources/Resources.h"
#include "Resources/Material.h"
 
#include "Scene/RayPick.h"
 
#include "Systems/ComponentSystem.h"
 
#include "TwinCadModelComponent.h"
#include "TwinVisualsSystem.h"
 
namespace Cogs::Core {
  class RenderSystem;
  struct TwinCadModelSystem;
 
  struct TwinCadModelRayPickExtension : MeshRayPickableBase
  {
    TwinCadModelRayPickExtension(RenderSystem* renderSystem, TwinCadModelSystem* cadModelSystem);
 
  private:
    bool pickImpl(Context* context,
                  const glm::mat4& worldPickMatrix,
                  const glm::mat4& rawViewProjection,
                  const glm::mat4& viewMatrix,
                  const RayPicking::RayPickFilter& filter,
                  PickingFlags pickingFlags,
                  PicksReturned returnFlag,
                  std::vector<RayPicking::RayPickHit>& hits) override;
 
    RenderSystem* renderSystem = nullptr;
    TwinCadModelSystem* cadModelSystem = nullptr;
  };
 
  struct TwinCadModelData
  {
    MaterialInstanceHandle materialInstance;
    TextureHandle attributeMap;
    std::vector<uint8_t> attributeMapData;
 
    std::string substance;
    TwinVisualsTextureWrapper baseColor;
    TwinVisualsTextureWrapper normal;
    TwinVisualsTextureWrapper orm;
 
    uint8_t attributeMapGen = 0;
    bool needUpdate = false;
    bool firstRun = true;
  };
 
  struct TwinCadModelSystem : public ComponentSystemWithDataPool<TwinCadModelComponent, TwinCadModelData>
  {
  public:
    TwinCadModelSystem(Memory::Allocator* allocator, SizeType capacity) : ComponentSystemWithDataPool(allocator, capacity) {}
 
    void preUpdate(Context * context) override;
 
    void update(Context* context) override;
 
    ComponentHandle createComponent() override;
 
    void destroyComponent(ComponentHandle handle) override;
 
  private:
    MaterialHandle material;
 
    VariableKey attributeMapKey = NoProperty;
    VariableKey emissiveColorKey = NoProperty;
    VariableKey diffuseColorKey = NoProperty;
    VariableKey specularColorKey = NoProperty;
    VariableKey specularPowerKey = NoProperty;
    VariableKey metallicKey = NoProperty;
    VariableKey roughnessKey = NoProperty;
    VariableKey substanceParamsKey = NoProperty;
    VariableKey parametersKey = NoProperty;
    VariableKey clippingKey = NoProperty;
    VariableKey minClippingKey = NoProperty;
    VariableKey maxClippingKey = NoProperty;
    VariableKey hasAttributeDataKey = NoProperty;
    VariableKey surfaceVarianceTextureKey = NoProperty;
    VariableKey baseColorMapKey = NoProperty;
    VariableKey normalMapKey = NoProperty;
    VariableKey occlusionRoughnessMetallicMapKey = NoProperty;
 
    size_t allowClippingVariantIndex = Material::NoVariantIndex;
    size_t customEffectVariantIndex = Material::NoVariantIndex;
    size_t selectionEffectVariantIndex = Material::NoVariantIndex;
    size_t surfaceVarianceVariantIndex = Material::NoVariantIndex;
    size_t lightModelVariantIndex = Material::NoVariantIndex;
    size_t numSubstanceDefinitionMapsVariantIndex = Material::NoVariantIndex;
 
    TwinCadModelRayPickExtension* rayPickExtension = nullptr;
    bool componentsCreated = false; // Set to true each time a component has been created, handled and cleared in preUpdate which does entity init.
  };
}