SequenceConstructor.hpp

#ifndef SEQUENCECONSTRUCTOR_HPP_
#define SEQUENCECONSTRUCTOR_HPP_

#include "TypeBuilder.hpp"
#include "../internal/DataSources.hpp"
#include <vector>

namespace RTT
{
    namespace types
    {
        template<class T>
        struct sequence_ctor: public std::unary_function<int, const T& >
        {
            typedef const T& ( Signature)(int);
            mutable boost::shared_ptr<T> ptr;
            sequence_ctor() :
                ptr(new T())
            {
            }
            const T& operator()(int size) const
            {
                ptr->resize(size);
                return *(ptr);
            }
        };

        template<class T>
        struct sequence_varargs_ctor
        {
            typedef const std::vector<T>& result_type;
            typedef T argument_type;
            result_type operator()(const std::vector<T>& args) const
            {
                return args;
            }
        };

        template<class T>
        struct sequence_constructor_datasource {
            typedef internal::NArityDataSource<sequence_varargs_ctor<T> > type;
        };

        template<class T>
        struct SequenceBuilder: public TypeBuilder
        {
            typedef typename T::value_type value_type;
            virtual base::DataSourceBase::shared_ptr build(const std::vector<
                    base::DataSourceBase::shared_ptr>& args) const
            {
                if (args.size() == 0)
                    return base::DataSourceBase::shared_ptr();
                typename sequence_constructor_datasource<value_type>::type::shared_ptr vds = new typename sequence_constructor_datasource<value_type>::type();
                for (unsigned int i = 0; i != args.size(); ++i)
                {
                    typename internal::DataSource<value_type>::shared_ptr dsd =
                            boost::dynamic_pointer_cast<internal::DataSource<value_type> >(
                                    args[i]);
                    if (dsd)
                        vds->add(dsd);
                    else
                        return base::DataSourceBase::shared_ptr();
                }
                return vds;
            }

        };

        template<class T>
        struct sequence_ctor2: public std::binary_function<int, typename T::value_type,
                const T&>
        {
            typedef const T& ( Signature)(int, typename T::value_type);
            mutable boost::shared_ptr<T> ptr;
            sequence_ctor2() :
                ptr(new T() )
            {
            }
            const T& operator()(int size, typename T::value_type value) const
            {
                ptr->resize(size);
                ptr->assign(size, value);
                return *(ptr);
            }
        };

    }
}

#endif /* SEQUENCECONSTRUCTOR_HPP_ */