WKernel.cpp 9.49 KB
Newer Older
ebaum's avatar
[ADD]  
ebaum committed
1 2
//---------------------------------------------------------------------------
//
3
// Project: OpenWalnut ( http://www.openwalnut.org )
ebaum's avatar
[ADD]  
ebaum committed
4
//
5 6
// Copyright 2009 OpenWalnut Community, BSV@Uni-Leipzig and CNCF@MPI-CBS
// For more information see http://www.openwalnut.org/copying
ebaum's avatar
[ADD]  
ebaum committed
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
//
// 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/>.
//
//---------------------------------------------------------------------------

25 26 27 28
#if defined(__APPLE__)
#include <mach-o/dyld.h>
#endif

ebaum's avatar
[ADD]  
ebaum committed
29
#include <iostream>
ebaum's avatar
ebaum committed
30
#include <list>
31 32
#include <string>
#include <vector>
ebaum's avatar
ebaum committed
33 34

#include <boost/thread/xtime.hpp>
ebaum's avatar
[ADD]  
ebaum committed
35

ebaum's avatar
ebaum committed
36
#include "WModule.h"
37
#include "../modules/data/WMData.hpp"
38 39
#include "../modules/navSlices/WMNavSlices.h"
#include "../modules/coordinateSystem/WMCoordinateSystem.h"
40 41 42
#include "../modules/fiberDisplay/WMFiberDisplay.h"
#include "../modules/fiberCulling/WMFiberCulling.h"
#include "../modules/fiberClustering/WMFiberClustering.h"
ebaum's avatar
[ADD]  
ebaum committed
43 44
#include "../common/WException.h"

ebaum's avatar
ebaum committed
45 46
#include "../graphicsEngine/WGraphicsEngine.h"

47
#include "WKernel.h"
48

49
/**
ebaum's avatar
ebaum committed
50 51 52 53
 * Used for program wide access to the kernel.
 */
WKernel* kernel = NULL;

54
WKernel::WKernel( int argc, char* argv[], boost::shared_ptr< WGUI > gui )
55
    : m_gui( gui )
ebaum's avatar
[ADD]  
ebaum committed
56
{
57
    WLogger::getLogger()->addLogMessage( "Initializing Kernel", "Kernel", LL_DEBUG );
ebaum's avatar
ebaum committed
58 59 60

    kernel = this;

ebaum's avatar
[ADD]  
ebaum committed
61
    // initialize members
ebaum's avatar
ebaum committed
62 63 64
    m_ArgC = argc;
    m_ArgV = argv;
    m_FinishRequested = false;
ebaum's avatar
[ADD]  
ebaum committed
65

ebaum's avatar
ebaum committed
66 67 68
    // init GE, DataHandler, ...
    init();

ebaum's avatar
[ADD]  
ebaum committed
69 70 71 72 73 74 75
    // load modules
    loadModules();
}

WKernel::~WKernel()
{
    // cleanup
76
    WLogger::getLogger()->addLogMessage( "Shutting down Kernel", "Kernel", LL_DEBUG );
77 78

    // finish running thread
79
    WLogger::getLogger()->wait( true );
80
    // write remaining log messages
81
    WLogger::getLogger()->processQueue();
ebaum's avatar
[ADD]  
ebaum committed
82 83 84 85 86 87 88
}

WKernel::WKernel( const WKernel& other )
{
    *this = other;
}

ebaum's avatar
ebaum committed
89 90 91 92 93
WKernel* WKernel::getRunningKernel()
{
    return kernel;
}

94
boost::shared_ptr< WGraphicsEngine > WKernel::getGraphicsEngine() const
ebaum's avatar
ebaum committed
95
{
96
    return m_graphicsEngine;
ebaum's avatar
ebaum committed
97 98
}

99
boost::shared_ptr< WDataHandler > WKernel::getDataHandler() const
100
{
101
    return m_dataHandler;
102 103
}

104
boost::shared_ptr< WGUI > WKernel::getGui()
105
{
106
    return m_gui;
107 108
}

109 110
void WKernel::setGui( boost::shared_ptr< WGUI > gui )
{
111
    m_gui = gui;
112
}
113

114
int WKernel::getArgumentCount() const
ebaum's avatar
ebaum committed
115 116 117 118
{
    return m_ArgC;
}

119
char** WKernel::getArguments() const
ebaum's avatar
ebaum committed
120 121 122 123 124 125
{
    return m_ArgV;
}

