dump.hpp

Go to the documentation of this file.

#ifndef __STAN__IO__DUMP_HPP__
#define __STAN__IO__DUMP_HPP__

#include <cctype>
#include <limits>
#include <map>
#include <vector>
#include <string>
#include <sstream>
#include <boost/throw_exception.hpp>
#include <stan/math/matrix.hpp>
#include <stan/io/var_context.hpp>

namespace stan {

  namespace io {

    namespace {
       size_t product(std::vector<size_t> dims) {
         size_t y = 1U;
         for (size_t i = 0; i < dims.size(); ++i)
           y *= dims[i];
         return y;
       }
    }

    class dump_writer {
    private:
      std::ostream& out_;

      // checks doesn't contain quote char
      void write_name_equals(const std::string& name) {
        for (size_t i = 0; i < name.size(); ++i)
          if (name.at(i) == '"')
            BOOST_THROW_EXCEPTION(
              std::invalid_argument ("name can not contain quote char"));
        out_ << '"' << name << '"' << " <- " << '\n';
      }


      // adds period at end of output if necessary for double
      void write_val(const double& x) {
        std::stringstream ss;
        ss << x;
        std::string s;
        ss >> s;
        for (std::string::iterator it = s.begin();
             it < s.end();
             ++it) {
          if (*it == '.' || *it == 'e' || *it == 'E') {
            out_ << s;
            return;
          }
        }
        out_ << s << ".";
      }

      void write_val(const unsigned long long int& n) {
        out_ << n;
      }

      void write_val(const unsigned long int& n) {
        out_ << n;
      }

      void write_val(const unsigned int& n) {
        out_ << n;
      }

      void write_val(const unsigned short& n) {
        out_ << n;
      }

      void write_val(const long long& n) {
        out_ << n;
      }

      void write_val(const long& n) {
        out_ << n;
      }

      void write_val(const int& n) {
        out_ << n;
      }

      void write_val(const short& n) {
        out_ << n;
      }

      void write_val(const char& n) {
        out_ << n;
      }


      template <typename T>
      void write_list(T xs) {
        out_ << "c(";
        for (typename T::size_type i = 0; i < xs.size(); ++i) {
          if (i > 0) out_ << ", ";
          write_val(xs[i]);
        }
        out_ << ")";
      }

      template <typename T>
      void write_structure(std::vector<T> xs,
                           std::vector<size_t> dims) {
        out_ << "structure(";
        write_list(xs);
        out_ << ',' << '\n';
        out_ << ".Dim = ";
        write_list(dims);
        out_ << ")";
      }
      

      void dims(double x, std::vector<size_t> ds) {
        // no op
      }

      void dims(int x, std::vector<size_t> ds) {
        // no op
      }

      template <typename T> 
      void dims(Eigen::Matrix<T,Eigen::Dynamic,Eigen::Dynamic> m, 
                std::vector<size_t> ds) {
        ds.push_back(m.rows());
        ds.push_back(m.cols());
      }

      template <typename T> 
      void dims(Eigen::Matrix<T,Eigen::Dynamic,1> v, 
                std::vector<size_t> ds) {
        ds.push_back(v.size());
      }

      template <typename T> 
      void dims(Eigen::Matrix<T,1,Eigen::Dynamic> rv, 
                std::vector<size_t> ds) {
        ds.push_back(rv.size());
      }

      template <typename T> 
      void dims(std::vector<T> x, std::vector<size_t> ds) {
        ds.push_back(x.size());
        if (x.size() > 0)
          dims(x[0],ds);
      }
      
      template <typename T>
      std::vector<size_t> dims(T x) {
        std::vector<size_t> ds;
        dims(x,ds);
        return ds;
      }

      bool increment(const std::vector<size_t>& dims,
                     std::vector<size_t>& idx) {
        for (size_t i = 0; i < dims.size(); ++i) {
          ++idx[i];
          if (idx[i] < dims[i]) return true;
          idx[i] = 0;
        }
        return false;
      }

      template <typename T>
      void write_stan_val(const std::vector<T>& x,
                          const std::vector<size_t>& idx,
                          const size_t pos) {
        size_t next_pos = pos + 1;
        write_stan_val(x[idx[pos]],idx,next_pos);
      }
      void write_stan_val(const std::vector<double>& x,
                          const std::vector<size_t>& idx,
                          const size_t pos) {
        write_val(x[idx[pos]]);
      }
      void write_stan_val(const std::vector<int>& x,
                          const std::vector<size_t>& idx,
                          const size_t pos) {
        write_val(x[idx[pos]]);
      }
      void write_stan_val(const Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic>& x,
                          const std::vector<size_t>& idx,
                          const size_t pos) {
        size_t next_pos = pos + 1;
        write_val(x(idx[pos],idx[next_pos]));
      }
      void write_stan_val(const Eigen::Matrix<double,1,Eigen::Dynamic>& x,
                          const std::vector<size_t>& idx,
                          const size_t pos) {
        write_val(x[idx[pos]]);
      }
      void write_stan_val(const Eigen::Matrix<double,Eigen::Dynamic,1>& x,
                          const std::vector<size_t>& idx,
                          const size_t pos) {
        write_val(x[idx[pos]]);
      }


