dmlc::FunctionRegEntryBase< EntryType, FunctionType > Class Template Reference

Common base class for function registry. More...

#include <registry.h>

Inheritance diagram for dmlc::FunctionRegEntryBase< EntryType, FunctionType >:

Public Member Functions
EntryType &	set_body (FunctionType body)
	Set the function body.
EntryType &	describe (const std::string &description)
	Describe the function.
EntryType &	add_argument (const std::string &name, const std::string &type, const std::string &description)
	Add argument information to the function.
EntryType &	add_arguments (const std::vector< ParamFieldInfo > &args)
	Append list if arguments to the end.
EntryType &	set_return_type (const std::string &type)
	Set the return type.

Data Fields
std::string	name
	name of the entry
std::string	description
	description of the entry
std::vector< ParamFieldInfo >	arguments
	additional arguments to the factory function
FunctionType	body
	Function body to create ProductType.
std::string	return_type
	Return type of the function.

Protected Member Functions
EntryType &	self ()

Detailed Description

template<typename EntryType, typename FunctionType>
class dmlc::FunctionRegEntryBase< EntryType, FunctionType >

Common base class for function registry.

// This example demonstrates how to use Registry to create a factory of trees.
struct TreeFactory :
    public FunctionRegEntryBase<TreeFactory, std::function<Tree*()> > {
};
 
// in a independent cc file
namespace dmlc {
DMLC_REGISTRY_ENABLE(TreeFactory);
}
// register binary tree constructor into the registry.
DMLC_REGISTRY_REGISTER(TreeFactory, TreeFactory, BinaryTree)
    .describe("Constructor of BinaryTree")
    .set_body([]() { return new BinaryTree(); });

Template Parameters

EntryType	The type of subclass that inheritate the base.
FunctionType	The function type this registry is registerd.

Member Function Documentation

add_argument()

template<typename EntryType , typename FunctionType >

EntryType & dmlc::FunctionRegEntryBase< EntryType, FunctionType >::add_argument ( const std::string &  name, const std::string &  type, const std::string &  description  )

inline

Add argument information to the function.

Parameters

name	Name of the argument.
type	Type of the argument.
description	Description of the argument.

Returns

reference to self.

add_arguments()

template<typename EntryType , typename FunctionType >

EntryType & dmlc::FunctionRegEntryBase< EntryType, FunctionType >::add_arguments ( const std::vector< ParamFieldInfo > &  args )

inline

Append list if arguments to the end.

Parameters

args	Additional list of arguments.

Returns

reference to self.

describe()

template<typename EntryType , typename FunctionType >

EntryType & dmlc::FunctionRegEntryBase< EntryType, FunctionType >::describe ( const std::string &  description )

inline

Describe the function.

Parameters

description

The description of the factory function.

Returns

reference to self.

self()

template<typename EntryType , typename FunctionType >

EntryType & dmlc::FunctionRegEntryBase< EntryType, FunctionType >::self ( )

inlineprotected

Returns

reference of self as derived type

set_body()

template<typename EntryType , typename FunctionType >

EntryType & dmlc::FunctionRegEntryBase< EntryType, FunctionType >::set_body ( FunctionType  body )

inline

Set the function body.

Parameters

body	Function body to set.

Returns

reference to self.

set_return_type()

template<typename EntryType , typename FunctionType >

EntryType & dmlc::FunctionRegEntryBase< EntryType, FunctionType >::set_return_type ( const std::string &  type )

inline

Set the return type.

Parameters

type	Return type of the function, could be Symbol or Symbol[]

Returns

reference to self.

---

The documentation for this class was generated from the following file:

• External/xgboost/dmlc-core/include/dmlc/registry.h