#include "RicSwapGridCrossPlotDataSetAxesFeature.h"

#include "RimGridCrossPlot.h"
#include "RimGridCrossPlotDataSet.h"

#include <cafSelectionManager.h>

#include <QAction>

CAF_CMD_SOURCE_INIT( RicSwapGridCrossPlotDataSetAxesFeature, "RicSwapGridCrossPlotDataSetAxesFeature" );

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RicSwapGridCrossPlotDataSetAxesFeature::isCommandEnabled()
{
    if ( caf::SelectionManager::instance()->selectedItemOfType<RimGridCrossPlotDataSet>() )
    {
        return true;
    }
    else if ( caf::SelectionManager::instance()->selectedItemOfType<RimGridCrossPlot>() )
    {
        auto plot = caf::SelectionManager::instance()->selectedItemOfType<RimGridCrossPlot>();
        if ( !plot->dataSets().empty() ) return true;
    }
    return false;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RicSwapGridCrossPlotDataSetAxesFeature::onActionTriggered( bool isChecked )
{
    if ( caf::SelectionManager::instance()->selectedItemOfType<RimGridCrossPlotDataSet>() )
    {
        auto dataSet = caf::SelectionManager::instance()->selectedItemOfType<RimGridCrossPlotDataSet>();
        dataSet->swapAxisProperties( true );
    }
    else if ( caf::SelectionManager::instance()->selectedItemOfType<RimGridCrossPlot>() )
    {
        auto plot = caf::SelectionManager::instance()->selectedItemOfType<RimGridCrossPlot>();
        plot->swapAxes();
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RicSwapGridCrossPlotDataSetAxesFeature::setupActionLook( QAction* actionToSetup )
{
    if ( caf::SelectionManager::instance()->selectedItemOfType<RimGridCrossPlotDataSet>() )
    {
        actionToSetup->setText( "Swap Axis Properties" );
    }
    else
    {
        actionToSetup->setText( "Swap Axis Properties for all Data Sets in Plot" );
    }
    actionToSetup->setIcon( QIcon( ":/Swap.png" ) );
}