num_elements.hpp

Go to the documentation of this file.

#ifndef STAN__MATH__MATRIX__NUM_ELEMENTS_HPP
#define STAN__MATH__MATRIX__NUM_ELEMENTS_HPP

#include <vector>
#include <stan/math/matrix/Eigen.hpp>

namespace stan {

  namespace math {

    template <typename T>
    inline int
    num_elements(const T& x) {
      return 1;
    }

    template <typename T, int R, int C>
    inline int 
    num_elements(const Eigen::Matrix<T,R,C>& m) {
      return m.size();
    }

    template <typename T>
    inline int
    num_elements(const std::vector<T>& v) {
      if (v.size() == 0) 
        return 0;
      return v.size() * num_elements(v[0]);
    }

  }
}
#endif