X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fencode_server.cc;h=f318da03b5c2878dc5be5f3d30e74244ebcdf57e;hb=3828baf56467224f5d44049bf1e7a7ed11f43a05;hp=3f30a361a898a49587202cb1a2a7f8b512193a29;hpb=17df947ac256397311a11894062070f8069c7e75;p=dcpomatic.git

diff --git a/src/lib/encode_server.cc b/src/lib/encode_server.cc
index 3f30a361a..f318da03b 100644
--- a/src/lib/encode_server.cc
+++ b/src/lib/encode_server.cc
@@ -1,19 +1,20 @@
 /*
     Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
-    This program is distributed in the hope that it will be useful,
+    DCP-o-matic 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 General Public License for more details.
 
     You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
@@ -65,11 +66,11 @@ using boost::optional;
 using dcp::Size;
 using dcp::Data;
 
-EncodeServer::EncodeServer (shared_ptr<Log> log, bool verbose)
-	: _terminate (false)
+EncodeServer::EncodeServer (shared_ptr<Log> log, bool verbose, int num_threads)
+	: Server (Config::instance()->server_port_base())
 	, _log (log)
 	, _verbose (verbose)
-	, _acceptor (_io_service, boost::asio::ip::tcp::endpoint (boost::asio::ip::tcp::v4(), Config::instance()->server_port_base()))
+	, _num_threads (num_threads)
 {
 
 }
@@ -77,7 +78,7 @@ EncodeServer::EncodeServer (shared_ptr<Log> log, bool verbose)
 EncodeServer::~EncodeServer ()
 {
 	{
-		boost::mutex::scoped_lock lm (_worker_mutex);
+		boost::mutex::scoped_lock lm (_mutex);
 		_terminate = true;
 		_empty_condition.notify_all ();
 		_full_condition.notify_all ();
@@ -89,8 +90,6 @@ EncodeServer::~EncodeServer ()
 		delete i;
 	}
 
-	_io_service.stop ();
-
 	_broadcast.io_service.stop ();
 	if (_broadcast.thread) {
 		DCPOMATIC_ASSERT (_broadcast.thread->joinable ());
@@ -146,7 +145,7 @@ void
 EncodeServer::worker_thread ()
 {
 	while (true) {
-		boost::mutex::scoped_lock lock (_worker_mutex);
+		boost::mutex::scoped_lock lock (_mutex);
 		while (_queue.empty () && !_terminate) {
 			_empty_condition.wait (lock);
 		}
@@ -209,21 +208,20 @@ EncodeServer::worker_thread ()
 }
 
 void
-EncodeServer::run (int num_threads)
+EncodeServer::run ()
 {
-	LOG_GENERAL ("Server starting with %1 threads", num_threads);
+	LOG_GENERAL ("Server starting with %1 threads", _num_threads);
 	if (_verbose) {
-		cout << "DCP-o-matic server starting with " << num_threads << " threads.\n";
+		cout << "DCP-o-matic server starting with " << _num_threads << " threads.\n";
 	}
 
-	for (int i = 0; i < num_threads; ++i) {
+	for (int i = 0; i < _num_threads; ++i) {
 		_worker_threads.push_back (new thread (bind (&EncodeServer::worker_thread, this)));
 	}
 
 	_broadcast.thread = new thread (bind (&EncodeServer::broadcast_thread, this));
 
-	start_accept ();
-	_io_service.run ();
+	Server::run ();
 }
 
 void
@@ -283,24 +281,9 @@ EncodeServer::broadcast_received ()
 }
 
 void
-EncodeServer::start_accept ()
+EncodeServer::handle (shared_ptr<Socket> socket)
 {
-	if (_terminate) {
-		return;
-	}
-
-	shared_ptr<Socket> socket (new Socket);
-	_acceptor.async_accept (socket->socket (), boost::bind (&EncodeServer::handle_accept, this, socket, boost::asio::placeholders::error));
-}
-
-void
-EncodeServer::handle_accept (shared_ptr<Socket> socket, boost::system::error_code const & error)
-{
-	if (error) {
-		return;
-	}
-
-	boost::mutex::scoped_lock lock (_worker_mutex);
+	boost::mutex::scoped_lock lock (_mutex);
 
 	/* Wait until the queue has gone down a bit */
 	while (_queue.size() >= _worker_threads.size() * 2 && !_terminate) {
@@ -309,6 +292,4 @@ EncodeServer::handle_accept (shared_ptr<Socket> socket, boost::system::error_cod
 
 	_queue.push_back (socket);
 	_empty_condition.notify_all ();
-
-	start_accept ();
 }