gsProperty.h

#pragma once

//== INCLUDES =================================================================


#include <vector>
#include <string>
#include <algorithm>
#include <typeinfo>


//== NAMESPACE ================================================================


namespace gismo {


//== CLASS DEFINITION =========================================================


class Base_property_array
{
public:

    Base_property_array(const std::string& name) : name_(name) {}

    virtual ~Base_property_array() {}

    virtual void reserve(size_t n) = 0;

    virtual void resize(size_t n) = 0;

    virtual void free_memory() = 0;

    virtual void push_back() = 0;

    virtual void swap(size_t i0, size_t i1) = 0;

    virtual Base_property_array* clone () const = 0;

    virtual const std::type_info& type() = 0;

    const std::string& name() const { return name_; }

    void rename(std::string newname)
    { name_ = give(newname) ; }

protected:

    std::string name_;
};



//== CLASS DEFINITION =========================================================


template <class T>
class gsProperty_array : public Base_property_array
{
public:

    typedef T                                       value_type;
    typedef std::vector<value_type>                 vector_type;
    typedef typename vector_type::reference         reference;
    typedef typename vector_type::const_reference   const_reference;

    gsProperty_array(const std::string& name, T t=T()) :
    Base_property_array(name), value_(give(t)) {}

public: // virtual interface of Base_property_array

    virtual void reserve(size_t n)
    {
        data_.reserve(n);
    }

    virtual void resize(size_t n)
    {
        data_.resize(n, value_);
    }

    virtual void push_back()
    {
        data_.push_back(value_);
    }

    virtual void free_memory()
    {
        vector_type(data_).swap(data_);
    }

    virtual void swap(size_t i0, size_t i1)
    {
        T d(data_[i0]);
        data_[i0]=data_[i1];
        data_[i1]=d;
    }

    virtual Base_property_array* clone() const
    {
        gsProperty_array<T>* p = new gsProperty_array<T>(name_, value_);
        p->data_ = data_;
        return p;
    }

    virtual const std::type_info& type() { return typeid(T); }


public:

    const T* data() const
    {
        return &data_[0];
    }


    std::vector<T>& vector()
    {
        return data_;
    }


    reference operator[](int _idx)
    {
        assert( size_t(_idx) < data_.size() );
        return data_[_idx];
    }

    const_reference operator[](int _idx) const
    {
        assert( size_t(_idx) < data_.size());
        return data_[_idx];
    }



private:
    vector_type data_;
    value_type  value_;
};


// specialization for bool properties
template <>
inline const bool*
gsProperty_array<bool>::data() const
{
    assert(false);
    return NULL;
}



//== CLASS DEFINITION =========================================================


template <class T>
class gsProperty
{
public:

    typedef typename gsProperty_array<T>::reference reference;
    typedef typename gsProperty_array<T>::const_reference const_reference;

    friend class gsProperty_container;
    friend class gsSurfMesh;


public:

    gsProperty(gsProperty_array<T>* p=NULL) : parray_(p) {}

    void reset()
    {
        parray_ = NULL;
    }

    operator bool() const
    {
        return parray_ != NULL;
    }

    reference operator[](int i)
    {
        assert(parray_ != NULL);
        return (*parray_)[i];
    }

    const_reference operator[](int i) const
    {
        assert(parray_ != NULL);
        return (*parray_)[i];
    }

    const T* data() const
    {
        assert(parray_ != NULL);
        return parray_->data();
    }


    std::vector<T>& vector()
    {
        assert(parray_ != NULL);
        return parray_->vector();
    }


private:

    gsProperty_array<T>& array()
    {
        assert(parray_ != NULL);
        return *parray_;
    }

    const gsProperty_array<T>& array() const
    {
        assert(parray_ != NULL);
        return *parray_;
    }


private:
    gsProperty_array<T>* parray_;
};



//== CLASS DEFINITION =========================================================


class gsProperty_container
{
public:

    // default constructor
    gsProperty_container() : size_(0) {}

    // destructor (deletes all property arrays)
    virtual ~gsProperty_container() { clear(); }

    // copy constructor: performs deep copy of property arrays
    gsProperty_container(const gsProperty_container& _rhs) { operator=(_rhs); }

    // assignment: performs deep copy of property arrays
    gsProperty_container& operator=(const gsProperty_container& _rhs)
    {
        if (this != &_rhs)
        {
            clear();
            parrays_.resize(_rhs.n_properties());
            size_ = _rhs.size();
            for (unsigned int i=0; i<parrays_.size(); ++i)
                parrays_[i] = _rhs.parrays_[i]->clone();
        }
        return *this;
    }

    // returns the current size of the property arrays
    size_t size() const { return size_; }

    // returns the number of property arrays
    size_t n_properties() const { return parrays_.size(); }

