159 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			159 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| "use strict";
 | |
| Object.defineProperty(exports, "__esModule", { value: true });
 | |
| exports.exportImage = void 0;
 | |
| const child_process = require("child_process");
 | |
| const fs = require("fs-extra");
 | |
| const path = require("path");
 | |
| const log = require("./log");
 | |
| function getWidthAndHeight(kha, exe, from, to, options, format, prealpha) {
 | |
|     return new Promise((resolve, reject) => {
 | |
|         let params = ['from=' + from, 'to=' + to, 'format=' + format, 'donothing'];
 | |
|         if (options.scale !== undefined && options.scale !== 1) {
 | |
|             params.push('scale=' + options.scale);
 | |
|         }
 | |
|         let process = child_process.spawn(exe, params);
 | |
|         let output = '';
 | |
|         process.stdout.on('data', (data) => {
 | |
|             output += data.toString();
 | |
|         });
 | |
|         process.stderr.on('data', (data) => {
 | |
|         });
 | |
|         process.on('close', (code) => {
 | |
|             if (code !== 0) {
 | |
|                 log.error('kraffiti process exited with code ' + code + ' when trying to get size of ' + path.parse(from).name);
 | |
|                 resolve({ w: 0, h: 0 });
 | |
|                 return;
 | |
|             }
 | |
|             const lines = output.split('\n');
 | |
|             for (let line of lines) {
 | |
|                 if (line.startsWith('#')) {
 | |
|                     let numbers = line.substring(1).split('x');
 | |
|                     resolve({ w: parseInt(numbers[0]), h: parseInt(numbers[1]) });
 | |
|                     return;
 | |
|                 }
 | |
|             }
 | |
|             resolve({ w: 0, h: 0 });
 | |
|         });
 | |
|     });
 | |
| }
 | |
| function convertImage(from, temp, to, kha, exe, params, options, cache) {
 | |
|     return new Promise((resolve, reject) => {
 | |
|         let process = child_process.spawn(exe, params);
 | |
|         let output = '';
 | |
|         process.stdout.on('data', (data) => {
 | |
|             output += data.toString();
 | |
|         });
 | |
|         process.stderr.on('data', (data) => {
 | |
|         });
 | |
|         process.on('close', (code) => {
 | |
|             if (code !== 0) {
 | |
|                 log.error('kraffiti process exited with code ' + code + ' when trying to convert ' + path.parse(from).name);
 | |
|                 resolve();
 | |
|                 return;
 | |
|             }
 | |
|             fs.renameSync(temp, to);
 | |
|             const lines = output.split('\n');
 | |
|             for (let line of lines) {
 | |
|                 if (line.startsWith('#')) {
 | |
|                     let numbers = line.substring(1).split('x');
 | |
|                     cache[to] = {};
 | |
|                     cache[to].original_width = options.original_width = parseInt(numbers[0]);
 | |
|                     cache[to].original_height = options.original_height = parseInt(numbers[1]);
 | |
|                     resolve();
 | |
|                     return;
 | |
|                 }
 | |
|             }
 | |
|             resolve();
 | |
|         });
 | |
|     });
 | |
| }
 | |
| async function exportImage(kha, exe, from, to, options, format, prealpha, poweroftwo, cache) {
 | |
|     if (format === undefined) {
 | |
|         if (from.toString().endsWith('.png'))
 | |
|             format = 'png';
 | |
|         else if (from.toString().endsWith('.hdr'))
 | |
|             format = 'hdr';
 | |
|         else
 | |
|             format = 'jpg';
 | |
|     }
 | |
|     if (format === 'jpg' && (options.scale === undefined || options.scale === 1) && options.background === undefined) {
 | |
|         to = to + '.jpg';
 | |
|     }
 | |
|     else if (format === 'pvr') {
 | |
|         to = to + '.pvr';
 | |
|     }
 | |
|     else if (format === 'ASTC') {
 | |
|         to = to + '.astc.k';
 | |
|     }
 | |
|     else if (format === 'DXT5') {
 | |
|         to = to + '.dxt5.k';
 | |
|     }
 | |
|     else if (format === 'hdr') {
 | |
|         to = to + '.hdr';
 | |
|     }
 | |
|     else if (format === 'lz4') {
 | |
|         to += '.k';
 | |
|     }
 | |
|     else {
 | |
|         format = 'png';
 | |
|         if (prealpha)
 | |
|             to = to + '.kng';
 | |
|         else
 | |
|             to = to + '.png';
 | |
|     }
 | |
|     let temp = to + '.temp';
 | |
|     let outputformat = format;
 | |
|     if (format === 'png' && prealpha) {
 | |
|         outputformat = 'kng';
 | |
|     }
 | |
|     if (format === 'lz4') {
 | |
|         outputformat = 'k';
 | |
|     }
 | |
|     if (format === 'ASTC') {
 | |
|         outputformat = 'astc.k';
 | |
|     }
 | |
|     if (format === 'DXT5') {
 | |
|         outputformat = 'dxt5.k';
 | |
|     }
 | |
|     if (fs.existsSync(to) && fs.statSync(to).mtime.getTime() > fs.statSync(from.toString()).mtime.getTime()) {
 | |
|         if (cache[to] !== undefined) {
 | |
|             const cachedOptions = cache[to];
 | |
|             options.original_width = cachedOptions.original_width;
 | |
|             options.original_height = cachedOptions.original_height;
 | |
|             return outputformat;
 | |
|         }
 | |
|         let wh = await getWidthAndHeight(kha, exe, from, to, options, format, prealpha);
 | |
|         cache[to] = {};
 | |
|         cache[to].original_width = options.original_width = wh.w;
 | |
|         cache[to].original_height = options.original_height = wh.h;
 | |
|         return outputformat;
 | |
|     }
 | |
|     fs.ensureDirSync(path.dirname(to));
 | |
|     if (format === 'jpg' || format === 'hdr') {
 | |
|         fs.copySync(from, temp, { overwrite: true });
 | |
|         fs.renameSync(temp, to);
 | |
|         let wh = await getWidthAndHeight(kha, exe, from, to, options, format, prealpha);
 | |
|         options.original_width = wh.w;
 | |
|         options.original_height = wh.h;
 | |
|         return outputformat;
 | |
|     }
 | |
|     let params = ['from=' + from, 'to=' + temp, 'format=' + format];
 | |
|     if (!poweroftwo) {
 | |
|         params.push('filter=nearest');
 | |
|     }
 | |
|     if (prealpha)
 | |
|         params.push('prealpha');
 | |
|     if (options.scale !== undefined && options.scale !== 1) {
 | |
|         params.push('scale=' + options.scale);
 | |
|     }
 | |
|     if (options.background !== undefined) {
 | |
|         params.push('transparent=' + ((options.background.red << 24) | (options.background.green << 16) | (options.background.blue << 8) | 0xff).toString(16));
 | |
|     }
 | |
|     if (poweroftwo) {
 | |
|         params.push('poweroftwo');
 | |
|     }
 | |
|     await convertImage(from, temp, to, kha, exe, params, options, cache);
 | |
|     return outputformat;
 | |
| }
 | |
| exports.exportImage = exportImage;
 | |
| //# sourceMappingURL=ImageTool.js.map
 |