lutfile.js

/* lutfile.js
* File I/O handling object for the LUTCalc Web App.
* 7th October 2014
*
* LUTCalc generates 1D and 3D Lookup Tables (LUTs) for video cameras that shoot log gammas, 
* principally the Sony CineAlta line.
*
* By Ben Turley, http://turley.tv
* First License: GPLv2
* Github: https://github.com/cameramanben/LUTCalc
*/
function LUTFile(inputs, messages) {
	this.inputs = inputs;
	this.messages = messages;
	this.filesaver = true;
	try {
		var isFileSaverSupported = !!new Blob();
	} catch(e){
		this.filesaver = false;
	}
	this.inputs.addInput('doSaveDialog',1);
}
LUTFile.prototype.save = function(data, fileName, extension, source) {
     if (this.inputs.appleApp) {
        var data = {
            data: data,
            filename: this.filename(fileName),
            extension: extension,
            doDialog: this.inputs.doSaveDialog,
            source: source
        }
        window.webkit.messageHandlers.saveText.postMessage(JSON.stringify(data));
    } else if (this.inputs.isApp) { // From native app detection in lutcalc.js
        return window.lutCalcApp.saveLUT(data, this.filename(fileName), extension, this.inputs.doSaveDialog, source);
    } else if (this.inputs.isChromeApp) { // Use Chrome App fileSystem API to select destination
		chrome.fileSystem.chooseEntry(
			{
				type: 'saveFile', 
				suggestedName: this.filename(fileName) + '.' + extension,
				accepts: [{
					extensions: [extension]
				}],
				acceptsAllTypes: false
			}, 
			function(writableFileEntry) {
				writableFileEntry.createWriter(
					function(writer) {
						writer.onwriteend = function(e) {
							savedFromApp(source, 1);
							notifyUser('LUTCalc',source + writableFileEntry.name + ' saved');
						};
						writer.write(new Blob([data],{type: 'text/plain;charset=UTF-8'})); 
					},
					function() {
						savedFromApp(source, 0);
						notifyUser('LUTCalc','File could not be saved');
					}
				);
			}
		);
    } else if (this.filesaver) { // Detect FileSaver.js applicability for browsers other than Safari and older IE
		saveAs(new Blob([data], {type: 'text/plain;charset=UTF-8'}), this.filename(fileName) + '.' + extension,true);
        this.messages.saved(source, true);
		return true;
	} else { // Fall back to opening LUT in a new tab for user to save with 'Save As...'
		window.open("data:text/plain," + encodeURIComponent(data),"_blank");
		this.messages.saved(source, true);
		return true;
	}
};
LUTFile.prototype.saveBinary = function(data, fileName, extension, source) {
    if (this.inputs.appleApp) { // WKWebView uses a different mechanism to WebView, so appleApp triggers the 'new' approach
        var data = {
            data: btoa(new Uint8Array(data).reduce((data, byte)=> {
                    return data + String.fromCharCode(byte);
                }, '')),
            filename: this.filename(fileName),
            extension: extension,
            doDialog: this.inputs.doSaveDialog,
            source: source
        }
        window.webkit.messageHandlers.saveBIN.postMessage(JSON.stringify(data));
    } else if (this.inputs.isApp) { // From native app detection in lutcalc.js
        return window.lutCalcApp.saveBIN(data, this.filename(fileName), extension, this.inputs.doSaveDialog, source);
    } else if (this.inputs.isChromeApp) { // Use Chrome App fileSystem API to select destination
		chrome.fileSystem.chooseEntry(
			{
				type: 'saveFile', 
				suggestedName: this.filename(fileName) + '.' + extension,
				accepts: [{
					extensions: [extension]
				}],
				acceptsAllTypes: false
			}, 
			function(writableFileEntry) {
				writableFileEntry.createWriter(
					function(writer) {
						writer.onwriteend = function(e) {
							savedFromApp(source, 1);
							notifyUser('LUTCalc',writableFileEntry.name + ' saved');
						};
						writer.write(new Blob([data],{type: 'application/octet-stream'})); 
					},
					function() {
						savedFromApp(source, 0);
						notifyUser('LUTCalc','File could not be saved');
					}
				);
			}
		);
    } else if (this.filesaver) { // Detect FileSaver.js applicability for browsers other than Safari and older IE
		saveAs(new Blob([data], {type: 'application/octet-stream'}), this.filename(fileName) + '.' + extension,true);
    	this.messages.saved(source, true);
		return true;
	} else { 
		console.log('Browser does not support file saving.');
    	this.messages.saved(source, false);
		return false;
	}
};
LUTFile.prototype.loadLUTFromInput = function(fileInput, extensions, isTxt, destination, parentObject, next) {
    if (this.inputs.appleApp) {
        var data = {
            extensions: extensions.toString(),
            destination: destination,
            parent: parentObject.p,
            next: next
        }
        window.webkit.messageHandlers.loadLUT.postMessage(JSON.stringify(data));
    } else if (this.inputs.isApp) {
		window.lutCalcApp.loadLUT(extensions.toString(), destination, parentObject.p, next);
	} else {
		var file = fileInput.files[0];
		var valid = false;
		var dot = file.name.lastIndexOf('.');
		var ext = '';
		var isBin = false;
		if (dot != -1) {
			dot++;
			ext = file.name.substr(dot).toLowerCase();
			var max = extensions.length;
			for (var i=0; i<max; i++) {
				if (ext === extensions[i]) {
					valid = true;
					isBin = !isTxt[i];
					break;
				}
			}
		}
		if (valid) {
			if (window.File && window.FileReader && window.FileList && window.Blob) {
				var reader = new FileReader();
				var localDestination = this.inputs[destination];
				localDestination.format = ext;
				localDestination.title = file.name.substr(0,file.name.length-ext.length-1);
				if (isBin) {
					reader.onload = (function(theFile){
						var theDestination = localDestination;
    					return function(e){
 	  						theDestination.isTxt = false;
    						theDestination.buff = e.target.result;
							parentObject.followUp(next);
    					};
    				})(file);
					reader.onerror = function(theFile) { return function() {
						alert("Error reading file.");
						inputBox.value = '';
					}; }(file);
					reader.readAsArrayBuffer(file);
				} else {
					reader.onload = (function(theFile){
						var theDestination = localDestination;
    					return function(e){
    						theDestination.isTxt = true;
    						theDestination.text = e.target.result.split(/[\n\u0085\u2028\u2029]|\r\n?/);
							parentObject.followUp(next);
    					};
    				})(file);
					reader.onerror = function(theFile) { return function() {
						alert("Error reading file.");
						inputBox.value = '';
					}; }(file);
					reader.readAsText(file);
				}
			} else {
				alert("Can't load LUT - your browser is not set to support Javascript File APIs.");
				fileInput.value = '';
			}
		} else {
			fileInput.value = '';
			alert("LUTCalc does not understand this file.");
		}
	}
};
LUTFile.prototype.loadImgFromInput = function(fileInput, extensions, destination, parentObject, next) {
    if (this.inputs.appleApp) {
        var data = {
            extensions: extensions.toString(),
            destination: destination,
            parent: parentObject.p,
            next: next
        }
        window.webkit.messageHandlers.loadImg.postMessage(JSON.stringify(data));
    } else 	if (this.inputs.isApp) {
		window.lutCalcApp.loadImg(extensions.toString(), destination, parentObject.p, next);
	} else {
		var file = fileInput.files[0];
		var valid = false;
		var dot = file.name.lastIndexOf('.');
		var ext = '';
		if (dot != -1) {
			dot++;
			ext = file.name.substr(dot).toLowerCase();
			var max = extensions.length;
			for (var i=0; i<max; i++) {
				if (ext === extensions[i]) {
					valid = true;
					break;
				}
			}
		}
		if (valid) {
			if (window.File && window.FileReader && window.FileList && window.Blob) {
				var reader = new FileReader();
				var localDestination = this.inputs[destination];
				localDestination.format = ext;
				localDestination.title = file.name.substr(0,file.name.length-ext.length-1);
				reader.onload = (function(theFile){
					var theDestination = localDestination;
    				return function(e){
    					theDestination.pic = new Image();
    					theDestination.pic.onload = function(e){
    						theDestination.w = theDestination.pic.width;
	    					theDestination.h = theDestination.pic.height;
	 						parentObject.followUp(next);
    					};
    					theDestination.pic.src = e.target.result;
    				};
    			})(file);
				reader.onerror = function(theFile) { return function() {
					alert("Error reading file.");
					inputBox.value = '';
				}; }(file);
				reader.readAsDataURL(file);
			} else {
				alert("Can't load image - your browser is not set to support Javascript File APIs.");
				fileInput.value = '';
			}
		} else {
			fileInput.value = '';
			alert("LUTCalc does not understand this image type.");
		}
	}
};
LUTFile.prototype.filename = function(filename) {
	return filename.replace(/[^a-z0-9_\-\ ]/gi, '').replace(/[^a-z0-9_\-]/gi, '_');
};
// Functions available to native apps
function loadLUTFromAppleApp(fileName, format, content64, destination, parentIdx, next) {
    lutInputs[destination].format = format;
    var content = atob(content64);
    if (format.toLowerCase() === 'labin') {
        var max = content.length;
        var data = new Uint8Array(max);
        for (var j=0; j<max; j++) {
            data[j] = content.charCodeAt(j);
        }
        lutInputs[destination].title = fileName;
        lutInputs[destination].buff = data.buffer;
        lutInputs[destination].isTxt = false;
    } else {
        lutInputs[destination].title = fileName;
        lutInputs[destination].text = content.split(/[\n\u0085\u2028\u2029]|\r\n?/);
        lutInputs[destination].isTxt = true;
    }
    switch (parseInt(parentIdx)) {
        case 5: lutGenerate.followUp(parseInt(next));
                break;
        case 10: lutTweaksBox.followUp(parseInt(parentIdx),parseInt(next));
                 break;
    }
}
function loadLUTFromApp(fileName, format, content, destination, parentIdx, next) {
    format = format.toLowerCase();
	lutInputs[destination].format = format;
	if (format.toLowerCase() === 'labin') {
        var max = content.length;
        var data = new Uint8Array(max);
        for (var j=0; j<max; j++) {
            data[j] = content[j];
        }
        lutInputs[destination].title = fileName;
		lutInputs[destination].buff = data.buffer;
		lutInputs[destination].isTxt = false;
	} else {
        lutInputs[destination].title = fileName;
		lutInputs[destination].text = content.split(/[\n\u0085\u2028\u2029]|\r\n?/);
		lutInputs[destination].isTxt = true;
	}
	switch (parseInt(parentIdx)) {
		case 5: lutGenerate.followUp(parseInt(next));
				break;
		case 10: lutTweaksBox.followUp(parseInt(parentIdx),parseInt(next));
				break;
	}
}
function loadImgFromAppleApp(format, title, content64, destination, parentIdx, next) {
    var theDestination = lutInputs[destination];
    var nextObject;
    switch (parseInt(parentIdx)) {
        case 8: nextObject = lutPreview;
                break;
    }
    theDestination.title = title;
    var contentString = atob(content64);

    var max = contentString.length;
    var data = new Uint8Array(max);
    for (var j=0; j<max; j++) {
        data[j] = contentString.charCodeAt(j);
    }
    var content = new Uint16Array(data.buffer);
    max = content.length;
    theDestination.imageData = new Float64Array(max);
    var sample;
     for (var j=0; j<max; j++) {
         sample = parseFloat(content[ j ])/65535;
         theDestination.imageData[ j ] = sample;
     }
     nextObject.followUp(parseInt(next));
}
function loadImgFromApp(format, title, content, destination, parentIdx, next) {
    var theDestination = lutInputs[destination];
    var nextObject;
    switch (parseInt(parentIdx)) {
        case 8: nextObject = lutPreview;
                break;
    }
    theDestination.title = title;
     var max = content.length;
     theDestination.imageData = new Float64Array(max);
    var sample;
     for (var j=0; j<max; j++) {
         sample = parseFloat(content[ j ])/65535;
         theDestination.imageData[ j ] = sample;
     }
     nextObject.followUp(parseInt(next));
}
function savedFromApp(source, success) {
    if (parseInt(success) === 1) {
        lutMessage.saved(parseInt(source), true);
    } else {
        lutMessage.saved(parseInt(source), false);
    }
}
// Loading progress bar
if (typeof splash !== 'undefined') {
	splashProg();
}