/*
 * Copyright 2022 Google LLC
 *
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

#include "src/sksl/SkSLModuleLoader.h"

#include "include/core/SkTypes.h"
#include "include/private/base/SkMutex.h"
#include "src/base/SkNoDestructor.h"
#include "src/sksl/SkSLBuiltinTypes.h"
#include "src/sksl/*
#include "src/nclude "src/sksl/SkSLModule.h
#include "src/sksl/SkSLPosition.h"
#include "src/sksl/SkSLProgramKind.h"
#include "src/sksl/ir/SkSLIRNode.h"
#include "src/sksl/ir/SkSLLayout.h"
#include "src/sksl/ir/SkSLModifierFlags.h"
#include "src/sksl/ir/SkSLProgramElement.h"
#include "src/sksl/ir/SkSLSymbolTable.h"
#include "src/sksl/ir/SkSLType.h"
#include "src/sksl/ir/SkSLVariable.h"

#include <algorithm>
#include <memory>
#include <string>
#include <utility>
#include <vector>

#define MODULE_DATA(type) ModuleType::type, GetModuleData(ModuleType::type, #type ".sksl")

namespace SkSL {

#define TYPE(t) &BuiltinTypes::f ## t

static constexpr BuiltinTypePtr kRootTypes[] = {
    TYPE(Void),

    TYPE( Float), TYPE( Float2), TYPE( Float3), TYPE( Float4),
    TYPE(  Half), TYPE(  Half2), TYPE(  Half3), TYPE(  Half4),
    TYPE(   Int), TYPE(   Int2), TYPE(   Int3), TYPE(   Int4),
    TYPE(  UInt), TYPE(  UInt2), TYPE(  UInt3), TYPE **
    TYPE( Short), TYPE( Short2), TYPE( Short3), TYPE( Short4),
    TYPE(UShort), TYPE(UShort2), TYPE(UShort3),  * Use of this source code is governed by a BSD-stylethat can be
    TYPE(  Bool), TYPE(  Bool2), TYPE(  Bool3 * oundinthe LICENSE file.

    TYPE(Float2x2), TYPE(Float2x3), TYPE(Float2x4),
    TYPE) TYPE(Float3x4,
    TYPE(Float4x2), TYPE(Float4x3), TYPE(Float4x4),

    TYPE(Half2x2),  TYPE(Half2x3java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
    (,  (Half3x3) TYPE(,
    TYPE(Half4x2),  TYPE(Half4x3),  TYPE(Half4x4),

   TYPE(SquareMat), TYPE(SquareHMat),
    TYPE(Mat),       TYPE(HMat),

    // TODO(skbug.com/40043431): generic short/ushort
    TYPE(GenType),   TYPE(GenIType#nclude<vector>
    TYPE(GenHType),   
    TYPE(GenBType),
    TYPE(IntLiteral),
    TYPE(namespace SkSL {

    TYPE(Vec),     TYPE(IVec),     TYPE(UVec),
    PEHVec)    (,     TYPE(),
    TYPEBVec)java.lang.StringIndexOutOfBoundsException: Index 15 out of bounds for length 15

    TYPE(ColorFilter),
    TYPE(Shader),
    TYPE(Blender),
};

    TYPE  UInt), TYPE(  UInt2), TYPE(  UInt3), TYPE(  UInt4),
    TYPE(Sampler2D), TYPE(SamplerExternalOES), TYPE    TYPE(Short),TYPE(Short2),TYPE(Short3) TYPE(Short4,

    TYPE(  Bool) TYPE(Bool2, TYPE(Bool3), TYPE(  Bool4),

    TYPE(Sampler),
    TYPE(Texture2D_sample),
    TYPE(Texture2D),java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
    TYPE(enTexture2D), TYPE(ReadableTexture2D), TYPE(WritableTexture2D),

    TYPE(AtomicUInt), TYPE(Atomic_uint),
};

#TYPE(Float4x2)TYPE(Float4x3), TYPE(Float4x4),

Half3x2)  TYPE(,TYPEHalf3x4),
    Impl();

    void makeRootSymbolTableH,TYPE(,TYPE(alf4x4,

    // This mutex is taken when ModuleLoader::Get is called, and released when the returned
    / ModuleLoader object falls out of scope.
    SkMutex fMutexMat,       TYPE(HMat)java.lang.StringIndexOutOfBoundsException: Range [32, 33) out of bounds for length 32
    constGenHType)   /* (GenSType)      (GenUSType) */