      template <typename T>
      void write_stan(const std::vector<T>& x) {
        std::vector<size_t> dims = dims(x);
        out_ << "structure(c(";
        std::vector<size_t> idx(dims.size(),0U);
        for (size_t count = 0; true; ++count) {
          if (count > 0) out_ << ", ";
          write_stan_val(x,idx);
          if (!increment(dims,idx)) break;
        }
        out_ << "), .Dim = ";
        write_list(dims);
        out_ << ")";
      }
      void write_stan(const std::vector<double>& x) {
        write_list(x);
      }
      void write_stan(const std::vector<int>& x) {
        write_list(x);
      }
      void write_stan(double x) {
        write_val(x);
      }
      void write_stan(int x) {
        write_val(x);
      }
      void write_stan(const Eigen::Matrix<double,1,Eigen::Dynamic>& x) {
        write_list(x);
      }
      void write_stan(const Eigen::Matrix<double,Eigen::Dynamic,1>& x) {
        write_list(x);
      }
      void write_stan(const Eigen::Matrix<double,
                                          Eigen::Dynamic,Eigen::Dynamic>& x) {
        typedef Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic>::Index Index;
        out_ << "structure(c(";
        std::vector<double> vals;
        for (Index m = 0; m < x.cols(); ++m) {
          for (Index n = 0; n < x.rows(); ++n) {
            if (m > 0 || n > 0) out_ << ", ";
            write_val(x(m,n));
          }
        }
        out_ << "), .Dim = c(";
        out_ << x.rows() << ", " << x.cols();
        out_ << "))";
      }
      
    public:

      dump_writer() : out_(std::cout) { }
      
      dump_writer(std::ostream& out) : out_(out) { }

      ~dump_writer() { }

      template <typename T>
      void write(const std::string& name,
                 const T& x) {
        write_name_equals(name);
        write_stan(x);
      }

      template <typename T>
      void dump_structure(std::string name,
                          std::vector<size_t> dims,
                          std::vector<T> xs) {
        if (xs.size() != product(dims)) 
          BOOST_THROW_EXCEPTION(
              std::invalid_argument("xs.size() != product(dims)"));
        write_structure(xs,dims);
      }

      template <typename T>
      void dump_list(std::string name,
                     std::vector<T> xs) {
        write_name_equals(name);
        write_list(xs);
      }

      template <typename T>
      void dump_var(std::string name,
                    T x) {
        write_name_equals(name);
        write_val(x);
      }
      
    };

    class dump_reader {
    private:
      std::string name_;
      std::vector<int> stack_i_;
      std::vector<double> stack_r_;
      std::vector<size_t> dims_;
      std::istream& in_;
      // stan::io::buffered_stream in_;

      bool scan_single_char(char c_expected) {
        int c = in_.peek();
        if (c != c_expected)
          return false;
        char c_skip;
        in_.get(c_skip);
        return true;
      }

      bool scan_optional_long() {
        if (scan_single_char('l'))
          return true;
        else if (scan_single_char('L'))
          return true;
        else
          return false;
      }

      bool scan_char(char c_expected) {
        char c;
        in_ >> c;
        if (c != c_expected) {
          in_.putback(c);
          return false;
        }
        return true;
      }

      bool scan_name_unquoted() {
        char c;
        in_ >> c; // 
        if (!std::isalpha(c)) return false;
        name_.push_back(c); 
        while (in_.get(c)) { // get turns off auto space skip
          if (std::isalpha(c) || std::isdigit(c) || c == '_' || c == '.') {
            name_.push_back(c);
          } else {
            in_.putback(c);
            return true;
          }
        }
        return true; // but hit eos
      }

      bool scan_name() {
        if (scan_char('"')) {
          if (!scan_name_unquoted()) return false;
          if (!scan_char('"')) return false;
        } else if (scan_char('\'')) {
          if (!scan_name_unquoted()) return false;
          if (!scan_char('\'')) return false;
        } else {
          if (!scan_name_unquoted()) return false;
        }
        return true;
      }


      bool scan_chars(std::string s, bool case_sensitive = true) {
        for (size_t i = 0; i < s.size(); ++i) {
          char c;
          if (!(in_ >> c)) {
            for (size_t j = 1; j < i; ++j) 
              in_.putback(s[i-j]);
            return false;
          }
          // all ASCII, so toupper is OK
          if ((case_sensitive && c != s[i])
              || (!case_sensitive && ::toupper(c) != ::toupper(s[i]))) {
            in_.putback(c);
            for (size_t j = 1; j < i; ++j) 
              in_.putback(s[i-j]);
            return false;
          }
        }
        return true;
      }

