#include "customGrid.h"
#include "globalFunctions.h"
#include "resources.h"
#include <wx/dc.h>
#include "../algorithm.h"

const unsigned int MinimumRows = 15;

//class containing pure grid data: basically the same as wxGridStringTable, but adds cell formatting
class CustomGridTableBase : public wxGridStringTable
{
public:
    CustomGridTableBase(int numRows, int numCols) :
            wxGridStringTable(numRows, numCols),
            lightBlue(80, 110, 255),
            lightGrey(212, 208, 200),
            gridIdentifier(0),
            gridRefUI(0),
            gridData(0),
            lastNrRows(MinimumRows) {}


    ~CustomGridTableBase() {}


    void setGridDataTable(GridView* gridRefUI, FileCompareResult* gridData)
    {
        this->gridRefUI = gridRefUI;
        this->gridData  = gridData;
    }


    void SetGridIdentifier(int id)
    {
        gridIdentifier = id;
    }

//###########################################################################
//grid standard input output methods, redirected directly to gridData to improve performance

    virtual int GetNumberRows()
    {
        if (gridRefUI)
            return max(gridRefUI->size(), MinimumRows);
        else
            return MinimumRows; //grid is initialized with this number of rows
    }


    virtual bool IsEmptyCell( int row, int col )
    {
        return (GetValue(row, col) == wxEmptyString);
    }


    wxString evaluateCmpResult(const CompareFilesResult result, const bool selectedForSynchronization)
    {
        if (selectedForSynchronization)
            switch (result)
            {
            case FILE_LEFT_SIDE_ONLY:
                return wxT("<|");
                break;
            case FILE_RIGHT_SIDE_ONLY:
                return wxT("|>");
                break;
            case FILE_RIGHT_NEWER:
                return wxT(">>");
                break;
            case FILE_LEFT_NEWER:
                return wxT("<<");
                break;
            case FILE_DIFFERENT:
                return wxT("!=");
                break;
            case FILE_EQUAL:
                return wxT("==");
                break;
            default:
                assert (false);
                return wxEmptyString;
            }
        else return wxT("(-)");
    }


    virtual wxString GetValue(int row, int col)
    {
        if (gridRefUI)
        {
            if (unsigned(row) < gridRefUI->size())
            {
                const FileCompareLine& gridLine = (*gridData)[(*gridRefUI)[row]];
                wxString fileSize; //tmp string

                switch (gridIdentifier)
                {
                case 1:
                    if (col < 4)
                    {
                        if (gridLine.fileDescrLeft.objType == FileDescrLine::TYPE_DIRECTORY)
                        {
                            switch (col)
                            {
                            case 0: //filename
                                return wxEmptyString;
                            case 1: //relative path
                                return gridLine.fileDescrLeft.relativeName;
                            case 2: //file size
                                return _("<Directory>");
                            case 3: //date
                                return wxEmptyString;
                            }
                        }
                        else if (gridLine.fileDescrLeft.objType == FileDescrLine::TYPE_FILE)
                        {
                            switch (col)
                            {
                            case 0: //filename
                                return gridLine.fileDescrLeft.relativeName.AfterLast(GlobalResources::FILE_NAME_SEPARATOR);
                            case 1: //relative path
                                return gridLine.fileDescrLeft.relativeName.BeforeLast(GlobalResources::FILE_NAME_SEPARATOR);
                            case 2: //file size
                                return globalFunctions::includeNumberSeparator(fileSize = gridLine.fileDescrLeft.fileSize.ToString());
                            case 3: //date
                                return FreeFileSync::utcTimeToLocalString(gridLine.fileDescrLeft.lastWriteTimeRaw);
                            }
                        }
                    }
                    else
                        assert (false);

                    break;

                case 2:
                    if (col < 4)
                    {
                        if (gridLine.fileDescrRight.objType == FileDescrLine::TYPE_DIRECTORY)
                        {
                            switch (col)
                            {
                            case 0: //filename
                                return wxEmptyString;
                            case 1: //relative path
                                return gridLine.fileDescrRight.relativeName;
                            case 2: //file size
                                return _("<Directory>");
                            case 3: //date
                                return wxEmptyString;
                            }
                        }
                        else if (gridLine.fileDescrRight.objType == FileDescrLine::TYPE_FILE)
                        {
                            switch (col)
                            {
                            case 0: //filename
                                return gridLine.fileDescrRight.relativeName.AfterLast(GlobalResources::FILE_NAME_SEPARATOR);
                            case 1: //relative path
                                return gridLine.fileDescrRight.relativeName.BeforeLast(GlobalResources::FILE_NAME_SEPARATOR);
                            case 2: //file size
                                return globalFunctions::includeNumberSeparator(fileSize = gridLine.fileDescrRight.fileSize.ToString());
                            case 3: //date
                                return FreeFileSync::utcTimeToLocalString(gridLine.fileDescrRight.lastWriteTimeRaw);
                            }
                        }
                    }
                    else
                        assert (false);

                    break;

                case 3:
                    if (col < 1)
                        return evaluateCmpResult(gridLine.cmpResult, gridLine.selectedForSynchronization);
                    else
                        assert (false);
                    break;

                default:
                    assert (false);
                    break;
                }
            }
        }
        //if data is not found:
        return wxEmptyString;
    }


