Exposed ArrayLengthProxyFn

Author: Ayuto

src/core/modules/entities/entities_props.cpp

@@ -169,6 +169,21 @@ BoostDataTableProxyFn SendPropSharedExt::get_data_table_proxy_function(SendProp
 	return BoostDataTableProxyFn(pSendProp->GetDataTableProxyFn());
 }
 
+BoostArrayLengthProxyFn SendPropSharedExt::get_array_length_proxy_function(SendProp *pSendProp)
+{
+	if (pSendProp->IsExcludeProp())
+		BOOST_RAISE_EXCEPTION(PyExc_TypeError, "%s is excluded.", pSendProp->GetName());
+
+	if (pSendProp->GetType() != DPT_Array)
+		BOOST_RAISE_EXCEPTION(PyExc_TypeError, "%s is not an array.", pSendProp->GetName());
+
+	if (!pSendProp->GetArrayLengthProxy()) {
+		BOOST_RAISE_EXCEPTION(PyExc_ValueError, "%s has no array length proxy function.", pSendProp->GetName());
+	}
+
+	return BoostArrayLengthProxyFn(pSendProp->GetArrayLengthProxy());
+}
+
 
 // ============================================================================
 // >> SendPropVariantExt

src/core/modules/entities/entities_props.h

@@ -42,6 +42,7 @@
 //-----------------------------------------------------------------------------
 BOOST_FUNCTION_TYPEDEF(void (const SendProp*, const void*, const void*, DVariant*, int, int), BoostSendVarProxyFn)
 BOOST_FUNCTION_TYPEDEF(void* (const SendProp*, const void*, const void*, CSendProxyRecipients*, int), BoostDataTableProxyFn)
+BOOST_FUNCTION_TYPEDEF(int (const void *pStruct, int objectID), BoostArrayLengthProxyFn)
 
 
 //-----------------------------------------------------------------------------
@@ -69,6 +70,7 @@ class SendPropSharedExt
 public:
 	static BoostSendVarProxyFn get_proxy_function(SendProp *pSendProp);
 	static BoostDataTableProxyFn get_data_table_proxy_function(SendProp *pSendProp);
+	static BoostArrayLengthProxyFn get_array_length_proxy_function(SendProp *pSendProp);
 };
 
 

src/core/modules/entities/entities_props_wrap.cpp

@@ -120,6 +120,7 @@ void export_send_prop(scope _props)
 {
 	EXPOSE_FUNCTION_TYPEDEF(BoostSendVarProxyFn, "SendVarProxyFn")
 	EXPOSE_FUNCTION_TYPEDEF_RET(BoostDataTableProxyFn, "DataTableProxyFn", return_by_value_policy())
+	EXPOSE_FUNCTION_TYPEDEF(BoostArrayLengthProxyFn, "ArrayLengthProxyFn")
 
 	class_<SendProp, SendProp *, boost::noncopyable> SendProp_("SendProp", no_init);
 
@@ -149,6 +150,7 @@ void export_send_prop(scope _props)
 	// TODO: Rename proxy_function to send_var_proxy_function
 	SendProp_.add_property("proxy_function", &SendPropSharedExt::get_proxy_function);
 	SendProp_.add_property("data_table_proxy_function", &SendPropSharedExt::get_data_table_proxy_function);
+	SendProp_.add_property("array_length_proxy_function", &SendPropSharedExt::get_array_length_proxy_function);
 
 	// CS:GO specific properties...
 	SendProp_.NOT_IMPLEMENTED_ATTR("priority");