NarrowbandCombinations.js

// ****************************************************************************
// NarrowbandCombinations — Standalone version of AutoIntegrate narrowband combinations
// ****************************************************************************

#engine v8
#feature-id    AutoIntegrate  > Narrowband Combinations
#feature-info  Narrowband Combinations using AutoIntegrate tools.

#ifndef NO_SOLVER_LIBRARY
#define NO_SOLVER_LIBRARY
#endif
#ifndef AUTOINTEGRATE_STANDALONE
#define AUTOINTEGRATE_STANDALONE
#endif

#include "AutoIntegrateGlobal.js"
#include "AutoIntegrateUtil.js"
#include "AutoIntegrateGUITools.js"
#include "AutoIntegrateCalibrate.js"
#include "AutoIntegrateEngine.js"
#include "AutoIntegratePreview.js"

// =============================================================================
//  Dummy this.flowchart routines
// =============================================================================

#ifndef AUTOINTEGRATEDUMMYFLOWCHART
#define AUTOINTEGRATEDUMMYFLOWCHART

class AutoIntegrateDummyFlowchart extends Object
{
    constructor() {
        super();
    }

    flowchartOperation() {};
    flowchartOperationEnd() {};
    flowchartParentBegin() {};
    flowchartParentEnd() {};
    flowchartChildBegin() {};
    flowchartChildEnd() {};
    flowchartMaskBegin() {};
    flowchartMaskEnd() {};
    flowchartInit() {};
    flowchartDone() {};
    flowchartReset() {};
    flowchartPrint() {};
}

#endif /* AUTOINTEGRATEDUMMYFLOWCHART */

// =============================================================================
//  DIALOG WITH PREVIEW
// =============================================================================

class AutoIntegrateNarrowbandCombinationsDialog extends Dialog {
    constructor() {
        super();

    this.title = "Narrowband Combinations";
    this.version = "1.00";

    this.windowTitle = this.title + " v" + this.version;
    // this.minWidth = 1000;

    this.global = new AutoIntegrateGlobal();

    for (let i = 0; i < Runtime.jsArguments.length; i++) {
        if (Runtime.jsArguments[i] == "do_not_read_settings") {
            console.writeln("do_not_read_settings");
            this.global.do_not_read_settings = true;
        }
    }

    this.util = new AutoIntegrateUtil(this.global);
    this.flowchart = new AutoIntegrateDummyFlowchart();
    this.engine = new AutoIntegrateEngine(this.global, this.util, this.flowchart);
    this.guitools = new AutoIntegrateGUITools(this, this.global, this.util, this.engine);

    // Read parameter default settings from persistent module settings.
    // These can be saved using the AutoIntegrate script.
    this.util.initStandalone();

    // Store original image for reset functionality
    this.previewImage = null;
    this.autoSTFPreviewImage = null;
    this.autoSTF = true;

    this.selectedMappings = null;

    this.initGUI();
    this.initGUI2();

} // End of constructor

   // -------------------------------------------------------------------------
   // Preview functions
   // -------------------------------------------------------------------------

    setPreviewImage(image)
    {
        if (this.global.debug) console.writeln("AutoIntegrateNarrowbandCombinationsDialog::setPreviewImage");
        this.previewControl.SetImage(image, "[Preview]");
    }

    // -------------------------------------------------------------------------
    // Image Selections
    // -------------------------------------------------------------------------

    getOpenWindows() {
        var windows = [];
        var allWindows = ImageWindow.windows;
        for (var i = 0; i < allWindows.length; i++) {
            if (!allWindows[i].isNull && !allWindows[i].mainView.isNull) {
                windows.push(allWindows[i]);
            }
        }
        return windows;
    }

    getComboBoxList(windows) {
        var list = [];
        if (windows.length === 0) {
            list.push("<No images>");
            return list;
        }
        list.push("<Select image>");
        for (var i = 0; i < windows.length; i++) {
            list.push(windows[i].mainView.id);
        }
        return list;
    }

    updateImageList(imageComboBox, combobox_list) {
        imageComboBox.clear();
        for (var i = 0; i < combobox_list.length; i++) {
            imageComboBox.addItem(combobox_list[i]);
        }
    }

