BrechtDeMan@753: /**
BrechtDeMan@753:  *  loundess.js
BrechtDeMan@753:  *  Loudness module for the Web Audio Evaluation Toolbox
BrechtDeMan@753:  *  Allows for automatic calculation of loudness of Web Audio API Buffer objects,
BrechtDeMan@753:  * 	return gain values to correct for a target loudness or match loudness between
BrechtDeMan@753:  *  multiple objects
BrechtDeMan@753:  */
BrechtDeMan@753: 
BrechtDeMan@753: var interval_cal_loudness_event = null;
BrechtDeMan@753: 
BrechtDeMan@753: function calculateLoudness(buffer, timescale, target, offlineContext)
BrechtDeMan@753: {
BrechtDeMan@753: 	// This function returns the EBU R 128 specification loudness model and sets the linear gain required to match -23 LUFS
BrechtDeMan@753: 	// buffer -> Web Audio API Buffer object
BrechtDeMan@753: 	// timescale -> M or Momentary (returns Array), S or Short (returns Array),
BrechtDeMan@753: 	//   I or Integrated (default, returns number)
BrechtDeMan@753: 	// target -> default is -23 LUFS but can be any LUFS measurement.
BrechtDeMan@753: 	
BrechtDeMan@753: 	if (buffer == undefined)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		return 0;
BrechtDeMan@753: 	}
BrechtDeMan@753: 	if (timescale == undefined)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		timescale = "I";
BrechtDeMan@753: 	}
BrechtDeMan@753: 	if (target == undefined)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		target = -23;
BrechtDeMan@753: 	}
BrechtDeMan@753: 	if (offlineContext == undefined)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		offlineContext = new OfflineAudioContext(buffer.numberOfChannels, buffer.length, buffer.sampleRate);
BrechtDeMan@753: 	}
BrechtDeMan@753: 	// Create the required filters
BrechtDeMan@753: 	var KFilter = offlineContext.createBiquadFilter();
BrechtDeMan@753: 	KFilter.type = "highshelf";
BrechtDeMan@753: 	KFilter.gain.value = 4;
BrechtDeMan@753: 	KFilter.frequency.value = 1480;
BrechtDeMan@753: 	
BrechtDeMan@753: 	var HPFilter = offlineContext.createBiquadFilter();
BrechtDeMan@753: 	HPFilter.type = "highpass";
BrechtDeMan@753: 	HPFilter.Q.value = 0.707;
BrechtDeMan@753: 	HPFilter.frequency.value = 60;
BrechtDeMan@753: 	// copy Data into the process buffer
BrechtDeMan@753: 	var processSource = offlineContext.createBufferSource();
BrechtDeMan@753: 	processSource.buffer = buffer;
BrechtDeMan@753: 	
BrechtDeMan@753: 	processSource.connect(KFilter);
BrechtDeMan@753: 	KFilter.connect(HPFilter);
BrechtDeMan@753: 	HPFilter.connect(offlineContext.destination);
BrechtDeMan@753: 	processSource.start();
BrechtDeMan@753: 	offlineContext.startRendering().then(function(renderedBuffer) {
BrechtDeMan@753: 		// Have the renderedBuffer information, now continue processing
BrechtDeMan@753: 		switch(timescale)
BrechtDeMan@753: 		{
BrechtDeMan@753: 		case "I":
BrechtDeMan@753: 			var blockEnergy = calculateProcessedLoudness(renderedBuffer, 400, 0.75);
BrechtDeMan@753: 			// Apply the absolute gate
BrechtDeMan@753: 			var loudness = calculateLoudnessFromChannelBlocks(blockEnergy);
BrechtDeMan@753: 			var absgatedEnergy = new Array(blockEnergy.length);
BrechtDeMan@753: 			for (var c=0; c<blockEnergy.length; c++)
BrechtDeMan@753: 			{
BrechtDeMan@753: 				absgatedEnergy[c] = [];
BrechtDeMan@753: 			}
BrechtDeMan@753: 			for (var i=0; i<loudness.length; i++)
BrechtDeMan@753: 			{
BrechtDeMan@753: 				if (loudness[i] >= -70)
BrechtDeMan@753: 				{
BrechtDeMan@753: 					for (var c=0; c<blockEnergy.length; c++)
BrechtDeMan@753: 					{
BrechtDeMan@753: 						absgatedEnergy[c].push(blockEnergy[c][i]);
BrechtDeMan@753: 					}
BrechtDeMan@753: 				}
BrechtDeMan@753: 			}
BrechtDeMan@753: 			var overallAbsLoudness = calculateOverallLoudnessFromChannelBlocks(absgatedEnergy);
BrechtDeMan@753: 			
BrechtDeMan@753: 			//applying the relative gate 8 dB down from overallAbsLoudness
BrechtDeMan@753: 			var relGateLevel = overallAbsLoudness - 8;
BrechtDeMan@753: 			var relgateEnergy = new Array(blockEnergy.length);
BrechtDeMan@753: 			for (var c=0; c<blockEnergy.length; c++)
BrechtDeMan@753: 			{
BrechtDeMan@753: 				relgateEnergy[c] = [];
BrechtDeMan@753: 			}
BrechtDeMan@753: 			for (var i=0; i<loudness.length; i++)
BrechtDeMan@753: 			{
BrechtDeMan@753: 				if (loudness[i] >= relGateLevel)
BrechtDeMan@753: 				{
BrechtDeMan@753: 					for (var c=0; c<blockEnergy.length; c++)
BrechtDeMan@753: 					{
BrechtDeMan@753: 						relgateEnergy[c].push(blockEnergy[c][i]);
BrechtDeMan@753: 					}
BrechtDeMan@753: 				}
BrechtDeMan@753: 			}
BrechtDeMan@753: 			var overallRelLoudness = calculateOverallLoudnessFromChannelBlocks(relgateEnergy);
BrechtDeMan@753: 			buffer.lufs =  overallRelLoudness;
BrechtDeMan@753: 		}
BrechtDeMan@753: 	}).catch(function(err) {
BrechtDeMan@753: 		console.log(err);
BrechtDeMan@753: 		buffer.lufs = 1;
BrechtDeMan@753: 	});
BrechtDeMan@753: }
BrechtDeMan@753: 
BrechtDeMan@753: function calculateProcessedLoudness(buffer, winDur, overlap)
BrechtDeMan@753: {
BrechtDeMan@753: 	// Buffer		Web Audio buffer node
BrechtDeMan@753: 	// winDur		Window Duration in milliseconds
BrechtDeMan@753: 	// overlap		Window overlap as normalised (0.5 = 50% overlap);
BrechtDeMan@753: 	if (buffer == undefined)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		return 0;
BrechtDeMan@753: 	}
BrechtDeMan@753: 	if (winDur == undefined)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		winDur = 400;
BrechtDeMan@753: 	}
BrechtDeMan@753: 	if (overlap == undefined)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		overlap = 0.5;
BrechtDeMan@753: 	}
BrechtDeMan@753: 	var winSize = buffer.sampleRate*winDur/1000;
BrechtDeMan@753: 	var olapSize = (1-overlap)*winSize;
BrechtDeMan@753: 	var numberOfFrames = Math.floor(buffer.length/olapSize - winSize/olapSize + 1);
BrechtDeMan@753: 	var blockEnergy = new Array(buffer.numberOfChannels);
BrechtDeMan@753: 	for (var channel = 0; channel < buffer.numberOfChannels; channel++)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		blockEnergy[channel] = new Float32Array(numberOfFrames);
BrechtDeMan@753: 		var data = buffer.getChannelData(channel);
BrechtDeMan@753: 		for (var i=0; i<numberOfFrames; i++)
BrechtDeMan@753: 		{
BrechtDeMan@753: 			var sigma = 0;
BrechtDeMan@753: 			for (var n=i*olapSize; n < i*olapSize+winSize; n++)
BrechtDeMan@753: 			{
BrechtDeMan@753: 				sigma += Math.pow(data[n],2);
BrechtDeMan@753: 			}
BrechtDeMan@753: 			blockEnergy[channel][i] = sigma/winSize;
BrechtDeMan@753: 		}
BrechtDeMan@753: 	}
BrechtDeMan@753: 	return blockEnergy;
BrechtDeMan@753: }
BrechtDeMan@753: function calculateLoudnessFromChannelBlocks(blockEnergy)
BrechtDeMan@753: {
BrechtDeMan@753: 	// Loudness
BrechtDeMan@753: 	var loudness = new Float32Array(blockEnergy[0].length);
BrechtDeMan@753: 	for (var i=0; i<blockEnergy[0].length; i++)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		var sigma = 0;
BrechtDeMan@753: 		for (var channel = 0; channel < blockEnergy.length; channel++)
BrechtDeMan@753: 		{
BrechtDeMan@753: 			var G = 1.0;
BrechtDeMan@753: 			if (channel >= 4) {G = 1.41;}
BrechtDeMan@753: 			sigma += blockEnergy[channel][i]*G;
BrechtDeMan@753: 		}
BrechtDeMan@753: 		loudness[i] = -0.691 + 10*Math.log10(sigma);
BrechtDeMan@753: 	}
BrechtDeMan@753: 	return loudness;
BrechtDeMan@753: }
BrechtDeMan@753: function calculateOverallLoudnessFromChannelBlocks(blockEnergy)
BrechtDeMan@753: {
BrechtDeMan@753: 	// Loudness
BrechtDeMan@753: 	var summation = 0;
BrechtDeMan@753: 	for (var channel = 0; channel < blockEnergy.length; channel++)
BrechtDeMan@753: 	{
BrechtDeMan@753: 		var G = 1.0;
BrechtDeMan@753: 		if (channel >= 4) {G = 1.41;}
BrechtDeMan@753: 		var sigma = 0;
BrechtDeMan@753: 		for (var i=0; i<blockEnergy[0].length; i++)
BrechtDeMan@753: 		{
BrechtDeMan@753: 			blockEnergy[channel][i] *= G;
BrechtDeMan@753: 			sigma += blockEnergy[channel][i];
BrechtDeMan@753: 		}
BrechtDeMan@753: 		sigma /= blockEnergy.length;
BrechtDeMan@753: 		summation+= sigma;
BrechtDeMan@753: 	}
BrechtDeMan@753: 	return -0.691 + 10*Math.log10(summation);;
BrechtDeMan@753: }