/*
 * Copyright 2006, Haiku.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Stephan Aßmus <superstippi@gmx.de>
 */

#include "Selection.h"

#include <stdio.h>

#include <debugger.h>

#include "Selectable.h"

// constructor
Selection::Selection()
        : fSelected(20)
{
}

// destructor
Selection::~Selection()
{
}

// Select
bool
Selection::Select(Selectable* object, bool extend)
{
        AutoNotificationSuspender _(this);

        if (!extend)
                _DeselectAllExcept(object);

        bool success = false;

        if (!object->IsSelected()) {

                #if DEBUG
                if (fSelected.HasItem((void*)object))
                        debugger("Selection::Select() - "
                                         "unselected object in list!");
                #endif

                if (fSelected.AddItem((void*)object)) {
                        object->_SetSelected(true);
                        success = true;

                        Notify();
                } else {
                        fprintf(stderr, "Selection::Select() - out of memory\n");
                }
        } else {

                #if DEBUG
                if (!fSelected.HasItem((void*)object))
                        debugger("Selection::Select() - "
                                         "already selected object not in list!");
                #endif

                success = true;
                // object already in list
        }

        return success;
}

// Deselect
void
Selection::Deselect(Selectable* object)
{
        if (object->IsSelected()) {
                if (!fSelected.RemoveItem((void*)object))
                        debugger("Selection::Deselect() - "
                                         "selected object not within list!");
                object->_SetSelected(false);

                Notify();
        }
}

// DeselectAll
void
Selection::DeselectAll()
{
        _DeselectAllExcept(NULL);
}

// #pragma mark -

// SelectableAt
Selectable*
Selection::SelectableAt(int32 index) const
{
        return (Selectable*)fSelected.ItemAt(index);
}

// SelectableAtFast
Selectable*
Selection::SelectableAtFast(int32 index) const
{
        return (Selectable*)fSelected.ItemAtFast(index);
}

// CountSelected
int32
Selection::CountSelected() const
{
        return fSelected.CountItems();
}

// #pragma mark -

// _DeselectAllExcept
void
Selection::_DeselectAllExcept(Selectable* except)
{
        bool notify = false;
        bool containedExcept = false;

        int32 count = fSelected.CountItems();
        for (int32 i = 0; i < count; i++) {
                Selectable* object = (Selectable*)fSelected.ItemAtFast(i);
                if (object != except) {
                        object->_SetSelected(false);
                        notify = true;
                } else {
                        containedExcept = true;
                }
        }

        fSelected.MakeEmpty();

        // if the "except" object was previously
        // in the selection, add it again after
        // making the selection list empty
        if (containedExcept)
                fSelected.AddItem(except);

        if (notify)
                Notify();
}