attribute_vector.h

//
//  attibute_vector.h
//  3D_DSC
//
//  Created by Morten Nobel-Jørgensen.
//  Copyright (c) 2014 Morten Nobel-Jørgensen. All rights reserved.
//

#pragma once

#include <vector>
#include <cassert>

namespace is_mesh {

template<typename ITEMID, typename ITEM>
class AttributeVector
{
    std::vector<ITEM> items;
public:
    AttributeVector(size_t size = 0, ITEM item = ITEM())
    {
        items = std::vector<ITEM>(size, item);
    }

    const ITEM& get(ITEMID ID) const
    {
        assert(static_cast<unsigned int>(ID) < items.size());
        return items[ID];
    }

    ITEM& get(ITEMID ID)
    {
        if(ID >= items.size())
        {
            items.resize(ID + 1);
        }
        return items[ID];
    }

    const ITEM& operator [](ITEMID ID) const
    {
        assert(static_cast<unsigned int>(ID) < items.size());
        return items[ID];
    }

    ITEM& operator [](ITEMID ID)
    {
        if(ID >= items.size())
        {
            items.resize(ID + 1);
        }
        return items[ID];
    }

    size_t size() const
    {
        return items.size();
    }

    // Set selected elements to the default value
    // Should be invoked when garbage collection on is_mesh is run (See is_mesh::add_gc_listener )
    // The garbage collection will destroy deleted elements and elements with same ids
    // can later be reused.
    void garbage_collect(const std::vector<ITEMID> & ids, ITEM item = ITEM()){
        for (auto id : ids){
            items[id] = item;
        }
    }
};
}