drivers/msd-vsi-vip/src/mapped_batch.h

// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MAPPED_BATCH_H
#define MAPPED_BATCH_H

#include <lib/magma/platform/platform_buffer.h>
#include <lib/magma/platform/platform_semaphore.h>
#include <lib/magma_service/util/mapped_batch.h>

#include "gpu_mapping.h"

class MsdVsiContext;

using MappedBatch = magma::MappedBatch<MsdVsiContext, MsdVsiBuffer>;

// Has no batch.
class NullBatch : public MappedBatch {
 public:
  uint64_t GetGpuAddress() const override { return 0; }
  uint64_t GetLength() const override { return 0; }
  void SetSequenceNumber(uint32_t sequence_number) override { seq_num_ = sequence_number; }
  uint32_t GetSequenceNumber() const override { return seq_num_; }
  const magma::GpuMappingView<MsdVsiBuffer>* GetBatchMapping() const override { return nullptr; }
  std::weak_ptr<MsdVsiContext> GetContext() const override { return {}; }

  uint32_t seq_num_ = 0;
};

// Signals the semaphores when destroyed.
class EventBatch : public NullBatch {
 public:
  EventBatch(std::shared_ptr<MsdVsiContext> context,
             std::vector<std::shared_ptr<magma::PlatformSemaphore>> wait_semaphores,
             std::vector<std::shared_ptr<magma::PlatformSemaphore>> signal_semaphores)
      : context_(std::move(context)),
        wait_semaphores_(std::move(wait_semaphores)),
        signal_semaphores_(std::move(signal_semaphores)) {}

  ~EventBatch() {
    for (auto& semaphore : signal_semaphores_) {
      semaphore->Signal();
    }
  }

  std::weak_ptr<MsdVsiContext> GetContext() const override { return context_; }

 private:
  std::shared_ptr<MsdVsiContext> context_;
  std::vector<std::shared_ptr<magma::PlatformSemaphore>> wait_semaphores_;
  std::vector<std::shared_ptr<magma::PlatformSemaphore>> signal_semaphores_;
};

// Releases the list of bus mappings when destroyed.
class MappingReleaseBatch : public NullBatch {
 public:
  MappingReleaseBatch(std::shared_ptr<MsdVsiContext> context,
                      std::vector<std::unique_ptr<magma::PlatformBusMapper::BusMapping>> mappings)
      : context_(std::move(context)), mappings_(std::move(mappings)) {}

  std::weak_ptr<MsdVsiContext> GetContext() const override { return context_; }

 private:
  std::shared_ptr<MsdVsiContext> context_;
  std::vector<std::unique_ptr<magma::PlatformBusMapper::BusMapping>> mappings_;
};

#endif  // MAPPED_BATCH_H