    initGUI() {

    this.windows = this.getOpenWindows();
    this.combobox_list = this.getComboBoxList(this.windows);

    this.channelComboBoxes = [];

    this.HSOSizer = new VerticalSizer;
    this.HSOSizer.spacing = 4;
    var channelNames = ["H", "S", "O"];
    for (var i = 0; i < channelNames.length; i++) {
        let channelName = channelNames[i];
        let channelComboBox = new ComboBox(this);
        channelComboBox.toolTip = "Select image for " + channelName + " channel.";
        this.updateImageList(channelComboBox, this.combobox_list);
        var channelSizer = new HorizontalSizer;
        channelSizer.spacing = 4;
        channelSizer.margin = 6;
        channelSizer.add(this.guitools.newLabel(this, channelName + ": ", channelComboBox.toolTip));
        channelSizer.add(channelComboBox);
        channelSizer.addStretch();
        this.HSOSizer.add(channelSizer);
        this.channelComboBoxes.push(channelComboBox);
    }

    this.RGBSizer = new VerticalSizer;
    this.RGBSizer.spacing = 4;
    var channelNames = ["R", "G", "B"];
    for (var i = 0; i < channelNames.length; i++) {
        let channelName = channelNames[i];
        let channelComboBox = new ComboBox(this);
        channelComboBox.toolTip = "Select image for " + channelName + " channel.";
        this.updateImageList(channelComboBox, this.combobox_list);
        var channelSizer = new HorizontalSizer;
        channelSizer.spacing = 4;
        channelSizer.margin = 6;
        channelSizer.add(this.guitools.newLabel(this, channelName + ": ", channelComboBox.toolTip));
        channelSizer.add(channelComboBox);
        channelSizer.addStretch();
        this.RGBSizer.add(channelSizer);
        this.channelComboBoxes.push(channelComboBox);
    }

    this.channelsSizer = new VerticalSizer;
    this.channelsSizer.spacing = 8;
    this.channelsSizer.add(this.HSOSizer);
    this.channelsSizer.addSpacing(12);
    this.channelsSizer.add(this.RGBSizer);

    this.channelsGroupBox = new GroupBox(this);
    this.channelsGroupBox.title = "Select images for Channels";
    this.channelsGroupBox.sizer = this.channelsSizer;

} // End of initGUI

    // -------------------------------------------------------------------------
    // Processing
    // -------------------------------------------------------------------------

    // Create [ 'channel', 'image id' ] mappings from selected images into an array
    // Skip images with <Select image> or <No images>
    generateMappingsFromSelection() {
        this.selectedMappings = [];
        var channelKeys = ['H', 'S', 'O', 'R', 'G', 'B'];
        for (var i = 0; i < this.channelComboBoxes.length; i++) {
            var comboBox = this.channelComboBoxes[i];
            var imageName = this.combobox_list[comboBox.currentItem];
            if (imageName !== "<Select image>" && imageName !== "<No images>") {
                this.selectedMappings.push( [ channelKeys[i], imageName ] );
            }
        }
        console.writeln("Selected mappings: " + JSON.stringify(this.selectedMappings));
        return this.selectedMappings;
    }

    applyPreview() {
        if (this.testMappings) {
            this.selectedMappings = this.testMappings;
            console.writeln("Using test mappings: " + JSON.stringify(this.selectedMappings));
        } else {
            this.selectedMappings = this.generateMappingsFromSelection();
            console.writeln("Using user selected mappings: " + JSON.stringify(this.selectedMappings));
        }
        if (!this.selectedMappings || this.selectedMappings.length === 0) {
            this.statusLabel.text = "No image loaded for preview.";
            return;
        }

        var narrowband_mappings = {
            target_palette: {
                name: this.global.par.narrowband_mapping.val,
                R: this.global.par.custom_R_mapping.val,
                G: this.global.par.custom_G_mapping.val,
                B: this.global.par.custom_B_mapping.val
            },
            mappings: this.selectedMappings
        };

        console.writeln("Narrowband mappings, " + JSON.stringify(narrowband_mappings));

        this.engine.standalone_narrowband_mappings = narrowband_mappings;

        console.show();
        console.writeln("Applying narrowband mapping to preview...");

        this.previewButton.enabled = false;
        this.statusLabel.text = "Processing preview...";
        CoreApplication.processEvents();

        try {
            // Pick the first channel images as a model for preview size
            var firstImageId = this.selectedMappings[0][1];
            console.writeln("Using first image for preview size: " + firstImageId);
            var firstImgWin = this.util.findWindow(firstImageId);

            // Create a temporary color image for processing
            var tempWindow = new ImageWindow(
                                    firstImgWin.mainView.image.width,
                                    firstImgWin.mainView.image.height,
                                    3,
                                    32,
                                    true,
                                    true,
                                    "NarrowbandCombinations_temp"
            );

            // Process the temp view
            this.engine.enhancementsProcessingEngine(this, tempWindow.mainView.id, true);

            // Copy result back to preview image
            this.previewImage = new Image(tempWindow.mainView.image);

            // make a stretched copy of the image
            this.engine.autoStretch(tempWindow);
            this.autoSTFPreviewImage = new Image(tempWindow.mainView.image);

            tempWindow.forceClose();

            // Update preview
            if (this.autoSTF) {
                this.setPreviewImage(this.autoSTFPreviewImage);
            } else {
                this.setPreviewImage(this.previewImage);
            }

            this.statusLabel.text = "Preview updated successfully.";
            console.writeln("Preview processing complete.");

        } catch (error) {
            this.statusLabel.text = "Error during preview: " + error.message;
            console.criticalln("Preview Error: " + error.message);
        }

        this.previewButton.enabled = true;
    };

