state_vector_multi.h

Go to the documentation of this file.

/*
This file is part of the Ristra Wonton project.
Please see the license file at the root of this repository, or at:
    https://github.com/laristra/wonton/blob/master/LICENSE
*/

#ifndef WONTON_STATE_VECTOR_MULTI_H_
#define WONTON_STATE_VECTOR_MULTI_H_

#include <string>
#include <typeinfo>
#include <vector>
#include <unordered_map>

#include "wonton/support/wonton.h"
#include "wonton/state/state_vector_base.h"

namespace Wonton {  

template <class T=double>
class StateVectorMulti : public StateVectorBase {

 public:
  
  StateVectorMulti(
      std::string name, 
      std::unordered_map<int, std::vector<T>> data = std::unordered_map<int, std::vector<T>>()
                   ) : StateVectorBase(name, Field_type::MULTIMATERIAL_FIELD, Entity_kind::CELL), 
                       data_(data) {}


  ~StateVectorMulti() {}

  // print
  std::ostream & print(std::ostream & os) const {
    os << "StateVectorMulti\n";
    return os;
  }

  // get the data type
  const std::type_info& data_type() const {
    const std::type_info& ti = typeid(T);
    return ti;
  }
    
    
  std::unordered_map<int, std::vector<T>>& get_data() { return data_; }
        
        
  std::unordered_map<int, std::vector<T>> const & get_data() const { return data_; }
        
        
  std::vector<T>& get_data(int m) { return data_[m]; }

  std::vector<T> const & get_data(int m) const { return data_.at(m); }

 private:
 
  std::unordered_map<int, std::vector<T>> data_;
 
};

}

#endif //WONTON_STATE_VECTOR_MULTI_H_