ScalarFieldBake.h

#pragma once
 
#include <cstddef>
#include <cstdint>
#include <fhsim_environment/ScalarField.h>
#include <vector>
 
namespace util
{
 
constexpr uint16_t kMaskedCell = 0;
 
constexpr std::size_t kMaxBakeCells = 33'554'432;
 
constexpr int kMaxAxisSize3D = 2048;
 
constexpr int kMaxAxisSize2D = 8192;
 
struct SamplingBox
{
   double origin[3];  
   double axis[3][3]; 
   int    counts[3];  
};
 
struct BakedField
{
   SamplingBox           box;      
   std::vector<uint16_t> values;   
   std::vector<uint8_t>  mask;     
   double                valueMin; 
   double                valueMax; 
};
 
struct ValueRange
{
   double valueMin; 
   double valueMax; 
   bool   valid;    
};
 
SamplingBox MakeAxisAlignedSamplingBox(const double min[3], const double max[3], const int counts[3]);
 
void CountsFromSpacing(const double extent[3], const double spacing[3], int counts[3]);
 
void DeriveSpacingForCellBudget(const double extent[3], std::size_t budget, double spacing[3]);
 
double EncodingPad(double lo, double hi);
 
void MakeEncodingRange(double lo, double hi, double& valueMin, double& valueMax);
 
ValueRange ProbeScalarFieldRange(environment::field::ScalarField& field,
      double                                                      time,
      const SamplingBox&                                          box,
      bool                                                        maskInaccurate,
      int                                                         samplesPerAxis = 32);
 
BakedField BakeScalarField(environment::field::ScalarField& field,
      double                                                time,
      const SamplingBox&                                    box,
      double                                                valueMin,
      double                                                valueMax,
      bool                                                  maskInaccurate,
      unsigned                                              numThreads = 1);
 
} // namespace util