      bool scan_number(bool negate_val) {

        // must take longest first!
        if (scan_chars("Inf")) { 
          scan_chars("inity"); // read past if there
          stack_r_.push_back(negate_val
                             ? -std::numeric_limits<double>::infinity()
                             : std::numeric_limits<double>::infinity());
          return true;
        }
        if (scan_chars("NaN",false)) {
          stack_r_.push_back(std::numeric_limits<double>::quiet_NaN());
          return true;
        }

        char c;
        bool is_double = false;
        std::string buf;
        while (in_.get(c)) {
          if (std::isdigit(c)) { // before pre-scan || c == '-' || c == '+') {
            buf.push_back(c);
          } else if (c == '.'
                     || c == 'e'
                     || c == 'E'
                     || c == '-'
                     || c == '+') {
            is_double = true;
            buf.push_back(c);
          } else {
            in_.putback(c);
            break;
          }
        }
        if (!is_double && stack_r_.size() == 0) {
          int n;
          if (!(std::stringstream(buf) >> n))
            return false;
          stack_i_.push_back(negate_val ? -n : n);
          scan_optional_long();
        } else {
          for (size_t j = 0; j < stack_i_.size(); ++j)
            stack_r_.push_back(negate_val 
                               ? -static_cast<double>(stack_i_[j])
                               : static_cast<double>(stack_i_[j]));
          stack_i_.clear();
          double x;
          if (!(std::stringstream(buf) >> x))
            return false;
          stack_r_.push_back(negate_val ? -x : x);
        }
        return true;
      }

      bool scan_number() {
        char c;
        while (in_.get(c)) {
          if (std::isspace(c)) continue;
          in_.putback(c);
          break;
        }
        bool negate_val = scan_char('-'); 
        if (!negate_val) scan_char('+'); // flush leading +
        return scan_number(negate_val);
      }


      bool scan_seq_value() {
        if (!scan_char('(')) return false;
        if (scan_char(')')) {
          dims_.push_back(0U);
          return true;
        }
        if (!scan_number()) return false;; // first entry
        while (scan_char(',')) {
          if (!scan_number()) return false;
        }
        dims_.push_back(stack_r_.size() + stack_i_.size());
        return scan_char(')');
      }

      bool scan_struct_value() {
        if (!scan_char('(')) return false;
        if (scan_char('c')) { 
          scan_seq_value();
        } else {
          size_t start;
          in_ >> start;
          if (!scan_char(':'))
            return false;
          size_t end;
          in_ >> end;
          if (start <= end) {
            for (size_t i = start; i <= end; ++i)
              stack_i_.push_back(i);
          } else {
            for (size_t i = start; i >= end; --i)
              stack_i_.push_back(i);
          }
        } 
        dims_.clear();
        if (!scan_char(',')) return false;
        if (!scan_char('.')) return false;
        if (!scan_chars("Dim")) return false;
        if (!scan_char('=')) return false;
        if (scan_char('c')) {
          if (!scan_char('(')) return false;
          size_t dim;
          in_ >> dim;
          scan_optional_long(); 
          dims_.push_back(dim);
          while (scan_char(',')) {
            in_ >> dim;
            scan_optional_long(); 
            dims_.push_back(dim);
          }
          if (!scan_char(')')) return false;
        } else {
          size_t start;
          in_ >> start;
          if (!scan_char(':'))
            return false;
          size_t end;
          in_ >> end;
          if (start < end) {
            for (size_t i = start; i <= end; ++i)
              dims_.push_back(i);
          } else {
            for (size_t i = start; i >= end; --i)
              dims_.push_back(i);
          }
        }
        if (!scan_char(')')) return false;
        return true;
      }

  

      bool scan_value() {
        if (scan_char('c'))
          return scan_seq_value();
        if (scan_chars("structure"))
          return scan_struct_value();
        if (!scan_number()) 
          return false;
        if (!scan_char(':'))
          return true;
        if (stack_i_.size() != 1)
          return false;
        if (!scan_number())
          return false; 
        if (stack_i_.size() != 2)
          return false;
        int start = stack_i_[0];
        int end = stack_i_[1];
        stack_i_.clear();
        if (start <= end) {
          for (int i = start; i <= end; ++i)
            stack_i_.push_back(i);
        } else {
          for (int i = start; i >= end; --i) 
            stack_i_.push_back(i);
        }
        dims_.push_back(stack_i_.size());
        return true;
      }

