/*

 * Copyright 2021 Google LLC

 *

 * Use of this source code is governed by a BSD-style license that can be

 * found in the LICENSE file.

 */

#ifndef skiatest_graphite_ContextFactory_DEFINED

#define skiatest_graphite_ContextFactory_DEFINED

#include "include/core/SkRefCnt.h"

#include "include/gpu/graphite/ContextOptions.h"

#include "include/gpu/graphite/GraphiteTypes.h"

#include "include/private/base/SkTArray.h"

#include "tools/gpu/ContextType.h"

#include "tools/graphite/GraphiteTestContext.h"

#include "tools/graphite/TestOptions.h"

namespace skgpu::graphite {

class Context;

}

namespace skiatest::graphite {

struct ContextInfo {

    GraphiteTestContext* fTestContext = nullptr;

    skgpu::graphite::Context* fContext = nullptr;

};

class ContextFactory {

public:

    explicit ContextFactory(const TestOptions&);

    ContextFactory() = default;

    ContextFactory(const ContextFactory&) = delete;

    ContextFactory& operator=(const ContextFactory&) = delete;

    ~ContextFactory() = default;

    ContextInfo getContextInfo(skgpu::ContextType);

private:

    struct OwnedContextInfo {

        OwnedContextInfo();

        OwnedContextInfo(skgpu::ContextType,

                         std::unique_ptr<GraphiteTestContext>,

                         std::unique_ptr<skgpu::graphite::Context>);

        ~OwnedContextInfo();

        OwnedContextInfo(OwnedContextInfo&&);

        OwnedContextInfo& operator=(OwnedContextInfo&&);

        // This holds the same data as ContextInfo, but uses unique_ptr to maintain ownership.

        skgpu::ContextType fType = skgpu::ContextType::kMock;

        std::unique_ptr<GraphiteTestContext> fTestContext;

        std::unique_ptr<skgpu::graphite::Context> fContext;

    };

    static ContextInfo AsContextInfo(const OwnedContextInfo& ctx);

    skia_private::TArray<OwnedContextInfo> fContexts;

    const TestOptions fOptions = {};

};

}  // namespace skiatest::graphite

#endif  // skiatest_graphite_ContextFactory_DEFINED