HaxeFoundation.haxe/std/python/_std/sys/thread/Thread.hx

/*
 * Copyright (C)2005-2019 Haxe Foundation
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */

package sys.thread;

class Thread {
	var nativeThread: NativeThread;
	var messages: Deque<Dynamic>;

	static var threads = new haxe.ds.ObjectMap<NativeThread, Thread>();
	static var threadsMutex: Mutex = new Mutex();
	static var mainThread: Thread;

	private function new(t:NativeThread) {
		nativeThread = t;
		messages = new Deque<Dynamic>();
	}

	public function sendMessage(msg:Dynamic):Void {
		messages.add(msg);
	}

	public static function current():Thread {
		threadsMutex.acquire();
		var ct = PyThreadingAPI.current_thread();
		if (ct == PyThreadingAPI.main_thread()) {
			if (mainThread == null) mainThread = new Thread(ct);
			threadsMutex.release();
			return mainThread;
		}
		// If the current thread was not created via the haxe API, it can still be wrapped
		if (!threads.exists(ct)) {
			threads.set(ct, new Thread(ct));
		}
		var t = threads.get(ct);
		threadsMutex.release();
		return t;
	}

	public static function create(callb:Void->Void):Thread {
		var nt:NativeThread = null;
		// Wrap the callback so it will clear the thread reference once the thread is finished
		var wrappedCallB = () -> { 
			callb();
			threadsMutex.acquire();
			threads.remove(nt);
			threadsMutex.release();
		}
		nt = new NativeThread(null, wrappedCallB);
		var t = new Thread(nt);
		threadsMutex.acquire();
		threads.set(nt, t);
		threadsMutex.release();
		nt.start();
		return t;
	}

	public static function readMessage(block:Bool):Dynamic {
		return current().messages.pop(block);
	}
}

@:pythonImport("threading", "Thread")
@:native("Thread")
private extern class NativeThread {
	function new(group:Dynamic, target:Void->Void);
	function start():Void;
}

@:pythonImport("threading")
@:native("threading")
private extern class PyThreadingAPI {
	static function current_thread():NativeThread;
	static function main_thread():NativeThread;
}