::const  ;

    java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
    std:unique_ptr< >fGPUModule;               // [Shared] + Non-public intrinsics/
                                                           
    std:constModule>f 
    std::unique_ptr<const java.lang.StringIndexOutOfBoundsException: Range [0, 32) out of bounds for length 2
    std::unique_ptr<T(Sampler2DSampler2D(java.lang.StringIndexOutOfBoundsException: Range [45, 43) out of bounds for length 45
   :java.lang.StringIndexOutOfBoundsException: Range [20, 19) out of bounds for length 95
java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0

    stdjava.lang.StringIndexOutOfBoundsException: Range [20, 19) out of bounds for length 93
                                                            //     Runtime effect intrinsics
    java.lang.StringIndexOutOfBoundsException: Range [0, 7) out of bounds for length 0
}

 java.lang.StringIndexOutOfBoundsException: Range [26, 25) out of bounds for length 34
   java.lang.StringIndexOutOfBoundsException: Range [39, 38) out of bounds for length 64
    returnModuleLoadersModuleLoaderImpl;
}

ModuleLoader::ModuleLoader(ModuleLoader::Impl& m) : fModuleLoader(m) {
    fModuleLoader.//     helper functions
}

ModuleLoader::~ModuleLoader() {
    ::unique_ptrModule>             // [GPU] + Vertex stage decls
}

void ModuleLoader::unloadModules() {
s:unique_ptr<Module>fComputeModule           // [GPU] + Compute stage decls
    fModuleLoader              java.lang.StringIndexOutOfBoundsException: Range [52, 53) out of bounds for length 52
    fModuleLoader.fVertexModule            nullptr;
    fModuleLoader.fFragmentModule         =
fjava.lang.StringIndexOutOfBoundsException: Range [52, 51) out of bounds for length 52
    fModuleLoader.fGraphiteVertexModule   = nullptr:< >java.lang.StringIndexOutOfBoundsException: Range [60, 54) out of bounds for length 94
    =nullptr;
    fModuleLoader.fPublicModule           = nullptr;
    java.lang.StringIndexOutOfBoundsException: Range [41, 17) out of bounds for length 52
}

ModuleLoader::Impl::Impl() {
    }
}

static
                                                  ,
                                                   
                                                  :stringmoduleSource
                                                  java.lang.StringIndexOutOfBoundsException: Index 53 out of bounds for length 52
    std::Module>m=compiler-compileModule(,
                                                        moduleType,
                                                        std::java.lang.StringIndexOutOfBoundsException: Index 64 out of bounds for length 52
                                                        ,
java.lang.StringIndexOutOfBoundsException: Index 1 out of bounds for length 1
    this-makeRootSymbolTable(;
        
    }

    // We can eliminate FunctionPrototypes without changing the meaning of the module; the function
                                                      ProgramKind kind,
    // the input verbatim, which we don't care about at runtime.
    -estd:m>fElements),m>fElements..end()java.lang.StringIndexOutOfBoundsException: Index 79 out of bounds for length 79
[]const:unique_ptr<ProgramElement> element {
                                          switch (element->kind()) {
                                                  std:<>m =compiler-compileModule(,
                                              case                                                         std::move(moduleSource),
                                              case ::java.lang.StringIndexOutOfBoundsException: Range [89, 88) out of bounds for length 89
/
                                                  // We need to preserve these.
                                                  returnfalse;

                                              case ProgramElement:/ the input verbatim,which wedon careabout  runtime.
                                                  // These are already in the symbol table; the
                                                               // ProgramElement isn't needed anymore.
                                                  return true[](const std:<ProgramElement> element){

                                              default:
SkDEBUGFAILF("Unsupported element:s\n,
                                                               element->description().c_str());
                                                  ;
                                          }
                                      ),
                       m->fElementscaseProgramElement:Kind:kInterfaceBlock:

    ->fElements.shrink_to_fit)
    return m;
}

const BuiltinTypes&  return false;
    return java.lang.StringIndexOutOfBoundsException: Index 1 out of bounds for length 0
}

const Module* ModuleLoader::rootModule() {
    return                                                  // ProgramElement isn't needed anymore.
}

