Files
solver/MbDCode/PartFrame.h
Aik-Siong Koh 371b13a9e0 runVelIC
2023-06-18 01:06:39 -06:00

97 lines
3.5 KiB
C++

#pragma once
#include <memory>
#include <vector>
#include <functional>
#include "CartesianFrame.h"
#include "EndFramec.h"
#include "FullColumn.h"
#include "EulerParameters.h"
#include "EulerParametersDot.h"
#include "CREATE.h"
namespace MbD {
class Part;
class MarkerFrame;
class EulerConstraint;
class AbsConstraint;
//class EulerParameters;
//class EulerParametersDot;
class PartFrame : public CartesianFrame
{
//ToDo: part iqX iqE qX qE qXdot qEdot qXddot qEddot aGeu aGabs markerFrames
public:
PartFrame();
PartFrame(const char* str);
void initialize();
void initializeLocally() override;
void initializeGlobally() override;
void asFixed();
void postInput() override;
void calcPostDynCorrectorIteration() override;
void setqX(FColDsptr x);
FColDsptr getqX();
void setqE(FColDsptr x);
FColDsptr getqE();
void setqXdot(FColDsptr x);
FColDsptr getqXdot();
void setomeOpO(FColDsptr x);
FColDsptr getomeOpO();
void setqXddot(FColDsptr x);
FColDsptr getqXddot();
void setqEddot(FColDsptr x);
FColDsptr getqEddot();
void setPart(Part* x);
Part* getPart();
void addMarkerFrame(std::shared_ptr<MarkerFrame> x);
EndFrmcptr endFrame(std::string name);
void aGabsDo(const std::function <void(std::shared_ptr<Constraint>)>& f);
void markerFramesDo(const std::function <void(std::shared_ptr<MarkerFrame>)>& f);
void removeRedundantConstraints(std::shared_ptr<std::vector<int>> redundantEqnNos) override;
void reactivateRedundantConstraints() override;
void constraintsReport() override;
void prePosIC() override;
void prePosKine() override;
FColDsptr rOpO();
FMatDsptr aAOp();
FColFMatDsptr pAOppE();
void fillEssenConstraints(std::shared_ptr<std::vector<std::shared_ptr<Constraint>>> essenConstraints) override;
void fillRedundantConstraints(std::shared_ptr<std::vector<std::shared_ptr<Constraint>>> redunConstraints) override;
void fillConstraints(std::shared_ptr<std::vector<std::shared_ptr<Constraint>>> allConstraints) override;
void fillqsu(FColDsptr col) override;
void fillqsuWeights(std::shared_ptr<DiagonalMatrix<double>> diagMat) override;
void fillqsulam(FColDsptr col) override;
void fillqsudot(FColDsptr col) override;
void fillqsudotWeights(std::shared_ptr<DiagonalMatrix<double>> diagMat) override;
void useEquationNumbers() override;
void setqsu(FColDsptr col) override;
void setqsulam(FColDsptr col) override;
void postPosICIteration() override;
void fillPosICError(FColDsptr col) override;
void fillPosICJacob(SpMatDsptr mat) override;
void postPosIC() override;
void outputStates() override;
void preDyn() override;
void storeDynState() override;
void fillPosKineError(FColDsptr col) override;
void preVelIC() override;
Part* part = nullptr; //Use raw pointer when pointing backwards.
int iqX = -1;
int iqE = -1; //Position index of frame variables qX and qE in system list of variables
FColDsptr qX = std::make_shared<FullColumn<double>>(3);
std::shared_ptr<EulerParameters<double>> qE = CREATE<EulerParameters<double>>::With(4);
FColDsptr qXdot = std::make_shared<FullColumn<double>>(3);
std::shared_ptr<EulerParametersDot<double>> qEdot = std::make_shared<EulerParametersDot<double>>(4);
FColDsptr qXddot = std::make_shared<FullColumn<double>>(3);
FColDsptr qEddot = std::make_shared<FullColumn<double>>(4);
std::shared_ptr<Constraint> aGeu;
std::shared_ptr<std::vector<std::shared_ptr<Constraint>>> aGabs;
std::shared_ptr<std::vector<std::shared_ptr<MarkerFrame>>> markerFrames;
};
}