jit.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_UTIL_JIT_H_
#define CORE_UTIL_JIT_H_
 
#include <functional>
#include <string>
#include <vector>
 
#include <TClass.h>
#include <TDataMember.h>
 
namespace bdm {
 
class Agent;
 
std::vector<TClass*> FindClassSlow(const std::string& class_name);
 
std::vector<TDataMember*> FindDataMemberSlow(TClass* tclass,
                                             const std::string& data_member);
 
// -----------------------------------------------------------------------------
class JitForEachDataMemberFunctor {
 public:
  JitForEachDataMemberFunctor(
      TClass* tclass, const std::vector<std::string>& dm_names,
      const std::string& functor_name,
      const std::function<std::string(const std::string&,
                                      const std::vector<TDataMember*>&)>&
          code_generation);
 
  void Compile() const;
 
  void* New(const std::string& parameter = "");
 
  template <typename TFunctor>
  TFunctor* New(const std::string& parameter = "") {
    return static_cast<TFunctor*>(New(parameter));
  }
 
 private:
  static std::atomic<int> counter_;
  std::string functor_name_;
  std::vector<TDataMember*> data_members_;
  std::function<std::string(const std::string&,
                            const std::vector<TDataMember*>&)>
      code_generator_;
};
 
// -----------------------------------------------------------------------------
class JitHeaders {
 public:
  static void Register(const std::string& header);
  static void IncludeIntoCling();
 
 private:
  static std::vector<std::string> headers_;
};
 
}  // namespace bdm
 
#endif  // CORE_UTIL_JIT_H_