absnormalform.h

Go to the documentation of this file.

#ifndef ADOLC_ABSNORMALFORM_H
#define ADOLC_ABSNORMALFORM_H
 
#include <adolc/adolcexport.h>
#include <adolc/drivers/absnormalformconcept.h>
#include <adolc/tape_interface.h>
#include <cmath>
#include <cstddef>
#include <vector>
 
namespace ADOLC {
 
struct DenseShape : ANFShape {
  size_t m{};
  size_t n{};
  size_t s{};
 
  constexpr DenseShape() = default;
  constexpr DenseShape(size_t m, size_t n, size_t s) : m(m), n(n), s(s) {}
};
 
DenseShape getShapeFromTape(short tapeId);
 
struct AbsNormalForm {
  using Shape = DenseShape;
 
  Shape shape{};
 
  std::vector<double *> Y;
  std::vector<double *> J;
  std::vector<double *> Z;
  std::vector<double *> L;
 
  std::vector<double> Y_storage;
  std::vector<double> J_storage;
  std::vector<double> Z_storage;
  std::vector<double> L_storage;
 
  std::vector<double> cz;
  std::vector<double> cy;
  std::vector<double> y;
  std::vector<double> z;
 
  ~AbsNormalForm() = default;
  AbsNormalForm() = default;
 
  explicit AbsNormalForm(Shape dims) { resize(dims); }
 
  AbsNormalForm(AbsNormalForm &&) = default;
  AbsNormalForm &operator=(AbsNormalForm &&) = default;
 
  AbsNormalForm(const AbsNormalForm &) = delete;
  AbsNormalForm &operator=(const AbsNormalForm &) = delete;
 
  static AbsNormalForm fromTape(short tapeId) {
    return AbsNormalForm(getShapeFromTape(tapeId));
  }
 
  bool empty() const { return shape.m == 0 && shape.n == 0 && shape.s == 0; }
 
  void clear();
 
  Shape dims() const { return shape; }
 
  void updateCy() {
    cy = y;
    for (size_t depRow = 0; depRow < dims().m; ++depRow) {
      for (size_t col = 0; col < dims().s; ++col) {
        cy[depRow] -= J[depRow][col] * std::fabs(z[col]);
      }
    }
  }
 
  void updateCz() {
    cz = z;
    for (size_t switchRow = 0; switchRow < dims().s; ++switchRow) {
      for (size_t col = 0; col < dims().s; ++col) {
        cz[switchRow] -= L[switchRow][col] * std::fabs(z[col]);
      }
    }
  }
 
  void resize(Shape dims);
};
 
} // namespace ADOLC
 
#endif // ADOLC_ABSNORMALFORM_H