UnstructuredGrid.h

Go to the documentation of this file.

#pragma once

#include <string>
#include <vector>
#include <Eigen/Core>

#include "nuto/visualize/Cell.h"
#include "nuto/visualize/Point.h"

struct UnstructuredGridCheck;

namespace NuTo
{
namespace Visualize
{

class UnstructuredGrid
{
    friend class XMLWriter;
    friend struct ::UnstructuredGridCheck;

public:
    void ExportVtuDataFile(const std::string& filename, bool asBinary = true) const;

    int AddPoint(Eigen::VectorXd coordinates);

    int AddCell(std::vector<int> pointIds, eCellTypes cellType);

    void DefinePointData(std::string name);

    void DefineCellData(std::string name);

    void SetPointData(int pointIndex, const std::string& name, double data);

    void SetPointData(int pointIndex, const std::string& name, Eigen::VectorXd data);

    void SetCellData(int cellIndex, const std::string& name, double data);

    void SetCellData(int cellIndex, const std::string& name, Eigen::VectorXd data);

private:
    std::vector<Point> mPoints;

    std::vector<Cell> mCells;

    std::vector<std::string> mPointDataNames;

    std::vector<std::string> mCellDataNames;

    void CheckPoints(std::vector<int> pointIds) const;

    int GetPointDataIndex(const std::string& name) const;

    int GetCellDataIndex(const std::string& name) const;
};
} // Visualize
} // NuTo