/* -*- 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_DATASOURCESURFACEWRAPPER_H_

#define MOZILLA_GFX_DATASOURCESURFACEWRAPPER_H_

#include "2D.h"

namespace mozilla {

namespace gfx {

// Wraps a DataSourceSurface and forwards all methods except for GetType(),

// from which it always returns SurfaceType::DATA.

class DataSourceSurfaceWrapper final : public DataSourceSurface

{

public:

  MOZ_DECLARE_REFCOUNTED_VIRTUAL_TYPENAME(DataSourceSurfaceWrapper, override)

  explicit DataSourceSurfaceWrapper(DataSourceSurface *aSurface)

   : mSurface(aSurface)

  {}

  bool Equals(SourceSurface* aOther, bool aSymmetric = true) override

  {

    return DataSourceSurface::Equals(aOther, aSymmetric) ||

           mSurface->Equals(aOther, aSymmetric);

  }

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

  virtual uint8_t *GetData() override { return mSurface->GetData(); }

  virtual int32_t Stride() override { return mSurface->Stride(); }

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

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

  virtual bool IsValid() const override { return mSurface->IsValid(); }

  bool Map(MapType aType, MappedSurface *aMappedSurface) override

  {

    return mSurface->Map(aType, aMappedSurface);

  }

  void Unmap() override { mSurface->Unmap(); }

private:

  RefPtr<DataSourceSurface> mSurface;

};

} // namespace gfx

} // namespace mozilla

#endif /* MOZILLA_GFX_DATASOURCESURFACEWRAPPER_H_ */