如何向向量添加或重载函数

Question

我想通过为我的向量创建一个运算符来简化我的编码。

我有一堂课Voxelgrid：

class Voxelgrid {
    struct Voxel{
        float dist;
        float weight;
        };
    vector<Voxel> grid;
    int RESOLUTION;
};

现在的事情是我想访问和写入这些体素很多，我将通过使用索引来访问它们：

grid[i + j*RESOLUTION + k*RESOLUTION*RESOLUTION].distance

例如。

是否可以编写一个运算符，以便我可以改为编写

grid(i,j,k).distance

改为。

Answer 1

例如

class Voxelgrid 
{
public:
   const Voxel & operator []( std::vector<Voxel>::size_type i ) const
   {
      return grid[i];
   }
//...
};

或者你可以定义一个成员函数

class Voxelgrid 
{
public:
   const Voxel & GetVoxel( size_type i, size_type j, size_type k ) const
   {
      return grid[i + j*RESOLUTION + k*RESOLUTION*RESOLUTION];
   }
//...
};