block_alloc.h

Go to the documentation of this file.

/***************************************************************************
 * include/stxxl/bits/mng/block_alloc.h
 *
 * Part of the STXXL. See http://stxxl.sourceforge.net
 *
 * Copyright (C) 2002-2007 Roman Dementiev <[email protected]>
 * Copyright (C) 2007-2009 Andreas Beckmann <[email protected]>
 *
 * Distributed under 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)
 **************************************************************************/

#ifndef STXXL_MNG_BLOCK_ALLOC_HEADER
#define STXXL_MNG_BLOCK_ALLOC_HEADER

#include <algorithm>
#include <stxxl/bits/parallel.h>
#include <stxxl/bits/common/rand.h>
#include <stxxl/bits/mng/config.h>


STXXL_BEGIN_NAMESPACE

//! \defgroup alloc Allocation Functors
//! \ingroup mnglayer
//! Standard allocation strategies encapsulated in functors.
//! \{

//! Example disk allocation scheme functor.
//! \remarks model of \b allocation_strategy concept
struct basic_allocation_strategy
{
 basic_allocation_strategy(int disks_begin, int disks_end);
 basic_allocation_strategy();
 int operator () (int i) const;
 static const char * name();
};

//! Striping disk allocation scheme functor.
//! \remarks model of \b allocation_strategy concept
struct striping
{
 unsigned_type begin, diff;

public:
 striping(unsigned_type b, unsigned_type e) : begin(b), diff(e - b)
 { }

 striping() : begin(0)
 {
 diff = config::get_instance()->disks_number();
 }

 unsigned_type operator () (unsigned_type i) const
 {
 return begin + i % diff;
 }

 static const char * name()
 {
 return "striping";
 }
};

//! Fully randomized disk allocation scheme functor.
//! \remarks model of \b allocation_strategy concept
struct FR : public striping
{
private:
 typedef random_number<random_uniform_fast> rnd_type;
 rnd_type rnd;

public:
 FR(unsigned_type b, unsigned_type e) : striping(b, e)
 { }

 FR() : striping()
 { }

 unsigned_type operator () (unsigned_type /*i*/) const
 {
 return begin + rnd(rnd_type::value_type(diff));
 }

 static const char * name()
 {
 return "fully randomized striping";
 }
};

//! Simple randomized disk allocation scheme functor.
//! \remarks model of \b allocation_strategy concept
struct SR : public striping
{
private:
 unsigned_type offset;

 typedef random_number<random_uniform_fast> rnd_type;

 void init()
 {
 rnd_type rnd;
 offset = rnd(rnd_type::value_type(diff));
 }

public:
 SR(unsigned_type b, unsigned_type e) : striping(b, e)
 {
 init();
 }

 SR() : striping()
 {
 init();
 }

 unsigned_type operator () (unsigned_type i) const
 {
 return begin + (i + offset) % diff;
 }

 static const char * name()
 {
 return "simple randomized striping";
 }
};

//! Randomized cycling disk allocation scheme functor.
//! \remarks model of \b allocation_strategy concept
struct RC : public striping
{
private:
 std::vector<unsigned_type> perm;

 void init()
 {
 for (unsigned_type i = 0; i < diff; i++)
 perm[i] = i;

 stxxl::random_number<random_uniform_fast> rnd;
 std::random_shuffle(perm.begin(), perm.end(), rnd _STXXL_FORCE_SEQUENTIAL);
 }

public:
 RC(unsigned_type b, unsigned_type e) : striping(b, e), perm(diff)
 {
 init();
 }

 RC() : striping(), perm(diff)
 {
 init();
 }

 unsigned_type operator () (unsigned_type i) const
 {
 return begin + perm[i % diff];
 }

 static const char * name()
 {
 return "randomized cycling striping";
 }
};

struct RC_disk : public RC
{
 RC_disk(unsigned_type b, unsigned_type e) : RC(b, e)
 { }

 RC_disk() : RC(config::get_instance()->regular_disk_range().first, config::get_instance()->regular_disk_range().second)
 { }

 static const char * name()
 {
 return "Randomized cycling striping on regular disks";
 }
};

struct RC_flash : public RC
{
 RC_flash(unsigned_type b, unsigned_type e) : RC(b, e)
 { }

 RC_flash() : RC(config::get_instance()->flash_range().first, config::get_instance()->flash_range().second)
 { }

 static const char * name()
 {
 return "Randomized cycling striping on flash devices";
 }
};

//! 'Single disk' disk allocation scheme functor.
//! \remarks model of \b allocation_strategy concept
struct single_disk
{
 unsigned_type disk;
 single_disk(unsigned_type d, unsigned_type = 0) : disk(d)
 { }

 single_disk() : disk(0)
 { }

 unsigned_type operator () (unsigned_type /*i*/) const
 {
 return disk;
 }

 static const char * name()
 {
 return "single disk";
 }
};

//! Allocator functor adaptor.
//!
//! Gives offset to disk number sequence defined in constructor
template <class BaseAllocator_>
struct offset_allocator
{
 BaseAllocator_ base;
 int_type offset;

 //! Creates functor based on instance of \c BaseAllocator_ functor
 //! with offset \c offset_.
 //! \param offset_ offset
 //! \param base_ used to create a copy
 offset_allocator(int_type offset_, const BaseAllocator_& base_) : base(base_), offset(offset_)
 { }

 //! Creates functor based on instance of \c BaseAllocator_ functor.
 //! \param base_ used to create a copy
 offset_allocator(const BaseAllocator_& base_) : base(base_), offset(0)
 { }

 //! Creates functor based on default \c BaseAllocator_ functor.
 offset_allocator() : offset(0)
 { }

 unsigned_type operator () (unsigned_type i) const
 {
 return base(offset + i);
 }

 int_type get_offset() const
 {
 return offset;
 }

 void set_offset(int_type i)
 {
 offset = i;
 }
};

#ifndef STXXL_DEFAULT_ALLOC_STRATEGY
 #define STXXL_DEFAULT_ALLOC_STRATEGY stxxl::RC
#endif

//! \}

STXXL_END_NAMESPACE

#endif // !STXXL_MNG_BLOCK_ALLOC_HEADER
// vim: et:ts=4:sw=4