-
Notifications
You must be signed in to change notification settings - Fork 11
/
lutCreatorJs.ts
114 lines (97 loc) · 4.21 KB
/
lutCreatorJs.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
class Hald {
lutSize: number;
constructor(lutSize: number) {
this.lutSize = lutSize
}
createHald(lutSize?: number) {
if (!lutSize) lutSize = this.lutSize;
let haldSize = Math.round((this.lutSize ** 3) ** 0.5);
let hald = document.createElement("canvas");
hald.width = hald.height = haldSize;
let haldData = hald.getContext("2d");
let r: number, g: number, b: number = g = r = 0;
function exportHald(target: HTMLCanvasElement, lutSize: number) {
let download = document.createElement('a');
download.download = "Neutral_" + lutSize + ".png";
download.href = target.toDataURL("image/png");
download.name = "Export HALD";
download.click();
}
function setPixel(target: CanvasRenderingContext2D, values: number[], position: number[]) {
function value(color: number) {
return Math.round((255 / (lutSize - 1)) * color)
}
target.fillStyle = "rgb(" + value(values[0]) + "," + value(values[1]) + "," + value(values[2]) + ")";
target.fillRect(position[0], position[1], 1, 1);
}
for (let y = 0; y < hald.height; y++) {
for (let x = 0; x < hald.width; x++) {
if (r >= this.lutSize) {
r = 0
g += 1
}
if (g >= this.lutSize) {
g = 0
b += 1
}
setPixel(haldData, [r, g, b], [x, y]);
r += 1
}
}
exportHald(hald, lutSize);
}
static exportCube(lutTitle?: string) {
if (!lutTitle) lutTitle = "lutCreatorJs";
let hald = document.createElement('canvas');
let haldData = hald.getContext('2d');
function exportCube(text: string, lutTitle: string) {
let exportFile = document.createElement('a');
let file = new Blob([text], { type: 'text/plain' });
exportFile.href = URL.createObjectURL(file);
exportFile.download = lutTitle + '.cube';
exportFile.click();
}
function setText(lutSize: number, lutTitle: string): string {
let text: string = "#Generated by lutCreator.js\n#Coded by SirSerch\n#https://github.com/SirSerch/LutCreatorJs\n";
text += ("TITLE " + lutTitle + "_" + lutSize + "\n");
text += ("LUT_3D_SIZE " + lutSize + "\n\n");
return text;
}
function generateCube(image: HTMLImageElement, lutTitle: string) {
hald.width = hald.height = image.width;
haldData.drawImage(image, 0, 0);
let lutSize = Math.round((hald.width * hald.height) ** 0.3333);
console.log(lutSize);
let cube = setText(lutSize, lutTitle);
for (let y = 0; y < hald.height; y++) {
for (let x = 0; x < hald.width; x++) {
var r = (haldData.getImageData(x, y, 1, 1).data[0] / 255).toFixed(6);
var g = (haldData.getImageData(x, y, 1, 1).data[1] / 255).toFixed(6);
var b = (haldData.getImageData(x, y, 1, 1).data[2] / 255).toFixed(6);
cube += r + " " + g + " " + b + "\n";
}
}
exportCube(cube, lutTitle);
}
function loadImage(file: FileList, lutTitle: string) {
let imageReader = new FileReader();
imageReader.onload = () => {
let imageHald = new Image();
imageHald.onload = () => {
generateCube(imageHald, lutTitle);
}
imageHald.src = String(imageReader.result);
}
imageReader.readAsDataURL(file[0]);
}
function inputHald(lutTitle: string) {
let selectFile = document.createElement('input');
selectFile.type = 'file';
selectFile.click();
selectFile.onchange = (e) => {
loadImage(selectFile.files, lutTitle);
}
}
inputHald(lutTitle);
}
}