LocomotorPrimitives: Check availability of control data before running

1 files changed, 8 insertions, 0 deletions

diff --git a/LocomotorPrimitivesController.h b/LocomotorPrimitivesController.h
index 702a3a5..c33745e 100644
--- a/LocomotorPrimitivesController.h
+++ b/LocomotorPrimitivesController.h
@@ -18,9 +18,17 @@ public:
 		delete[] _muscle_act;
 	}
 
+	/**
+	 * Check availability of control data for all controls in the model.
+	 *
+	 * @return    0 if control data is available for all controls, negative
+	 *            number of missing control data vectors on error
+	 */
+	int checkControls();
 	void computeControls(const SimTK::State &s, SimTK::Vector &controls) const;
 private:
 	double _alpha;
+	/* used to store actuation data in compute controls */
 	double *_muscle_act;
 
 	OpenSim::Storage _act;