/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* vim: set ts=8 sts=2 et sw=2 tw=80: */

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef MOZILLA_GFX_SOURCESURFACEDUAL_H_

#define MOZILLA_GFX_SOURCESURFACEDUAL_H_

#include "2D.h"

namespace mozilla {

namespace gfx {

class DualSurface;

class DualPattern;

class SourceSurfaceDual : public SourceSurface

{

public:

  MOZ_DECLARE_REFCOUNTED_VIRTUAL_TYPENAME(SourceSurfaceDual, override)

  SourceSurfaceDual(DrawTarget *aDTA, DrawTarget *aDTB)

    : mA(aDTA->Snapshot())

    , mB(aDTB->Snapshot())

  { }

  SourceSurfaceDual(SourceSurface *aSourceA, SourceSurface *aSourceB)

    : mA(aSourceA)

    , mB(aSourceB)

  { }

  virtual SurfaceType GetType() const override { return SurfaceType::DUAL_DT; }

  virtual IntSize GetSize() const override { return mA->GetSize(); }

  virtual SurfaceFormat GetFormat() const override { return mA->GetFormat(); }

  // TODO: This is probably wrong as this was originally only

  // used for debugging purposes, but now has legacy relying on

  // giving the first type only.

  virtual already_AddRefed<DataSourceSurface> GetDataSurface() override {

    return mA->GetDataSurface();

  }

  SourceSurface* GetFirstSurface() {

    MOZ_ASSERT(mA->GetType() == mB->GetType());

    return mA;

  }

  bool SameSurfaceTypes() {

    return mA->GetType() == mB->GetType();

  }

private:

  friend class DualSurface;

  friend class DualPattern;

  RefPtr<SourceSurface> mA;

  RefPtr<SourceSurface> mB;

};

} // namespace gfx

} // namespace mozilla

#endif /* MOZILLA_GFX_SOURCESURFACEDUAL_H_ */