#ifndef STAN_LANG_AST_LOCAL_ARRAY_TYPE_DEF_HPP
#define STAN_LANG_AST_LOCAL_ARRAY_TYPE_DEF_HPP

#include <stan/lang/ast.hpp>
#include <vector>

namespace stan {
namespace lang {

local_array_type::local_array_type(const local_var_type& el_type,
                                   const expression& len)
    : element_type_(el_type), array_len_(len) {}

local_array_type::local_array_type()
    : local_array_type(ill_formed_type(), nil()) {}

local_array_type::local_array_type(const local_var_type& el_type,
                                   const std::vector<expression>& lens)
    : element_type_(el_type), array_len_(lens[0]) {
  if (lens.size() == 1) {
    return;
  }
  if (lens.size() == 0) {
    element_type_ = ill_formed_type();
    return;
  }
  local_array_type tmp(el_type, lens[lens.size() - 1]);
  for (size_t i = lens.size() - 2; i > 0; --i) {
    tmp = local_array_type(tmp, lens[i]);
  }
  element_type_ = tmp;
}

int local_array_type::dims() const {
  int total = 1;
  for (local_var_type cur_type(element_type_); cur_type.is_array_type();
       cur_type = cur_type.array_element_type())
    ++total;
  return total;
}

local_var_type local_array_type::contains() const {
  local_var_type cur_type(element_type_);
  while (cur_type.is_array_type()) {
    cur_type = cur_type.array_element_type();
  }
  return cur_type;
}

local_var_type local_array_type::element_type() const { return element_type_; }

expression local_array_type::array_len() const { return array_len_; }

std::vector<expression> local_array_type::array_lens() const {
  std::vector<expression> result = {array_len_};
  for (local_var_type cur_type(element_type_); cur_type.is_array_type();
       cur_type = cur_type.array_element_type())
    result.push_back(cur_type.array_len());
  return result;
}
}  // namespace lang
}  // namespace stan
#endif