    processFinal() {
        if (!this.previewImage) {
            this.statusLabel.text = "No processed image.";
            return;
        }

        console.show();
        console.writeln("Creating final processed image...");

        this.processButton.enabled = false;
        CoreApplication.processEvents();

        try {
            // Create new window with processed result
            // Use the selected narrowband mapping name as window title
            // Some mappings start with a number so in that case add _ at the start
            var windowId = this.util.mapBadChars(this.global.par.narrowband_mapping.val);
            if (/^\d/.test(windowId)) {
                windowId = "_" + windowId;
            }
            console.writeln("Creating final image window "+ windowId);
            var targetWindow = new ImageWindow(
                                    this.previewImage.width,
                                    this.previewImage.height,
                                    this.previewImage.numberOfChannels,
                                    32,
                                    true,
                                    this.previewImage.colorSpace != ColorSpace.Gray,
                                    windowId);

            console.writeln("Assigning processed image to final window...");
            targetWindow.mainView.beginProcess(UndoFlag.NoSwapFile);
            targetWindow.mainView.image.assign(this.previewImage);
            targetWindow.mainView.endProcess();

            targetWindow.show();

            this.statusLabel.text = "Final image created: " + targetWindow.mainView.id;
            console.writeln("Final processing complete.");

        } catch (error) {
            this.statusLabel.text = "Error during final processing: " + error.message;
            console.criticalln("Processing Error: " + error.message);
        }

        this.processButton.enabled = true;
    };

