frame.h

Go to the documentation of this file.

#ifndef ION_GFXUTILS_FRAME_H_
#define ION_GFXUTILS_FRAME_H_

#include <functional>
#include <string>

#include "base/integral_types.h"
#include "ion/base/referent.h"
#include "ion/base/stlalloc/allocmap.h"

namespace ion {
namespace gfxutils {

class ION_API Frame : public base::Referent {
 public:
  typedef std::function<void(const Frame& frame)> Callback;

  Frame();

  uint64 GetCounter() const { return counter_; }
  void ResetCounter() { counter_ = 0; }

  void Begin();

  void End();

  bool IsInFrame() const { return in_frame_; }

  void AddPreFrameCallback(const std::string& key, const Callback& callback);
  void AddPostFrameCallback(const std::string& key, const Callback& callback);

  bool RemovePreFrameCallback(const std::string& key);
  bool RemovePostFrameCallback(const std::string& key);

 private:
  typedef base::AllocMap<std::string, Callback> CallbackMap;

  ~Frame() override;

  uint64 counter_;
  bool in_frame_;
  CallbackMap pre_frame_callbacks_;
  CallbackMap post_frame_callbacks_;
};

typedef base::ReferentPtr<Frame>::Type FramePtr;

}  // namespace gfxutils
}  // namespace ion

#endif  // ION_GFXUTILS_FRAME_H_