PipelineStage.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_PIPELINESTAGE_H
#define TALVOS_PIPELINESTAGE_H

#include <map>
#include <memory>
#include <vector>

#include "talvos/Dim3.h"
#include "talvos/Object.h"

namespace talvos
{

class Device;
class EntryPoint;
class Module;

typedef std::map<uint32_t, Object> SpecConstantMap;

class PipelineStage
{
public:
  PipelineStage(Device &D, std::shared_ptr<const Module> M,
                const EntryPoint *EP, const SpecConstantMap &SM = {});

  // Do not allow PipelineStage objects to be copied.
  PipelineStage(const PipelineStage &) = delete;
  PipelineStage &operator=(const PipelineStage &) = delete;

  const EntryPoint *getEntryPoint() const { return EP; }

  Dim3 getGroupSize() const { return GroupSize; }

  std::shared_ptr<const Module> getModule() const { return Mod; }

  const std::vector<Object> &getObjects() const { return Objects; };

private:
  std::shared_ptr<const Module> Mod;

  const EntryPoint *EP; 
  Dim3 GroupSize;       

  std::vector<Object> Objects;
};

} // namespace talvos

#endif