AverageGeometries.h

Go to the documentation of this file.

#pragma once

#include <vector>
#include "nuto/visualize/VisualizeEnum.h"
#include "Eigen/Core"

namespace NuTo
{
namespace Visualize
{

struct AverageGeometry
{
    std::vector<Eigen::VectorXd> cornerCoordinates;
    eCellTypes cellType;
};

inline AverageGeometry AverageGeometryLine()
{
    std::vector<Eigen::VectorXd> cornerCoordinates = {Eigen::VectorXd::Constant(1, -1),
                                                      Eigen::VectorXd::Constant(1, 1)};
    return {cornerCoordinates, eCellTypes::LINE};
}

inline AverageGeometry AverageGeometryQuad()
{
    std::vector<Eigen::VectorXd> cornerCoordinates = {Eigen::Vector2d(-1, -1), Eigen::Vector2d(1, -1),
                                                      Eigen::Vector2d(1, 1), Eigen::Vector2d(-1, 1)};
    return {cornerCoordinates, eCellTypes::QUAD};
}

inline AverageGeometry AverageGeometryTriangle()
{
    std::vector<Eigen::VectorXd> cornerCoordinates = {Eigen::Vector2d(0, 0), Eigen::Vector2d(1, 0),
                                                      Eigen::Vector2d(0, 1)};
    return {cornerCoordinates, eCellTypes::TRIANGLE};
}

inline AverageGeometry AverageGeometryTetrahedron()
{
    std::vector<Eigen::VectorXd> cornerCoordinates = {Eigen::Vector3d(0, 0, 0), Eigen::Vector3d(1, 0, 0),
                                                      Eigen::Vector3d(0, 1, 0), Eigen::Vector3d(0, 0, 1)};
    return {cornerCoordinates, eCellTypes::TETRAEDER};
}

inline AverageGeometry AverageGeometryPrism()
{
    std::vector<Eigen::VectorXd> cornerCoordinates = {Eigen::Vector3d(0, 0, -1), Eigen::Vector3d(1, 0, -1),
                                                      Eigen::Vector3d(0, 1, -1), Eigen::Vector3d(0, 0, 1),
                                                      Eigen::Vector3d(1, 0, 1),  Eigen::Vector3d(0, 1, 1)};
    return {cornerCoordinates, eCellTypes::WEDGE};
}
} // namespace Visualize
} // namespace NuTo