/*******************************************************************************
 * task.cpp
 *
 * ---------------------------------------------------------------------------
 * Persistence of Vision Ray Tracer ('POV-Ray') version 3.7.
 * Copyright 1991-2013 Persistence of Vision Raytracer Pty. Ltd.
 *
 * POV-Ray is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * POV-Ray 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * ---------------------------------------------------------------------------
 * POV-Ray is based on the popular DKB raytracer version 2.12.
 * DKBTrace was originally written by David K. Buck.
 * DKBTrace Ver 2.0-2.12 were written by David K. Buck & Aaron A. Collins.
 * ---------------------------------------------------------------------------
 * $File: //depot/public/povray/3.x/source/backend/support/task.cpp $
 * $Revision: #1 $
 * $Change: 6069 $
 * $DateTime: 2013/11/06 11:59:40 $
 * $Author: chrisc $
 *******************************************************************************/

#include <cassert>
#include <stdexcept>

#include <boost/thread.hpp>
#include <boost/bind.hpp>

// frame.h must always be the first POV file included (pulls in platform config)
#include "backend/frame.h"

#include "base/types.h"
#include "base/timer.h"

#include "backend/support/task.h"

// this must be the last file included
#include "base/povdebug.h"

namespace pov
{

using namespace pov_base;

IMPLEMENT_THREAD_LOCAL_PTR(Task::TaskData, Task::taskDataPtr, TaskDataCleanup);

Task::Task(TaskData *td, const boost::function1<void, Exception&>& f) :
	taskData(td),
	fatalErrorHandler(f),
	stopRequested(false),
	paused(false),
	done(false),
	failed(kNoError),
	timer(NULL),
	realTime(-1),
	cpuTime(-1),
	taskThread(NULL),
	povmsContext(NULL)
{
	if (td == NULL)
		throw POV_EXCEPTION_STRING("Internal error: TaskData is NULL in Task constructor");
	td->task = this;
	SET_THREAD_LOCAL_PTR(taskDataPtr, td);
}

Task::~Task()
{
	Stop();

	// NOTE: Freeing taskData is the responsiblity of the task creator!
	taskData = NULL;
	SET_THREAD_LOCAL_PTR(taskDataPtr, NULL);
}

int Task::FailureCode(int defval)
{
	if(failed == kNoError)
		return defval;
	else
		return failed;
}

POV_LONG Task::ConsumedRealTime() const
{
	return realTime;
}

POV_LONG Task::ConsumedCPUTime() const
{
	return cpuTime;
}

void Task::Start(const boost::function0<void>& completion)
{
	if((done == false) && (taskThread == NULL))
#ifndef USE_OFFICIAL_BOOST
		taskThread = new boost::thread(boost::bind(&Task::TaskThread, this, completion), 1024 * 1024 * 2); // TODO - make stack size definable
#else
		taskThread = new boost::thread(boost::bind(&Task::TaskThread, this, completion));
#endif
}

void Task::RequestStop()
{
	stopRequested = true;
}

void Task::Stop()
{
	stopRequested = true;

	if(taskThread != NULL)
	{
		taskThread->join();
		delete taskThread;
		taskThread = NULL;
	}
}

void Task::Pause()
{
	paused = true;
}

void Task::Resume()
{
	paused = false;
}

POV_LONG Task::ElapsedRealTime() const
{
	assert(timer != NULL);
	return timer->ElapsedRealTime();
}

POV_LONG Task::ElapsedCPUTime() const
{
	assert(timer != NULL);
	return timer->ElapsedCPUTime();
}

void Task::TaskThread(const boost::function0<void>& completion)
{
	int result;

	if((result = POVMS_OpenContext(&povmsContext)) != kNoErr)
	{
		failed = result;
		timer = NULL;
		done = true;
		return;
	}

	POV_SYS_THREAD_STARTUP

	Timer tasktime(true); // only keep CPU time of this thread (if supported)

	timer = &tasktime;

	try
	{
		Run();
	}
	catch(StopThreadException&)
	{
		try
		{
			Stopped();
		}
		catch(pov_base::Exception& e)
		{
			FatalErrorHandler(e);
			failed = e.code(kUncategorizedError);
		}
		catch(std::exception& e)
		{
			FatalErrorHandler(POV_EXCEPTION_STRING(e.what()));
			failed = kUncategorizedError;
		}
		catch(...)
		{
			FatalErrorHandler(POV_EXCEPTION_STRING("An unknown error occured stopping a task!"));
			failed = kUncategorizedError;
		}
	}
	catch(bad_alloc&)
	{
		// can't raise a new exception object here as the alloc will probably fail
#if 0
		FatalErrorHandler(POV_EXCEPTION_STRING("Out of memory! There is not enough memory available for\n"
		                                       "POV-Ray to complete a task. Usually this suggests the scene\n"
		                                       "you are trying to render is too complex for your hardware\n"
		                                       "to handle. Refer to the platform-specific documentation for\n"
		                                       "details, it might contain hints how to work around this.\n"));
#endif
		failed = kOutOfMemoryErr;
	}
	catch(pov_base::Exception& e)
	{
		FatalErrorHandler(e);
		failed = e.code(kUncategorizedError);
	}
	catch(std::exception& e)
	{
		FatalErrorHandler(POV_EXCEPTION_STRING(e.what()));
		failed = kUncategorizedError;
	}
	catch(...)
	{
		FatalErrorHandler(POV_EXCEPTION_STRING("An unknown error occured running a task!"));
		failed = kUncategorizedError;
	}

	realTime = tasktime.ElapsedRealTime();
	if(tasktime.HasValidCPUTime() == true)
		cpuTime = tasktime.ElapsedCPUTime();
	else
		cpuTime = -1;

	try
	{
		Finish();
	}
	catch(pov_base::Exception& e)
	{
		FatalErrorHandler(e);
	}
	catch(std::exception& e)
	{
		FatalErrorHandler(POV_EXCEPTION_STRING(e.what()));
	}
	catch(...)
	{
		FatalErrorHandler(POV_EXCEPTION_STRING("An unknown error occured finishing a task!"));
	}

	timer = NULL;
	done = true;

	POV_SYS_THREAD_CLEANUP

	(void)POVMS_CloseContext(povmsContext);

	completion();
}

}
