mirror of
https://github.com/isledecomp/SIEdit.git
synced 2024-11-23 15:48:03 -05:00
146 lines
3.3 KiB
C++
146 lines
3.3 KiB
C++
#ifndef TYPES_H
|
|
#define TYPES_H
|
|
|
|
#include <cstring>
|
|
#include <map>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#if defined(_MSC_VER) && (_MSC_VER < 1600)
|
|
// Declare types for MSVC versions less than 2010 (1600) which lacked a stdint.h
|
|
typedef unsigned char uint8_t;
|
|
typedef char int8_t;
|
|
typedef unsigned short uint16_t;
|
|
typedef short int16_t;
|
|
typedef unsigned int uint32_t;
|
|
typedef int int32_t;
|
|
typedef unsigned long long uint64_t;
|
|
typedef long long int64_t;
|
|
#else
|
|
#include <stdint.h>
|
|
#endif
|
|
|
|
namespace si {
|
|
|
|
class bytearray : public std::vector<char>
|
|
{
|
|
public:
|
|
bytearray(){}
|
|
|
|
template <typename T>
|
|
T *cast() { return reinterpret_cast<T*>(data()); }
|
|
|
|
template <typename T>
|
|
const T *cast() const { return reinterpret_cast<const T*>(data()); }
|
|
|
|
void append(const char *data, size_t size)
|
|
{
|
|
size_t current = this->size();
|
|
this->resize(current + size);
|
|
memcpy(this->data() + current, data, size);
|
|
}
|
|
|
|
void append(const bytearray &other)
|
|
{
|
|
size_t current = this->size();
|
|
this->resize(current + other.size());
|
|
memcpy(this->data() + current, other.data(), other.size());
|
|
}
|
|
|
|
void fill(char c)
|
|
{
|
|
memset(this->data(), c, this->size());
|
|
}
|
|
|
|
};
|
|
|
|
class Vector3
|
|
{
|
|
public:
|
|
Vector3(){}
|
|
Vector3(double ix, double iy, double iz)
|
|
{
|
|
x = ix;
|
|
y = iy;
|
|
z = iz;
|
|
}
|
|
|
|
double x;
|
|
double y;
|
|
double z;
|
|
};
|
|
|
|
class Data
|
|
{
|
|
public:
|
|
inline Data()
|
|
{
|
|
data_.resize(sizeof(uint32_t));
|
|
memset(data_.data(), 0, data_.size());
|
|
}
|
|
|
|
inline Data(const uint32_t &u) { set(u); }
|
|
inline Data(const Vector3 &u) { set(u); }
|
|
inline Data(const bytearray &u) { set(u); }
|
|
inline Data(const std::string &u)
|
|
{
|
|
data_.resize(u.size());
|
|
memcpy(data_.data(), u.data(), u.size());
|
|
}
|
|
|
|
inline operator uint32_t() const { return toU32(); }
|
|
inline operator const char *() const { return data(); }
|
|
inline operator Vector3() const { return toVector3(); }
|
|
inline operator bytearray() const { return data_; }
|
|
inline operator std::string() const { return toString(); }
|
|
|
|
inline uint16_t toU16() const { return *data_.cast<uint16_t>(); }
|
|
inline int16_t toS16() const { return *data_.cast<int16_t>(); }
|
|
inline uint32_t toU32() const { return *data_.cast<uint32_t>(); }
|
|
inline int32_t toS32() const { return *data_.cast<int32_t>(); }
|
|
inline Vector3 toVector3() const { return *data_.cast<Vector3>(); }
|
|
inline const char *data() const { return data_.data(); };
|
|
inline char *data() { return data_.data(); };
|
|
inline const char *c_str() const { return this->data(); };
|
|
inline size_t size() const { return data_.size(); }
|
|
inline const std::string toString() const
|
|
{
|
|
// Subtract 1 from size, assuming the last character is a null terminator
|
|
return std::string(data_.data(), std::max(size_t(0), data_.size()-1));
|
|
}
|
|
|
|
inline bool operator==(int u) const
|
|
{
|
|
return get<int>() == u;
|
|
}
|
|
|
|
inline bool operator==(uint32_t u) const
|
|
{
|
|
return get<uint32_t>() == u;
|
|
}
|
|
|
|
template <typename T>
|
|
inline const T &get() const
|
|
{
|
|
return *data_.cast<T>();
|
|
}
|
|
|
|
template <typename T>
|
|
inline void set(const T &value)
|
|
{
|
|
data_.resize(sizeof(T));
|
|
memcpy(data_.data(), &value, sizeof(T));
|
|
}
|
|
|
|
inline void set(const bytearray &value) { data_ = value; }
|
|
|
|
private:
|
|
bytearray data_;
|
|
|
|
};
|
|
|
|
typedef std::map<std::string, Data> DataMap;
|
|
|
|
}
|
|
|
|
#endif // TYPES_H
|