binary_data_wrapper.h

Go to the documentation of this file.

/*=========================================================================

  Project:   AtilaCalculatorSoftware
  File:      binary_data_wrapper.h

  Copyright (c) 2020
  All rights reserved.

=========================================================================*/
#ifndef ATILACALCULATORSOFTWARE_BINARY_DATA_WRAPPER_H
#define ATILACALCULATORSOFTWARE_BINARY_DATA_WRAPPER_H

#include "core/binary_data.h"
#include "logger/logger.h"

#include <QString>
#include <QStringList>
#include <fstream>
#include <string>
#include <vtkCell.h>
#include <vtkCellArray.h>
#include <vtkFloatArray.h>
#include <vtkHexahedron.h>
#include <vtkLine.h>
#include <vtkPointData.h>
#include <vtkPoints.h>
#include <vtkPolygon.h>
#include <vtkPolyhedron.h>
#include <vtkPyramid.h>
#include <vtkQuad.h>
#include <vtkQuadraticPyramid.h>
#include <vtkQuadraticTriangle.h>
#include <vtkSmartPointer.h>
#include <vtkUnstructuredGrid.h>

class BinaryDataWrapper : public BinaryData {
  public:
    explicit BinaryDataWrapper(const std::string& file);

    ~BinaryDataWrapper();

    void loadResult(Result& result, const int& component);

    void reload();

    void hideElement(const std::string& element);

    void showElement(const std::string& element);

    void toggleElement(const std::string& element);

    void showAllElements();

    vtkSmartPointer<vtkUnstructuredGrid> getUnstructuredGrid() const;

    vtkSmartPointer<vtkFloatArray> getScalars() const;

    std::set<std::string> getElements() const;

    std::set<std::string> getHiddenElements() const;

    bool isElementHidden(const std::string& element) const;

    QStringList getInformationList() const;

  private:
    void loadMeshInformation(Mesh& mesh);

    void convertFromGiD();

    static vtkSmartPointer<vtkCell> getPolygonVTKCell(const std::string& meshElement,
                                                      const unsigned int& dimCount);

    vtkSmartPointer<vtkUnstructuredGrid> unstructuredGrid;

    vtkSmartPointer<vtkPoints> points;

    vtkSmartPointer<vtkFloatArray> scalars;

    std::set<std::string> elements;

    std::set<std::string> hiddenElements;

    QStringList informationList;
};

#endif // ATILACALCULATORSOFTWARE_BINARY_DATA_WRAPPER_H