onnxruntime

1
// Copyright (c) Microsoft Corporation. All rights reserved.
2
// Licensed under the MIT License.
3

4
import { getGlChannels } from '../utils';
5

6
export function getVecChannels(name: string, rank: number): string[] {
7
  return getGlChannels(rank).map((d) => `${name}.${d}`);
8
}
9

10
export function getChannels(name: string, rank: number): string[] {
11
  if (rank === 1) {
12
    return [name];
13
  }
14
  return getVecChannels(name, rank);
15
}
16

17
export function unpackFromChannel(): string {
18
  return `
19
    float getChannel(vec4 frag, int dim) {
20
      int modCoord = imod(dim, 2);
21
      return modCoord == 0 ? frag.r : frag.g;
22
    }
23

24
    float getChannel(vec4 frag, vec2 innerDims) {
25
      vec2 modCoord = mod(innerDims, 2.);
26
      return modCoord.x == 0. ?
27
        (modCoord.y == 0. ? frag.r : frag.g) :
28
        (modCoord.y == 0. ? frag.b : frag.a);
29
    }
30
  `;
31
}
32