multithreading.helpers.hpp

Go to the documentation of this file.

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use these files except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Copyright 2012 - Gonzalo Iglesias, Adrià de Gispert, William Byrne

#ifndef MULTITHREADING_HELPERS_HPP
#define MULTITHREADING_HELPERS_HPP

#include <multithreading.hpp>

namespace HifstConstants {
const std::string kNThreads = "nthreads";
const std::string kServerEnable = "server.enable";
}

namespace ucam {
namespace util {

template<class SingleThreadedFunctorT, class MultiThreadedFunctorT>
class Runner {
 private:
  //Use multithreading or not
  bool multithread_;
  //How many threads is the user requesting
  unsigned threadcount_;
  //Command-line/config file options
  const RegistryPO& rg_;

  SingleThreadedFunctorT sf_;
  MultiThreadedFunctorT mtf_;

 public:
  Runner ( const RegistryPO& rg ) :
    rg_ ( rg ),
    multithread_ ( rg.exists ( HifstConstants::kNThreads ) ),
    threadcount_ ( multithread_ ? rg.get<unsigned> ( HifstConstants::kNThreads ) :
                   0 ) {
    if ( multithread_ ) {
      LINFO ( "Multithreading:" << threadcount_ << " threads" );
    } else {
      LINFO ( "Multithreading: no" );
    }
  };

  inline void operator() () {
    if ( multithread_ ) mtf_ ( rg_, threadcount_ );
    else sf_ ( rg_ );
  };

};

template<class SingleThreadedFunctorT, class MultiThreadedFunctorT >
class Runner2 {
 private:
  //Use multithreading or not
  bool multithread_;
  //Command-line/config file options
  const RegistryPO& rg_;

 public:
  Runner2 ( const RegistryPO& rg ) :
    rg_ ( rg ),
    multithread_ ( rg.exists ( HifstConstants::kNThreads ) )

  {
    if ( multithread_ ) {
      LINFO ( "Multithreading:yes" );
    } else {
      LINFO ( "Multithreading:no" );
    }
  };

  inline void operator() () {
    if ( multithread_ ) {
      MultiThreadedFunctorT mtf ( rg_ );
      mtf();
      return;
    }
    SingleThreadedFunctorT sf ( rg_ );
    sf();
  };

};

template<class SingleThreadedFunctorT, class MultiThreadedFunctorT, class ServerFunctorT >
class Runner3 {
 private:
  //Use multithreading or not
  bool multithread_;
  //Command-line/config file options
  const RegistryPO& rg_;

  bool server_;

 public:
  Runner3 ( const RegistryPO& rg ) :
    rg_ ( rg ),
    multithread_ ( rg.exists ( HifstConstants::kNThreads ) ),
    server_ ( rg.getBool ( HifstConstants::kServerEnable ) )

  {
    if ( server_ ) {
      LINFO ( "Initializing server..." );
    } else if ( multithread_ ) {
      LINFO ( "Multithreading:yes" );
    } else {
      LINFO ( "Multithreading:no" );
    }
  };

  inline void operator() () {
    if ( server_ ) {
      ServerFunctorT serv ( rg_ );
      serv();
      return;
    }
    if ( multithread_ ) {
      MultiThreadedFunctorT mtf ( rg_ );
      mtf();
      return;
    }
    SingleThreadedFunctorT sf ( rg_ );
    sf();
  };

};

}
} // end namespaces
#endif