    virtual void SetValue( int row, int col, const wxString& value )
    {
        assert (false); //should not be used, since values are retrieved directly from gridRefUI
    }
    virtual void Clear()
    {
        assert (false); // we don't want to use this, since the visible grid is directly connected to gridRefUI}
    }
    virtual bool InsertRows( size_t pos = 0, size_t numRows = 1 )
    {
        assert (false); // we don't want to use this, since the visible grid is directly connected to gridRefUI}
        return true;
    }
    virtual bool AppendRows( size_t numRows = 1 )
    {
        assert (false); // we don't want to use this, since the visible grid is directly connected to gridRefUI}
        return true;
    }
    virtual bool DeleteRows( size_t pos = 0, size_t numRows = 1 )
    {
        assert (false); // we don't want to use this, since the visible grid is directly connected to gridRefUI}
        return true;
    }

    //update dimensions of grid: no need for InsertRows, AppendRows, DeleteRows anymore!!!
    void updateGridSizes()
    {
        if (gridRefUI)
        {
            int currentNrRows = GetNumberRows();

            if (lastNrRows < currentNrRows)
            {
                if (GetView())
                {
                    wxGridTableMessage msg(this,
                                           wxGRIDTABLE_NOTIFY_ROWS_APPENDED,
                                           currentNrRows - lastNrRows);

                    GetView()->ProcessTableMessage( msg );
                }
            }
            else if (lastNrRows > currentNrRows)
            {
                if (GetView())
                {
                    wxGridTableMessage msg(this,
                                           wxGRIDTABLE_NOTIFY_ROWS_DELETED,
                                           0,
                                           lastNrRows - currentNrRows);

                    GetView()->ProcessTableMessage( msg );
                }
            }
            lastNrRows = currentNrRows;
        }
    }
//###########################################################################

    enum RowColor {BLUE, GREY, NONE};

    inline  //redundant command
    RowColor getRowColor(int row) //rows that are filtered out are shown in different color
    {
        if (gridRefUI)
        {
            if (unsigned(row) < gridRefUI->size())
            {
                const FileCompareLine cmpLine = (*gridData)[(*gridRefUI)[row]];

                //mark filtered rows
                if (!cmpLine.selectedForSynchronization)
                    return BLUE;
                //mark directories
                else if (gridIdentifier == 1 && cmpLine.fileDescrLeft.objType == FileDescrLine::TYPE_DIRECTORY)
                    return GREY;
                else if (gridIdentifier == 2 && cmpLine.fileDescrRight.objType == FileDescrLine::TYPE_DIRECTORY)
                    return GREY;
                else
                    return NONE;
            }
        }
        return NONE;
    }


    wxGridCellAttr* GetAttr(int row, int col, wxGridCellAttr::wxAttrKind  kind)
    {
        wxGridCellAttr* result = wxGridTableBase::GetAttr(row, col, kind);

        //  MARK FILTERED ROWS:
        if (result)
        {   //if kind is not a cell or row attribute, we have to clone the cell attribute, since we don't want to change e.g. all column attribs
            if (result->GetKind() != wxGridCellAttr::Cell && result->GetKind() != wxGridCellAttr::Row)
            {
                wxGridCellAttr* attr = result->Clone();
                result->DecRef();
                result = attr;
            }

            RowColor color = getRowColor(row);
            if (color == BLUE)
                result->SetBackgroundColour(lightBlue);
            else if (color == GREY)
                result->SetBackgroundColour(lightGrey);
            else
                result->SetBackgroundColour(*wxWHITE);
        }
        else
        {
            result = new wxGridCellAttr;

            RowColor color = getRowColor(row);
            if (color == BLUE)
                result->SetBackgroundColour(lightBlue);
            else if (color == GREY)
                result->SetBackgroundColour(lightGrey);
        }

        return result;
    }

private:
    wxColour  lightBlue;
    wxColour  lightGrey;
    int       gridIdentifier;
    GridView* gridRefUI; //(very fast) access to underlying grid data :)
    FileCompareResult* gridData;
    int       lastNrRows;
};



//########################################################################################################

CustomGrid::CustomGrid(wxWindow *parent,
                       wxWindowID id,
                       const wxPoint& pos,
                       const wxSize& size,
                       long style,
                       const wxString& name) :
        wxGrid(parent, id, pos, size, style, name),
        scrollbarsEnabled(true),
        m_gridLeft(NULL), m_gridRight(NULL), m_gridMiddle(NULL),
        gridDataTable(NULL),
        currentSortColumn(-1),
        sortMarker(NULL) {}


CustomGrid::~CustomGrid() {}