int WKernel::run()
{
126
    WLogger::getLogger()->addLogMessage( "Starting Kernel", "Kernel", LL_DEBUG );
ebaum's avatar
ebaum committed
127 128

    // TODO(ebaum): add separate graphics thread here
129
    m_graphicsEngine->run();
ebaum's avatar
ebaum committed
130 131 132

    // run Gui
    // TODO(all): clean up this option handler mess
133
    m_gui->run();
ebaum's avatar
ebaum committed
134 135 136

    // run? data handler stuff?

137 138 139 140
    // **************************************************************************************************************************
    // This part will be exchanged by some kind of ModuleContainer managing module execution.
    // TODO(ebaum): replace by ModuleContainer
    // **************************************************************************************************************************
141

ebaum's avatar
ebaum committed
142 143 144
    // run module execution threads
    // TODO(ebaum): after having modules loaded they should be started here.
    // currently this is just the test module
145
    WLogger::getLogger()->addLogMessage( "*** Starting modules:", "Kernel", LL_DEBUG );
146
    for( std::list< boost::shared_ptr< WModule > >::iterator list_iter = m_modules.begin(); list_iter != m_modules.end();
wiebel's avatar
[STYLE]  
wiebel committed
147
            ++list_iter )
ebaum's avatar
ebaum committed
148
    {
149
        WLogger::getLogger()->addLogMessage( "Starting module: " + ( *list_iter )->getName(), "Kernel", LL_DEBUG );
ebaum's avatar
ebaum committed
150 151 152
        ( *list_iter )->run();
    }

153
    // TODO(schurade): this must be moved somewhere else, and realize the wait loop in another fashion
154
    while ( !m_gui->isInitalized() )
155 156
    {
    }
157
    m_gui->getLoadButtonSignal()->connect( boost::bind( &WKernel::doLoadDataSets, this, _1 ) );
Sebastian Eichelbaum's avatar
[MERGE]  
Sebastian Eichelbaum committed
158

159
    for( std::list< boost::shared_ptr< WModule > >::iterator list_iter = m_modules.begin(); list_iter != m_modules.end();
160 161 162 163 164
                ++list_iter )
    {
        ( *list_iter )->connectToGui();
    }

ebaum's avatar
ebaum committed
165
    // wait
166
    // TODO(ebaum): this is not the optimal. It would be better to quit OSG, GE and so on in the right order.
167
    m_gui->wait( false );
ebaum's avatar
ebaum committed
168 169 170
    m_FinishRequested = true;

    // wait for modules to finish
171
    for( std::list< boost::shared_ptr< WModule > >::iterator list_iter = m_modules.begin(); list_iter != m_modules.end();
wiebel's avatar
[STYLE]  
wiebel committed
172
            ++list_iter )
ebaum's avatar
ebaum committed
173 174 175 176 177
    {
        ( *list_iter )->wait( true );
    }

    // finally GE
178
    m_graphicsEngine->wait( true );
ebaum's avatar
ebaum committed
179 180 181 182 183

    // how to get QT return code from its thread?
    return 0;
}

ebaum's avatar
[ADD]  
ebaum committed
184 185
void WKernel::loadModules()
{
ebaum's avatar
ebaum committed
186
    // TODO(ebaum): add dynamic loading here
187
    WLogger::getLogger()->addLogMessage( "*** Loading Modules: ", "Kernel", LL_DEBUG );
ebaum's avatar
ebaum committed
188 189
    m_modules.clear();

190
    using boost::shared_ptr;
191 192 193 194
    shared_ptr< WModule > m1 = shared_ptr< WModule >( new WMNavSlices() );
    // shared_ptr< WModule > m = shared_ptr< WModule >( new WMFiberDisplay() );
    // shared_ptr< WModule > m = shared_ptr< WModule >( new WMFiberCulling() );
    // shared_ptr< WModule > m = shared_ptr< WModule >( new WMFiberClustering() );
195
    WLogger::getLogger()->addLogMessage( "Loading module: " + m1->getName(), "Kernel", LL_DEBUG );
ebaum's avatar
ebaum committed
196

197 198
    m_modules.push_back( m1 );

199
    shared_ptr< WModule > m2 = shared_ptr< WModule >( new WMCoordinateSystem() );
200 201
    WLogger::getLogger()->addLogMessage( "Loading module: " + m2->getName(), "Kernel", LL_DEBUG );
    m_modules.push_back( m2 );
ebaum's avatar
ebaum committed
202 203 204 205
}