    // returns a vector of all property names
    std::vector<std::string> properties() const
    {
        std::vector<std::string> names;
        for (unsigned int i=0; i<parrays_.size(); ++i)
            names.push_back(parrays_[i]->name());
        return names;
    }


    // add a property with name \c name and default value \c t
    template <class T> gsProperty<T> add(const std::string& name, T t=T())
    {
        // if a property with this name already exists, return an invalid property
        for (unsigned int i=0; i<parrays_.size(); ++i)
        {
            if (parrays_[i]->name() == name)
            {
                std::cerr << "[gsProperty_container] A property with name \""
                          << name << "\" already exists. Returning invalid property.\n";
                return gsProperty<T>();
            }
        }

        // otherwise add the property
// #pragma GCC diagnostic push
// #pragma GCC diagnostic ignored "-Wuninitialized"
        gsProperty_array<T>* p = new gsProperty_array<T>(name, give(t));//warn unin.
// #pragma GCC diagnostic pop
// #pragma GCC diagnostic pop //restores cmdline options
        p->resize(size_);
        parrays_.push_back(p);
        return gsProperty<T>(p);
    }


    // get a property by its name. returns invalid property if it does not exist.
    template <class T> gsProperty<T> get(const std::string& name) const
    {
        for (unsigned int i=0; i<parrays_.size(); ++i)
            if (parrays_[i]->name() == name)
                return gsProperty<T>(dynamic_cast<gsProperty_array<T>*>(parrays_[i]));
        return gsProperty<T>();
    }


    // returns a property if it exists, otherwise it creates it first.
    template <class T> gsProperty<T> get_or_add(const std::string& name, const T t=T())
    {
        gsProperty<T> p = get<T>(name);
        if (!p) p = add<T>(name, give(t));
        return p;
    }

    // Returns true if a property labeled \a name exists.
    bool has(const std::string& name) const
    {
        for (unsigned int i=0; i<parrays_.size(); ++i)
            if (parrays_[i]->name() == name)
                return true;
        return false;
    }

    // get the type of property by its name. returns typeid(void) if
    // it does not exist.
    const std::type_info& get_type(const std::string& name)
    {
        for (unsigned int i=0; i<parrays_.size(); ++i)
            if (parrays_[i]->name() == name)
                return parrays_[i]->type();
        return typeid(void);
    }

    // Returns true if a property labeled \a name exists.
    void swap(const std::string& name1, const std::string& name2) const
    {
        std::pair<int,int> spair(-1,-1);
        for (unsigned int i=0; i<parrays_.size(); ++i)
        {
            if (parrays_[i]->name() == name1)
                spair.first = i;
            if (parrays_[i]->name() == name2)
                spair.second = i;
        }
        GISMO_ASSERT(spair.first!=-1 && spair.second!=-1,"error");
        parrays_[spair.first ]->rename(name2);
        parrays_[spair.second]->rename(name1);
    }

    // rename a property (the name should not exist already)
    template <class T> void rename(gsProperty<T>& h, std::string newname)
    {
        GISMO_ASSERT( !has(newname), "There is already a property with this name.");
        h.parray_->rename(give(newname));
    }

    // delete a property
    template <class T> void remove(gsProperty<T>& h)
    {
        std::vector<Base_property_array*>::iterator
            it=parrays_.begin(),
            end=parrays_.end();
        for (; it!=end; ++it)
        {
            if (*it == h.parray_)
            {
                delete *it;
                parrays_.erase(it);
                h.reset();
                break;
            }
        }
    }

    // delete all properties
    void clear()
    {
        for (unsigned int i=0; i<parrays_.size(); ++i)
            delete parrays_[i];
        parrays_.clear();
        size_ = 0;
    }


    // reserve memory for n entries in all arrays
    void reserve(size_t n) const
    {
        for (unsigned int i=0; i<parrays_.size(); ++i)
            parrays_[i]->reserve(n);
    }

    // resize all arrays to size n
    void resize(size_t n)
    {
        for (unsigned int i=0; i<parrays_.size(); ++i)
            parrays_[i]->resize(n);
        size_ = n;
    }

    // free unused space in all arrays
    void free_memory() const
    {
        for (unsigned int i=0; i<parrays_.size(); ++i)
            parrays_[i]->free_memory();
    }

    // add a new element to each vector
    void push_back()
    {
        for (unsigned int i=0; i<parrays_.size(); ++i)
            parrays_[i]->push_back();
        ++size_;
    }

    // swap elements i0 and i1 in all arrays
    void swap(size_t i0, size_t i1) const
    {
        for (unsigned int i=0; i<parrays_.size(); ++i)
            parrays_[i]->swap(i0, i1);
    }


private:
    std::vector<Base_property_array*>  parrays_;
    size_t  size_;
};


//=============================================================================
} // namespace gismo