Files
solver/MbDCode/PartFrame.h
2023-06-03 13:51:16 -06:00

76 lines
2.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 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<AbsConstraint>)>& f);
void markerFramesDo(const std::function <void(std::shared_ptr<MarkerFrame>)>& f);
void prePosIC() override;
FColDsptr rOpO();
FMatDsptr aAOp();
FColFMatDsptr pAOppE();
void fillEssenConstraints(std::shared_ptr<std::vector<std::shared_ptr<Constraint>>> essenConstraints) override;
void fillqsu(FColDsptr col) override;
void fillqsuWeights(std::shared_ptr<DiagonalMatrix<double>> diagMat) override;
void fillqsulam(FColDsptr col) override;
void useEquationNumbers() override;
void setqsulam(FColDsptr col) override;
void postPosICIteration() override;
Part* part = nullptr; //Use raw pointer when pointing backwards.
size_t iqX = -1;
size_t 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<EulerConstraint> aGeu;
std::shared_ptr<std::vector<std::shared_ptr<AbsConstraint>>> aGabs;
std::shared_ptr<std::vector<std::shared_ptr<MarkerFrame>>> markerFrames;
};
}