1+ /**
2+ * @license
3+ * Copyright 2026 Google Inc.
4+ * Licensed under the Apache License, Version 2.0 (the "License");
5+ * you may not use this file except in compliance with the License.
6+ * You may obtain a copy of the License at
7+ *
8+ * http://www.apache.org/licenses/LICENSE-2.0
9+ *
10+ * Unless required by applicable law or agreed to in writing, software
11+ * distributed under the License is distributed on an "AS IS" BASIS,
12+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+ * See the License for the specific language governing permissions and
14+ * limitations under the License.
15+ */
16+
17+ export type ShaderStringLiteralIdMap = Map < string , number > ;
18+
19+ export interface ShaderSourcePreprocessingData {
20+ stringLiteralIds : ShaderStringLiteralIdMap ;
21+ }
22+
23+ export interface ShaderStringPreprocessingResult {
24+ code : string ;
25+ stringLiteralIds : ShaderStringLiteralIdMap ;
26+ }
27+
28+ export function preprocessStrings (
29+ userShader : string ,
30+ ) : ShaderStringPreprocessingResult {
31+ const stringLiteralIds = new Map < string , number > ( ) ;
32+ const code = userShader . replace ( / " (?: \\ .| [ ^ \\ " ] ) * " / g, ( token ) => {
33+ const value = JSON . parse ( token ) as string ;
34+ let index = stringLiteralIds . get ( value ) ;
35+ if ( index === undefined ) {
36+ index = stringLiteralIds . size + 1 ;
37+ stringLiteralIds . set ( value , index ) ;
38+ }
39+ return `${ index } u` ;
40+ } ) ;
41+ return { code, stringLiteralIds } ;
42+ }
0 commit comments