tmbutils/vectorize.hpp

Go to the documentation of this file.

#ifdef TMBAD_FRAMEWORK

namespace vectorize {

template <class Type>
struct vector : TMBad::ad_segment {
  typedef TMBad::ad_segment Base;
  using Base::Base;
  vector() {} // Surprisingly, needed on some systems.
  vector(tmbutils::vector<Type> x) :
    Base(x.data(), x.size()) { }
  vector(const Base &x) :
    Base(x) { }
  operator std::vector<Type>() const {
    std::vector<Type> x(this->size());
    for (size_t i=0; i<this->size(); i++)
      x[i] = (*this)[i];
    return x;
  }
  operator tmbutils::vector<Type>() const {
    tmbutils::vector<Type> x(this->size());
    for (size_t i=0; i<this->size(); i++)
      x[i] = (*this)[i];
    return x;
  }
  tmbutils::vector<Type> array() const {
    return *this;
  }
  Type sum() const {
    return TMBad::sum(*this);
  }
};
template <>
struct vector<double> : tmbutils::vector<double> {
  typedef tmbutils::vector<double> Base;
  using Base::Base;
  vector() {} // Surprisingly, needed on some systems.
  vector(tmbutils::vector<double> x) : Base(x) { }
};

template<class Type, class T1, class T2>
vector<Type> dnorm(vector<Type> x, T1 mean, T2 sd, int give_log=0)
{
  vector<Type> logres;
  x = (x - mean) / sd;
  logres = -log(Type(sqrt(2*M_PI)) * sd) - Type(.5) * x * x;
  if(give_log) return logres;
  else return exp(logres);
}

} // End namespace

#endif