caffe InternalThread类学习

InternalThread

InternalThread 这个类是使用与DataLayer里面的。InternalThread使用的是boost库的thread(基本的使用方法找一下其他博客)。

class InternalThread {
 public:
  InternalThread() : thread_() {}
  virtual ~InternalThread();

  /** Returns true if the thread was successfully started. **/
  bool StartInternalThread();

  /** Will not return until the internal thread has exited. */
  bool WaitForInternalThreadToExit();

  bool is_started() const;

 protected:
  /* Implement this method in your subclass
      with the code you want your thread to run. */
  virtual void InternalThreadEntry() {}//这个是线程处理的函数，需要子类自行定义。

  shared_ptr<boost::thread> thread_;//boost::thread智能指针
};

InternalThread::~InternalThread() {
  WaitForInternalThreadToExit();
}

bool InternalThread::is_started() const {
  return thread_.get() != NULL && thread_->joinable();
}


bool InternalThread::StartInternalThread() {
  if (!WaitForInternalThreadToExit()) {
    return false;
  }
  try {
    thread_.reset(
        new boost::thread(&InternalThread::InternalThreadEntry, this));//对thread_进行初始化，线程处理的是InternalThreadEntry这个函数。
  } catch (...) {
    return false;
  }
  return true;
}

/** Will not return until the internal thread has exited. */
bool InternalThread::WaitForInternalThreadToExit() {
  if (is_started()) {
    try {
      thread_->join();
    } catch (...) {
      return false;
    }
  }
  return true;
}