Workgroup.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_WORKGROUP_H
#define TALVOS_WORKGROUP_H

#include <memory>
#include <vector>

#include "talvos/Dim3.h"

namespace talvos
{

class Device;
class Invocation;
class Memory;
class Object;
class PipelineExecutor;

class Workgroup
{
public:
  typedef std::vector<std::unique_ptr<Invocation>> WorkItemList;

  typedef std::vector<std::pair<uint32_t, Object>> VariableList;

  Workgroup(Device &Dev, const PipelineExecutor &Executor, Dim3 GroupId);

  ~Workgroup();

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

  void addWorkItem(std::unique_ptr<Invocation> WorkItem);

  Dim3 getGroupId() const { return GroupId; }

  Memory &getLocalMemory() { return *LocalMemory; }

  const WorkItemList &getWorkItems() const { return WorkItems; }

  const VariableList &getVariables() const { return Variables; }

private:
  Dim3 GroupId; 

  Memory *LocalMemory; 

  WorkItemList WorkItems; 

  VariableList Variables; 
};

} // namespace talvos

#endif