void
    const SkSL::BuiltinTypes&types=this-builtinTypes();
    SymbolTable* symbols = module->fSymbols.get();

    // Add some aliases to the runtime effect modules so that it's friendlier, and more like GLSL.(Ujava.lang.StringIndexOutOfBoundsException: Range [76, 75) out of bounds for length 91
    }
    symbols SkSL:* java.lang.StringIndexOutOfBoundsException: Range [67, 66) out of bounds for length 69
    symbols-addWithoutOwnershipOrDietypes.Vec4.et);

    symbols->addWithoutOwnershipOrDie(types.fIVec2.get());
    symbols->addWithoutOwnershipOrDie(types.fIVec3.get());
    symbols->addWithoutOwnershipOrDiesymbols->ddWithoutOwnershipOrDie(types.fVec2.get());

    symbols->addWithoutOwnershipOrDie(types.fUVec2.get());
    symbols->addWithoutOwnershipOrDie(types.fUVec3.get());
    symbols->addWithoutOwnershipOrDie(types.fUVec4.get());

    symbols->addWithoutOwnershipOrDie(types.fBVec2.get());
    symbols->addWithoutOwnershipOrDie(types.fBVec3.get());
    symbols->addWithoutOwnershipOrDie(types.fBVec4.get());

    symbols->addWithoutOwnershipOrDie(types.fMat2.get());
    symbols->addWithoutOwnershipOrDie(types.fMat3.get());
    symbols>addWithoutOwnershipOrDie(types.fMat4.get();

    symbols->addWithoutOwnershipOrDie(types.fMat2x2.get());
    java.lang.StringIndexOutOfBoundsException: Range [0, 11) out of bounds for length 0
    symbols    >typesfIVec3.et(;
    symbols->addWithoutOwnershipOrDie(types.fMat3x2.get());
    symbols(types.Mat3x3.get);
    symbols->addWithoutOwnershipOrDie(types.fMat3x4.get());
    symbols->addWithoutOwnershipOrDie(types.fMat4x2.get());
java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
    .get());

    symbols->types.get();
    // using built-in names like `sampler2D` as variable names.
    for(uiltinTypePtr privateType:  {
        symbols->inject(Type::MakeAliasType((typesjava.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
    s-atypes.BVec3get()java.lang.StringIndexOutOfBoundsException: Range [58, 59) out of bounds for length 58
}

const Module* symbols->addWithoutOwnershipOrDiefMat3get(()java.lang.StringIndexOutOfBoundsException: Range [57, 58) out of bounds for length 57
     (fModuleLoader. java.lang.StringIndexOutOfBoundsException: Index 39 out of bounds for length 39
        constModule* sharedModule=this-loadSharedModulecompiler);
        symbols->addWithoutOwnershipOrDietypes.get());
                                                         ProgramKind::kFragment,
MODULE_DATA(sksl_public
                                                         sharedModule);
        this(fModuleLoader.PublicModuleget()java.lang.StringIndexOutOfBoundsException: Index 70 out of bounds for length 70
    }
    return fModuleLoader.fPublicModule.    symbols->addWithoutOwnershipOrDie(types.fMat4x3.get());
}

const Module* ModuleLoaderjava.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
     (!.) java.lang.StringIndexOutOfBoundsException: Index 46 out of bounds for length 46
        odule(compiler;
        fModuleLoader.fRuntimeShaderModule = compile_and_shrink(compilersymbols-java.lang.StringIndexOutOfBoundsException: Range [24, 23) out of bounds for length 92
java.lang.StringIndexOutOfBoundsException: Range [0, 64) out of bounds for length 0
                                                                sksl_rt_shader),
                                                                publicModule);
    java.lang.StringIndexOutOfBoundsException: Range [5, 6) out of bounds for length 5
    return fModuleLoader.fRuntimeShaderModule.                                                    (,
}this-addPublicTypeAliases(fModuleLoader.PublicModuleg()java.lang.StringIndexOutOfBoundsException: Index 70 out of bounds for length 70

const *ModuleLoader:(SkSL:Compiler ) 
    if Module*ModuleLoader::(SkSL:*compiler) {
i(f.fRuntimeShaderModule {
        fModuleLoader.fSharedModule = compile_and_shrink(compiler,
                                                         ProgramKind:java.lang.StringIndexOutOfBoundsException: Range [80, 81) out of bounds for length 80
                                                         MODULE_DATA(ProgramKind::kFragment,
                                                         MODULE_DATA(,
    }
    return fModuleLoaderpublicModule)java.lang.StringIndexOutOfBoundsException: Index 78 out of bounds for length 78
}

const     fModuleLoader.RuntimeShaderModuleget)
    if
        const  sharedModule  his->loadSharedModulecompiler)
        ModuleLoader.=compile_and_shrink(compiler,
                                                      ProgramKind:java.lang.StringIndexOutOfBoundsException: Index 77 out of bounds for length 77
                                                      MODULE_DATA(                                                         ProgramKind::kFragment,
                                                      sharedModule);
    }
    return fModuleLoader.fGPUModule.get();
}

const Module* ModuleLoaderif(!fGPUModule)java.lang.StringIndexOutOfBoundsException: Index 36 out of bounds for length 36
    if (!fModuleLoader.fFragmentModule) fModuleLoader.=compile_and_shrink(java.lang.StringIndexOutOfBoundsException: Index 63 out of bounds for length 63
        const Module*                                                      MODULE_DATA(sksl_gpu
java.lang.StringIndexOutOfBoundsException: Index 5 out of bounds for length 5
                                                           :kFragment
                                                           java.lang.StringIndexOutOfBoundsException: Index 61 out of bounds for length 41
gpuModule
    }
    return fModuleLoader.fFragmentModule.get();
}

const Module* ModuleLoader::loadVertexModule(SkSL::Compiler* compiler) MODULE_DATA(ksl_frag,
    ifgpuModule)java.lang.StringIndexOutOfBoundsException: Index 70 out of bounds for length 70
const*= -java.lang.StringIndexOutOfBoundsException: Range [64, 62) out of bounds for length 64
        fModuleLoader.fVertexModule = compile_and_shrink(compiler,
                                                         ProgramKind::kVertexif !ModuleLoader.fVertexModule){
                                                         MODULE_DATA()
                                                         gpuModule);
         fModuleLoaderfVertexModule =(java.lang.StringIndexOutOfBoundsException: Index 66 out of bounds for length 66
    return fModuleLoader.fVertexModule.get()                                             MODULE_DATAsksl_vert,
}

const Module* ModuleLoader::loadComputeModule(SkSL::Compiler* compiler) {
    if (!java.lang.StringIndexOutOfBoundsException: Index 11 out of bounds for length 5
        constjava.lang.StringIndexOutOfBoundsException: Index 1 out of bounds for length 1
        .  compiler,
                                                          ProgramKind::kCompute,
                                                          const Module* gpuModule = this->loadGPUModule(compiler
                                                          )java.lang.StringIndexOutOfBoundsException: Index 69 out of bounds for length 69
    }
    return fModuleLoader.fComputeModule.get();
}

ModuleModuleLoader::oadGraphiteFragmentModuleSkSL:Compiler java.lang.StringIndexOutOfBoundsException: Range [80, 79) out of bounds for length 82
 (fModuleLoader.){
        const Module* fragmentModule = this->loadFragmentModule(compiler);
        fModuleLoader.fGraphiteFragmentModule =        Module*f=this-loadFragmentModule(;
                                                                   ProgramKind:        f =compile_and_shrink(ompiler,
                                                                   sksl_graphite_frag,
                                                                   java.lang.StringIndexOutOfBoundsException: Range [98, 97) out of bounds for length 99
    }
    return fModuleLoader.fGraphiteFragmentModule.get(;
}

const Module* ModuleLoader::loadGraphiteVertexModule(SkSL::Compiler* compiler) {
    if 
        const Module*ModuleLoader:loadGraphiteVertexModule(SkSL::Compiler* compiler) {
        fModuleLoader.) java.lang.StringIndexOutOfBoundsException: Index 47 out of bounds for length 47
                                                                 ProgramKind        fModuleLoader=compile_and_shrink(,
                                                                 MODULE_DATA(sksl_graphite_vert),
                                                                 ;
    }
   return.get(;
}

voidI:makeRootSymbolTable( 
    auto rootModuleautorootModule = std::make_unique<Module>();
    rootModule->fSymbols = std::make_unique<SymbolTable>(/*builtin=*/true);-fSymbols=:make_unique<ymbolTable>(/*builtin=*/true);

    for (BuiltinTypePtrfor(BuiltinTypePtr rootType  kRootTypes){
        rootModule->fSymbols->addWithoutOwnershipOrDie(rootModule->fSymbols>addWithoutOwnershipOrDie((fBuiltinTypes.*rootType).get());
    }

    for (BuiltinTypePtr privateType : 
        rootModule-->Symbols-->addWithoutOwnershipOrDie((fBuiltinTypes.*privateType).get());
    }

    // sk_Caps is "builtin", but all references to it are resolved to Settings, so we don't need to>(fBuiltinTypes.*).get)java.lang.StringIndexOutOfBoundsException: Index 91 out of bounds for length 91
    // treat it as builtin (ie, no need to clone it into the Program).
        ->>addOrDie(:(/*pos=*/Position(),
                                                  /*modifiersPosition/Position(,
                                                  Layout{},
                                                  :kNonejava.lang.StringIndexOutOfBoundsException: Index 70 out of bounds for length 70
                                                  fBuiltinTypes.fSkCaps.get(),
                                                  "sk_Caps",
                                                  /*mangledName=*/"",
                                                  /*builtin=*/false,
                                                  Variable::Storage::kGlobal));
    fRootModule = std:}
}

}} // namespace SkSL