test_host.h

#ifndef NXDK_PGRAPH_TESTS_TEST_HOST_H
#define NXDK_PGRAPH_TESTS_TEST_HOST_H
 
#include <debug_output.h>
#include <ftp_logger.h>
#include <pbkit/pbkit.h>
#include <printf/printf.h>
 
#include <cstdint>
#include <memory>
 
#include "nv2astate.h"
#include "nxdk_ext.h"
#include "pushbuffer.h"
#include "string"
#include "texture_format.h"
#include "texture_stage.h"
#include "vertex_buffer.h"
#include "xbox_math_types.h"
 
using namespace XboxMath;
using namespace PBKitPlusPlus;
 
namespace PBKitPlusPlus {
class VertexShaderProgram;
struct Vertex;
class VertexBuffer;
}  // namespace PBKitPlusPlus
 
class TestHost : public NV2AState {
 public:
  TestHost(std::shared_ptr<FTPLogger> ftp_logger, uint32_t framebuffer_width, uint32_t framebuffer_height,
           uint32_t max_texture_width, uint32_t max_texture_height, uint32_t max_texture_depth = 4);
 
  void FinishDraw(bool allow_saving, const std::string &output_directory, const std::string &suite_name,
                  const std::string &name, bool save_zbuffer = false);
 
  [[nodiscard]] bool GetSaveResults() const { return save_results_; }
  void SetSaveResults(bool enable = true) { save_results_ = enable; }
 
  [[nodiscard]] std::shared_ptr<FTPLogger> GetFTPLogger() const { return ftp_logger_; }
 
  static std::string SaveTexture(const std::string &output_directory, const std::string &name, const uint8_t *texture,
                                 uint32_t width, uint32_t height, uint32_t pitch, uint32_t bits_per_pixel,
                                 SDL_PixelFormatEnum format);
  static std::string SaveRawTexture(const std::string &output_directory, const std::string &name,
                                    const uint8_t *texture, uint32_t width, uint32_t height, uint32_t pitch,
                                    uint32_t bits_per_pixel);
  [[nodiscard]] std::string SaveZBuffer(const std::string &output_directory, const std::string &name) const;
 
  static void EnsureFolderExists(const std::string &folder_path);
 
  inline float CenterX(float item_width, bool pixel_align = true) {
    float ret = (GetFramebufferWidthF() - item_width) * 0.5f;
    if (pixel_align) {
      return floorf(ret);
    }
 
    return ret;
  }
 
  inline float CenterY(float item_height, bool pixel_align = true) {
    float ret = (GetFramebufferHeightF() - item_height) * 0.5f;
    if (pixel_align) {
      return floorf(ret);
    }
    return ret;
  }
 
  static std::string GetDrawPrimitiveName(DrawPrimitive primitive);
 
 private:
  static std::string PrepareSaveFile(std::string output_directory, const std::string &filename,
                                     const std::string &ext = ".png");
  static std::string SaveBackBuffer(const std::string &output_directory, const std::string &name);
 
 private:
  bool save_results_{true};
 
  std::shared_ptr<FTPLogger> ftp_logger_;
};
 
#endif  // NXDK_PGRAPH_TESTS_TEST_HOST_H