<?php
namespace Std\Lib\Syntax;

class Glsl extends Language {
  protected bool $preprocessor = true;
  protected array $headerDirectives = ['include'];
  protected array $strings = ['"'];
  protected string $digitSep = '';

  protected array $tokens = [
    'keyword' => [
      'attribute', 'const', 'uniform', 'varying', 'buffer', 'shared', 'layout', 'centroid', 'flat', 'smooth', 'noperspective',
      'patch', 'sample', 'break', 'continue', 'do', 'for', 'while', 'switch', 'case', 'default', 'if', 'else', 'subroutine',
      'in', 'out', 'inout', 'invariant', 'precise', 'discard', 'return', 'struct', 'precision', 'highp', 'mediump', 'lowp',
      'coherent', 'volatile', 'restrict', 'readonly', 'writeonly', 'demote', 'terminateInvocation', 'rayPayloadEXT', 'hitAttributeEXT',
      'callableDataEXT', 'taskPayloadSharedEXT', 'perprimitiveEXT',
      // layout()の中（「offset」「points」等、変数名にも使う単語は除く）
      'location', 'binding', 'set', 'push_constant', 'std140', 'std430', 'local_size_x', 'local_size_y', 'local_size_z',
      'input_attachment_index', 'constant_id', 'row_major', 'column_major', 'early_fragment_tests', 'max_vertices',
    ],
    'constant' => [
      'true', 'false', '__FILE__', '__LINE__', '__VERSION__', 'GL_ES',
    ],
    'type' => [
      'void', 'bool', 'int', 'uint', 'float', 'double', 'atomic_uint', 'genType', 'genIType', 'genUType', 'genBType', 'genDType',
      'int8_t', 'int16_t', 'int32_t', 'int64_t', 'uint8_t', 'uint16_t', 'uint32_t', 'uint64_t', 'float16_t', 'float32_t', 'float64_t',
      'accelerationStructureEXT', 'rayQueryEXT',
      'subpassInput', 'subpassInputMS', 'isubpassInput', 'isubpassInputMS', 'usubpassInput', 'usubpassInputMS', 'sampler', 'samplerShadow',
    ],
    'builtin' => [
      // 角度・三角関数
      'radians', 'degrees', 'sin', 'cos', 'tan', 'asin', 'acos', 'atan', 'sinh', 'cosh', 'tanh', 'asinh', 'acosh', 'atanh',
      // 指数
      'pow', 'exp', 'log', 'exp2', 'log2', 'sqrt', 'inversesqrt',
      // 共通
      'abs', 'sign', 'floor', 'trunc', 'round', 'roundEven', 'ceil', 'fract', 'mod', 'modf', 'min', 'max', 'clamp', 'mix', 'step', 'smoothstep',
      'isnan', 'isinf', 'floatBitsToInt', 'floatBitsToUint', 'intBitsToFloat', 'uintBitsToFloat', 'fma', 'frexp', 'ldexp',
      'packUnorm2x16', 'packSnorm2x16', 'packUnorm4x8', 'packSnorm4x8', 'unpackUnorm2x16', 'unpackSnorm2x16', 'unpackUnorm4x8', 'unpackSnorm4x8',
      'packHalf2x16', 'unpackHalf2x16', 'packDouble2x32', 'unpackDouble2x32',
      // 幾何
      'length', 'distance', 'dot', 'cross', 'normalize', 'faceforward', 'reflect', 'refract', 'ftransform',
      // 行列・ベクトル比較
      'matrixCompMult', 'outerProduct', 'transpose', 'determinant', 'inverse',
      'lessThan', 'lessThanEqual', 'greaterThan', 'greaterThanEqual', 'equal', 'notEqual', 'any', 'all', 'not',
      // 整数
      'uaddCarry', 'usubBorrow', 'umulExtended', 'imulExtended', 'bitfieldExtract', 'bitfieldInsert', 'bitfieldReverse', 'bitCount', 'findLSB', 'findMSB',
      // テクスチャ
      'textureSize', 'textureQueryLod', 'textureQueryLevels', 'textureSamples', 'texture', 'textureProj', 'textureLod', 'textureOffset',
      'texelFetch', 'texelFetchOffset', 'textureProjOffset', 'textureLodOffset', 'textureProjLod', 'textureProjLodOffset',
      'textureGrad', 'textureGradOffset', 'textureProjGrad', 'textureProjGradOffset', 'textureGather', 'textureGatherOffset', 'textureGatherOffsets',
      'texture1D', 'texture1DProj', 'texture1DLod', 'texture1DProjLod', 'texture2D', 'texture2DProj', 'texture2DLod', 'texture2DProjLod',
      'texture3D', 'texture3DProj', 'texture3DLod', 'texture3DProjLod', 'textureCube', 'textureCubeLod',
      'shadow1D', 'shadow1DProj', 'shadow1DLod', 'shadow1DProjLod', 'shadow2D', 'shadow2DProj', 'shadow2DLod', 'shadow2DProjLod',
      // 原子操作・画像
      'atomicCounterIncrement', 'atomicCounterDecrement', 'atomicCounter', 'atomicAdd', 'atomicMin', 'atomicMax', 'atomicAnd', 'atomicOr',
      'atomicXor', 'atomicExchange', 'atomicCompSwap', 'imageSize', 'imageSamples', 'imageLoad', 'imageStore', 'imageAtomicAdd',
      'imageAtomicMin', 'imageAtomicMax', 'imageAtomicAnd', 'imageAtomicOr', 'imageAtomicXor', 'imageAtomicExchange', 'imageAtomicCompSwap',
      // 微分・補間・ジオメトリ・同期
      'dFdx', 'dFdy', 'dFdxFine', 'dFdyFine', 'dFdxCoarse', 'dFdyCoarse', 'fwidth', 'fwidthFine', 'fwidthCoarse',
      'interpolateAtCentroid', 'interpolateAtSample', 'interpolateAtOffset', 'EmitStreamVertex', 'EndStreamPrimitive', 'EmitVertex', 'EndPrimitive',
      'barrier', 'memoryBarrier', 'memoryBarrierAtomicCounter', 'memoryBarrierBuffer', 'memoryBarrierShared', 'memoryBarrierImage', 'groupMemoryBarrier',
      'subpassLoad', 'noise1', 'noise2', 'noise3', 'noise4',
      // サブグループ・レイトレーシング
      'subgroupBarrier', 'subgroupElect', 'subgroupAll', 'subgroupAny', 'subgroupBroadcast', 'subgroupBroadcastFirst', 'subgroupBallot',
      'subgroupAdd', 'subgroupMul', 'subgroupMin', 'subgroupMax', 'subgroupShuffle', 'subgroupShuffleXor',
      'traceRayEXT', 'reportIntersectionEXT', 'executeCallableEXT', 'ignoreIntersectionEXT', 'terminateRayEXT', 'EmitMeshTasksEXT', 'SetMeshOutputsEXT',
    ],
  ];

