id.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_ID_H
#define FREE_TENSOR_ID_H
 
#include <atomic>
#include <functional>
#include <iostream>
#include <optional>
#include <serialize/stream_utils.h>
 
namespace freetensor {
 
class ID {
    friend class StmtNode;
 
    uint64_t id_;
 
    static std::atomic_uint64_t globalIdCnt_;
    explicit ID(uint64_t id) : id_(id) {}
 
  public:
    ID() : id_(0) {}
    ID(std::nullopt_t) : id_(0) {}
 
    static ID make() { return ID(globalIdCnt_++); }
    static ID make(uint64_t id) { return ID(id); }
 
    bool isValid() const { return id_ != 0; }
 
    operator uint64_t() const { return id_; }
 
    friend std::ostream &operator<<(std::ostream &os, const ID &id);
    friend bool operator==(const ID &lhs, const ID &rhs);
    friend struct ::std::hash<ID>;
};
 
extern int OSTREAM_NO_ID_SIGN;
std::function<std::ostream &(std::ostream &)> manipNoIdSign(bool flag);
std::ostream &operator<<(std::ostream &os, const ID &id);
 
} // namespace freetensor
 
namespace std {
 
template <> struct hash<freetensor::ID> {
    size_t operator()(const freetensor::ID &id) const;
};
 
} // namespace std
 
#endif // FREE_TENSOR_ID_H