#include <common.h>
#include <wxPcbStruct.h>
#include <pcb_netlist.h>
#include <dialog_update_pcb.h>
#include <wx_html_report_panel.h>
#include <board_netlist_updater.h>
#include <tool/tool_manager.h>
#include <tools/common_actions.h>
#include <class_draw_panel_gal.h>
#include <class_board.h>
#include <ratsnest_data.h>

#include <functional>
using namespace std::placeholders;

DIALOG_UPDATE_PCB::DIALOG_UPDATE_PCB( PCB_EDIT_FRAME* aParent, NETLIST *aNetlist ) :
    DIALOG_UPDATE_PCB_BASE ( aParent ),
    m_frame (aParent),
    m_netlist (aNetlist)
{
    m_messagePanel->SetLabel( _("Changes to be applied:") );
    m_messagePanel->SetLazyUpdate( true );
    m_netlist->SortByReference();
    m_btnPerformUpdate->SetFocus();

    m_messagePanel->SetVisibleSeverities( REPORTER::RPT_WARNING | REPORTER::RPT_ERROR | REPORTER::RPT_ACTION );
}


DIALOG_UPDATE_PCB::~DIALOG_UPDATE_PCB()
{
}


void DIALOG_UPDATE_PCB::PerformUpdate( bool aDryRun )
{
    m_messagePanel->Clear();

    REPORTER& reporter = m_messagePanel->Reporter();
    TOOL_MANAGER* toolManager = m_frame->GetToolManager();
    BOARD* board = m_frame->GetBoard();

    if( !aDryRun )
    {

        // Clear selection, just in case a selected item has to be removed
        toolManager->RunAction( COMMON_ACTIONS::selectionClear, true );
    }

    m_netlist->SetDeleteExtraFootprints( true );
    m_netlist->SetFindByTimeStamp( m_matchByTimestamp->GetValue() );
    m_netlist->SetReplaceFootprints( true );

    try
    {
        m_frame->LoadFootprints( *m_netlist, &reporter );
    }
    catch( IO_ERROR &error )
    {
        wxString msg;

        reporter.Report( _( "Failed to load one or more footprints. Please add the missing libraries in PCBNew configuration. "
                            "The PCB will not update completely." ), REPORTER::RPT_ERROR );
        reporter.Report( error.What(), REPORTER::RPT_INFO );
    }

    BOARD_NETLIST_UPDATER updater( m_frame, m_frame->GetBoard() );
    updater.SetReporter ( &reporter );
    updater.SetIsDryRun( aDryRun);
    updater.SetLookupByTimestamp( m_matchByTimestamp->GetValue() );
    updater.SetDeleteUnusedComponents ( true );
    updater.SetReplaceFootprints( true );
    updater.SetDeleteSinglePadNets( false );
    updater.UpdateNetlist( *m_netlist );

    m_messagePanel->Flush();

    if( aDryRun )
        return;

    m_frame->SetCurItem( NULL );
    m_frame->SetMsgPanel( board );

    if( m_frame->IsGalCanvasActive() )
    {
        std::vector<MODULE*> newFootprints = updater.GetAddedComponents();

        // Place the new modules
        m_frame->SpreadFootprints( &newFootprints, false, false, m_frame->GetCrossHairPosition() );

        if( !newFootprints.empty() )
        {
            for( MODULE* footprint : newFootprints )
            {
                toolManager->RunAction( COMMON_ACTIONS::selectItem, true, footprint );
            }

            toolManager->InvokeTool( "pcbnew.InteractiveEdit" );
        }
    }

    m_btnPerformUpdate->Enable( false );
    m_btnPerformUpdate->SetLabel( _( "Update complete" ) );
    m_btnCancel->SetLabel( _( "Close" ) );
    m_btnCancel->SetFocus();
}


void DIALOG_UPDATE_PCB::OnMatchChange( wxCommandEvent& event )
{
    PerformUpdate( true );
}


void DIALOG_UPDATE_PCB::OnCancelClick( wxCommandEvent& event )
{
    EndModal( wxID_CANCEL );
}


void DIALOG_UPDATE_PCB::OnUpdateClick( wxCommandEvent& event )
{
    m_messagePanel->SetLabel( _( "Changes applied to the PCB:" ) );
    PerformUpdate( false );
    m_btnCancel->SetFocus();
}