OpenWalnut  1.4.0
WSelectorBranch.cpp
00001 //---------------------------------------------------------------------------
00002 //
00003 // Project: OpenWalnut ( http://www.openwalnut.org )
00004 //
00005 // Copyright 2009 OpenWalnut Community, BSV@Uni-Leipzig and CNCF@MPI-CBS
00006 // For more information see http://www.openwalnut.org/copying
00007 //
00008 // This file is part of OpenWalnut.
00009 //
00010 // OpenWalnut is free software: you can redistribute it and/or modify
00011 // it under the terms of the GNU Lesser General Public License as published by
00012 // the Free Software Foundation, either version 3 of the License, or
00013 // (at your option) any later version.
00014 //
00015 // OpenWalnut is distributed in the hope that it will be useful,
00016 // but WITHOUT ANY WARRANTY; without even the implied warranty of
00017 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
00018 // GNU Lesser General Public License for more details.
00019 //
00020 // You should have received a copy of the GNU Lesser General Public License
00021 // along with OpenWalnut. If not, see <http://www.gnu.org/licenses/>.
00022 //
00023 //---------------------------------------------------------------------------
00024 
00025 #include <list>
00026 #include <vector>
00027 
00028 #include "WSelectorBranch.h"
00029 
00030 WSelectorBranch::WSelectorBranch( boost::shared_ptr< const WDataSetFibers > fibers, boost::shared_ptr<WRMBranch> branch ) :
00031     m_fibers( fibers ),
00032     m_size( fibers->size() ),
00033     m_dirty( true ),
00034     m_branch( branch )
00035 {
00036     m_bitField = boost::shared_ptr< std::vector<bool> >( new std::vector<bool>( m_size, false ) );
00037 
00038     m_changeSignal =
00039         boost::shared_ptr< boost::function< void() > >( new boost::function< void() >( boost::bind( &WSelectorBranch::setDirty, this ) ) );
00040     m_branch->addChangeNotifier( m_changeSignal );
00041 
00042     m_changeRoiSignal =
00043         boost::shared_ptr< boost::function< void() > >( new boost::function< void() >( boost::bind( &WSelectorBranch::setDirty, this ) ) );
00044 }
00045 
00046 WSelectorBranch::~WSelectorBranch()
00047 {
00048     m_branch->removeChangeNotifier( m_changeSignal );
00049 
00050     // We need the following because not all ROIs are removed per slot below
00051     for( std::list< boost::shared_ptr< WSelectorRoi > >::iterator roiIter = m_rois.begin(); roiIter != m_rois.end(); ++roiIter )
00052     {
00053         ( *roiIter )->getRoi()->removeROIChangeNotifier( m_changeRoiSignal );
00054     }
00055 }
00056 
00057 void WSelectorBranch::addRoi( boost::shared_ptr< WSelectorRoi > roi )
00058 {
00059     m_rois.push_back( roi );
00060     roi->getRoi()->addROIChangeNotifier( m_changeRoiSignal );
00061 }
00062 
00063 std::list< boost::shared_ptr< WSelectorRoi > > WSelectorBranch::getROIs()
00064 {
00065     return m_rois;
00066 }
00067 
00068 void WSelectorBranch::setDirty()
00069 {
00070     m_dirty = true;
00071 }
00072 
00073 void WSelectorBranch::removeRoi( osg::ref_ptr< WROI > roi )
00074 {
00075     for( std::list< boost::shared_ptr< WSelectorRoi > >::iterator iter = m_rois.begin(); iter != m_rois.end(); ++iter )
00076     {
00077         if( ( *iter )->getRoi() == roi )
00078         {
00079             ( *iter )->getRoi()->removeROIChangeNotifier( m_changeRoiSignal );
00080             m_rois.erase( iter );
00081             break;
00082         }
00083     }
00084 }
00085 
00086 void WSelectorBranch::recalculate()
00087 {
00088     bool atLeastOneActive = false;
00089 
00090     for( std::list< boost::shared_ptr< WSelectorRoi > >::iterator iter = m_rois.begin(); iter != m_rois.end(); ++iter )
00091     {
00092         if( ( *iter )->getRoi()->active() )
00093         {
00094             atLeastOneActive = true;
00095         }
00096     }
00097 
00098     if( atLeastOneActive )
00099     {
00100         m_workerBitfield = boost::shared_ptr< std::vector< bool > >( new std::vector< bool >( m_size, true ) );
00101 
00102         for( std::list< boost::shared_ptr< WSelectorRoi > >::iterator iter = m_rois.begin(); iter != m_rois.end(); ++iter )
00103         {
00104             if( ( *iter )->getRoi()->active() )
00105             {
00106                 boost::shared_ptr< std::vector<bool> > bf = ( *iter )->getBitField();
00107                 bool isnot = ( *iter )->getRoi()->isNot();
00108                 if( !isnot )
00109                 {
00110                     for( size_t i = 0 ; i < m_size ; ++i )
00111                     {
00112                         ( *m_workerBitfield )[i] = ( *m_workerBitfield )[i] & ( *bf )[i];
00113                     }
00114                 }
00115                 else
00116                 {
00117                     for( size_t i = 0 ; i < m_size ; ++i )
00118                     {
00119                         ( *m_workerBitfield )[i] = ( *m_workerBitfield )[i] & !( *bf )[i];
00120                     }
00121                 }
00122             }
00123         }
00124 
00125         if( m_branch->isNot() )
00126         {
00127            for( size_t i = 0 ; i < m_size ; ++i )
00128             {
00129                 ( *m_workerBitfield )[i] = !( *m_workerBitfield )[i];
00130             }
00131         }
00132     }
00133     else
00134     {
00135         m_workerBitfield = boost::shared_ptr< std::vector< bool > >( new std::vector< bool >( m_size, false ) );
00136     }
00137 
00138     m_bitField = m_workerBitfield;
00139 }
00140 
00141 WColor WSelectorBranch::getBranchColor() const
00142 {
00143     return m_branch->colorProperty()->toPropColor()->get( true );
00144 }