PipelineContext.h

Go to the documentation of this file.

// Copyright (c) 2018 the Talvos developers. All rights reserved.
//
// This file is distributed under a three-clause BSD license. For full license
// terms please see the LICENSE file distributed with this source code.


#ifndef TALVOS_PIPELINECONTEXT_H
#define TALVOS_PIPELINECONTEXT_H

#include <map>
#include <vector>

#include "vulkan/vulkan_core.h"

namespace talvos
{

class ComputePipeline;
class GraphicsPipeline;

typedef std::map<uint32_t, uint64_t> VertexBindingMap;

struct BindingInfo
{
  uint64_t Address;  
  uint64_t NumBytes; 
};

typedef std::map<std::pair<uint32_t, uint32_t>, BindingInfo> DescriptorSet;

typedef std::map<uint32_t, talvos::DescriptorSet> DescriptorSetMap;

class PipelineContext
{
public:
  static const uint32_t PUSH_CONSTANT_MEM_SIZE = 128;

  void bindComputeDescriptors(const DescriptorSetMap &DSM);

  void bindComputePipeline(const ComputePipeline *PL);

  void bindGraphicsDescriptors(const DescriptorSetMap &DSM);

  void bindGraphicsPipeline(const GraphicsPipeline *PL);

  void bindVertexBuffer(uint32_t Binding, uint64_t Address);

  void clear();

  const DescriptorSetMap &getComputeDescriptors() const { return ComputeDSM; }

  const ComputePipeline *getComputePipeline() const { return ComputePL; }

  const DescriptorSetMap &getGraphicsDescriptors() const { return GraphicsDSM; }

  const GraphicsPipeline *getGraphicsPipeline() const { return GraphicsPL; }

  const uint8_t *getPushConstantData() const { return PushConstantData; }

  const std::vector<VkRect2D> &getScissors() const { return Scissors; }

  const VertexBindingMap &getVertexBindings() const { return VertexBindings; }

  const std::vector<VkViewport> &getViewports() const { return Viewports; }

  void setPushConstantData(uint32_t Offset, uint32_t NumBytes,
                           const uint8_t *Data);

private:
  const ComputePipeline *ComputePL = nullptr;   
  const GraphicsPipeline *GraphicsPL = nullptr; 

  DescriptorSetMap ComputeDSM;  
  DescriptorSetMap GraphicsDSM; 

  uint8_t PushConstantData[PUSH_CONSTANT_MEM_SIZE]; 

  std::vector<VkViewport> Viewports; 

  std::vector<VkRect2D> Scissors; 

  VertexBindingMap VertexBindings; 
};

}; // namespace talvos

#endif