element.h

Go to the documentation of this file.

#ifndef ELEMENT_H
#define ELEMENT_H

#include <omp.h>

#include <alias/eigen.h>
#include <alias/pybind.h>
#include <core/generator.h>

class Element : public Generator, public std::enable_shared_from_this<Element> {
  static int n_element_instances;
  int n_bookings = 0;
  int elementid;
  Json data;
  Vec<Str> datakeys;
  Str accessing_classname;
  omp_lock_t omp_lock;
  void increment_bookings();
  void decrement_bookings();
  void update_data(
    const ShPtr<Updater> &upd);
  Vec<std::pair<Str,int>> get_distances_between_keys(
    const Json &key_);
  friend class Generator;
 public:
  Element();
  virtual ~Element() = default;
  virtual ShPtr<Element> get_element(
    const Json &name = nullptr) override;
  virtual ShPtr<Generator> get_generator(
    const Json &name = nullptr) override;
  virtual void accessed_by_instance_of(
    const Str &classname) override;
  ShPtr<Element> append_updater(
    const ShPtr<Updater> &upd);
  const Json &get_data();
  Set<Str> get_keys();
  Json make_reduced_data(
    const Json &key_);
  template <typename T>
  void make_1darray_from_data(
    T &array,
    const Str &key);
  template <typename T>
  void make_2darray_from_data(
    T &array,
    const Vec<Str> &keys);
  void update_keys();
  void check_required_keys(
    const Json &key_);
  bool check_optional_keys(
    const Json &key_);

  /* Methods for bridging to Python */

  const Json &get_data_py();
  Set<Str> get_keys_py();
  ArrayXi get_1d_int_py(
    const Str &key);
  ArrayXd get_1d_float_py(
    const Str &key);
  ArrayXXi get_2d_int_py(
    const py::args &args);
  ArrayXXd get_2d_float_py(
    const py::args &args);
 private:
  void init_for_python();
};

using ElPtr = ShPtr<Element>;

#endif