Added support for exporing plot object data from Python.

Author: SteveMaas1978

FEBioStudio/DataFieldSelector.cpp

@@ -223,10 +223,9 @@ CPlotObjectDataSelector::CPlotObjectDataSelector(Post::FEPostModel::PlotObject*
 
 void CPlotObjectDataSelector::BuildMenu(QMenu* menu)
 {
-	std::vector<PlotObjectData*>& data = m_po->m_data;
-	for (int i = 0; i < (int)data.size(); ++i)
+	for (int i = 0; i < (int) m_po->DataCount(); ++i)
 	{
-		ModelDataField& d = *data[i];
+		PlotObjectData& d = *m_po->GetData(i);
 		int dataClass = d.DataClass();
 		int dataComponents = d.components(TENSOR_SCALAR);
 		if (dataComponents > 0)

FEBioStudio/GraphWindow.cpp

@@ -2262,44 +2262,7 @@ void CModelGraphWindow::TrackObjectHistory(int nobj, float* pval, int nfield, in
 
 	for (int j = 0; j < nn; ++j)
 	{
-		Post::FEState* state = fem.GetState(j + nmin);
-		Post::OBJECTDATA& pointData = state->GetObjectData(nobj);
-
-		Post::ObjectData* data = pointData.data;
-
-		// get the data ID
-		int ndata = FIELD_CODE(nfield);
-
-		// get the component
-		int ncomp = FIELD_COMP(nfield);
-
-		Post::ModelDataField* dataField = po->m_data[ndata];
-
-		float val = 0.f;
-
-		switch (dataField->Type())
-		{
-		case DATA_SCALAR:
-		{
-			val = data->get<float>(ndata);
-		}
-		break;
-		case DATA_VEC3:
-		{
-			vec3f v = data->get<vec3f>(ndata);
-			val = component(v, ncomp);
-		}
-		break;
-		case DATA_MAT3:
-		{
-			mat3f v = data->get<mat3f>(ndata);
-			val = component(v, ncomp);
-		}
-		break;
-		default:
-			assert(false);
-		}
-
+		float val = fem.EvaluatePlotObjectData(nobj, j + nmin, nfield);
 		pval[j] = val;
 	}
 }

PostGL/GLPlotObjectGlyph.cpp

@@ -38,9 +38,9 @@ Post::GLPlotObjectVector::GLPlotObjectVector(Post::FEPostModel::PointObject* po)
 
 	std::vector<std::string> nameList;
 	size_t l = 0;
-	for (int i = 0; i < po->m_data.size(); ++i)
+	for (int i = 0; i < po->DataCount(); ++i)
 	{
-		Post::PlotObjectData* pd = po->m_data[i];
+		Post::PlotObjectData* pd = po->GetData(i);
 		if (pd->Type() == DATA_VEC3)
 		{
 			std::string name = pd->GetName();
@@ -81,9 +81,9 @@ void Post::GLPlotObjectVector::Update(int ntime, float dt, bool breset)
 	Post::OBJ_POINT_DATA& data = state->m_objPt[m_po->m_id];
 
 	int nvec = GetIntValue(DATA_FIELD);
-	for (int i = 0, n = 0; i < m_po->m_data.size(); ++i)
+	for (int i = 0, n = 0; i < m_po->DataCount(); ++i)
 	{
-		Post::PlotObjectData* pd = m_po->m_data[i];
+		Post::PlotObjectData* pd = m_po->GetData(i);
 		if (pd->Type() == DATA_VEC3)
 		{
 			if (nvec == n)

PostLib/FEDataField.cpp

@@ -41,6 +41,7 @@ ModelDataField::ModelDataField(FEPostModel* fem, DATA_TYPE ntype, DATA_FORMAT nf
 	m_nclass = ncls;
 	m_flag = flag;
 	m_arraySize = 0;
+	m_nfield = -1;
 }
 
 ModelDataField::~ModelDataField() {}

PostLib/FEKinemat.cpp

@@ -167,11 +167,11 @@ bool FEKinemat::BuildStates(Post::FEPostModel* pfem)
 
 		PlotObjectData* data1 = new PlotObjectData(&fem, DATA_VEC3);
 		data1->SetName("position");
-		ob->m_data.push_back(data1);
+		ob->AddData(data1);
 
 		PlotObjectData* data2 = new PlotObjectData(&fem, DATA_MAT3);
 		data2->SetName("rotation");
-		ob->m_data.push_back(data2);
+		ob->AddData(data2);
 
 		fem.AddPointObject(ob);
 	}

PostLib/FEPostModel.cpp

@@ -1477,6 +1477,37 @@ FEPostModel::PlotObject* FEPostModel::GetPlotObject(int n)
 	return nullptr;
 }
 
+int FEPostModel::GetPlotObjectIndex(FEPostModel::PlotObject* po)
+{
+	for (int i = 0; i < PointObjects(); ++i)
+	{
+		if (m_Points[i].get() == po) return i;
+	}
+
+	int N0 = PointObjects();
+	for (int i = 0; i < LineObjects(); ++i)
+	{
+		if (m_Lines[i].get() == po) return i + N0;
+	}
+
+	return -1;
+}
+
+FEPostModel::PlotObject* FEPostModel::FindPlotObject(const std::string& name)
+{
+	for (int i=0; i<PointObjects(); ++i)
+	{ 
+		if (m_Points[i]->GetName() == name) return m_Points[i].get();
+	}
+
+	for (int i = 0; i < LineObjects(); ++i)
+	{
+		if (m_Lines[i]->GetName() == name) return m_Lines[i].get();
+	}
+
+	return nullptr;
+}
+
 int FEPostModel::PointObjects() const
 {
 	return (int)m_Points.size();

PostLib/FEPostModel.h

@@ -34,6 +34,7 @@ SOFTWARE.*/
 #include <FSCore/box.h>
 #include <vector>
 #include <memory>
+#include "constants.h"
 
 namespace Post {
 
@@ -84,6 +85,24 @@ class FEPostModel
 		//! Find the index of object data with the given name
 		int FindObjectDataIndex(const std::string& name) const;
 
+		ModelDataField* GetObjectData(int n) { return m_data[n]; }
+		ModelDataField* FindObjectData(const std::string& name)
+		{ 
+			int n = FindObjectDataIndex(name);
+			if (n < 0) return nullptr;
+			return m_data[n]; 
+		}
+
+		void AddData(PlotObjectData* po)
+		{
+			po->SetFieldID(BUILD_FIELD(DATA_CLASS::OBJECT_DATA, m_data.size(), 0));
+			m_data.push_back(po);
+		}
+
+		PlotObjectData* GetData(int n) { return m_data[n]; }
+
+		size_t DataCount() const { return m_data.size(); }
+
 	public:
 		//! Unique identifier for the plot object
 		int				m_id;
@@ -94,6 +113,7 @@ class FEPostModel
 		//! Rotation of the plot object
 		quatd			m_rot;
 
+	private:
 		//! Vector of data associated with the plot object
 		std::vector<PlotObjectData*>	m_data;
 	};
@@ -299,6 +319,9 @@ class FEPostModel
 	//! Check if the field code is valid for the given state
 	bool IsValidFieldCode(int nfield, int nstate);
 
+	float EvaluatePlotObjectData(int nobj, int ntime, int nfield);
+	float EvaluatePlotObject(PlotObject* po, ModelDataField& data, int comp, int ntime);
+
 public:
 	//! Add a dependant object that will be notified of model changes
 	void AddDependant(FEModelDependant* pc);
@@ -345,6 +368,10 @@ class FEPostModel
 	//! Get a plot object by index
 	PlotObject* GetPlotObject(int n);
 
+	int GetPlotObjectIndex(PlotObject* po);
+
+	PlotObject* FindPlotObject(const std::string& name);
+
 	//! Get the number of point objects
 	int PointObjects() const;
 	//! Add a point object

PostLib/FEState.cpp

@@ -165,11 +165,11 @@ FEState::FEState(float time, FEPostModel* fem, FSMesh* pmesh) : m_fem(fem), m_me
 		di.m_r1 = po.m_r1;
 		di.m_r2 = po.m_r2;
 
-		int ndata = (int)po.m_data.size();
+		int ndata = (int)po.DataCount();
 		di.data = new ObjectData;
 		for (int j = 0; j < ndata; ++j)
 		{
-			Post::PlotObjectData& dj = *po.m_data[j];
+			Post::PlotObjectData& dj = *po.GetData(j);
 
 			switch (dj.Type())
 			{
@@ -209,11 +209,11 @@ void FEState::AddPointObjectData()
 
 		di.m_rt = po.m_rt;
 
-		int ndata = (int)po.m_data.size();
+		int ndata = (int)po.DataCount();
 		di.data = new ObjectData;
 		for (int j = 0; j < ndata; ++j)
 		{
-			Post::PlotObjectData& dj = *po.m_data[j];
+			Post::PlotObjectData& dj = *po.GetData(j);
 
 			switch (dj.Type())
 			{

PostLib/evaluate.cpp

@@ -2932,3 +2932,56 @@ mat3f FEPostModel::EvaluateElemTensor(int n, int ntime, int nten, int ntype)
 
 	return m;
 }
+
+float FEPostModel::EvaluatePlotObjectData(int nobj, int ntime, int nfield)
+{
+	PlotObject* po = GetPlotObject(nobj);
+
+	Post::FEState* state = GetState(ntime);
+	Post::OBJECTDATA& pointData = state->GetObjectData(nobj);
+
+	Post::ObjectData* data = pointData.data;
+
+	// get the data ID
+	int ndata = FIELD_CODE(nfield);
+
+	// get the component
+	int ncomp = FIELD_COMP(nfield);
+
+	Post::ModelDataField* dataField = po->GetData(ndata);
+
+	float val = 0.f;
+
+	switch (dataField->Type())
+	{
+	case DATA_SCALAR:
+	{
+		val = data->get<float>(ndata);
+	}
+	break;
+	case DATA_VEC3:
+	{
+		vec3f v = data->get<vec3f>(ndata);
+		val = component(v, ncomp);
+	}
+	break;
+	case DATA_MAT3:
+	{
+		mat3f v = data->get<mat3f>(ndata);
+		val = component(v, ncomp);
+	}
+	break;
+	default:
+		assert(false);
+	}
+
+	return val;
+}
+
+float FEPostModel::EvaluatePlotObject(FEPostModel::PlotObject* po, ModelDataField& data, int comp, int ntime)
+{
+	int nobj = GetPlotObjectIndex(po);
+	if (nobj < 0) return 0.f;
+	int nfield = data.GetFieldID() + comp;
+	return EvaluatePlotObjectData(nobj, ntime, nfield);
+}

PyLib/PyFBSPost.cpp

@@ -173,6 +173,11 @@ void init_FBSPost(py::module& m)
         .def("Enabled", &Material::enabled, "Checks if the material is enabled.")
         ;
 
+	py::class_<FEPostModel::PlotObject>(post, "PlotObject")
+		.def("Name", &FEPostModel::PlotObject::GetName, "Returns the name of the plot object.")
+		.def("GetDataField", &FEPostModel::PlotObject::FindObjectData, "Returns the name of the plot object.")
+		;
+
 	py::class_<FEPostModel>(post, "PostModel", DOC(Post, FEPostModel))
 		.def("Materials", &FEPostModel::Materials, DOC(Post, FEPostModel, Materials))
 		.def("Material", &FEPostModel::GetMaterial, DOC(Post, FEPostModel, GetMaterial), py::return_value_policy::reference)
@@ -194,7 +199,10 @@ void init_FBSPost(py::module& m)
                 self.Evaluate(field.GetFieldID() | component, time);
                 return self.GetState(time);
             }, "Evaluates the model at a specific time step. Returns a reference to the state and updates the values stored in the state's member variables.", 
-            py::return_value_policy::reference);
+            py::return_value_policy::reference)
+		.def("GetPlotObject", &FEPostModel::FindPlotObject, py::return_value_policy::reference)
+		.def("EvaluatePlotObject", &FEPostModel::EvaluatePlotObject, py::return_value_policy::reference)
+		;
 
 	py::enum_<Data_Tensor_Type>(post, "DataTensorType")
         .value("DATA_SCALAR", Data_Tensor_Type::TENSOR_SCALAR)