CPGEquations.h

Go to the documentation of this file.

/*
 * Copyright © 2012, United States Government, as represented by the
 * Administrator of the National Aeronautics and Space Administration.
 * All rights reserved.
 * 
 * The NASA Tensegrity Robotics Toolkit (NTRT) v1 platform is licensed
 * under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * http://www.apache.org/licenses/LICENSE-2.0.
 * 
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied. See the License for the specific language
 * governing permissions and limitations under the License.
*/

#ifndef SRC_UTIL_CPGS_CPGEQUATIONS
#define SRC_UTIL_CPGS_CPGEQUATIONS

#include <vector>
#include <assert.h>
#include <sstream>

#include "CPGNode.h"
#include "CPGEdge.h"

class CPGEquations
{
 public:
                
                CPGEquations();

                CPGEquations(std::vector<CPGNode*> newNodeList);
                
                ~CPGEquations();
                
                int addNode(std::vector<double> newParams);
                                                                                
                void connectNode(int nodeIndex,
                                                 std::vector<CPGEdge*> connectivityList);
                
                 void defineConnections (int nodeIndex,
                                                                 std::vector<int> connections,
                                                                 std::vector<double> newWeights,
                                                                 std::vector<double> newPhaseOffsets);
                
                double operator[](const int i)
                {                               
                                return nodeList[i]->nodeValue;
                }

                std::vector<double> getXVars();
                
                std::vector<double> getDXVars();
                
                void updateNodes(std::vector<double> descCom);
                
                void updateNodeData(std::vector<double> newXVals);
                
                void update(std::vector<double> descCom, double dt);
                
                std::string toString(const std::string& prefix = "") const;
                
protected:
                
                std::vector<CPGNode*> nodeList;
                
                double stepSize;
};

inline std::ostream&
operator<<(std::ostream& os, const CPGEquations& obj)
{
    os << obj.toString() << std::endl;
                return os;
}

#endif // SIMULATOR_SRC_LIB_MODELS_SNAKE_CPGS_CPGEQUATIONS