You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

244 lines
5.9 KiB

function splitOnSpaceHandleQuotesWithEscapes(str, splits = ' \t\n\r') {
const strings = [];
let quoteType;
let escape;
let s = [];
for (let i = 0; i < str.length; ++i) {
const c = str[i];
if (escape) {
escape = false;
s.push(c);
} else {
if (quoteType) { // we're inside quotes
if (c === quoteType) {
quoteType = undefined;
strings.push(s.join(''));
s = [];
} else if (c === '\\') {
escape = true;
} else {
s.push(c);
}
} else { // we're not in quotes
if (splits.indexOf(c) >= 0) {
if (s.length) {
strings.push(s.join(''));
s = [];
}
} else if (c === '"' || c === '\'') {
if (s.length) { // its in th middle of a word
s.push(c);
} else {
quoteType = c;
}
} else {
s.push(c);
}
}
}
}
if (s.length || strings.length === 0) {
strings.push(s.join(''));
}
return strings;
}
const startWhitespaceRE = /^\s/;
const intRE = /^\d+$/;
const isNum = s => intRE.test(s);
const quotesRE = /^".*"$/;
function trimQuotes(s) {
return quotesRE.test(s) ? s.slice(1, -1) : s;
}
const splitToNumbers = s => s.split(' ').map(parseFloat);
export function parseCSP(str) {
const data = [];
const lut = {
name: 'unknown',
type: '1D',
size: 0,
data,
min: [0, 0, 0],
max: [1, 1, 1],
};
const lines = str.split('\n').map(s => s.trim()).filter(s => s.length > 0 && !startWhitespaceRE.test(s));
// check header
lut.type = lines[1];
if (lines[0] !== 'CSPLUTV100' ||
(lut.type !== '1D' && lut.type !== '3D')) {
throw new Error('not CSP');
}
// skip meta (read to first number)
let lineNdx = 2;
for (; lineNdx < lines.length; ++lineNdx) {
const line = lines[lineNdx];
if (isNum(line)) {
break;
}
if (line.startsWith('TITLE ')) {
lut.name = trimQuotes(line.slice(6).trim());
}
}
// read ranges
for (let i = 0; i < 3; ++i) {
++lineNdx;
const input = splitToNumbers(lines[lineNdx++]);
const output = splitToNumbers(lines[lineNdx++]);
if (input.length !== 2 || output.length !== 2 ||
input[0] !== 0 || input[1] !== 1 ||
output[0] !== 0 || output[1] !== 1) {
throw new Error('mapped ranges not support');
}
}
// read sizes
const sizes = splitToNumbers(lines[lineNdx++]);
if (sizes[0] !== sizes[1] || sizes[0] !== sizes[2]) {
throw new Error('only cubic sizes supported');
}
lut.size = sizes[0];
// read data
for (; lineNdx < lines.length; ++lineNdx) {
const parts = splitToNumbers(lines[lineNdx]);
if (parts.length !== 3) {
throw new Error('malformed file');
}
data.push(...parts);
}
return lut;
}
export function parseCUBE(str) {
const data = [];
const lut = {
name: 'unknown',
type: '1D',
size: 0,
data,
min: [0, 0, 0],
max: [1, 1, 1],
};
const lines = str.split('\n');
for (const origLine of lines) {
const hashNdx = origLine.indexOf('#');
const line = hashNdx >= 0 ? origLine.substring(0, hashNdx) : origLine;
const parts = splitOnSpaceHandleQuotesWithEscapes(line);
switch (parts[0].toUpperCase()) {
case 'TITLE':
lut.name = parts[1];
break;
case 'LUT_1D_SIZE':
lut.size = parseInt(parts[1]);
lut.type = '1D';
break;
case 'LUT_3D_SIZE':
lut.size = parseInt(parts[1]);
lut.type = '3D';
break;
case 'DOMAIN_MIN':
lut.min = parts.slice(1).map(parseFloat);
break;
case 'DOMAIN_MAX':
lut.max = parts.slice(1).map(parseFloat);
break;
default:
if (parts.length === 3) {
data.push(...parts.map(parseFloat));
}
break;
}
}
if (!lut.size) {
lut.size = lut.type === '1D'
? (data.length / 3)
: Math.cbrt(data.length / 3);
}
return lut;
}
function lerp(a, b, t) {
return a + (b - a) * t;
}
function lut1Dto3D(lut) {
let src = lut.data;
if (src.length / 3 !== lut.size) {
src = [];
for (let i = 0; i < lut.size; ++i) {
const u = i / lut.size * lut.data.length;
const i0 = (u | 0) * 3;
const i1 = i0 + 3;
const t = u % 1;
src.push(
lerp(lut.data[i0 + 0], lut.data[i1 + 0], t),
lerp(lut.data[i0 + 0], lut.data[i1 + 1], t),
lerp(lut.data[i0 + 0], lut.data[i1 + 2], t),
);
}
}
const data = [];
for (let i = 0; i < lut.size * lut.size; ++i) {
data.push(...src);
}
return {...lut, data};
}
const parsers = {
'cube': parseCUBE,
'csp': parseCSP,
};
// for backward compatibility
export function parse(str, format = 'cube') {
const parser = parsers[format.toLowerCase()];
if (!parser) {
throw new Error(`no parser for format: ${format}`);
}
return parser(str);
}
export function lutTo2D3Drgba8(lut) {
if (lut.type === '1D') {
lut = lut1Dto3D(lut);
}
const {min, max, size} = lut;
const range = min.map((min, ndx) => {
return max[ndx] - min;
});
const src = lut.data;
const data = new Uint8Array(size*size*size * 4);
const srcOffset = (offX, offY, offZ) => {
return (offX + offY * size + offZ * size * size) * 3;
};
const dOffset = (offX, offY, offZ) => {
return (offX + offY * size + offZ * size * size) * 4;
};
for (let dz = 0; dz < size; ++dz) {
for (let dy = 0; dy < size; ++dy) {
for (let dx = 0; dx < size; ++dx) {
const sx = dx;
const sy = dz;
const sz = dy;
const sOff = srcOffset(sx, sy, sz);
const dOff = dOffset(dx, dy, dz);
data[dOff + 0] = (src[sOff + 0] - min[0]) / range[0] * 255;
data[dOff + 1] = (src[sOff + 1] - min[1]) / range[1] * 255;
data[dOff + 2] = (src[sOff + 2] - min[2]) / range[2] * 255;
data[dOff + 3] = 255;
}
}
}
return {...lut, data};
}