      void print() {
        std::cout << "var name=|" << name_ << "|" << std::endl;
        std:: cout << "dims=(";
        for (size_t i = 0; i < dims_.size(); ++i) {
          if (i > 0)
            std::cout << ",";
          std::cout << dims_[i];
        }
        std::cout << ")" << std::endl;
        std::cout << "float stack:" << std::endl;
        for (size_t i = 0; i < stack_r_.size(); ++i)
          std::cout << "  [" << i << "] " << stack_r_[i] << std::endl;
        std::cout << "int stack" << std::endl;
        for (size_t i = 0; i < stack_i_.size(); ++i)
          std::cout << "  [" << i << "] " << stack_i_[i] << std::endl;
      }


    public:
      dump_reader() : in_(std::cin) { }

      dump_reader(std::istream& in) : in_(in) { }

      ~dump_reader() { }


      std::string name() {
        return name_;
      }

      std::vector<size_t> dims() {
        return dims_;
      }

      bool is_int() {
        return stack_i_.size() > 0;
      }

      std::vector<int> int_values() {
        return stack_i_;
      }

      std::vector<double> double_values() {
        return stack_r_;
      }

      bool next() {
        stack_r_.clear();
        stack_i_.clear();
        dims_.clear();
        name_.erase();
        if (!scan_name())  // set name
          return false;
        if (!scan_char('<')) // set <- 
          return false;
        if (!scan_char('-')) 
          return false;
        if (!scan_value()) // set stack_r_, stack_i_, dims_
          return false;
        return true;
      }
  
    };



    class dump : public stan::io::var_context {
    private: 
      std::map<std::string, 
               std::pair<std::vector<double>,
                         std::vector<size_t> > > vars_r_;
      std::map<std::string, 
               std::pair<std::vector<int>, 
                         std::vector<size_t> > > vars_i_;
      std::vector<double> const empty_vec_r_;
      std::vector<int> const empty_vec_i_;
      std::vector<size_t> const empty_vec_ui_;
      bool contains_r_only(const std::string& name) const {
        return vars_r_.find(name) != vars_r_.end();
      }
    public: 

      dump(std::istream& in) {
        dump_reader reader(in);
        while (reader.next()) {
          if (reader.is_int()) {
            vars_i_[reader.name()] 
              = std::pair<std::vector<int>, 
                          std::vector<size_t> >(reader.int_values(), 
                                                      reader.dims());
            
          } else {
            vars_r_[reader.name()] 
              = std::pair<std::vector<double>, 
                          std::vector<size_t> >(reader.double_values(), 
                                                      reader.dims());
          }
        }
      }

      bool contains_r(const std::string& name) const {
        return contains_r_only(name) || contains_i(name);
      }

      bool contains_i(const std::string& name) const {
        return vars_i_.find(name) != vars_i_.end();
      }

      std::vector<double> vals_r(const std::string& name) const {
        if (contains_r_only(name)) {
          return (vars_r_.find(name)->second).first;
        } else if (contains_i(name)) {
          std::vector<int> vec_int = (vars_i_.find(name)->second).first;
          std::vector<double> vec_r(vec_int.size());
          for (size_t ii = 0; ii < vec_int.size(); ii++) {
            vec_r[ii] = vec_int[ii];
          }
          return vec_r;
        }
        return empty_vec_r_;
      }
      
      std::vector<size_t> dims_r(const std::string& name) const {
        if (contains_r_only(name)) {
          return (vars_r_.find(name)->second).second;
        } else if (contains_i(name)) {
          return (vars_i_.find(name)->second).second;
        }
        return empty_vec_ui_;
      }

      std::vector<int> vals_i(const std::string& name) const {
        if (contains_i(name)) {
          return (vars_i_.find(name)->second).first;
        }
        return empty_vec_i_;
      }
      
      std::vector<size_t> dims_i(const std::string& name) const {
        if (contains_i(name)) {
          return (vars_i_.find(name)->second).second;
        }
        return empty_vec_ui_;
      }

      virtual void names_r(std::vector<std::string>& names) const {
        names.resize(0);        
        for (std::map<std::string, 
                      std::pair<std::vector<double>,
                                std::vector<size_t> > >
                 ::const_iterator it = vars_r_.begin();
             it != vars_r_.end(); ++it)
          names.push_back((*it).first);
      }

      virtual void names_i(std::vector<std::string>& names) const {
        names.resize(0);        
        for (std::map<std::string, 
                      std::pair<std::vector<int>, 
                                std::vector<size_t> > >
                 ::const_iterator it = vars_i_.begin();
             it != vars_i_.end(); ++it)
          names.push_back((*it).first);
      }

      bool remove(const std::string& name) {
        return (vars_i_.erase(name) > 0) 
          || (vars_r_.erase(name) > 0);
      }
      
    };
    
  }

}
#endif