mirror of
https://github.com/netbymatt/ws4kp.git
synced 2026-04-21 11:09:30 -07:00
radar processed in web worker
This commit is contained in:
80
server/scripts/modules/radar-worker.mjs
Normal file
80
server/scripts/modules/radar-worker.mjs
Normal file
@@ -0,0 +1,80 @@
|
|||||||
|
import * as utils from './radar-utils.mjs';
|
||||||
|
|
||||||
|
const fetchAsBlob = async (url) => {
|
||||||
|
const response = await fetch(url);
|
||||||
|
return response.blob();
|
||||||
|
};
|
||||||
|
|
||||||
|
onmessage = async (e) => {
|
||||||
|
const baseMapImage = createImageBitmap(await fetchAsBlob('/images/maps/radar.webp'));
|
||||||
|
|
||||||
|
const {
|
||||||
|
url, index, RADAR_HOST, OVERRIDES, radarSourceXY, sourceXY, offsetX, offsetY,
|
||||||
|
} = e.data;
|
||||||
|
|
||||||
|
// calculate offsets and sizes
|
||||||
|
const width = 2550;
|
||||||
|
const height = 1600;
|
||||||
|
const radarOffsetX = 120;
|
||||||
|
const radarOffsetY = 70;
|
||||||
|
const radarSourceX = Math.round(radarSourceXY.x / 2);
|
||||||
|
const radarSourceY = Math.round(radarSourceXY.y / 2);
|
||||||
|
|
||||||
|
// create destination context
|
||||||
|
const baseCanvas = new OffscreenCanvas(640, 367);
|
||||||
|
const baseContext = baseCanvas.getContext('2d', { alpha: false });
|
||||||
|
baseContext.imageSmoothingEnabled = false;
|
||||||
|
|
||||||
|
// create working context for manipulation
|
||||||
|
const radarCanvas = new OffscreenCanvas(width, height);
|
||||||
|
const radarContext = radarCanvas.getContext('2d', { alpha: false });
|
||||||
|
radarContext.imageSmoothingEnabled = false;
|
||||||
|
|
||||||
|
// get the image
|
||||||
|
const modifiedUrl = OVERRIDES.RADAR_HOST ? url.replace(RADAR_HOST, OVERRIDES.RADAR_HOST) : url;
|
||||||
|
console.time(`Radar-${index}-fetch`);
|
||||||
|
const response = await fetch(modifiedUrl);
|
||||||
|
console.timeEnd(`Radar-${index}-fetch`);
|
||||||
|
|
||||||
|
// test response
|
||||||
|
if (!response.ok) throw new Error(`Unable to fetch radar error ${response.status} ${response.statusText} from ${response.url}`);
|
||||||
|
|
||||||
|
// get the blob
|
||||||
|
console.time(`Radar-${index}-blob`);
|
||||||
|
const radarImgBlob = await response.blob();
|
||||||
|
console.timeEnd(`Radar-${index}-blob`);
|
||||||
|
|
||||||
|
// assign to an html image element
|
||||||
|
console.time(`Radar-${index}-loadimg-element`);
|
||||||
|
const radarImgElement = await createImageBitmap(radarImgBlob);
|
||||||
|
console.timeEnd(`Radar-${index}-loadimg-element`);
|
||||||
|
// draw the entire image
|
||||||
|
radarContext.clearRect(0, 0, width, 1600);
|
||||||
|
console.time(`Radar-${index}-drawimage`);
|
||||||
|
radarContext.drawImage(radarImgElement, 0, 0, width, 1600);
|
||||||
|
console.timeEnd(`Radar-${index}-drawimage`);
|
||||||
|
// get the base map
|
||||||
|
console.time(`Radar-${index}-drawbasemap`);
|
||||||
|
baseContext.drawImage(await baseMapImage, sourceXY.x, sourceXY.y, offsetX * 2, offsetY * 2, 0, 0, 640, 367);
|
||||||
|
console.timeEnd(`Radar-${index}-drawbasemap`);
|
||||||
|
// crop the radar image
|
||||||
|
const cropCanvas = new OffscreenCanvas(640, 367);
|
||||||
|
const cropContext = cropCanvas.getContext('2d', { willReadFrequently: true });
|
||||||
|
cropContext.imageSmoothingEnabled = false;
|
||||||
|
console.time(`Radar-${index}-copy-radar`);
|
||||||
|
cropContext.drawImage(radarCanvas, radarSourceX, radarSourceY, (radarOffsetX * 2), Math.round(radarOffsetY * 2.33), 0, 0, 640, 367);
|
||||||
|
console.timeEnd(`Radar-${index}-copy-radar`);
|
||||||
|
// clean the image
|
||||||
|
console.time(`Radar-${index}-clean-image`);
|
||||||
|
utils.removeDopplerRadarImageNoise(cropContext);
|
||||||
|
console.timeEnd(`Radar-${index}-clean-image`);
|
||||||
|
|
||||||
|
// merge the radar and map
|
||||||
|
console.time(`Radar-${index}-merge`);
|
||||||
|
utils.mergeDopplerRadarImage(baseContext, cropContext);
|
||||||
|
console.timeEnd(`Radar-${index}-merge`);
|
||||||
|
|
||||||
|
const processedRadar = baseCanvas.transferToImageBitmap();
|
||||||
|
|
||||||
|
postMessage(processedRadar, [processedRadar]);
|
||||||
|
};
|
||||||
@@ -1,9 +1,8 @@
|
|||||||
// current weather conditions display
|
// current weather conditions display
|
||||||
import STATUS from './status.mjs';
|
import STATUS from './status.mjs';
|
||||||
import { DateTime } from '../vendor/auto/luxon.mjs';
|
import { DateTime } from '../vendor/auto/luxon.mjs';
|
||||||
import { loadImgElement, loadImg } from './utils/image.mjs';
|
import { loadImgElement } from './utils/image.mjs';
|
||||||
import { text } from './utils/fetch.mjs';
|
import { text } from './utils/fetch.mjs';
|
||||||
import { rewriteUrl } from './utils/cors.mjs';
|
|
||||||
import WeatherDisplay from './weatherdisplay.mjs';
|
import WeatherDisplay from './weatherdisplay.mjs';
|
||||||
import { registerDisplay, timeZone } from './navigation.mjs';
|
import { registerDisplay, timeZone } from './navigation.mjs';
|
||||||
import * as utils from './radar-utils.mjs';
|
import * as utils from './radar-utils.mjs';
|
||||||
@@ -41,6 +40,9 @@ class Radar extends WeatherDisplay {
|
|||||||
{ time: 1, si: 4 },
|
{ time: 1, si: 4 },
|
||||||
{ time: 12, si: 5 },
|
{ time: 12, si: 5 },
|
||||||
];
|
];
|
||||||
|
|
||||||
|
// get some web workers started
|
||||||
|
this.workers = (new Array(this.dopplerRadarImageMax)).fill(null).map(() => radarWorker());
|
||||||
}
|
}
|
||||||
|
|
||||||
async getData(weatherParameters, refresh) {
|
async getData(weatherParameters, refresh) {
|
||||||
@@ -52,10 +54,6 @@ class Radar extends WeatherDisplay {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// get the base map
|
|
||||||
const src = 'images/maps/radar.webp';
|
|
||||||
this.baseMapImageElem = await loadImgElement(src);
|
|
||||||
|
|
||||||
const baseUrl = `https://${RADAR_HOST}/archive/data/`;
|
const baseUrl = `https://${RADAR_HOST}/archive/data/`;
|
||||||
const baseUrlEnd = '/GIS/uscomp/?F=0&P=n0r*.png';
|
const baseUrlEnd = '/GIS/uscomp/?F=0&P=n0r*.png';
|
||||||
const baseUrls = [];
|
const baseUrls = [];
|
||||||
@@ -105,104 +103,50 @@ class Radar extends WeatherDisplay {
|
|||||||
// calculate offsets and sizes
|
// calculate offsets and sizes
|
||||||
let offsetX = 120;
|
let offsetX = 120;
|
||||||
let offsetY = 69;
|
let offsetY = 69;
|
||||||
const width = 2550;
|
|
||||||
const height = 1600;
|
|
||||||
offsetX *= 2;
|
offsetX *= 2;
|
||||||
offsetY *= 2;
|
offsetY *= 2;
|
||||||
const sourceXY = utils.getXYFromLatitudeLongitudeMap(this.weatherParameters, offsetX, offsetY);
|
const sourceXY = utils.getXYFromLatitudeLongitudeMap(this.weatherParameters, offsetX, offsetY);
|
||||||
|
|
||||||
// calculate radar offsets
|
|
||||||
const radarOffsetX = 120;
|
|
||||||
const radarOffsetY = 70;
|
|
||||||
const radarSourceXY = utils.getXYFromLatitudeLongitudeDoppler(this.weatherParameters, offsetX, offsetY);
|
const radarSourceXY = utils.getXYFromLatitudeLongitudeDoppler(this.weatherParameters, offsetX, offsetY);
|
||||||
const radarSourceX = Math.round(radarSourceXY.x / 2);
|
|
||||||
const radarSourceY = Math.round(radarSourceXY.y / 2);
|
|
||||||
|
|
||||||
// Load the most recent doppler radar images.
|
// Load the most recent doppler radar images.
|
||||||
const radarInfo = await Promise.all(urls.map(async (url, index) => {
|
const radarInfo = await Promise.all(urls.map(async (url, index) => {
|
||||||
console.time(`Radar-${index}`);
|
console.time(`Radar-${index}`);
|
||||||
// create destination context
|
const processedRadar = await this.workers[index].processRadar({
|
||||||
const baseCanvas = new OffscreenCanvas(640, 367);
|
url,
|
||||||
const baseContext = baseCanvas.getContext('2d', { alpha: false });
|
index,
|
||||||
baseContext.imageSmoothingEnabled = false;
|
RADAR_HOST,
|
||||||
|
OVERRIDES,
|
||||||
// create working context for manipulation
|
sourceXY,
|
||||||
const radarCanvas = new OffscreenCanvas(width, height);
|
radarSourceXY,
|
||||||
const radarContext = radarCanvas.getContext('2d', { alpha: false });
|
offsetX,
|
||||||
radarContext.imageSmoothingEnabled = false;
|
offsetY,
|
||||||
|
});
|
||||||
// get the image
|
|
||||||
const modifiedUrl = OVERRIDES.RADAR_HOST ? url.replace(RADAR_HOST, OVERRIDES.RADAR_HOST) : url;
|
|
||||||
console.time(`Radar-${index}-fetch`);
|
|
||||||
const response = await fetch(rewriteUrl(modifiedUrl));
|
|
||||||
console.timeEnd(`Radar-${index}-fetch`);
|
|
||||||
|
|
||||||
// test response
|
|
||||||
if (!response.ok) throw new Error(`Unable to fetch radar error ${response.status} ${response.statusText} from ${response.url}`);
|
|
||||||
|
|
||||||
// get the blob
|
|
||||||
console.time(`Radar-${index}-blob`);
|
|
||||||
const radarImgBlob = await response.blob();
|
|
||||||
console.timeEnd(`Radar-${index}-blob`);
|
|
||||||
|
|
||||||
// store the time
|
// store the time
|
||||||
const timeMatch = url.match(/_(\d{4})(\d\d)(\d\d)(\d\d)(\d\d)\./);
|
const timeMatch = url.match(/_(\d{4})(\d\d)(\d\d)(\d\d)(\d\d)\./);
|
||||||
let time;
|
|
||||||
if (timeMatch) {
|
|
||||||
const [, year, month, day, hour, minute] = timeMatch;
|
|
||||||
time = DateTime.fromObject({
|
|
||||||
year,
|
|
||||||
month,
|
|
||||||
day,
|
|
||||||
hour,
|
|
||||||
minute,
|
|
||||||
}, {
|
|
||||||
zone: 'UTC',
|
|
||||||
}).setZone(timeZone());
|
|
||||||
} else {
|
|
||||||
time = DateTime.fromHTTP(response.headers.get('last-modified')).setZone(timeZone());
|
|
||||||
}
|
|
||||||
|
|
||||||
// assign to an html image element
|
const [, year, month, day, hour, minute] = timeMatch;
|
||||||
console.time(`Radar-${index}-loadimg-element`);
|
const time = DateTime.fromObject({
|
||||||
const radarImgElement = await loadImg(radarImgBlob);
|
year,
|
||||||
console.timeEnd(`Radar-${index}-loadimg-element`);
|
month,
|
||||||
// draw the entire image
|
day,
|
||||||
radarContext.clearRect(0, 0, width, 1600);
|
hour,
|
||||||
console.time(`Radar-${index}-drawimage`);
|
minute,
|
||||||
radarContext.drawImage(radarImgElement, 0, 0, width, 1600);
|
}, {
|
||||||
console.timeEnd(`Radar-${index}-drawimage`);
|
zone: 'UTC',
|
||||||
// get the base map
|
}).setZone(timeZone());
|
||||||
console.time(`Radar-${index}-drawbasemap`);
|
|
||||||
baseContext.drawImage(this.baseMapImageElem, sourceXY.x, sourceXY.y, offsetX * 2, offsetY * 2, 0, 0, 640, 367);
|
|
||||||
console.timeEnd(`Radar-${index}-drawbasemap`);
|
|
||||||
// crop the radar image
|
|
||||||
const cropCanvas = new OffscreenCanvas(640, 367);
|
|
||||||
const cropContext = cropCanvas.getContext('2d', { willReadFrequently: true });
|
|
||||||
cropContext.imageSmoothingEnabled = false;
|
|
||||||
console.time(`Radar-${index}-copy-radar`);
|
|
||||||
cropContext.drawImage(radarCanvas, radarSourceX, radarSourceY, (radarOffsetX * 2), Math.round(radarOffsetY * 2.33), 0, 0, 640, 367);
|
|
||||||
console.timeEnd(`Radar-${index}-copy-radar`);
|
|
||||||
// clean the image
|
|
||||||
console.time(`Radar-${index}-clean-image`);
|
|
||||||
utils.removeDopplerRadarImageNoise(cropContext);
|
|
||||||
console.timeEnd(`Radar-${index}-clean-image`);
|
|
||||||
|
|
||||||
// merge the radar and map
|
|
||||||
console.time(`Radar-${index}-merge`);
|
|
||||||
utils.mergeDopplerRadarImage(baseContext, cropContext);
|
|
||||||
console.timeEnd(`Radar-${index}-merge`);
|
|
||||||
|
|
||||||
console.time(`Radar-${index}-transfer-canvas`);
|
console.time(`Radar-${index}-transfer-canvas`);
|
||||||
const onscreenCanvas = document.createElement('canvas');
|
const onscreenCanvas = document.createElement('canvas');
|
||||||
onscreenCanvas.width = baseCanvas.width;
|
onscreenCanvas.width = 640;
|
||||||
onscreenCanvas.height = baseCanvas.height;
|
onscreenCanvas.height = 367;
|
||||||
onscreenCanvas.getContext('bitmaprenderer').transferFromImageBitmap(baseCanvas.transferToImageBitmap());
|
const onscreenContext = onscreenCanvas.getContext('bitmaprenderer');
|
||||||
|
onscreenContext.transferFromImageBitmap(processedRadar);
|
||||||
|
|
||||||
const elem = this.fillTemplate('frame', { map: { type: 'canvas', canvas: onscreenCanvas } });
|
const elem = this.fillTemplate('frame', { map: { type: 'canvas', canvas: onscreenCanvas } });
|
||||||
console.timeEnd(`Radar-${index}-transfer-canvas`);
|
console.timeEnd(`Radar-${index}-transfer-canvas`);
|
||||||
console.timeEnd(`Radar-${index}`);
|
console.timeEnd(`Radar-${index}`);
|
||||||
return {
|
return {
|
||||||
canvas: baseCanvas,
|
|
||||||
time,
|
time,
|
||||||
elem,
|
elem,
|
||||||
};
|
};
|
||||||
@@ -215,8 +159,6 @@ class Radar extends WeatherDisplay {
|
|||||||
|
|
||||||
// set max length
|
// set max length
|
||||||
this.timing.totalScreens = radarInfo.length;
|
this.timing.totalScreens = radarInfo.length;
|
||||||
// store the images
|
|
||||||
this.data = radarInfo.map((radar) => radar.canvas);
|
|
||||||
|
|
||||||
this.times = radarInfo.map((radar) => radar.time);
|
this.times = radarInfo.map((radar) => radar.time);
|
||||||
this.setStatus(STATUS.loaded);
|
this.setStatus(STATUS.loaded);
|
||||||
@@ -239,5 +181,30 @@ class Radar extends WeatherDisplay {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// create a radar worker with helper functions
|
||||||
|
const radarWorker = () => {
|
||||||
|
// create the worker
|
||||||
|
const worker = new Worker('scripts/modules/radar-worker.mjs', { type: 'module' });
|
||||||
|
|
||||||
|
const processRadar = (url) => new Promise((resolve, reject) => {
|
||||||
|
// prepare for done message
|
||||||
|
worker.onmessage = (e) => {
|
||||||
|
if (e?.data instanceof Error) {
|
||||||
|
reject(e.data);
|
||||||
|
} else if (e?.data instanceof ImageBitmap) {
|
||||||
|
resolve(e.data);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// start up the worker
|
||||||
|
worker.postMessage(url);
|
||||||
|
});
|
||||||
|
|
||||||
|
// return the object
|
||||||
|
return {
|
||||||
|
processRadar,
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
// register display
|
// register display
|
||||||
registerDisplay(new Radar(11, 'radar'));
|
registerDisplay(new Radar(11, 'radar'));
|
||||||
|
|||||||
Reference in New Issue
Block a user