Skip to content

Commit

Permalink
Get user input
Browse files Browse the repository at this point in the history
  • Loading branch information
@Razzmatazzz
Razzmatazzz committed Jun 27, 2023
1 parent a600a96 commit 5e40663
Show file tree
Hide file tree
Showing 5 changed files with 265 additions and 127 deletions.
2 changes: 1 addition & 1 deletion .prettierrc.json
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
{
"printWidth": 80,
"tabWidth": 2,
"tabWidth": 4,
"useTabs": false,
"semi": true,
"singleQuote": true,
Expand Down
6 changes: 3 additions & 3 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,8 @@ npm start "path/to/image.png"

You can also create a `.env` file which specifies any of the following variables:

- `IMAGE_PATH`: The path to the large image that will be used to generate tiles. If `IMAGE_PATH` is provided, it is not necessary to supply a path via command line argument
- `MAP_NAME`: The folder name in which to store generated tiles
- `IMAGE_PATH`: The path to the large image that will be used to generate tiles. If `IMAGE_PATH` is provided, it is not necessary to supply a path via command line argument. If the path is to a folder, the user is presented with a list of images in that folder to use
- `MAP_NAME`: The folder name in which to store generated tiles; can be chosen at runtime
- `MAX_ZOOM`: The maximum zoom level for which to generate tiles (zero-indexed)
- `MIN_ZOOM`: The minimum zoom level for which to generate tiles (zero-indexed)
- `TILE_SIZE`: The size of tiles, defaults to 256
- `TILE_SIZE`: The size of tiles, defaults to an optimal value at runtime, can be chosen at runtime, and falls back to 256
355 changes: 232 additions & 123 deletions index.mjs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,142 +3,251 @@ import path from 'path';

import sharp from 'sharp';
import ora from 'ora';
import promptSync from 'prompt-sync';
import dotenv from 'dotenv';

dotenv.config();
const prompt = promptSync();

(async () => {
const tileSize = isNaN(process.env.TILE_SIZE)
? 256
: parseInt(process.env.TILE_SIZE);
const imagePath = process.env.IMAGE_PATH || process.argv[2];
let inputImage = sharp(imagePath, {
unlimited: true,
limitInputPixels: false
}).png();

let metadata = await inputImage.metadata();

let fullSize = Math.max(metadata.width, metadata.height);
let extend = fullSize % tileSize;
if (((fullSize + extend) / tileSize) % 2) {
extend += tileSize;
}
fullSize += extend;

if (extend) {
console.log(
`Padding source image from ${metadata.width}x${metadata.height} to ${fullSize}x${fullSize}`
);
inputImage = sharp(
await inputImage
.resize({
width: fullSize,
height: fullSize,
fit: sharp.fit.contain,
position: sharp.gravity.northwest,
background: {r: 1, g: 0, b: 0, alpha: 0}
})
.toBuffer()
);
}

let zoomLevels = 0;
for (let imageWidth = fullSize; imageWidth >= tileSize; imageWidth /= 2) {
zoomLevels++;
}

const maxZoom = Math.min(
isNaN(process.env.MAX_ZOOM) ? zoomLevels : parseInt(process.env.MAX_ZOOM),
zoomLevels - 1
);
const minZoom = process.env.MIN_ZOOM || 0;

await fs.mkdir('output').catch(error => {
if (!error.code === 'EEXIST') {
console.log(error);
let imagePath = process.env.IMAGE_PATH || process.argv[2];
if ((await fs.lstat(imagePath)).isDirectory()) {
const imageExtensions = ['jpg', 'jpeg', 'png', 'webp'];
const files = await fs
.readdir(imagePath)
.then(files =>
files.filter(filename =>
imageExtensions.some(ext =>
filename.toLocaleLowerCase().endsWith(`.${ext}`)
)
)
);
if (files.length < 1) {
console.log(`The folder ${imagePath} does not contain any images`);
return;
}
console.log('Select input image:');
for (let i = 0; i < files.length; i++) {
console.log(`${i + 1}. ${files[i]}`);
}
const index = prompt(`[1-${files.length}]: `);
if (isNaN(index) || index < 1 || index > files.length) {
console.log(`${index} is an invalid selection`);
return;
}
imagePath = path.join(imagePath, files[parseInt(index - 1)]);
}
});
let inputImage = sharp(imagePath, {
unlimited: true,
limitInputPixels: false
}).png();

const mapName =
process.env.MAP_NAME ||
imagePath.substring(
imagePath.lastIndexOf(path.sep) + 1,
imagePath.lastIndexOf('.')
);
await fs.mkdir(`output/${mapName}`).catch(error => {
if (!error.code === 'EEXIST') {
console.log(error);
let mapName =
process.env.MAP_NAME ||
imagePath.substring(
imagePath.lastIndexOf(path.sep) + 1,
imagePath.lastIndexOf('.')
);
const newMapName = prompt(`Output folder (${mapName}): `, mapName);
mapName = newMapName.replace(' ', '_');

let metadata = await inputImage.metadata();
console.log(`Image size: ${metadata.width}x${metadata.height}`);

let fullSize = Math.max(metadata.width, metadata.height);

let tileSize = 0;
const tileConfig = {difference: Number.MAX_SAFE_INTEGER};
tileLoop: for (let i = 200; i <= 300; i++) {
for (let pow = 0; pow < 100; pow++) {
if (i * Math.pow(2, pow) === fullSize) {
tileConfig.size = i;
tileConfig.pow = pow;
tileConfig.difference = 0;
tileSize = i;
break tileLoop;
}
}
}
if (!tileSize) {
for (let i = 200; i <= 300; i++) {
for (let pow = 0; pow < 100; pow++) {
const diff = Math.abs(i * Math.pow(2, pow) - fullSize);
if (diff < tileConfig.difference) {
tileConfig.size = i;
tileConfig.pow = pow;
tileConfig.difference = diff;
}
}
}
if (tileConfig.difference < Number.MAX_SAFE_INTEGER) {
tileSize = tileConfig.size;
} else {
tileSize = isNaN(process.env.TILE_SIZE)
? 256
: parseInt(process.env.TILE_SIZE);
}
}
});

let totalTiles = 0;
for (let z = minZoom; z <= maxZoom; z++) {
totalTiles += Math.pow((tileSize * Math.pow(2, z)) / tileSize, 2);
}

const tiles = [];
const tileCheck = async () => {
if (tiles.length > 32) {
await Promise.all(tiles);
tiles.length = 0;
const newTileSize = prompt(`Tile size (${tileSize}): `, tileSize);
if (isNaN(newTileSize)) {
console.log(`${newTileSize} is not a valid tile size`);
return;
}
};
let completedTiles = 0;
const zoomSpinner = ora({text: mapName, prefixText: '0.00%'});
for (let z = minZoom; z <= maxZoom; z++) {
const scaledSize = tileSize * Math.pow(2, z);
zoomSpinner.start(`${mapName} | z ${z}/${maxZoom}`);
const scaledMap = sharp(
await inputImage
.clone()
.resize({
width: scaledSize,
height: scaledSize,
fit: sharp.fit.contain,
position: sharp.gravity.northwest,
background: {r: 1, g: 0, b: 0, alpha: 0}
})
.toBuffer(),
{unlimited: true, limitInputPixels: false}
if (parseInt(newTileSize) !== tileConfig.size) {
tileConfig.size = parseInt(newTileSize);
tileConfig.difference = Number.MAX_SAFE_INTEGER;
tileConfig.pow = 0;
for (const pow = 0; pow < 100; pow++) {
const diff = Math.abs(
tileConfig.size * Math.pow(2, pow) - fullSize
);
if (diff < tileConfig.difference) {
tileConfig.pow = pow;
tileConfig.difference = diff;
}
}
}

tileSize = tileConfig.size;

if (tileConfig.difference) {
const resized = tileSize * Math.pow(2, tileConfig.pow);
if (resized < fullSize) {
console.log(`Shrinking source image to fit ${resized}x${resized}`);
inputImage = sharp(
await inputImage
.resize({
width: fullSize,
height: fullSize,
fit: sharp.fit.contain,
position: sharp.gravity.northwest,
background: {r: 1, g: 0, b: 0, alpha: 0}
})
.toBuffer()
);
} else {
console.log(`Padding source image to fit ${resized}x${resized}`);
inputImage = sharp(
await inputImage
.extend({
right: resized - metadata.width,
height: resized - metadata.height,
background: {r: 1, g: 0, b: 0, alpha: 0}
})
.toBuffer()
);
}
fullSize = resized;
}

let zoomLevels = 0;
for (let imageWidth = fullSize; imageWidth >= tileSize; imageWidth /= 2) {
zoomLevels++;
}

const maxZoom = Math.min(
isNaN(process.env.MAX_ZOOM)
? zoomLevels
: parseInt(process.env.MAX_ZOOM),
zoomLevels - 1
);
await fs.mkdir(`output/${mapName}/${z}`).catch(error => {
if (!error.code === 'EEXIST') {
console.log(error);
}
const minZoom = process.env.MIN_ZOOM || 0;

await fs.mkdir('output').catch(error => {
if (!error.code === 'EEXIST') {
console.log(error);
}
});
for (let x = 0; x < scaledSize / tileSize; x++) {
await fs.mkdir(`output/${mapName}/${z}/${x}`).catch(error => {

await fs.mkdir(`output/${mapName}`).catch(error => {
if (!error.code === 'EEXIST') {
console.log(error);
console.log(error);
}
});
for (let y = 0; y < scaledSize / tileSize; y++) {
tiles.push(
scaledMap
.clone()
.extract({
left: x * tileSize,
top: y * tileSize,
width: tileSize,
height: tileSize
})
.toFile(`output/${mapName}/${z}/${x}/${y}.png`)
.then(() => {
zoomSpinner.suffixText = `| x ${x}/${
scaledSize / tileSize - 1
} | y ${y}/${scaledSize / tileSize - 1}`;
completedTiles++;
zoomSpinner.prefixText = `${(
Math.round((completedTiles / totalTiles) * 10000) / 100
).toFixed(2)}%`;
})
});

await fs.writeFile(
`output/${mapName}/config.json`,
JSON.stringify(
{
tileSize: tileSize,
minZoom: minZoom,
maxZoom: maxZoom
},
null,
4
)
);

let totalTiles = 0;
for (let z = minZoom; z <= maxZoom; z++) {
totalTiles += Math.pow((tileSize * Math.pow(2, z)) / tileSize, 2);
}

const tiles = [];
const tileCheck = async () => {
if (tiles.length > 32) {
await Promise.all(tiles);
tiles.length = 0;
}
};
let completedTiles = 0;
const zoomSpinner = ora({text: mapName, prefixText: '0.00%'});
zoomSpinner.start();
for (let z = minZoom; z <= maxZoom; z++) {
const scaledSize = tileSize * Math.pow(2, z);
zoomSpinner.suffixText = `| Resizing to ${scaledSize}`;
const scaledMap = sharp(
await inputImage
.clone()
.resize({
width: scaledSize,
height: scaledSize,
fit: sharp.fit.contain,
position: sharp.gravity.northwest,
background: {r: 1, g: 0, b: 0, alpha: 0}
})
.toBuffer(),
{unlimited: true, limitInputPixels: false}
);
await tileCheck();
}
await fs.mkdir(`output/${mapName}/${z}`).catch(error => {
if (!error.code === 'EEXIST') {
console.log(error);
}
});
for (let x = 0; x < scaledSize / tileSize; x++) {
await fs.mkdir(`output/${mapName}/${z}/${x}`).catch(error => {
if (!error.code === 'EEXIST') {
console.log(error);
}
});
for (let y = 0; y < scaledSize / tileSize; y++) {
tiles.push(
scaledMap
.clone()
.extract({
left: x * tileSize,
top: y * tileSize,
width: tileSize,
height: tileSize
})
.toFile(`output/${mapName}/${z}/${x}/${y}.png`)
.then(() => {
zoomSpinner.suffixText = `| z ${z}/${maxZoom} | x ${x}/${
scaledSize / tileSize - 1
} | y ${y}/${scaledSize / tileSize - 1}`;
completedTiles++;
zoomSpinner.prefixText = `${(
Math.round(
(completedTiles / totalTiles) * 10000
) / 100
).toFixed(2)}%`;
})
);
await tileCheck();
}
}
zoomSpinner.suffixText = '';
zoomSpinner.prefixText = '';
}
zoomSpinner.suffixText = '';
zoomSpinner.prefixText = '';
}
zoomSpinner.succeed(mapName);
zoomSpinner.succeed();
})();
Loading

0 comments on commit 5e40663

Please sign in to comment.