OpenWalnut  1.3.1
WSelectorBranch.cpp
1 //---------------------------------------------------------------------------
2 //
3 // Project: OpenWalnut ( http://www.openwalnut.org )
4 //
5 // Copyright 2009 OpenWalnut Community, BSV@Uni-Leipzig and CNCF@MPI-CBS
6 // For more information see http://www.openwalnut.org/copying
7 //
8 // This file is part of OpenWalnut.
9 //
10 // OpenWalnut is free software: you can redistribute it and/or modify
11 // it under the terms of the GNU Lesser General Public License as published by
12 // the Free Software Foundation, either version 3 of the License, or
13 // (at your option) any later version.
14 //
15 // OpenWalnut is distributed in the hope that it will be useful,
16 // but WITHOUT ANY WARRANTY; without even the implied warranty of
17 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 // GNU Lesser General Public License for more details.
19 //
20 // You should have received a copy of the GNU Lesser General Public License
21 // along with OpenWalnut. If not, see <http://www.gnu.org/licenses/>.
22 //
23 //---------------------------------------------------------------------------
24 
25 #include <list>
26 #include <vector>
27 
28 #include "WSelectorBranch.h"
29 
30 WSelectorBranch::WSelectorBranch( boost::shared_ptr< const WDataSetFibers > fibers, boost::shared_ptr<WRMBranch> branch ) :
31  m_fibers( fibers ),
32  m_size( fibers->size() ),
33  m_dirty( true ),
34  m_branch( branch )
35 {
36  m_bitField = boost::shared_ptr< std::vector<bool> >( new std::vector<bool>( m_size, false ) );
37 
39  boost::shared_ptr< boost::function< void() > >( new boost::function< void() >( boost::bind( &WSelectorBranch::setDirty, this ) ) );
40  m_branch->addChangeNotifier( m_changeSignal );
41 
43  boost::shared_ptr< boost::function< void() > >( new boost::function< void() >( boost::bind( &WSelectorBranch::setDirty, this ) ) );
44 }
45 
47 {
48  m_branch->removeChangeNotifier( m_changeSignal );
49 
50  // We need the following because not all ROIs are removed per slot below
51  for( std::list< boost::shared_ptr< WSelectorRoi > >::iterator roiIter = m_rois.begin(); roiIter != m_rois.end(); ++roiIter )
52  {
53  ( *roiIter )->getRoi()->removeROIChangeNotifier( m_changeRoiSignal );
54  }
55 }
56 
57 void WSelectorBranch::addRoi( boost::shared_ptr< WSelectorRoi > roi )
58 {
59  m_rois.push_back( roi );
60  roi->getRoi()->addROIChangeNotifier( m_changeRoiSignal );
61 }
62 
63 std::list< boost::shared_ptr< WSelectorRoi > > WSelectorBranch::getROIs()
64 {
65  return m_rois;
66 }
67 
69 {
70  m_dirty = true;
71  if( m_branch->getProperties()->getProperty( "Bundle color" )->toPropColor()->changed() )
72  {
73  colorChanged();
74  }
75 }
76 
77 void WSelectorBranch::removeRoi( osg::ref_ptr< WROI > roi )
78 {
79  for( std::list< boost::shared_ptr< WSelectorRoi > >::iterator iter = m_rois.begin(); iter != m_rois.end(); ++iter )
80  {
81  if( ( *iter )->getRoi() == roi )
82  {
83  ( *iter )->getRoi()->removeROIChangeNotifier( m_changeRoiSignal );
84  m_rois.erase( iter );
85  break;
86  }
87  }
88 }
89 
91 {
92  bool atLeastOneActive = false;
93 
94  for( std::list< boost::shared_ptr< WSelectorRoi > >::iterator iter = m_rois.begin(); iter != m_rois.end(); ++iter )
95  {
96  if( ( *iter )->getRoi()->active() )
97  {
98  atLeastOneActive = true;
99  }
100  }
101 
102  if( atLeastOneActive )
103  {
104  m_workerBitfield = boost::shared_ptr< std::vector< bool > >( new std::vector< bool >( m_size, true ) );
105 
106  for( std::list< boost::shared_ptr< WSelectorRoi > >::iterator iter = m_rois.begin(); iter != m_rois.end(); ++iter )
107  {
108  if( ( *iter )->getRoi()->active() )
109  {
110  boost::shared_ptr< std::vector<bool> > bf = ( *iter )->getBitField();
111  bool isnot = ( *iter )->getRoi()->isNot();
112  if( !isnot )
113  {
114  for( size_t i = 0 ; i < m_size ; ++i )
115  {
116  ( *m_workerBitfield )[i] = ( *m_workerBitfield )[i] & ( *bf )[i];
117  }
118  }
119  else
120  {
121  for( size_t i = 0 ; i < m_size ; ++i )
122  {
123  ( *m_workerBitfield )[i] = ( *m_workerBitfield )[i] & !( *bf )[i];
124  }
125  }
126  }
127  }
128 
129  if( m_branch->isNot() )
130  {
131  for( size_t i = 0 ; i < m_size ; ++i )
132  {
133  ( *m_workerBitfield )[i] = !( *m_workerBitfield )[i];
134  }
135  }
136  }
137  else
138  {
139  m_workerBitfield = boost::shared_ptr< std::vector< bool > >( new std::vector< bool >( m_size, false ) );
140  }
141 
143 }
144 
146 {
147  WColor color = m_branch->getProperties()->getProperty( "Bundle color" )->toPropColor()->get( true );
148 
149  boost::shared_ptr<std::vector<float> > array = m_fibers->getColorScheme( "Custom Color" )->getColor();
150 
151  boost::shared_ptr< std::vector< size_t > > startIndexes = m_fibers->getLineStartIndexes();
152  boost::shared_ptr< std::vector< size_t > > pointsPerLine = m_fibers->getLineLengths();
153 
154  for( size_t i = 0; i < m_size; ++i )
155  {
156  if( ( *m_bitField )[i] )
157  {
158  size_t idx = ( *startIndexes )[i] * 3;
159  for( size_t k = 0; k < ( *pointsPerLine )[i]; ++k )
160  {
161  ( *array )[idx++] = color[0];
162  ( *array )[idx++] = color[1];
163  ( *array )[idx++] = color[2];
164  }
165  }
166  }
167 }