VM polymorphism for getter and setter

youknowone · youknowone · commit 226a2a6cb9f5 · 2020-02-06T03:36:05.000+09:00
diff --git a/vm/src/obj/objgetset.rs b/vm/src/obj/objgetset.rs
@@ -12,11 +12,11 @@ use crate::vm::VirtualMachine;
 pub type PyGetterFunc = Box<dyn Fn(&VirtualMachine, PyObjectRef) -> PyResult>;
 pub type PySetterFunc = Box<dyn Fn(&VirtualMachine, PyObjectRef, PyObjectRef) -> PyResult<()>>;
 
-pub trait IntoPyGetterFunc<T, R> {
+pub trait IntoPyGetterFunc<T> {
     fn into_getter(self) -> PyGetterFunc;
 }
 
-impl<F, T, R> IntoPyGetterFunc<OwnedParam<T>, R> for F
+impl<F, T, R> IntoPyGetterFunc<(OwnedParam<T>, R, VirtualMachine)> for F
 where
     F: Fn(T, &VirtualMachine) -> R + 'static,
     T: TryFromObject,
@@ -30,7 +30,7 @@ where
     }
 }
 
-impl<F, S, R> IntoPyGetterFunc<RefParam<S>, R> for F
+impl<F, S, R> IntoPyGetterFunc<(RefParam<S>, R, VirtualMachine)> for F
 where
     F: Fn(&S, &VirtualMachine) -> R + 'static,
     S: PyValue,
@@ -44,6 +44,28 @@ where
     }
 }
 
+impl<F, T, R> IntoPyGetterFunc<(OwnedParam<T>, R)> for F
+where
+    F: Fn(T) -> R + 'static,
+    T: TryFromObject,
+    R: IntoPyObject,
+{
+    fn into_getter(self) -> PyGetterFunc {
+        IntoPyGetterFunc::into_getter(move |obj, _vm: &VirtualMachine| (self)(obj))
+    }
+}
+
+impl<F, S, R> IntoPyGetterFunc<(RefParam<S>, R)> for F
+where
+    F: Fn(&S) -> R + 'static,
+    S: PyValue,
+    R: IntoPyObject,
+{
+    fn into_getter(self) -> PyGetterFunc {
+        IntoPyGetterFunc::into_getter(move |zelf: &S, _vm: &VirtualMachine| (self)(zelf))
+    }
+}
+
 pub trait IntoPyNoResult {
     fn into_noresult(self) -> PyResult<()>;
 }
@@ -60,14 +82,11 @@ impl IntoPyNoResult for PyResult<()> {
     }
 }
 
-pub trait IntoPySetterFunc<T, V, R>
-where
-    R: IntoPyNoResult,
-{
+pub trait IntoPySetterFunc<T> {
     fn into_setter(self) -> PySetterFunc;
 }
 
-impl<F, T, V, R> IntoPySetterFunc<OwnedParam<T>, V, R> for F
+impl<F, T, V, R> IntoPySetterFunc<(OwnedParam<T>, V, R, VirtualMachine)> for F
 where
     F: Fn(T, V, &VirtualMachine) -> R + 'static,
     T: TryFromObject,
@@ -83,7 +102,7 @@ where
     }
 }
 
-impl<F, S, V, R> IntoPySetterFunc<RefParam<S>, V, R> for F
+impl<F, S, V, R> IntoPySetterFunc<(RefParam<S>, V, R, VirtualMachine)> for F
 where
     F: Fn(&S, V, &VirtualMachine) -> R + 'static,
     S: PyValue,
@@ -99,6 +118,30 @@ where
     }
 }
 
+impl<F, T, V, R> IntoPySetterFunc<(OwnedParam<T>, V, R)> for F
+where
+    F: Fn(T, V) -> R + 'static,
+    T: TryFromObject,
+    V: TryFromObject,
+    R: IntoPyNoResult,
+{
+    fn into_setter(self) -> PySetterFunc {
+        IntoPySetterFunc::into_setter(move |obj, v, _vm: &VirtualMachine| (self)(obj, v))
+    }
+}
+
+impl<F, S, V, R> IntoPySetterFunc<(RefParam<S>, V, R)> for F
+where
+    F: Fn(&S, V) -> R + 'static,
+    S: PyValue,
+    V: TryFromObject,
+    R: IntoPyNoResult,
+{
+    fn into_setter(self) -> PySetterFunc {
+        IntoPySetterFunc::into_setter(move |zelf: &S, v, _vm: &VirtualMachine| (self)(zelf, v))
+    }
+}
+
 #[pyclass]
 pub struct PyGetSet {
     name: String,
@@ -155,9 +198,9 @@ impl PyBuiltinDescriptor for PyGetSet {
 }
 
 impl PyGetSet {
-    pub fn with_get<G, T, R>(name: String, getter: G) -> Self
+    pub fn with_get<G, X>(name: String, getter: G) -> Self
     where
-        G: IntoPyGetterFunc<T, R>,
+        G: IntoPyGetterFunc<X>,
     {
         Self {
             name,
@@ -166,11 +209,10 @@ impl PyGetSet {
         }
     }
 
-    pub fn with_get_set<G, S, GT, GR, ST, SV, SR>(name: String, getter: G, setter: S) -> Self
+    pub fn with_get_set<G, S, X, Y>(name: String, getter: G, setter: S) -> Self
     where
-        G: IntoPyGetterFunc<GT, GR>,
-        S: IntoPySetterFunc<ST, SV, SR>,
-        SR: IntoPyNoResult,
+        G: IntoPyGetterFunc<X>,
+        S: IntoPySetterFunc<Y>,
     {
         Self {
             name,
diff --git a/vm/src/obj/objproperty.rs b/vm/src/obj/objproperty.rs
@@ -268,7 +268,7 @@ impl<'a> PropertyBuilder<'a> {
         }
     }
 
-    pub fn add_getter<I, V, VM, F: IntoPyNativeFunc<I, V, VM>>(self, func: F) -> Self {
+    pub fn add_getter<I, R, VM, F: IntoPyNativeFunc<I, R, VM>>(self, func: F) -> Self {
         let func = self.ctx.new_method(func);
         Self {
             ctx: self.ctx,

Original file line number	Diff line number	Diff line change
`@@ -268,7 +268,7 @@ impl<'a> PropertyBuilder<'a> {`
`268`	`268`	`}`
`269`	`269`	`}`
`270`	`270`
`271`		`- pub fn add_getter<I, V, VM, F: IntoPyNativeFunc<I, V, VM>>(self, func: F) -> Self {`
	`271`	`+ pub fn add_getter<I, R, VM, F: IntoPyNativeFunc<I, R, VM>>(self, func: F) -> Self {`
`272`	`272`	`let func = self.ctx.new_method(func);`
`273`	`273`	`Self {`
`274`	`274`	`ctx: self.ctx,`