    initGUI2() {

    // -------------------------------------------------------------------------
    // Left Side: Preview Control
    // -------------------------------------------------------------------------

    this.autoSTFCheckBox = new CheckBox(this);
    this.autoSTFCheckBox.text = "AutoSTF";
    this.autoSTFCheckBox.toolTip = "Automatically apply Screen Transfer Function (STF) to preview image.";
    this.autoSTFCheckBox.checked = true;
    this.autoSTFCheckBox.onCheck = (checked) => {
        this.autoSTF = checked;
        // Update preview
        if (this.autoSTF) {
            this.setPreviewImage(this.autoSTFPreviewImage);
        } else {
            this.setPreviewImage(this.previewImage);
        }
    };

    // Preview buttons
    this.previewButton = new PushButton(this);
    this.previewButton.text = "Update Preview";
    this.previewButton.icon = this.scaledResource(":/icons/play.png");
    this.previewButton.toolTip = "Apply narrowband combination to preview image.";
    this.previewButton.onClick = () => {
        this.applyPreview();
    };

    this.previewButtonsSizer = new HorizontalSizer;
    this.previewButtonsSizer.spacing = 4;
    this.previewButtonsSizer.addStretch();
    this.previewButtonsSizer.add(this.autoSTFCheckBox);
    this.previewButtonsSizer.add(this.previewButton, 50);

    // Preview control
    this.previewControl = new AutoIntegratePreviewControl(this, "narrowband_preview", this.engine, this.util, this.global, 600, 600, false);

    this.leftSizer = new VerticalSizer;
    this.leftSizer.spacing = 4;
    this.leftSizer.add(this.previewControl, 600);
    this.leftSizer.add(this.previewButtonsSizer);

    // -------------------------------------------------------------------------
    // Right Side: Title and Controls
    // -------------------------------------------------------------------------

    this.titleLabel = new Label(this);
    this.titleLabel.text = this.title + " v" + this.version;
    this.titleLabel.textAlignment = TextAlignment.Center;
    this.titleLabel.styleSheet = "font-size: 14pt; font-weight: bold; color: #4488FF;";

    this.subtitleLabel = new Label(this);
    this.subtitleLabel.text = "Select channel images and apply narrowband palette combination.";
    this.subtitleLabel.textAlignment = TextAlignment.Center;
    this.subtitleLabel.styleSheet = "font-size: 9pt; color: #888888; font-style: italic;";

   // -------------------------------------------------------------------------
   // Load and save JSON controls
   // -------------------------------------------------------------------------

   var obj = this.guitools.newJsonSizerObj(this, null, "NarrowbandCombinationsSettings.json");
   this.loadSaveSizer = obj.sizer;

   // -------------------------------------------------------------------------
   // Narrowband palette selection
   // -------------------------------------------------------------------------

    if (this.global.debug) console.writeln("AutoIntegrateNarrowbandCombinationsDialog:: creating narrowbandCustomPaletteGroupBox");

    this.narrowbandCustomPaletteSizer = this.guitools.createNarrowbandCustomPaletteSizer(this);

    this.narrowbandCustomPaletteGroupBox = new GroupBox(this);
    this.narrowbandCustomPaletteGroupBox.title = "Narrowband Palette";
    this.narrowbandCustomPaletteGroupBox.sizer = this.narrowbandCustomPaletteSizer;

   // -------------------------------------------------------------------------
   // Status
   // -------------------------------------------------------------------------

   this.statusLabel = new Label(this);
   this.statusLabel.text = "";
   this.statusLabel.textAlignment = TextAlignment.Center;
   this.statusLabel.styleSheet = "color: #AAAAAA;";

   // -------------------------------------------------------------------------
   // Buttons
   // -------------------------------------------------------------------------

    this.resetButton = new PushButton(this);
    this.resetButton.text = "Reset";
    this.resetButton.toolTip = "Reset all parameters to defaults.";
    this.resetButton.onClick = () => {
            this.util.setParameterDefaults();
    };

    this.processButton = new PushButton(this);
    this.processButton.text = "Process Final";
    this.processButton.icon = this.scaledResource(":/icons/power.png");
    this.processButton.toolTip = "Create final processed image (does not overwrite original).";
    this.processButton.onClick = () => {
        this.processFinal();
    };

    this.closeButton = new PushButton(this);
    this.closeButton.text = "Close";
    this.closeButton.icon = this.scaledResource(":/icons/close.png");
    this.closeButton.onClick = () => {
        this.ok();
    };

    this.buttonsSizer = new HorizontalSizer;
    this.buttonsSizer.spacing = 6;
    this.buttonsSizer.add(this.resetButton);
    this.buttonsSizer.addStretch();
    this.buttonsSizer.add(this.processButton);
    this.buttonsSizer.add(this.closeButton);

   // -------------------------------------------------------------------------
   // Right Side Layout
   // -------------------------------------------------------------------------

   this.rightSizer = new VerticalSizer;
   this.rightSizer.spacing = 8;
   this.rightSizer.add(this.titleLabel);
   this.rightSizer.add(this.subtitleLabel);
   this.rightSizer.add(this.statusLabel);
   this.rightSizer.add(this.loadSaveSizer);
   this.rightSizer.add(this.channelsGroupBox);
   this.rightSizer.add(this.narrowbandCustomPaletteGroupBox);
   this.rightSizer.add(this.buttonsSizer);

   // -------------------------------------------------------------------------
   // Main Layout (Preview on Left, Controls on Right)
   // -------------------------------------------------------------------------

   this.sizer = new HorizontalSizer;
   this.sizer.margin = 8;
   this.sizer.spacing = 8;
   this.sizer.add(this.leftSizer);
   this.sizer.add(this.rightSizer);

} // End of initGUI2
}

// =============================================================================
//  MAIN ENTRY POINT
// =============================================================================

#ifndef AUTOINTEGRATE_NO_MAIN

function main() {
   console.show();

   var dialog = new AutoIntegrateNarrowbandCombinationsDialog();
   dialog.execute();
}

main();

#endif /* AUTOINTEGRATE_NO_MAIN */