make a broadcast_array for rev and cleanup includes

Author: SteveBronder

stan/math/fwd/functor/broadcast_array.hpp

@@ -1,9 +1,9 @@
-#ifndef STAN_MATH_FWD_META_BROADCAST_ARRAY_HPP
-#define STAN_MATH_FWD_META_BROADCAST_ARRAY_HPP
+#ifndef STAN_MATH_FWD_FUNCTOR_BROADCAST_ARRAY_HPP
+#define STAN_MATH_FWD_FUNCTOR_BROADCAST_ARRAY_HPP
 
 #include <stan/math/prim/fun/Eigen.hpp>
 #include <stan/math/fwd/fun/sum.hpp>
-#include <stan/math/prim/functor/broadcast_array_fwd.hpp>
+#include <stan/math/prim/functor/broadcast_array.hpp>
 #include <stan/math/prim/meta.hpp>
 
 namespace stan {

stan/math/prim/functor/broadcast_array.hpp

@@ -1,5 +1,5 @@
-#ifndef STAN_MATH_PRIM_META_BROADCAST_ARRAY_HPP
-#define STAN_MATH_PRIM_META_BROADCAST_ARRAY_HPP
+#ifndef STAN_MATH_PRIM_FUNCTOR_BROADCAST_ARRAY_HPP
+#define STAN_MATH_PRIM_FUNCTOR_BROADCAST_ARRAY_HPP
 
 #include <stan/math/prim/functor/broadcast_array_fwd.hpp>
 #include <stan/math/prim/meta/is_eigen.hpp>
@@ -16,7 +16,7 @@ namespace math {
 namespace internal {
 
 template <typename T>
-class broadcast_array<T, require_t<is_var_or_arithmetic<T>>> {
+class broadcast_array<T, require_st_arithmetic<T>> {
  private:
   std::reference_wrapper<T> prim_;
 

stan/math/rev/functor.hpp

@@ -8,6 +8,7 @@
 #include <stan/math/rev/functor/apply_scalar_unary.hpp>
 #include <stan/math/rev/functor/apply_scalar_binary.hpp>
 #include <stan/math/rev/functor/apply_vector_unary.hpp>
+#include <stan/math/rev/functor/broadcast_array.hpp>
 #include <stan/math/rev/functor/coupled_ode_system.hpp>
 #include <stan/math/rev/functor/cvodes_integrator.hpp>
 #include <stan/math/rev/functor/cvodes_utils.hpp>

stan/math/rev/functor/broadcast_array.hpp

@@ -0,0 +1,38 @@
+#ifndef STAN_MATH_REV_FUNCTOR_BROADCAST_ARRAY_HPP
+#define STAN_MATH_REV_FUNCTOR_BROADCAST_ARRAY_HPP
+
+#include <stan/math/prim/fun/Eigen.hpp>
+#include <stan/math/rev/fun/sum.hpp>
+#include <stan/math/prim/functor/broadcast_array.hpp>
+#include <stan/math/prim/meta.hpp>
+
+namespace stan {
+namespace math {
+namespace internal {
+
+template <typename T>
+class broadcast_array<T, require_st_var<T>> {
+ private:
+  std::reference_wrapper<T> prim_;
+
+ public:
+  template <typename TT>
+  explicit broadcast_array(TT&& prim) : prim_(std::forward<TT>(prim)) {}
+
+  T& operator[](int /*i*/) { return prim_.get(); }
+
+  /** \ingroup type_trait
+   * Broadcast array can be assigned a scalar or a vector. If assigned a scalar,
+   * it will be used directly. If assigned a vector, the argument will be summed
+   * first.
+   */
+  template <typename Y>
+  void operator=(const Y& m) {
+    prim_.get() = sum(m);
+  }
+};
+
+}  // namespace internal
+}  // namespace math
+}  // namespace stan
+#endif

stan/math/rev/functor/operands_and_partials.hpp

@@ -7,13 +7,14 @@
 #include <stan/math/rev/core/var.hpp>
 #include <stan/math/rev/core/vari.hpp>
 #include <stan/math/rev/core/typedefs.hpp>
+#include <stan/math/rev/functor/broadcast_array.hpp>
 #include <stan/math/prim/meta/is_eigen.hpp>
 #include <stan/math/prim/meta/is_vector_like.hpp>
 #include <stan/math/prim/meta/compiler_attributes.hpp>
 #include <stan/math/prim/meta/promote_scalar_type.hpp>
 #include <stan/math/prim/fun/size.hpp>
-#include <stan/math/prim/functor/broadcast_array.hpp>
 #include <stan/math/prim/functor/operands_and_partials.hpp>
+#include <stan/math/prim/functor/broadcast_array.hpp>
 #include <stan/math/prim/functor/apply.hpp>
 #include <stan/math/prim/functor/for_each.hpp>
 #include <vector>

stan/math/rev/functor/partials_propagator.hpp

@@ -5,6 +5,7 @@
 #include <stan/math/rev/core/var.hpp>
 #include <stan/math/rev/core/typedefs.hpp>
 #include <stan/math/rev/functor/operands_and_partials.hpp>
+#include <stan/math/rev/functor/broadcast_array.hpp>
 #include <stan/math/prim/functor/for_each.hpp>
 #include <vector>
 #include <tuple>