void WKernel::init()
{
206
     // initialize
207
    findAppPath();
ebaum's avatar
ebaum committed
208 209 210

    // initialize graphics engine
    // this also includes initialization of WGEScene and OpenSceneGraph
211
    m_graphicsEngine = boost::shared_ptr< WGraphicsEngine >( new WGraphicsEngine( m_shaderPath ) );
212 213

    // initialize Datahandler
214
    m_dataHandler = boost::shared_ptr< WDataHandler >( new WDataHandler() );
215

216 217
    // m_gui->createMainWindow();

218
    m_dataHandler->getSignalAddDataset()->connect( boost::bind( &WKernel::slotFinishLoadData, this, _1 ) );
219 220 221 222 223 224
}

bool WKernel::findAppPath()
{
    // FIXME (schurade)
    // this should work on linux, have to implement it for windows and mac later
Alexander Wiebel's avatar
Alexander Wiebel committed
225
#ifdef __linux__
226 227 228 229 230 231 232 233
    int length;
    char appPath[255];

    length = readlink( "/proc/self/exe", appPath, sizeof( appPath ) );

    // Catch some errors
    if ( length < 0 )
    {
234
        WLogger::getLogger()->addLogMessage( "Error resolving symlink /proc/self/exe.", "Kernel", LL_ERROR );
235 236 237 238
        return false;
    }
    if ( length >= 255 )
    {
239
        WLogger::getLogger()->addLogMessage( "Path too long. Truncated.", "Kernel", LL_ERROR );
240 241 242 243 244 245 246 247 248 249 250
        return false;
    }

    // the string this readlink() function returns is appended with a '@'.
    appPath[length] = '\0';

    // strip off the executable name
    while ( appPath[length] != '/' )
    {
        appPath[length] = '\0';
        --length;
schurade's avatar
schurade committed
251
        assert( length >= 0 );
252 253 254 255 256
    }

    m_AppPath = appPath;

    std::string shaderPath( appPath );
257
    m_shaderPath = shaderPath + "shaders/";
258 259

    return true;
schurade's avatar
schurade committed
260
#elif defined( __APPLE__ )
261
    char path[1024];
schurade's avatar
schurade committed
262
    uint32_t size = sizeof( path );
263
    if( _NSGetExecutablePath( path, &size ) == 0 )
264
    {
265 266 267
        WLogger::getLogger()->addLogMessage( "Executable path is " + std::string( path ), "Kernel", LL_ERROR );

        int i = strlen( path );
schurade's avatar
schurade committed
268 269 270 271 272 273
        while( path[i] != '/' )
        {
            path[i] = '\0';
            i--;
            assert( i >= 0 );
        }
274
        WLogger::getLogger()->addLogMessage( "Application path is " + std::string( path ), "Kernel", LL_ERROR );
schurade's avatar
schurade committed
275 276 277 278 279 280
        m_AppPath = path;

        std::string shaderPath( path );
        m_shaderPath = shaderPath + "shaders/";

        return true;
281 282 283
    }
    else
    {
284
        WLogger::getLogger()->addLogMessage( "Buffer too small; need size " + boost::lexical_cast< std::string >( size ),
285
                                             "Kernel", LL_ERROR );
schurade's avatar
schurade committed
286
        assert( size <= sizeof( path ) );
287 288 289 290 291 292
    }
#else
#error findAppPath not implemented for this platform
#endif

    return false;
ebaum's avatar
ebaum committed
293 294
}

295
bool WKernel::isFinishRequested() const
ebaum's avatar
ebaum committed
296 297
{
    return m_FinishRequested;
ebaum's avatar
[ADD]  
ebaum committed
298 299
}

300 301
void WKernel::doLoadDataSets( std::vector< std::string > fileNames )
{
302
    m_dataHandler->loadDataSets( fileNames );
303
}
304

305 306
void WKernel::slotFinishLoadData( boost::shared_ptr< WDataSet > dataSet )
{
307
    boost::shared_ptr< WModule > module = boost::shared_ptr< WModule >( new WMData< int >( dataSet ) );
308

309
    module->getProperties()->addBool( "active", true );
310
    module->getProperties()->hideProperty( "active" );
311
    module->getProperties()->addBool( "interpolation", true );
312 313
    module->getProperties()->addInt( "threshold", 0 );
    module->getProperties()->addInt( "alpha", 100 );
314
    module->getProperties()->setMax( "alpha", 100 );
315
    module->getProperties()->setValue( "name", dataSet->getFileName() );
316

317
    m_gui->addDatasetToBrowser( module, 0 );
318 319
}

320
boost::shared_ptr< WDataHandler > WKernel::getDataHandler()
321
{
322
    return m_dataHandler;
323 324
}

325 326 327 328 329 330 331
std::string WKernel::getAppPath()
{
    return m_AppPath;
}

std::string WKernel::getShaderPath()
{
332
    return m_shaderPath;
333
}