summaryrefslogtreecommitdiff
path: root/src/factories/sm64/TrajectoryFactory.h
blob: 45f317d44369d47434c1a03dc05ea4b3aad7a570 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#pragma once

#include "factories/BaseFactory.h"
#include "types/Vec3D.h"

namespace SM64 {

struct Trajectory {
    int16_t trajId;
    int16_t posX;
    int16_t posY;
    int16_t posZ;
    Trajectory(int16_t trajId, int16_t posX, int16_t posY, int16_t posZ) : trajId(trajId), posX(posX), posY(posY), posZ(posZ) {}
};

class TrajectoryData : public IParsedData {
public:
    std::vector<Trajectory> mTrajectoryData;

    TrajectoryData(std::vector<Trajectory> trajectoryData) : mTrajectoryData(std::move(trajectoryData)) {}
};

class TrajectoryHeaderExporter : public BaseExporter {
    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
};

class TrajectoryBinaryExporter : public BaseExporter {
    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
};

class TrajectoryCodeExporter : public BaseExporter {
    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
};

class TrajectoryFactory : public BaseFactory {
public:
    bool CanPreviewCode() override { return true; }
    std::optional<std::shared_ptr<IParsedData>> parse(std::vector<uint8_t>& buffer, YAML::Node& data) override;
    inline std::unordered_map<ExportType, std::shared_ptr<BaseExporter>> GetExporters() override {
        return {
            REGISTER(Code, TrajectoryCodeExporter)
            REGISTER(Header, TrajectoryHeaderExporter)
            REGISTER(Binary, TrajectoryBinaryExporter)
        };
    }
};

#ifdef BUILD_UI
// Previews the trajectory as a 3D path.
class TrajectoryFactoryUI : public BaseFactoryUI {
public:
    float GetItemHeight(const ParseResultData& data) override;
    void DrawUI(const ParseResultData& data) override;
};
#endif
}