/**
 * Copyright (c) 2014-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

#ifndef FBX2GLTF_MATERIALDATA_H
#define FBX2GLTF_MATERIALDATA_H

#include <string>

#include "Raw2Gltf.h"

struct Tex
{
    static std::unique_ptr<Tex> ref(const TextureData *tex, uint32_t texCoord = 0);
    explicit Tex(uint32_t texRef, uint32_t texCoord);

    const uint32_t texRef;
    const uint32_t texCoord;
};

struct KHRCommonMats
{
    enum MaterialType
    {
        Blinn,
        Phong,
        Lambert,
        Constant,
    };

    KHRCommonMats(
        MaterialType type,
        const TextureData *shininessTexture, float shininess,
        const TextureData *ambientTexture, const Vec3f &ambientFactor,
        const TextureData *diffuseTexture, const Vec4f &diffuseFactor,
        const TextureData *specularTexture, const Vec3f &specularFactor);

    static std::string typeDesc(MaterialType type);

    const MaterialType         type;
    const std::unique_ptr<Tex> shininessTexture;
    const float                shininess;
    const std::unique_ptr<Tex> ambientTexture;
    const Vec3f                ambientFactor;
    const std::unique_ptr<Tex> diffuseTexture;
    const Vec4f                diffuseFactor;
    const std::unique_ptr<Tex> specularTexture;
    const Vec3f                specularFactor;
};

struct PBRSpecularGlossiness
{
    PBRSpecularGlossiness(
        const TextureData *diffuseTexture, const Vec4f &diffuseFactor,
        const TextureData *specularGlossinessTexture,
        const Vec3f &specularFactor, float glossinessFactor);

    std::unique_ptr<Tex> diffuseTexture;
    const Vec4f          diffuseFactor;
    std::unique_ptr<Tex> specularGlossinessTexture;
    const Vec3f          specularFactor;
    const float          glossinessFactor;
};

struct PBRMetallicRoughness
{
    PBRMetallicRoughness(
        const TextureData *baseColorTexture, const Vec4f &baseolorFactor,
        float metallic = 0.1f, float roughness = 0.4f);

    std::unique_ptr<Tex> baseColorTexture;
    const Vec4f          baseColorFactor;
    const float          metallic;
    const float          roughness;
};

struct MaterialData : Holdable
{
    MaterialData(
        std::string name, bool isTransparent, const TextureData *normalTexture,
        const TextureData *emissiveTexture, const Vec3f &emissiveFactor,
        std::shared_ptr<KHRCommonMats> const khrCommonMats,
        std::shared_ptr<PBRMetallicRoughness> const pbrMetallicRoughness,
        std::shared_ptr<PBRSpecularGlossiness> const pbrSpecularGlossiness);

    json serialize() const override;

    const std::string                name;
    const bool                       isTransparent;
    const std::unique_ptr<const Tex> normalTexture;
    const std::unique_ptr<const Tex> emissiveTexture;
    const Vec3f                      emissiveFactor;

    const std::shared_ptr<const KHRCommonMats>         khrCommonMats;
    const std::shared_ptr<const PBRMetallicRoughness>  pbrMetallicRoughness;
    const std::shared_ptr<const PBRSpecularGlossiness> pbrSpecularGlossiness;
};

void to_json(json &j, const Tex &data);
void to_json(json &j, const KHRCommonMats &d);
void to_json(json &j, const PBRSpecularGlossiness &d);
void to_json(json &j, const PBRMetallicRoughness &d);

#endif //FBX2GLTF_MATERIALDATA_H