<?php
namespace Std\Lib\Syntax;

/**
 * Slang（HLSLの上位互換）
 */
class Slang extends Hlsl {
  protected array $rules = [
    // 属性 [[vk::binding(0, 0)]]
    ['re' => '\[\[[^\]\n]*\]\]', 'class' => 'attribute'],
  ];

  /** @var array HLSLに加えるSlangのトークン */
  protected array $slangTokens = [
    'keyword' => [
      'import', 'module', 'implementing', '__include', '__exported', 'public', 'private', 'internal', 'extension', 'associatedtype',
      'where', 'let', 'var', 'func', 'property', 'get', 'set', '__init', '__subscript', 'This', 'typealias', 'throw', 'throws',
      'try', 'catch', 'defer', 'is', 'as', 'new', 'some', 'dyn', 'ref', '__ref', '__generic', '__target_switch', 'spirv_asm',
      '__intrinsic_asm', 'constexpr', 'override', 'no_diff', 'fwd_diff', 'bwd_diff', 'alignof', 'countof',
    ],
    'constant' => [
      'none', 'nullptr',
    ],
    'type' => [
      'Ptr', 'Optional', 'Tuple', 'Array', 'DescriptorHandle', 'ParameterBlock', 'NativeString', 'String', 'int8_t', 'uint8_t',
      'Sampler1D', 'Sampler2D', 'Sampler3D', 'SamplerCube', 'Sampler1DArray', 'Sampler2DArray', 'SamplerCubeArray', 'Sampler2DShadow',
      'Sampler2DMS', 'Sampler2DMSArray', 'Texture2DShadow', 'TextureCubeShadow', 'RWTexture2DArray', 'Atomic',
      'IDifferentiable', 'IDifferentiablePtrType', 'DifferentialPair', 'IArithmetic', 'IFloat', 'IInteger', 'ILogical', 'IComparable',
      'IRangedValue', 'IArray', 'IRWArray', 'IBufferDataLayout', 'DefaultDataLayout', 'Std140DataLayout', 'Std430DataLayout', 'ScalarDataLayout',
    ],
    'builtin' => [
      'detach', 'diffPair', 'updatePrimal', 'updateDiff', 'getDiff', 'getPrimal', 'isDifferentialNull',
      'printf', 'bit_cast', 'reinterpret', 'createDynamicObject', 'getStringHash', 'static_assert',
    ],
  ];

  protected array $attributes = [
    // HLSL
    'numthreads', 'unroll', 'loop', 'branch', 'flatten', 'fastopt', 'allow_uav_condition', 'call', 'forcecase', 'earlydepthstencil',
    'maxvertexcount', 'domain', 'partitioning', 'outputtopology', 'outputcontrolpoints', 'patchconstantfunc', 'maxtessfactor',
    'instance', 'RootSignature', 'shader', 'WaveSize', 'NumThreads', 'vk',
    // Slang
    'Differentiable', 'ForwardDifferentiable', 'BackwardDifferentiable', 'BackwardDerivative', 'ForwardDerivative', 'PrimalSubstitute',
    'PreferRecompute', 'PreferCheckpoint', 'TreatAsDifferentiable', 'mutating', 'nonmutating', 'ForceInline', 'noinline', 'require',
    'SpecializationConstant', 'push_constant', 'Payload', 'raypayload', 'AutoPyBindCUDA', 'CudaKernel', 'TorchEntryPoint',
    'Format', 'OverloadRank', 'Deprecated', 'KnownBuiltin', 'anyValueSize', 'COM', 'Flags', 'UnscopedEnum', 'ForceUnroll',
  ];

  public function prepare(string $code): void {
    $tokens = $this->tokens;
    foreach ($this->slangTokens as $k => $list) {
      foreach ($list as $w) $tokens[$k][] = $w;
    }
    foreach (self::vectorTypes($this->scalars) as $t) $tokens['type'][] = $t;
    foreach (self::vectorTypes(['int8_t', 'uint8_t']) as $t) $tokens['type'][] = $t;
    $this->words = $this->buildWords($tokens);
  }
}