VtkRayCasting.h

Go to the documentation of this file.

#ifndef VTK_RAY_CASTING_H
#define VTK_RAY_CASTING_H
 
#ifdef VTK_PRESENT
#include <vtkActor.h>
#include <vtkBYUReader.h>
#include <vtkCamera.h>
#include <vtkCellCenters.h>
#include <vtkCellData.h>
#include <vtkDoubleArray.h>
#include <vtkIdList.h>
#include <vtkMeshQuality.h>
#include <vtkOBBTree.h>
#include <vtkOBJReader.h>
#include <vtkPLYReader.h>
#include <vtkPoints.h>
#include <vtkPolyDataMapper.h>
#include <vtkPolyDataNormals.h>
#include <vtkPolyDataReader.h>
#include <vtkProperty.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkSTLReader.h>
#include <vtkSmartPointer.h>
#include <vtkSphereSource.h>
#include <vtkXMLPolyDataReader.h>
 
#include <algorithm>
#include <array>
#include <cmath>
#include <fstream>
#include <iomanip>
#include <numbers>
#include <ostream>
#include <random>
#include <string>
#include <vector>
#include <vtksys/SystemTools.hxx>
 
#include "CsrMatrix.h"
 
class VtkRayCasting
{
    private:
        int m_rows{};
        int m_cols{};
        CsrSparseMatrix<double> m_viewfactor_matrix;
        int m_starting_facet{0};
        int m_num_local_facets{0};
        std::vector<double> m_solar_irradiation;
        vtkSmartPointer<vtkPolyData> m_base_mesh;
        vtkSmartPointer<vtkOBBTree> m_ray_mesh;
        vtkSmartPointer<vtkPolyData> m_centroids;
        vtkSmartPointer<vtkDataArray> m_normals;
        vtkSmartPointer<vtkDataArray> m_areas;
 
        static void vectorSub(const double vector_1[3], const double vector_2[3], double res[3]);
        static void vectorAdd(const double vector_1[3], const double vector_2[3], double res[3]);
        static void vectorMult(const double vector_1[3], double mult, double res[3]);
        static double dotProduct(const double vector_1[3], const double vector_2[3]);
        static double dotProductInv(const double vector_1[3], const double vector_2[3]);
        static double l2Norm(double vector_1[3], double vector_2[3]);
        static void vectorDirection(double vector_1[3], double vector_2[3], double direction[3]);
        void calculate_data_from_mesh();
        void calculateViewfactorMatrix(const vtkSmartPointer<vtkOBBTree>& mesh,
                                       const vtkSmartPointer<vtkPolyData>& centroids,
                                       const vtkSmartPointer<vtkDataArray>& normals,
                                       const vtkSmartPointer<vtkDataArray>& areas);
        void calculateSolarIrradiation(const vtkSmartPointer<vtkOBBTree>& mesh,
                                       const vtkSmartPointer<vtkPolyData>& centroids,
                                       const vtkSmartPointer<vtkDataArray>& normals,
                                       double sol_vec[3], double Q_0, std::vector<double>& Q_sol);
 
        static vtkSmartPointer<vtkPolyData> readPolyData(const char* fileName);
 
    public:
        VtkRayCasting(const char* fileName, bool calculate_vf_matrix);
        VtkRayCasting(const char* fileName, int starting_facet, int num_local_facets,
                      bool calculate_vf_matrix);
 
        const CsrSparseMatrix<double>& getViewfactorMatrix() const { return m_viewfactor_matrix; }
        const std::vector<double>& getSolarIrradiation(double sol_vec[3], double Q_0);
        int getFacetNumber() const { return m_rows; };
        int getMatrixElementNumber() const { return m_viewfactor_matrix.getNonZeroElementNumber(); }
        std::vector<std::vector<double>> getVertices();
        std::vector<std::vector<int>> getFaces();
        std::vector<std::vector<double>> getNormals();
        std::vector<double> getAreas();
};
#endif
 
#endif