use std::mem;
use std::ops::{Deref, DerefMut};
pub trait Free {
unsafe fn godot_free(self);
}
pub trait QueueFree {
unsafe fn godot_queue_free(&mut self);
}
pub struct FreeOnDrop<T: Free + Clone> {
ptr: T,
}
impl<T> FreeOnDrop<T> where T: Free + Clone {
pub unsafe fn new(ptr: T) -> Self {
FreeOnDrop { ptr }
}
pub fn forget(self) -> T {
let ptr = self.ptr.clone();
mem::forget(self);
ptr
}
}
impl<T> Drop for FreeOnDrop<T> where T: Free + Clone {
fn drop(&mut self) {
unsafe { self.ptr.clone().godot_free(); }
}
}
impl<T> Deref for FreeOnDrop<T> where T: Free + Clone {
type Target = T;
fn deref(&self) -> &T { &self.ptr }
}
impl<T> DerefMut for FreeOnDrop<T> where T: Free + Clone {
fn deref_mut(&mut self) -> &mut T { &mut self.ptr }
}
pub struct QueueFreeOnDrop<T: QueueFree + Clone> {
ptr: T,
}
impl<T> QueueFreeOnDrop<T> where T: QueueFree + Clone {
pub unsafe fn new(ptr: T) -> Self {
QueueFreeOnDrop { ptr }
}
pub fn forget(self) -> T {
let ptr = self.ptr.clone();
mem::forget(self);
ptr
}
}
impl<T> Drop for QueueFreeOnDrop<T> where T: QueueFree + Clone {
fn drop(&mut self) {
unsafe { self.ptr.godot_queue_free(); }
}
}
impl<T> Deref for QueueFreeOnDrop<T> where T: QueueFree + Clone{
type Target = T;
fn deref(&self) -> &T { &self.ptr }
}
impl<T> DerefMut for QueueFreeOnDrop<T> where T: QueueFree + Clone {
fn deref_mut(&mut self) -> &mut T { &mut self.ptr }
}