  public function prepare(string $code): void {
    $tokens = $this->tokens;
    // ベクトル・行列
    foreach (['vec', 'ivec', 'uvec', 'bvec', 'dvec', 'i8vec', 'u8vec', 'i16vec', 'u16vec', 'i64vec', 'u64vec', 'f16vec', 'f64vec'] as $v) {
      for ($i = 2; $i <= 4; $i++) $tokens['type'][] = "{$v}{$i}";
    }
    foreach (['mat', 'dmat', 'f16mat', 'f64mat'] as $m) {
      for ($i = 2; $i <= 4; $i++) {
        $tokens['type'][] = "{$m}{$i}";
        for ($j = 2; $j <= 4; $j++) $tokens['type'][] = "{$m}{$i}x{$j}";
      }
    }
    // サンプラー・テクスチャ・画像
    $dims = ['1D', '2D', '3D', 'Cube', '2DRect', '1DArray', '2DArray', 'CubeArray', 'Buffer', '2DMS', '2DMSArray'];
    foreach (['', 'i', 'u'] as $p) {
      foreach ($dims as $d) {
        $tokens['type'][] = "{$p}sampler{$d}";
        $tokens['type'][] = "{$p}texture{$d}";
        $tokens['type'][] = "{$p}image{$d}";
      }
    }
    foreach (['1D', '2D', 'Cube', '2DRect', '1DArray', '2DArray', 'CubeArray'] as $d) $tokens['type'][] = "sampler{$d}Shadow";
    $this->words = $this->buildWords($tokens);
  }

  public function classify(string $word, array $ctx): ?string {
    // 組み込み変数（gl_Position, gl_GlobalInvocationID等）
    if (str_starts_with($word, 'gl_')) return 'variable';
    return parent::classify($word, $ctx);
  }
}