gov.redhawk.frontend.ui.internal.AllocateHandler.java Source code

Introduction

Here is the source code for gov.redhawk.frontend.ui.internal.AllocateHandler.java

Source

/** 
 * This file is protected by Copyright. 
 * Please refer to the COPYRIGHT file distributed with this source distribution.
 * 
 * This file is part of REDHAWK IDE.
 * 
 * All rights reserved.  This program and the accompanying materials are made available under 
 * the terms of the Eclipse Public License v1.0 which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html.
 *
 */
package gov.redhawk.frontend.ui.internal;

import gov.redhawk.frontend.FrontendFactory;
import gov.redhawk.frontend.TunerContainer;
import gov.redhawk.frontend.TunerStatus;
import gov.redhawk.frontend.UnallocatedTunerContainer;
import gov.redhawk.frontend.ui.wizard.TunerAllocationWizard;
import gov.redhawk.frontend.util.TunerUtils;
import gov.redhawk.model.sca.ScaDevice;
import gov.redhawk.sca.util.PluginUtil;

import java.util.ArrayList;
import java.util.List;

import mil.jpeojtrs.sca.util.ScaEcoreUtils;

import org.eclipse.core.commands.AbstractHandler;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.commands.IHandler;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.wizard.WizardDialog;
import org.eclipse.ui.handlers.HandlerUtil;

/**
 *
 */
public class AllocateHandler extends AbstractHandler implements IHandler {

    /* (non-Javadoc)
     * @see org.eclipse.core.commands.IHandler#execute(org.eclipse.core.commands.ExecutionEvent)
     */
    @Override
    public Object execute(ExecutionEvent event) throws ExecutionException {
        IStructuredSelection selection = (IStructuredSelection) HandlerUtil.getActiveMenuSelection(event);
        if (selection == null) {
            selection = (IStructuredSelection) HandlerUtil.getCurrentSelection(event);
        }

        if (selection.getFirstElement() instanceof TunerStatus && selection.size() > 1) {
            Object[] items = selection.toArray();
            TunerStatus[] tuners = castArray(items, new TunerStatus[0]);
            if (tuners.length > 0) {
                WizardDialog dialog = new WizardDialog(HandlerUtil.getActiveShell(event),
                        new TunerAllocationWizard(tuners[0]));
                dialog.open();
            }
        } else {
            Object obj = selection.getFirstElement();
            if (obj instanceof UnallocatedTunerContainer) {
                UnallocatedTunerContainer container = (UnallocatedTunerContainer) obj;
                TunerStatus[] tuners = getUnallocatedTunersOfType(container.getTunerContainer(),
                        container.getTunerType());
                if (tuners.length > 0) {
                    WizardDialog dialog = new WizardDialog(HandlerUtil.getActiveShell(event),
                            new TunerAllocationWizard(tuners[0]));
                    dialog.open();
                }
            } else if (obj instanceof TunerContainer) {
                TunerContainer container = (TunerContainer) obj;
                TunerStatus[] tuners = container.getTunerStatus().toArray(new TunerStatus[0]);
                if (tuners.length > 0) {
                    WizardDialog dialog = new WizardDialog(HandlerUtil.getActiveShell(event),
                            new TunerAllocationWizard(tuners[0]));
                    dialog.open();
                } else {
                    ScaDevice<?> device = ScaEcoreUtils.getEContainerOfType(container, ScaDevice.class);
                    if (device != null) {
                        WizardDialog dialog = new WizardDialog(HandlerUtil.getActiveShell(event),
                                new TunerAllocationWizard(FrontendFactory.eINSTANCE.createTunerStatus(), device));
                        warnNoTuners(event);
                        dialog.open();
                    } else {
                        MessageDialog warning = new MessageDialog(HandlerUtil.getActiveShell(event),
                                "Error - No Device Found", null, "The device could not be found.",
                                MessageDialog.ERROR, new String[] { "OK" }, 0);
                        warning.open();
                        return null;
                    }
                }
            } else if (obj instanceof ScaDevice<?>) {
                ScaDevice<?> device = (ScaDevice<?>) obj;
                TunerContainer container = TunerUtils.INSTANCE.getTunerContainer(device);
                if (container == null) {
                    MessageDialog warning = new MessageDialog(HandlerUtil.getActiveShell(event),
                            "Error - No Tuner Container Found", null,
                            "The device's tuner container could not be found.  Make sure the device's \"Data Providers Enabled\" property is set to \"true\".",
                            MessageDialog.ERROR, new String[] { "OK" }, 0);
                    warning.open();
                    return null;
                }
                TunerStatus[] tuners = container.getTunerStatus().toArray(new TunerStatus[0]);
                if (tuners.length > 0) {
                    WizardDialog dialog = new WizardDialog(HandlerUtil.getActiveShell(event),
                            new TunerAllocationWizard(tuners[0]));
                    dialog.open();
                } else {
                    WizardDialog dialog = new WizardDialog(HandlerUtil.getActiveShell(event),
                            new TunerAllocationWizard(FrontendFactory.eINSTANCE.createTunerStatus(), device));
                    warnNoTuners(event);
                    dialog.open();
                }
            }
        }

        return null;
    }

    private void warnNoTuners(ExecutionEvent event) {
        MessageDialog warning = new MessageDialog(HandlerUtil.getActiveShell(event),
                "Warning - No Tuners Available", null,
                "The selected device has no tuners.  Dynamic tuner creation may not be supported.",
                MessageDialog.WARNING, new String[] { "OK" }, 0);
        warning.open();
    }

    private TunerStatus[] getUnallocatedTunersOfType(TunerContainer container, String tunerType) {
        List<TunerStatus> tuners = new ArrayList<TunerStatus>();
        for (TunerStatus tuner : container.getTunerStatus()) {
            if ((tuner.getAllocationID() == null || tuner.getAllocationID().length() == 0)
                    && PluginUtil.equals(tuner.getTunerType(), tunerType)) {
                tuners.add(tuner);
            }
        }
        return tuners.toArray(new TunerStatus[0]);
    }

    @SuppressWarnings("unchecked")
    private <T> T[] castArray(Object[] objects, T[] array) {
        List<T> list = new ArrayList<T>();
        for (Object obj : objects) {
            list.add((T) obj);
        }
        return list.toArray(array);
    }

}