// Copyright (c) 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef UI_ACCELERATED_WIDGET_MAC_IO_SURFACE_CONTEXT_H_
#define UI_ACCELERATED_WIDGET_MAC_IO_SURFACE_CONTEXT_H_

#include <OpenGL/OpenGL.h>
#include <map>

#include "base/lazy_instance.h"
#include "base/mac/scoped_nsobject.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "ui/accelerated_widget_mac/accelerated_widget_mac_export.h"
#include "ui/gl/gpu_switching_observer.h"
#include "ui/gl/scoped_cgl.h"

namespace ui {

class IOSurfaceContext
    : public base::RefCounted<IOSurfaceContext>,
      public ui::GpuSwitchingObserver {
 public:
  enum Type {
    // The number used to look up the context used for async readback and for
    // initializing the IOSurface.
    kOffscreenContext = -2,
    // The number used to look up the context used by CAOpenGLLayers.
    kCALayerContext = -3,
  };

  // Get or create a GL context of the specified type. Share these GL contexts
  // as much as possible because creating and destroying them can be expensive.
  // http://crbug.com/180463
  ACCELERATED_WIDGET_MAC_EXPORT static scoped_refptr<IOSurfaceContext> Get(
      Type type);

  // Mark that all the GL contexts in the same sharegroup as this context as
  // invalid, so they shouldn't be returned anymore by Get, but rather, new
  // contexts should be created. This is called as a precaution when unexpected
  // GL errors occur, or after a GPU switch.
  void PoisonContextAndSharegroup();
  bool HasBeenPoisoned() const { return poisoned_; }

  CGLContextObj cgl_context() const { return cgl_context_; }

  // ui::GpuSwitchingObserver implementation.
  void OnGpuSwitched() override;

 private:
  friend class base::RefCounted<IOSurfaceContext>;

  IOSurfaceContext(
      Type type,
      base::ScopedTypeRef<CGLContextObj> clg_context_strong);
  virtual ~IOSurfaceContext();

  Type type_;
  base::ScopedTypeRef<CGLContextObj> cgl_context_;

  bool poisoned_;

  // The global map from window number and window ordering to
  // context data.
  typedef std::map<Type, IOSurfaceContext*> TypeMap;
  static base::LazyInstance<TypeMap> type_map_;
  static TypeMap* type_map();
};

}  // namespace ui

#endif  // UI_ACCELERATED_WIDGET_MAC_IO_SURFACE_CONTEXT_H_