Boost C++ Libraries

libs/ptr_container/test/simple_test.cpp

//
// Boost.Pointer Container
//
//  Copyright Thorsten Ottosen 2003-2005. Use, modification and
//  distribution is subject to the Boost Software License, Version
//  1.0. (See accompanying file LICENSE_1_0.txt or copy at
//  http://www.boost.org/LICENSE_1_0.txt)
//
// For more information, see http://www.boost.org/libs/ptr_container/
//

#include <boost/ptr_container/ptr_vector.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/lambda/lambda.hpp>
#include <algorithm>

using namespace std;
                       
//
// A simple polymorphic class
//
class Poly
{
    int        i_;
    static int cnt_;
    
public:
    Poly() : i_( cnt_++ )  { }
    virtual ~Poly()        { }
    void foo()             { doFoo(); }
    
private:    
    virtual void doFoo()   { ++i_; }
    
public:
    friend inline bool operator>( const Poly& l, const Poly r )
    {
        return l.i_ > r.i_;
    }
};

int Poly::cnt_ = 0;

//
// Normally we need something like this to compare pointers to objects
//
template< typename T >
struct sptr_greater
{
    bool operator()( const boost::shared_ptr<T>& l, const boost::shared_ptr<T>& r ) const
    {
        return *l > *r;
    }
};
                                                         
//
// one doesn't need to introduce new names or live with long ones
//                                                         
typedef boost::shared_ptr<Poly> PolyPtr;


void simple_test()
{
    enum { size = 2000 };
    typedef vector<PolyPtr>          vector_t;
    typedef boost::ptr_vector<Poly>  ptr_vector_t;
    vector_t                         svec;   
    ptr_vector_t                     pvec;
    
    for( int i = 0; i < size; ++i ) 
    {
        svec.push_back( PolyPtr( new Poly ) ); 
        pvec.push_back( new Poly );  // no extra syntax      
    }
                   
    for( int i = 0; i < size; ++i )
    {
        svec[i]->foo();
        pvec[i].foo(); // automatic indirection
        svec[i] = PolyPtr( new Poly );
        pvec.replace( i, new Poly ); // direct pointer assignment not possible, original element is deleted
    }
    
    for( vector_t::iterator i = svec.begin(); i != svec.end(); ++i )
        (*i)->foo();
 
    for( ptr_vector_t::iterator i = pvec.begin(); i != pvec.end(); ++i )
        i->foo(); // automatic indirection
}
 
#include <boost/test/unit_test.hpp>
using boost::unit_test::test_suite;

test_suite* init_unit_test_suite( int argc, char* argv[] )
{
    test_suite* test = BOOST_TEST_SUITE( "Pointer Container Test Suite" );

    test->add( BOOST_TEST_CASE( &simple_test ) );

    return test;
}