146 lines
3.4 KiB
Python
146 lines
3.4 KiB
Python
"""
|
|
Central Add Node menu manifest.
|
|
|
|
Edit MENU_LAYOUT to rearrange which nodes appear under each menu leaf and
|
|
their order within that leaf. Node classes not listed here fall back to their
|
|
class CATEGORY.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
|
|
MENU_LAYOUT: dict[str, list[str]] = {
|
|
"Input": [
|
|
"Image",
|
|
"Note",
|
|
"ImageDemo",
|
|
"Folder",
|
|
"RangeSlider",
|
|
"Coordinate",
|
|
"CoordinatePair",
|
|
],
|
|
"Display": [
|
|
"ColorMap",
|
|
"Font",
|
|
"ColormapAdjust",
|
|
"PreviewImage",
|
|
"ValueIO",
|
|
"View3D",
|
|
"Save",
|
|
"SaveImage",
|
|
"PrintTable",
|
|
],
|
|
"Overlay": [
|
|
"Markup",
|
|
"Annotations",
|
|
"AngleMeasure",
|
|
],
|
|
"Geometry": [
|
|
"CropResizeField",
|
|
"RotateField",
|
|
"FlipField",
|
|
"Resample",
|
|
"FieldArithmetic",
|
|
],
|
|
"Filter": [
|
|
"GaussianFilter",
|
|
"MedianFilter",
|
|
"EdgeDetect",
|
|
"Gradient",
|
|
"FFTFilter",
|
|
"ScarRemoval",
|
|
],
|
|
"Spectral": [
|
|
"FFT2D",
|
|
"FFT2DInverse",
|
|
"FFTFilter",
|
|
"ACF2D",
|
|
"ACF1D",
|
|
"PSDF",
|
|
],
|
|
"Level & Correct": [
|
|
"FixZero",
|
|
"PlaneLevelField",
|
|
"PolyLevelField",
|
|
"FacetLevelField",
|
|
"LineCorrection",
|
|
"ScarRemoval",
|
|
],
|
|
"Measure": [
|
|
"FFT1D",
|
|
"AngleMeasure",
|
|
"CrossSection",
|
|
"Histogram",
|
|
"Cursors",
|
|
"Curvature",
|
|
"FractalDimension",
|
|
"ACF2D",
|
|
"ACF1D",
|
|
"PSDF",
|
|
"SlopeDistribution",
|
|
"RadialProfile",
|
|
"Statistics",
|
|
"Stats",
|
|
],
|
|
"Mask": [
|
|
"DrawMask",
|
|
"ThresholdMask",
|
|
"MaskMorphology",
|
|
"MaskInvert",
|
|
"MaskOperations",
|
|
"GrainDistanceTransform",
|
|
"WatershedSegmentation",
|
|
],
|
|
"Grains": [
|
|
"GrainDistanceTransform",
|
|
"WatershedSegmentation",
|
|
"GrainAnalysis",
|
|
"GrainFilter",
|
|
],
|
|
"Tip": [
|
|
"TipModel",
|
|
"TipDeconvolution",
|
|
"BlindTipEstimate",
|
|
],
|
|
}
|
|
|
|
|
|
_CATEGORY_ORDER = {category: index for index, category in enumerate(MENU_LAYOUT)}
|
|
_NODE_METADATA: dict[str, dict[str, Any]] = {}
|
|
for category, class_names in MENU_LAYOUT.items():
|
|
for node_order, class_name in enumerate(class_names):
|
|
metadata = _NODE_METADATA.setdefault(class_name, {
|
|
"category": category,
|
|
"category_order": _CATEGORY_ORDER[category],
|
|
"menu_order": node_order,
|
|
"menu_categories": [],
|
|
})
|
|
metadata["menu_categories"].append({
|
|
"category": category,
|
|
"category_order": _CATEGORY_ORDER[category],
|
|
"menu_order": node_order,
|
|
})
|
|
|
|
|
|
def get_menu_metadata(class_name: str, cls: type | None = None) -> dict[str, Any]:
|
|
metadata = _NODE_METADATA.get(class_name)
|
|
if metadata is not None:
|
|
return dict(metadata)
|
|
|
|
# Nodes not listed in MENU_LAYOUT (e.g. plugins) can declare their own
|
|
# menu category via a CATEGORY class attribute. Falls back to "Unsorted".
|
|
category = getattr(cls, "CATEGORY", "Unsorted") if cls else "Unsorted"
|
|
order = len(_CATEGORY_ORDER)
|
|
return {
|
|
"category": category,
|
|
"category_order": order,
|
|
"menu_order": 10_000,
|
|
"menu_categories": [{
|
|
"category": category,
|
|
"category_order": order,
|
|
"menu_order": 10_000,
|
|
}],
|
|
}
|