int main (int argc, char * argv[])
{
std::string session( "default" );
std::string hostname;
#ifdef GPUSD_USE_BOOST
arg::variables_map vm;
arg::options_description desc("GPU service discovery daemon");
desc.add_options()
("help", "output this help message")
("session,s", arg::value< std::string >(), "set session name")
("hostname,h", arg::value< std::string >(), "set hostname")
("daemon,d", "run as daemon")
("width", arg::value< int >(), "maximum width to report")
("height", arg::value< int >(), "maximum width to report");
try
{
arg::store( arg::parse_command_line( argc, argv, desc ), vm );
arg::notify( vm );
}
catch(...)
{
std::cout << desc << std::endl;
return EXIT_SUCCESS;
}
if( vm.count( "help" ))
{
std::cout << desc << std::endl;
return EXIT_SUCCESS;
}
if( vm.count( "session" ))
session = vm["session"].as< std::string >();
if( vm.count( "hostname" ))
hostname = vm["hostname"].as< std::string >();
const bool daemon = vm.count( "daemon" ) > 0;
#else
if( argc > 1 )
std::cerr << "Ignoring command line options, compiled without "
<< "boost::program_options support" << std::endl;
const bool daemon = false;
#endif
#ifdef GPUSD_CGL
gpusd::cgl::Module::use();
#endif
#ifdef GPUSD_GLX
gpusd::glx::Module::use();
#endif
#ifdef GPUSD_WGL
gpusd::wgl::Module::use();
#endif
GPUInfos gpus = gpusd::Module::discoverGPUs();
if( gpus.empty( ))
{
std::cerr << "No GPUs found, quitting" << std::endl;
return EXIT_FAILURE;
}
if( vm.count( "width" ) )
{
const int width = vm["width"].as< int >();
if( width > 0 )
{
for( GPUInfos::iterator gpu = gpus.begin();
gpu != gpus.end(); ++gpu )
gpu->pvp[2] = std::min( gpu->pvp[2], width );
}
else
{
std::cerr << "Ignoring negative width" << std::endl;
}
}
if( vm.count( "height" ) )
{
const int height = vm["height"].as< int >();
if( height > 0 )
{
for( GPUInfos::iterator gpu = gpus.begin();
gpu != gpus.end(); ++gpu )
gpu->pvp[3] = std::min( gpu->pvp[3], height );
}
else
{
std::cerr << "Ignoring negative height" << std::endl;
}
}
lunchbox::Servus service( "_gpu-sd._tcp" );
setKeys( service, gpus, session, hostname );
if( !service.announce( 4242, "" ))
{
std::cerr << "Service announcement failed" << std::endl;
return EXIT_FAILURE;
}
if( daemon )
{
#if LUNCHBOX_VERSION_GT(1, 5, 0)
if( lunchbox::Log::setOutput( "gpusd.log" ))
lunchbox::daemonize();
#else
LBWARN << "Ignoring daemon request, need Lunchbox >= 1.5.1, got "
<< lunchbox::Version::getString() << std::endl;
#endif
for( ;; )
lunchbox::sleep( 0xFFFFFFFFu );
}
std::cout << "Press <Enter> to quit" << std::endl;
getchar();
return EXIT_SUCCESS;
}