bool CustomGrid::CreateGrid(int numRows, int numCols, wxGrid::wxGridSelectionModes selmode)
{
    //use custom wxGridTableBase class for management of large sets of formatted data.
    //This is done in CreateGrid instead of SetTable method since source code is generated and wxFormbuilder invokes CreatedGrid by default.

    gridDataTable = new CustomGridTableBase(numRows, numCols);
    SetTable(gridDataTable, true, selmode);  //give ownership to wxGrid: gridDataTable is deleted automatically in wxGrid destructor
    return true;
}


void CustomGrid::deactivateScrollbars()
{
    scrollbarsEnabled = false;
}


//overwrite virtual method to finally get rid of the scrollbars
void CustomGrid::SetScrollbar(int orientation, int position, int thumbSize, int range, bool refresh)
{
    if (scrollbarsEnabled)
        wxWindow::SetScrollbar(orientation, position, thumbSize, range, refresh);
    else
        wxWindow::SetScrollbar(orientation, 0, 0, 0, refresh);
}


//workaround: ensure that all grids are properly aligned: add some extra window space to grids that have no horizontal scrollbar
void CustomGrid::adjustGridHeights() //m_gridLeft, m_gridRight, m_gridMiddle are not NULL in this context
{
    int y1 = 0;
    int y2 = 0;
    int y3 = 0;
    int dummy = 0;

    m_gridLeft->GetViewStart(&dummy, &y1);
    m_gridRight->GetViewStart(&dummy, &y2);
    m_gridMiddle->GetViewStart(&dummy, &y3);

    if (y1 != y2 || y2 != y3)
    {
        int yMax = max(y1, max(y2, y3));

        if (::leadGrid == m_gridLeft)  //do not handle case (y1 == yMax) here!!! Avoid back coupling!
            m_gridLeft->SetMargins(0, 0);
        else if (y1 < yMax)
            m_gridLeft->SetMargins(0, 50);

        if (::leadGrid == m_gridRight)
            m_gridRight->SetMargins(0, 0);
        else if (y2 < yMax)
            m_gridRight->SetMargins(0, 50);

        if (::leadGrid == m_gridMiddle)
            m_gridMiddle->SetMargins(0, 0);
        else if (y3 < yMax)
            m_gridMiddle->SetMargins(0, 50);

        m_gridLeft->ForceRefresh();
        m_gridRight->ForceRefresh();
        m_gridMiddle->ForceRefresh();
    }
}


//this method is called when grid view changes: useful for parallel updating of multiple grids
void CustomGrid::DoPrepareDC(wxDC& dc)
{
    wxScrollHelper::DoPrepareDC(dc);

    int x, y = 0;
    if (this == ::leadGrid)  //avoid back coupling
    {
        if (this == m_gridLeft)
        {
            GetViewStart(&x, &y);
            m_gridRight->Scroll(x, y);
            m_gridMiddle->Scroll(-1, y); //scroll in y-direction only
            adjustGridHeights(); //keep here to ensure m_gridLeft, m_gridRight, m_gridMiddle != NULL
        }
        else if (this == m_gridRight)
        {
            GetViewStart(&x, &y);
            m_gridLeft->Scroll(x, y);
            m_gridMiddle->Scroll(-1, y);
            adjustGridHeights(); //keep here to ensure m_gridLeft, m_gridRight, m_gridMiddle != NULL
        }
        else if (this == m_gridMiddle)
        {
            GetViewStart(&x, &y);
            m_gridLeft->Scroll(-1, y);
            m_gridRight->Scroll(-1, y);
            adjustGridHeights(); //keep here to ensure m_gridLeft, m_gridRight, m_gridMiddle != NULL
        }
    }
}


//these classes will scroll together, hence the name ;)
void CustomGrid::setScrollFriends(CustomGrid* gridLeft, CustomGrid* gridRight, CustomGrid* gridMiddle)
{
    assert(gridLeft && gridRight && gridMiddle);

    m_gridLeft   = gridLeft;
    m_gridRight  = gridRight;
    m_gridMiddle = gridMiddle;

    assert(gridDataTable);
    if (this == m_gridLeft)
        gridDataTable->SetGridIdentifier(1);
    else if (this == m_gridRight)
        gridDataTable->SetGridIdentifier(2);
    else if (this == m_gridMiddle)
        gridDataTable->SetGridIdentifier(3);
    else
        assert (false);
}


void CustomGrid::setGridDataTable(GridView* gridRefUI, FileCompareResult* gridData)
{   //set underlying grid data
    assert(gridDataTable);
    gridDataTable->setGridDataTable(gridRefUI, gridData);
}


void CustomGrid::updateGridSizes()
{
    assert(gridDataTable);
    gridDataTable->updateGridSizes();
}


void CustomGrid::setSortMarker(const int sortColumn, const wxBitmap* bitmap)
{
    currentSortColumn = sortColumn;
    sortMarker        = bitmap;
}


void CustomGrid::DrawColLabel(wxDC& dc, int col)
{
    assert(0 <= col && col < 4);

    wxGrid::DrawColLabel(dc, col);

    if (col == currentSortColumn)
    {
        dc.DrawBitmap(*sortMarker, GetColRight(col) - 16 - 2, 2, true); //respect 2-pixel border
    }
}