//---------------------------------------------------------------------------
//
// Project: OpenWalnut ( http://www.openwalnut.org )
//
// Copyright 2009 OpenWalnut Community, BSV-Leipzig and CNCF-CBS
// For more information see http://www.openwalnut.org/copying
//
// This file is part of OpenWalnut.
//
// OpenWalnut is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// OpenWalnut is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with OpenWalnut. If not, see <http://www.gnu.org/licenses/>.
//
//---------------------------------------------------------------------------

#include <string>

#include "core/kernel/WKernel.h"
#include "WMDetTract2GPConvert.h"

WMDetTract2GPConvert::WMDetTract2GPConvert():
    WModule()
{
}

WMDetTract2GPConvert::~WMDetTract2GPConvert()
{
}

boost::shared_ptr< WModule > WMDetTract2GPConvert::factory() const
{
    return boost::shared_ptr< WModule >( new WMDetTract2GPConvert() );
}

const std::string WMDetTract2GPConvert::getName() const
{
    return "Deterministic Tract to Gaussian Process Converter";
}

const std::string WMDetTract2GPConvert::getDescription() const
{
    return "Converts deterministic tracts to Gaussian processes as described in the paper of Wassermann: "
           "http://dx.doi.org/10.1016/j.neuroimage.2010.01.004";
}

void WMDetTract2GPConvert::connectors()
{
    m_tractIC = WModuleInputData< WDataSetFibers >::createAndAdd( shared_from_this(), "tractInput", "The deterministic tracts" );
    m_tensorIC = WModuleInputData< WDataSetDTI >::createAndAdd( shared_from_this(), "tensorInput", "The 2nd order symmetric diffusion tensors" );
    m_gpOC = WModuleOutputData< WDataSetGP >::createAndAdd( shared_from_this(), "gpOutput", "The Gaussian processes" );

    WModule::connectors();
}

void WMDetTract2GPConvert::properties()
{
    WModule::properties();
}

void WMDetTract2GPConvert::moduleMain()
{
    m_moduleState.setResetable( true, true ); // remember actions when actually not waiting for actions
    m_moduleState.add( m_tractIC->getDataChangedCondition() );
    m_moduleState.add( m_tensorIC->getDataChangedCondition() );

    ready();

    while( !m_shutdownFlag() ) // loop until the module container requests the module to quit
    {
        debugLog() << "Waiting..";
        m_moduleState.wait();
        // Demain Wasserman said, using tensors is not realy worth its effort
        // if( !m_tractIC->getData().get() || !m_tensorIC->getData().get() ) // ok, the output has not yet sent data
        if( !m_tractIC->getData().get() ) // ok, the output has not yet sent data
        {
            continue;
        }

        // bool dataUpdated = m_tractIC->handledUpdate() || m_tensorIC->handledUpdate();
        boost::shared_ptr< WDataSetFibers > tracts = m_tractIC->getData();
        boost::shared_ptr< WDataSetDTI > tensors = m_tensorIC->getData();
        // Demain Wasserman said, using tensors is not realy worth its effort
        // bool dataValid = tracts && tensors;
        bool dataValid = static_cast< bool >( tracts );
        if( !dataValid )
        {
            continue;
        }
        boost::shared_ptr< WProgress > progress1 = boost::shared_ptr< WProgress >( new WProgress( "Converting tracts into Gaussian processes.", tracts->size() ) ); // NOLINT line length
        m_progress->addSubProgress( progress1 );
        m_gpOC->updateData( boost::shared_ptr< WDataSetGP >( new WDataSetGP( tracts, tensors, m_shutdownFlag, progress1 ) ) );
        progress1->finish();
        m_progress->removeSubProgress( progress1 );
        debugLog() << "done";
    }
}