Visualizer.h

Go to the documentation of this file.

#pragma once

#include "nuto/base/Group.h"
#include "nuto/mechanics/cell/CellInterface.h"
#include "nuto/visualize/HandlerInterface.h"
#include "nuto/visualize/UnstructuredGrid.h"

namespace NuTo
{
namespace Visualize
{

class Visualizer
{
public:
    Visualizer(const Group<CellInterface>& cells, const HandlerInterface& handler);

    Visualizer(Visualizer&&) = default;

    Visualizer& operator=(Visualizer&&) = default;

    void DofValues(DofType dof);

    void CellData(std::function<Eigen::VectorXd(const CellIpData&)> f, std::string name);

    void PointData(std::function<Eigen::VectorXd(Eigen::VectorXd)> f, std::string name);

    void WriteVtuFile(std::string filename, bool asBinary = true);

private:
    Group<CellInterface> mCells;
    std::vector<std::vector<int>> mPointIds;
    std::unique_ptr<HandlerInterface> mHandler;
    UnstructuredGrid mGrid;

    void WriteGeometry();
};
} // namespace Visualize
} // namespace NuTo