IdentityGenerator.h

Go to the documentation of this file.

#ifndef CAROTID_IDENTITYGENERATOR_H
#define CAROTID_IDENTITYGENERATOR_H
 
#include "Utility.h"
 
namespace carotid {
 
constexpr uint32_t CAROTID_SPATIAL_DIMENSIONS = 3;
 
constexpr uint32_t CAROTID_VERTEX_COUNT_LVL1 = 12466;
constexpr uint32_t CAROTID_VERTEX_COUNT_LVL2 = 49751;
constexpr uint32_t CAROTID_VERTEX_COUNT_LVL3 = 198781;
 
class IdOptions {
 
public:
  uint32_t vertexCount;
 
  uint32_t dim;
 
public:
  explicit IdOptions(const uint32_t dim)
      : vertexCount(CAROTID_VERTEX_COUNT_LVL2), dim(dim) {}
};
 
class IdentityGenerator {
 
public:
  IdOptions opts;
 
  Vec<float> cur_model;
 
  Vec<float> cur_params;
 
public:
  [[nodiscard]] explicit IdentityGenerator(const IdOptions opts) : opts(opts) {}
 
  inline uint32_t getNumRows() const {
    return CAROTID_SPATIAL_DIMENSIONS * opts.vertexCount;
  }
 
  inline uint32_t getNumCols() const { return opts.dim; }
 
  inline std::pair<uint32_t, uint32_t> getSize() const {
    return {getNumRows(), getNumCols()};
  }
 
  void loadModelFromBinary(const Path &avgModelPath, const Path &eigenvecsPath,
                           const Path &eigenvalsPath);
 
  [[nodiscard]] bool validateModel() const;
 
  [[nodiscard]] Vec<float> generateBaseModel() const;
 
  [[nodiscard]] Vec<float> generateModel(const Vec<float> &eigenvals) const;
 
private:
  Vec<float> avgModel;
  Mat<float> eigenvecs;
  Vec<float> eigenvals;
};
 
} // namespace carotid
 
#endif // CAROTID_IDENTITYGENERATOR_H