#include "rice/Data_Type.hpp"
#include "rice/Constructor.hpp"

using namespace Rice;

class ClassTwo
{
public:
    ClassTwo() {};
    virtual ~ClassTwo() {};
};

class ClassOne
{
public:
    ClassOne() {};
    virtual ~ClassOne() {};
    ClassTwo *getClassTwo() { return &myClassTwo; }
    Object getClassTwo2() { return Data_Object<ClassTwo>(&myClassTwo); }

private:
    ClassTwo myClassTwo;
};


extern "C"
void Init_ricetest()
{
    Data_Type<ClassTwo> rb_cClassTwo = define_class<ClassTwo>("ClassTwo")
                                       .define_constructor(Constructor<ClassTwo>())
                                       ;

    Data_Type<ClassOne> rb_cClassOne = define_class<ClassOne>("ClassOne")
                                       .define_constructor(Constructor<ClassOne>())
                                       .define_method("class_two", &ClassOne::getClassTwo)
                                       .define_method("class_two_2", &ClassOne::getClassTwo2)
                                       ;
}