Initial commit
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
dinlo
+294
@@ -0,0 +1,294 @@
|
||||
"use strict";
|
||||
Object.defineProperty(exports, "__esModule", { value: true });
|
||||
var IconItem_js_1 = require("../data/IconItem.js");
|
||||
var RawIconItem_js_1 = require("../data/RawIconItem.js");
|
||||
var functions_js_1 = require("../util/functions.js");
|
||||
function generateEntryBinary(icons) {
|
||||
var count = icons.length;
|
||||
if (count > 65535) {
|
||||
count = 65535;
|
||||
}
|
||||
var size = 6 + 14 * icons.length;
|
||||
var bin = new ArrayBuffer(size);
|
||||
var view = new DataView(bin);
|
||||
view.setUint16(0, 0, true); // reserved
|
||||
view.setUint16(2, 1, true); // icon type
|
||||
view.setUint16(4, count, true);
|
||||
var offset = 6;
|
||||
icons.forEach(function (icon) {
|
||||
view.setUint8(offset, icon.width >= 256 ? 0 : icon.width);
|
||||
view.setUint8(offset + 1, icon.height >= 256 ? 0 : icon.height);
|
||||
view.setUint8(offset + 2, icon.colors >= 256 ? 0 : icon.colors);
|
||||
view.setUint8(offset + 3, 0);
|
||||
view.setUint16(offset + 4, icon.planes, true);
|
||||
view.setUint16(offset + 6, icon.bitCount, true);
|
||||
view.setUint32(offset + 8, icon.dataSize, true);
|
||||
view.setUint16(offset + 12, icon.iconID, true);
|
||||
offset += 14;
|
||||
});
|
||||
return bin;
|
||||
}
|
||||
function findUnusedIconID(entries, lang, isCursor) {
|
||||
var type = isCursor ? 1 : 3;
|
||||
// (ignore string id)
|
||||
var filteredIDs = entries
|
||||
.filter(function (e) { return e.type === type && e.lang === lang && typeof e.id === 'number'; })
|
||||
.map(function (e) { return e.id; })
|
||||
.sort(function (a, b) { return a - b; });
|
||||
var idCurrent = 1;
|
||||
for (var _i = 0, filteredIDs_1 = filteredIDs; _i < filteredIDs_1.length; _i++) {
|
||||
var id = filteredIDs_1[_i];
|
||||
if (idCurrent < id) {
|
||||
return {
|
||||
id: idCurrent,
|
||||
last: false,
|
||||
};
|
||||
}
|
||||
else if (idCurrent === id) {
|
||||
++idCurrent;
|
||||
}
|
||||
}
|
||||
return {
|
||||
id: idCurrent,
|
||||
last: true,
|
||||
};
|
||||
}
|
||||
/**
|
||||
* A class that treats icon-group resource data (`RT_ICON_GROUP`).
|
||||
* Note that this class does not treat `RT_ICON` data.
|
||||
*
|
||||
* - To pick all icons, use `IconGroupEntry.fromEntries`
|
||||
* and `IconGroupEntry.prototype.getIconItemsFromEntries`.
|
||||
* - The easiest way to add/replace icons is using `IconGroupEntry.replaceIconsForResource`,
|
||||
* which treats both `RT_ICON_GROUP` and `RT_ICON` entries.
|
||||
*/
|
||||
var IconGroupEntry = /** @class */ (function () {
|
||||
function IconGroupEntry(groupEntry) {
|
||||
var view = new DataView(groupEntry.bin);
|
||||
var totalSize = view.byteLength;
|
||||
var icons = [];
|
||||
if (view.getUint16(2, true) === 1) {
|
||||
var count = view.getUint16(4, true);
|
||||
var offset = 6;
|
||||
for (var i = 0; i < count; ++i) {
|
||||
icons.push({
|
||||
width: functions_js_1.readUint8WithLastOffset(view, offset, totalSize),
|
||||
height: functions_js_1.readUint8WithLastOffset(view, offset + 1, totalSize),
|
||||
colors: functions_js_1.readUint8WithLastOffset(view, offset + 2, totalSize),
|
||||
planes: functions_js_1.readUint16WithLastOffset(view, offset + 4, totalSize),
|
||||
bitCount: functions_js_1.readUint16WithLastOffset(view, offset + 6, totalSize),
|
||||
dataSize: functions_js_1.readUint32WithLastOffset(view, offset + 8, totalSize),
|
||||
iconID: functions_js_1.readUint16WithLastOffset(view, offset + 12, totalSize),
|
||||
});
|
||||
offset += 14; // 16 for .ico file, but 14 for resource data
|
||||
}
|
||||
}
|
||||
this.id = groupEntry.id;
|
||||
this.lang = groupEntry.lang;
|
||||
this.icons = icons;
|
||||
}
|
||||
IconGroupEntry.fromEntries = function (entries) {
|
||||
return entries
|
||||
.filter(function (e) { return e.type === 14; })
|
||||
.map(function (e) { return new IconGroupEntry(e); });
|
||||
};
|
||||
IconGroupEntry.prototype.generateEntry = function () {
|
||||
var bin = generateEntryBinary(this.icons);
|
||||
return {
|
||||
type: 14,
|
||||
id: this.id,
|
||||
lang: this.lang,
|
||||
codepage: 0,
|
||||
bin: bin,
|
||||
};
|
||||
};
|
||||
/**
|
||||
* Return an array of `IconItem` / `RawIconItem`, which are in the group of this `IconGroupEntry` instance,
|
||||
* from specified resource entries.
|
||||
*/
|
||||
IconGroupEntry.prototype.getIconItemsFromEntries = function (entries) {
|
||||
var _this = this;
|
||||
return entries
|
||||
.map(function (e) {
|
||||
if (e.type !== 3 || e.lang !== _this.lang) {
|
||||
return null;
|
||||
}
|
||||
var c = _this.icons
|
||||
.filter(function (icon) { return e.id === icon.iconID; })
|
||||
.shift();
|
||||
if (!c) {
|
||||
return null;
|
||||
}
|
||||
return {
|
||||
entry: e,
|
||||
icon: c,
|
||||
};
|
||||
})
|
||||
.filter(function (item) { return !!item; })
|
||||
.map(function (item) {
|
||||
var bin = item.entry.bin;
|
||||
var view = new DataView(bin);
|
||||
if (view.getUint32(0, true) === 0x28) {
|
||||
return IconItem_js_1.default.from(bin);
|
||||
}
|
||||
else {
|
||||
var c = item.icon;
|
||||
return RawIconItem_js_1.default.from(bin, c.width, c.height, c.bitCount);
|
||||
}
|
||||
});
|
||||
};
|
||||
/**
|
||||
* Add or replace icon resource entries with specified icon data.
|
||||
* The IDs of individual icon resources (`RT_ICON`) are calculated automatically.
|
||||
* @param destEntries base (destination) resource entries.
|
||||
* @param iconGroupID the icon ID for the new resource data.
|
||||
* If the icon-group resource of the ID and 'lang' value already exists,
|
||||
* the resource data is replaced; otherwise the resource data is appended.
|
||||
* @param lang the language for specified icons (0 for neutral, 0x409 for en-US)
|
||||
* @param icons the icons to replace
|
||||
*/
|
||||
IconGroupEntry.replaceIconsForResource = function (destEntries, iconGroupID, lang, icons) {
|
||||
// find existing entry
|
||||
var entry = destEntries
|
||||
.filter(function (e) { return e.type === 14 && e.id === iconGroupID && e.lang === lang; })
|
||||
.shift();
|
||||
var tmpIconArray = icons.map(function (icon) {
|
||||
if (icon.isIcon()) {
|
||||
var width = icon.width, height = icon.height;
|
||||
if (width === null) {
|
||||
width = icon.bitmapInfo.width;
|
||||
}
|
||||
if (height === null) {
|
||||
height = icon.bitmapInfo.height;
|
||||
// if mask is specified, the icon height must be the half of bitmap height
|
||||
if (icon.masks !== null) {
|
||||
height = Math.floor(height / 2);
|
||||
}
|
||||
}
|
||||
return {
|
||||
base: icon,
|
||||
bm: {
|
||||
width: width,
|
||||
height: height,
|
||||
planes: icon.bitmapInfo.planes,
|
||||
bitCount: icon.bitmapInfo.bitCount,
|
||||
},
|
||||
bin: icon.generate(),
|
||||
id: 0,
|
||||
};
|
||||
}
|
||||
else {
|
||||
return {
|
||||
base: icon,
|
||||
bm: {
|
||||
width: icon.width,
|
||||
height: icon.height,
|
||||
planes: 1,
|
||||
bitCount: icon.bitCount,
|
||||
},
|
||||
bin: icon.bin,
|
||||
id: 0,
|
||||
};
|
||||
}
|
||||
});
|
||||
if (entry) {
|
||||
// remove unused icon data
|
||||
for (var i = destEntries.length - 1; i >= 0; --i) {
|
||||
var e = destEntries[i];
|
||||
if (e != null && e.type === 3) {
|
||||
// RT_ICON
|
||||
if (!isIconUsed(e, destEntries, entry)) {
|
||||
destEntries.splice(i, 1);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
else {
|
||||
// create new entry
|
||||
entry = {
|
||||
type: 14,
|
||||
id: iconGroupID,
|
||||
lang: lang,
|
||||
codepage: 0,
|
||||
// set later
|
||||
bin: null,
|
||||
};
|
||||
destEntries.push(entry);
|
||||
}
|
||||
// append icons
|
||||
var idInfo;
|
||||
tmpIconArray.forEach(function (icon) {
|
||||
if (!(idInfo === null || idInfo === void 0 ? void 0 : idInfo.last)) {
|
||||
idInfo = findUnusedIconID(destEntries, lang, false);
|
||||
}
|
||||
else {
|
||||
++idInfo.id;
|
||||
}
|
||||
destEntries.push({
|
||||
type: 3,
|
||||
id: idInfo.id,
|
||||
lang: lang,
|
||||
codepage: 0,
|
||||
bin: icon.bin,
|
||||
});
|
||||
// set 'id' field to use in generateEntryBinary
|
||||
icon.id = idInfo.id;
|
||||
});
|
||||
var binEntry = generateEntryBinary(tmpIconArray.map(function (icon) {
|
||||
var width = Math.abs(icon.bm.width);
|
||||
if (width >= 256) {
|
||||
width = 0;
|
||||
}
|
||||
var height = Math.abs(icon.bm.height);
|
||||
if (height >= 256) {
|
||||
height = 0;
|
||||
}
|
||||
var colors = 0;
|
||||
if (icon.base.isIcon()) {
|
||||
var bmBase = icon.base.bitmapInfo;
|
||||
colors = bmBase.colorUsed || bmBase.colors.length;
|
||||
if (!colors) {
|
||||
switch (bmBase.bitCount) {
|
||||
case 1:
|
||||
colors = 2;
|
||||
break;
|
||||
case 4:
|
||||
colors = 16;
|
||||
break;
|
||||
// case 8:
|
||||
// colors = 256;
|
||||
// break;
|
||||
}
|
||||
}
|
||||
if (colors >= 256) {
|
||||
colors = 0;
|
||||
}
|
||||
}
|
||||
return {
|
||||
width: width,
|
||||
height: height,
|
||||
colors: colors,
|
||||
planes: icon.bm.planes,
|
||||
bitCount: icon.bm.bitCount,
|
||||
dataSize: icon.bin.byteLength,
|
||||
iconID: icon.id,
|
||||
};
|
||||
}));
|
||||
// rewrite entry
|
||||
entry.bin = binEntry;
|
||||
function isIconUsed(icon, allEntries, excludeGroup) {
|
||||
return allEntries.some(function (e) {
|
||||
if (e.type !== 14 ||
|
||||
(e.id === excludeGroup.id && e.lang === excludeGroup.lang)) {
|
||||
return false;
|
||||
}
|
||||
var g = new IconGroupEntry(e);
|
||||
return g.icons.some(function (c) {
|
||||
return c.iconID === icon.id;
|
||||
});
|
||||
});
|
||||
}
|
||||
};
|
||||
return IconGroupEntry;
|
||||
}());
|
||||
exports.default = IconGroupEntry;
|
||||
Reference in New Issue
Block a user