/****************************************************************************
 ** Copyright (C) 2001-2006 Klarälvdalens Datakonsult AB.  All rights reserved.
 **
 ** This file is part of the KD Gantt library.
 **
 ** This file may be distributed and/or modified under the terms of the
 ** GNU General Public License version 2 as published by the Free Software
 ** Foundation and appearing in the file LICENSE.GPL included in the
 ** packaging of this file.
 **
 ** Licensees holding valid commercial KD Gantt licenses may use this file in
 ** accordance with the KD Gantt Commercial License Agreement provided with
 ** the Software.
 **
 ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 **
 ** See http://www.kdab.net/kdgantt for
 **   information about KD Gantt Commercial License Agreements.
 **
 ** Contact info@kdab.net if any conditions of this
 ** licensing are not clear to you.
 **
 **********************************************************************/
#include "kdganttconstraintproxy.h"
#include "kdganttconstraintmodel.h"

#include <QAbstractProxyModel>

using namespace KDGantt;

/*!\class KDGantt::ConstraintProxy
 * \internal
 */

ConstraintProxy::ConstraintProxy( QObject* parent )
    : QObject( parent )
{
}

ConstraintProxy::~ConstraintProxy()
{
}

void ConstraintProxy::setSourceModel( ConstraintModel* src )
{
    if ( m_source ) disconnect( m_source );
    m_source = src;

    copyFromSource();

    connect( m_source, SIGNAL(constraintAdded(Constraint)),
             this, SLOT(slotSourceConstraintAdded(Constraint)) );
    connect( m_source, SIGNAL(constraintRemoved(Constraint)),
             this, SLOT(slotSourceConstraintRemoved(Constraint)) );
}

void ConstraintProxy::setDestinationModel( ConstraintModel* dest )
{
    if ( m_destination ) disconnect( m_destination );
    m_destination = dest;

    copyFromSource();

    connect( m_destination, SIGNAL(constraintAdded(Constraint)),
             this, SLOT(slotDestinationConstraintAdded(Constraint)) );
    connect( m_destination, SIGNAL(constraintRemoved(Constraint)),
             this, SLOT(slotDestinationConstraintRemoved(Constraint)) );
}

void ConstraintProxy::setProxyModel( QAbstractProxyModel* proxy )
{
    m_proxy = proxy;
}

ConstraintModel* ConstraintProxy::sourceModel() const { return m_source; }
ConstraintModel* ConstraintProxy::destinationModel() const { return m_destination; }
QAbstractProxyModel* ConstraintProxy::proxyModel() const { return m_proxy; }


void ConstraintProxy::copyFromSource()
{
    if ( m_destination ) {
        m_destination->clear();
        if ( !m_source ) return;
        QList<Constraint> lst = m_source->constraints();
        Q_FOREACH( const Constraint& c, lst ) {
            m_destination->addConstraint( Constraint( m_proxy->mapFromSource( c.startIndex() ),
                                                      m_proxy->mapFromSource( c.endIndex() ) ) );
        }
    }
}

void ConstraintProxy::slotSourceConstraintAdded( const Constraint& c )
{
    if ( m_destination ) m_destination->addConstraint( Constraint( m_proxy->mapFromSource( c.startIndex() ),
                                                                   m_proxy->mapFromSource( c.endIndex() ) ) );
}

void ConstraintProxy::slotSourceConstraintRemoved( const Constraint& c )
{
    if ( m_destination ) m_destination->removeConstraint( Constraint( m_proxy->mapFromSource( c.startIndex() ),
                                                                      m_proxy->mapFromSource( c.endIndex() ) ) );
}

void ConstraintProxy::slotDestinationConstraintAdded( const Constraint& c )
{
    if ( m_source ) m_source->addConstraint( Constraint( m_proxy->mapToSource( c.startIndex() ),
                                                         m_proxy->mapToSource( c.endIndex() ) ) );
}

void ConstraintProxy::slotDestinationConstraintRemoved( const Constraint& c )
{
    if ( m_source ) m_source->removeConstraint( Constraint( m_proxy->mapToSource( c.startIndex() ),
                                                            m_proxy->mapToSource( c.endIndex() ) ) );
}