add load_vector from full

Author: akaszynski

pyansys/common.py

@@ -79,7 +79,7 @@ def read_record(self, pointer, return_bufsize=False):
         Parameters
         ----------
         pointer : int
-            ANSYS file position (n words from start of file.  A word
+            ANSYS file position (n words from start of file).  A word
             is four bytes.
 
         return_bufsize : bool, optional

pyansys/full.py

@@ -236,9 +236,6 @@ def load_km(self, as_sparse=True, sort=False):
                 raise ImportError('Unable to load scipy, use ``load_km`` with '
                                   '``as_sparse=False``')
 
-        # Get header details
-        neqn = self._header['neqn']  # Number of equations
-
         # number of terms in stiffness matrix
         ntermK = two_ints_to_long(self._header['ntermKl'], self._header['ntermKh'])
 
@@ -272,7 +269,7 @@ def load_km(self, as_sparse=True, sort=False):
             krow, kcol, kdata = _binary_reader.read_array(self.filename,
                                                           ptrSTF,
                                                           ntermK,
-                                                          neqn,
+                                                          self.neqn,
                                                           const)
         else:
             warnings.warn('Missing stiffness matrix')
@@ -282,7 +279,7 @@ def load_km(self, as_sparse=True, sort=False):
             mrow, mcol, mdata = _binary_reader.read_array(self.filename,
                                                           ptrMAS,
                                                           ntermM,
-                                                          neqn,
+                                                          self.neqn,
                                                           const)
         else:
             warnings.warn('Missing mass matrix')
@@ -305,7 +302,8 @@ def load_km(self, as_sparse=True, sort=False):
 
 
         # sort nodal equivalence
-        dof_ref, index, nref, dref = _binary_reader.sort_nodal_eqlv(neqn, neqv, ndof)
+        dof_ref, index, nref, dref = _binary_reader.sort_nodal_eqlv(self.neqn,
+                                                                    neqv, ndof)
 
         # store constrained dof information
         unsort_dof_ref = np.vstack((nref, dref)).T
@@ -339,7 +337,7 @@ def load_km(self, as_sparse=True, sort=False):
         if as_sparse:
 
             if kdata is not None:
-                k = coo_matrix((neqn,) * 2)
+                k = coo_matrix((self.neqn,) * 2)
                 k.data = kdata  # data has to be set first
                 k.row = krow
                 k.col = kcol
@@ -350,7 +348,7 @@ def load_km(self, as_sparse=True, sort=False):
                 k = None
 
             if mdata is not None:
-                m = coo_matrix((neqn,) * 2)
+                m = coo_matrix((self.neqn,) * 2)
                 m.data = mdata
                 m.row = mrow
                 m.col = mcol
@@ -362,19 +360,41 @@ def load_km(self, as_sparse=True, sort=False):
 
         else:
             if kdata is not None:
-                k = np.zeros((neqn,) * 2)
+                k = np.zeros((self.neqn,) * 2)
                 k[krow, kcol] = kdata
             else:
                 k = None
 
             if mdata is not None:
-                m = np.zeros((neqn,) * 2)
+                m = np.zeros((self.neqn,) * 2)
                 m[mrow, mcol] = mdata
             else:
                 m = None
 
         return dof_ref, k, m
 
+    @property
+    def neqn(self):
+        """Number of equations
+
+        Examples
+        --------
+        >>> full.neqn
+        963
+        """
+        return self._header['neqn']
+
+    @property
+    def load_vector(self):
+        """The load vector
+
+        Examples
+        --------
+        >>> full.load_vector
+        array([0., 0., 0., ..., 0., 0., 0.])
+        """
+        return self.read_record(self._header['ptrRHS'])[:self.neqn]
+
     def _load_km(self):
         """Loads the matrices with sorted DOF"""
         self._dof_ref, self._k, self._m = self.load_km(sort=True)

tests/test_full.py

@@ -63,3 +63,7 @@ def test_full_load_km(sparse_full):
 
     # make sure internal values are not overwritten
     assert (np.diff(sparse_full.dof_ref[:, 0]) >= 0).all()
+
+
+def test_load_vector(sparse_full):
+    assert not sparse_full.load_vector.any()