// Copyright 2014 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 CC_TEST_TEST_OCCLUSION_TRACKER_H_
#define CC_TEST_TEST_OCCLUSION_TRACKER_H_

#include "cc/layers/render_surface_impl.h"
#include "cc/trees/occlusion_tracker.h"

namespace cc {

// A subclass to expose the total current occlusion.
class TestOcclusionTracker : public OcclusionTracker {
 public:
  explicit TestOcclusionTracker(const gfx::Rect& screen_scissor_rect)
      : OcclusionTracker(screen_scissor_rect) {}

  SimpleEnclosedRegion occlusion_from_inside_target() const {
    return stack_.back().occlusion_from_inside_target;
  }
  SimpleEnclosedRegion occlusion_from_outside_target() const {
    return stack_.back().occlusion_from_outside_target;
  }

  SimpleEnclosedRegion occlusion_on_contributing_surface_from_inside_target()
      const {
    size_t stack_size = stack_.size();
    if (stack_size < 2)
      return SimpleEnclosedRegion();
    return stack_[stack_size - 2].occlusion_from_inside_target;
  }
  SimpleEnclosedRegion occlusion_on_contributing_surface_from_outside_target()
      const {
    size_t stack_size = stack_.size();
    if (stack_size < 2)
      return SimpleEnclosedRegion();
    return stack_[stack_size - 2].occlusion_from_outside_target;
  }

  void set_occlusion_from_outside_target(const SimpleEnclosedRegion& region) {
    stack_.back().occlusion_from_outside_target = region;
  }
  void set_occlusion_from_inside_target(const SimpleEnclosedRegion& region) {
    stack_.back().occlusion_from_inside_target = region;
  }

  void set_occlusion_on_contributing_surface_from_outside_target(
      const SimpleEnclosedRegion& region) {
    size_t stack_size = stack_.size();
    DCHECK_GE(stack_size, 2u);
    stack_[stack_size - 2].occlusion_from_outside_target = region;
  }
  void set_occlusion_on_contributing_surface_from_inside_target(
      const SimpleEnclosedRegion& region) {
    size_t stack_size = stack_.size();
    DCHECK_GE(stack_size, 2u);
    stack_[stack_size - 2].occlusion_from_inside_target = region;
  }
};

}  // namespace cc

#endif  // CC_TEST_TEST_OCCLUSION_TRACKER_H_