operation_registry.h

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//
// Copyright (C) 2021 CERN & University of Surrey for the benefit of the
// BioDynaMo collaboration. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
//
// See the LICENSE file distributed with this work for details.
// See the NOTICE file distributed with this work for additional information
// regarding copyright ownership.
//
// -----------------------------------------------------------------------------
 
#ifndef CORE_OPERATION_OPERATION_REGISTRY_H_
#define CORE_OPERATION_OPERATION_REGISTRY_H_
 
#include "core/operation/operation.h"
 
#include <unordered_map>
 
namespace bdm {
 
class OperationRegistry {
 public:
  static OperationRegistry *GetInstance();
 
  Operation *GetOperation(const std::string &op_name);
 
  bool AddOperationImpl(const std::string &op_name, OpComputeTarget target,
                        OperationImpl *impl, uint32_t frequency = 1);
 
 private:
  std::unordered_map<std::string, Operation *> operations_;
 
  OperationRegistry();
  ~OperationRegistry();
};
 
#define BDM_REGISTER_OP(op, name, target)                                    \
  bool op::registered_ = OperationRegistry::GetInstance()->AddOperationImpl( \
      name, OpComputeTarget::target, new op());
 
#define BDM_REGISTER_OP_WITH_FREQ(op, name, target, frequency)               \
  bool op::registered_ = OperationRegistry::GetInstance()->AddOperationImpl( \
      name, OpComputeTarget::target, new op(), frequency);
 
#define BDM_REGISTER_TEMPLATE_OP(op, T, name, target)     \
  template <>                                             \
  bool op<T>::registered_ =                               \
      OperationRegistry::GetInstance()->AddOperationImpl( \
          name, OpComputeTarget::target, new op<T>());
 
inline Operation *NewOperation(const std::string &name) {
  return OperationRegistry::GetInstance()->GetOperation(name)->Clone();
}
 
#define BDM_OP_HEADER(class_name) \
 private:                         \
  static bool registered_;        \
                                  \
 public:                          \
  class_name *Clone() override { return new class_name(*this); }
 
}  // namespace bdm
 
#endif  // CORE_OPERATION_OPERATION_REGISTRY_H_