hilbert_sfc.h

// -*-c++-*-
#ifndef STORMM_HILBERT_SFC_H
#define STORMM_HILBERT_SFC_H
 
#include <vector>
#include "copyright.h"
#include "Constants/behavior.h"
#include "DataTypes/stormm_vector_types.h"
#include "math_enumerators.h"
 
namespace stormm {
namespace stmath {
 
using constants::CartesianDimension;
using constants::UnitCellAxis;
  
const int3 hilbert_cube_hll[] = { { 0, 0, 0 }, { 0, 1, 0 }, { 0, 1, 1 }, { 0, 0, 1 },
                                  { 1, 0, 1 }, { 1, 1, 1 }, { 1, 1, 0 }, { 1, 0, 0 } };

const int3 hilbert_cube_hhh[] = { { 0, 0, 0 }, { 0, 1, 0 }, { 1, 1, 0 }, { 1, 0, 0 },
                                  { 1, 0, 1 }, { 0, 0, 1 }, { 0, 1, 1 }, { 1, 1, 1 } };

const int3 hilbert_square_hl[] = { { 0, 0, 0 }, { 0, 1, 0 }, { 1, 1, 0 }, { 1, 0, 0 } };
const int3 peano_square[] = { { 0, 0, 0 }, { 0, 1, 0 }, { 0, 2, 0 },
                              { 1, 2, 0 }, { 1, 1, 0 }, { 1, 0, 0 },
                              { 2, 0, 0 }, { 2, 1, 0 }, { 2, 2, 0 } };
const int3 hileano_square[] = { { 0, 0, 0 }, { 1, 0, 0 }, { 1, 1, 0 },
                                { 0, 1, 0 }, { 0, 2, 0 }, { 1, 2, 0 },
                                { 2, 2, 0 }, { 2, 1, 0 }, { 2, 0, 0 } };
    
  
class HilbertSFC {
public:

  HilbertSFC(int x_span_in, int y_span_in, int z_span_in,
             HilbertCurveMode method = HilbertCurveMode::OVERSPAN);

  int3 getDimensions() const;

  int getDimension(CartesianDimension dim) const;
 
  int getDimension(UnitCellAxis dim) const;

  int getCurveIndex(int grid_x, int grid_y, int grid_z) const;

  const std::vector<int3>& getCurve() const;
  
private:
 
  int x_span;                      
  int y_span;                      
  int z_span;                      
  std::vector<int3> curve;         
  std::vector<int> curve_indices;  
};

void propagateHilbertCurve(int s, std::vector<int3> *curve, int x = 0, int y = 0, int z = 0,
                           int dx = 1, int dy = 0, int dz = 0, int dx2 = 0, int dy2 = 1,
                           int dz2 = 0, int dx3 = 0, int dy3 = 0, int dz3 = 1);

std::vector<int3> drawHilbertSpaceCurve(int order);
  
  
} // namespace stmath
} // namespace stormm
 
#endif