sc2_map_info.h

Go to the documentation of this file.

#pragma once

#include "sc2_common.h"
#include "sc2_gametypes.h"

#include "s2clientprotocol/sc2api.pb.h"

#include <string>
#include <vector>

namespace sc2 {

struct ImageData {
    int width;
    int height;
    int bits_per_pixel;
    std::string data;

    ImageData();
};

struct RenderedFrame {
    ImageData map;
    ImageData minimap;
};

struct SpatialSetup {
    float camera_width;
    int map_resolution_x;
    int map_resolution_y;
    int minimap_resolution_x;
    int minimap_resolution_y;
};

struct InterfaceOptions {
    bool raw;
    SpatialSetup feature_layer;
    SpatialSetup render;
};

struct PlayerInfo {
    uint32_t player_id;
    PlayerType player_type;
    Race race_requested;
    Race race_actual;
    Difficulty difficulty;
    AIBuild ai_build;
    std::string player_name;

    PlayerInfo(uint32_t player_id, PlayerType player_type,
        Race race_requested, Race race_actual,
        Difficulty difficulty, AIBuild ai_build,
        const std::string& player_name):
        player_id(player_id),
        player_type(player_type),
        race_requested(race_requested),
        race_actual(race_actual),
        difficulty(difficulty),
        ai_build(ai_build),
        player_name(player_name) {};
};

struct GameInfo {
    std::string map_name;
    std::string local_map_path;
    int width;
    int height;
    ImageData pathing_grid;
    ImageData terrain_height;
    ImageData placement_grid;
    Point2D playable_min;
    Point2D playable_max;
    std::vector<Point2D> enemy_start_locations;
    std::vector<Point2D> start_locations;

    InterfaceOptions options;

    std::vector<PlayerInfo> player_info;

    GameInfo();
};

struct SampleImage {
    explicit SampleImage(const SC2APIProtocol::ImageData& data);

    explicit SampleImage(const ImageData& data);

    bool GetBit(const Point2DI& point, bool* dst) const;

    bool GetBit(const Point2DI& point, unsigned char* dst) const;

    int BPP() const;

    Rect2DI Area() const;

private:
    const std::string& data_;
    Rect2DI area_;

    // NOTE (alkurbatov): Possible bits per pixel values specified in
    // protocol/s2clientprotocol/raw.proto
    int bits_per_pixel_;
};

struct PathingGrid {
    explicit PathingGrid(const GameInfo& info);

    bool IsPathable(const Point2DI& point) const;

    void Dump(const std::string& file_path) const;

private:
    SampleImage pathing_grid_;
};

struct PlacementGrid {
    explicit PlacementGrid(const GameInfo& info);

    bool IsPlacable(const Point2DI& point) const;

    void Dump(const std::string& file_path) const;

private:
    SampleImage placement_grid_;
};

struct HeightMap {
    explicit HeightMap(const GameInfo& info);

    float TerrainHeight(const Point2DI& point) const;

    void Dump(const std::string& file_path) const;

private:
    SampleImage height_map_;
};

}