model_builder.h

#ifndef NXDK_PGRAPH_TESTS_SRC_MODELS_MODEL_BUILDER_H_
#define NXDK_PGRAPH_TESTS_SRC_MODELS_MODEL_BUILDER_H_
 
#include <cstdint>
#include <vector>
 
#include "vertex_buffer.h"
#include "xbox_math_types.h"
 
class ModelBuilder {
 public:
  virtual ~ModelBuilder() = default;
 
  [[nodiscard]] virtual uint32_t GetVertexCount() const = 0;
 
  virtual void PopulateVertexBuffer(const std::shared_ptr<VertexBuffer> &vertices);
 
  virtual void PopulateVertexBuffer(const std::shared_ptr<VertexBuffer> &vertices, const float *transformation);
 
 protected:
  [[nodiscard]] virtual const float *GetVertexPositions() = 0;
  [[nodiscard]] virtual const float *GetVertexNormals() = 0;
 
  // Called after vertex arrays have been consumed.
  virtual void ReleaseData() {};
};
 
class SolidColorModelBuilder : public ModelBuilder {
 public:
  SolidColorModelBuilder() = default;
  SolidColorModelBuilder(const vector_t &diffuse, const vector_t &specular);
  SolidColorModelBuilder(const vector_t &diffuse, const vector_t &specular, const vector_t &back_diffuse,
                         const vector_t &back_specular);
 
  void PopulateVertexBuffer(const std::shared_ptr<VertexBuffer> &vertices) override;
 
  void PopulateVertexBuffer(const std::shared_ptr<VertexBuffer> &vertices, const float *transformation) override;
 
 private:
  void ApplyColors(const std::shared_ptr<VertexBuffer> &vertices) const;
 
 protected:
  vector_t diffuse_{1.0f, 1.0f, 1.0f, 1.0f};
  vector_t specular_{0.0f, 0.0f, 0.0f, 1.0f};
  vector_t back_diffuse_{1.0f, 1.0f, 1.0f, 1.0f};
  vector_t back_specular_{0.0f, 0.0f, 0.0f, 1.0f};
};
 
#endif  // NXDK_PGRAPH_TESTS_SRC_MODELS_MODEL_BUILDER_H_