MooseVec can now handle fieldElement as well.

Author: Dilawar Singh

pybind11/MooseVec.cpp

@@ -58,7 +58,19 @@ unsigned int MooseVec::len()
 
 ObjId MooseVec::getItem(const size_t i) const
 {
-    return ObjId(oid_.path(), i, 0);
+    if (oid_.element()->hasFields())
+        return getFieldItem(i);
+    return getDataItem(i);
+}
+
+ObjId MooseVec::getDataItem(const size_t i) const
+{
+    return ObjId(oid_.path(), i, oid_.fieldIndex);
+}
+
+ObjId MooseVec::getFieldItem(const size_t i) const
+{
+    return ObjId(oid_.path(), oid_.dataIndex, i);
 }
 
 void MooseVec::setAttrOneToAll(const string& name, const py::object& val)

pybind11/MooseVec.h

@@ -29,7 +29,10 @@ class MooseVec {
 
     const ObjId& getItemRef(const size_t i) const;
 
+    // Get vector element. Vector element could be `dataIndex` or `fieldIndex`.
     ObjId getItem(const size_t i) const;
+    ObjId getDataItem(const size_t i) const;
+    ObjId getFieldItem(const size_t i) const;
 
     void setAttrOneToAll(const string& name, const py::object& val);
 

tests/py_moose/test_api.py

@@ -1,9 +1,10 @@
-__author__ = "Dilawar Singh"
-__copyright__ = "Copyright 2019-, Dilawar Singh"
+__author__     = "Dilawar Singh"
+__copyright__  = "Copyright 2019-, Dilawar Singh"
 __maintainer__ = "Dilawar Singh"
-__email__ = "[email protected]"
+__email__      = "[email protected]"
 
 import moose
+import numpy as np
 
 
 def test_children():
@@ -57,7 +58,8 @@ def test_finfos():
     print(syns.weight)
 
     # this is a shorthand for above for loop.
-    #  syns.weight = 9.0
+    syns.weight = 11.121
+    assert np.allclose(syns.weight, 11.121), syns.weight
 
      # try:
      #     print(syns[11])