NodeConcepts.hh

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef avro_NodeConcepts_hh__
#define avro_NodeConcepts_hh__

#include <vector>
#include <map>
#include "Exception.hh"

namespace avro {



namespace concepts {

template <typename Attribute>
struct NoAttribute
{
    static const bool hasAttribute = false;

    size_t size() const {
        return 0;
    }

    void add( const Attribute &attr) {
        // There must be an add function for the generic NodeImpl, but the
        // Node APIs ensure that it is never called, the throw here is
        // just in case
        throw Exception("This type does not have attribute");
    }

    const Attribute &get(size_t index = 0) const {
        // There must be an get function for the generic NodeImpl, but the
        // Node APIs ensure that it is never called, the throw here is
        // just in case
        throw Exception("This type does not have attribute");
        // even though this code is unreachable the compiler requires it
        static const Attribute empty = Attribute();
        return empty;
    }

};

template<typename Attribute>
struct SingleAttribute
{
    static const bool hasAttribute = true;

    SingleAttribute() : attr_(), size_(0)
    { }

    // copy constructing from another single attribute is allowed
    SingleAttribute(const SingleAttribute<Attribute> &rhs) : 
        attr_(rhs.attr_), size_(rhs.size_)
    { }

    // copy constructing from a no attribute is allowed
    SingleAttribute(const NoAttribute<Attribute> &rhs) : 
        attr_(), size_(0)
    { }

    size_t size() const {
        return size_;
    }

    void add(const Attribute &attr) {
        if(size_ == 0) {
            size_ = 1;
        }
        else {
            throw Exception("SingleAttribute can only be set once");
        }
        attr_ = attr;
    }

    const Attribute &get(size_t index = 0) const {
        if(index != 0) {
            throw Exception("SingleAttribute has only 1 value");
        }
        return attr_;
    }

  private:

    template<typename T> friend class MultiAttribute;

    Attribute attr_;
    int       size_;
};

template<typename Attribute>
struct MultiAttribute
{
    static const bool hasAttribute = true;

    MultiAttribute() 
    { }

    // copy constructing from another single attribute is allowed, it
    // pushes the attribute
    MultiAttribute(const SingleAttribute<Attribute> &rhs) 
    { 
        // since map is the only type that does this we know it's
        // final size will be two, so reserve 
        attrs_.reserve(2);
        attrs_.push_back(rhs.attr_);
    }

    MultiAttribute(const MultiAttribute<Attribute> &rhs)  :
        attrs_(rhs.attrs_)
    { }

    MultiAttribute(const NoAttribute<Attribute> &rhs)
    {}

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

    void add(const Attribute &attr) {
        attrs_.push_back(attr); 
    }

    const Attribute &get(size_t index = 0) const {
        return attrs_.at(index);
    }

    Attribute &get(size_t index) {
        return attrs_.at(index);
    }

  private:

    std::vector<Attribute> attrs_;
};


template<typename T>
struct NameIndexConcept {

    bool lookup(const std::string &name, size_t &index) const {
        throw Exception("Name index does not exist");
        return 0;
    }

    bool add(const::std::string &name, size_t index) {
        throw Exception("Name index does not exist");
        return false;
    }
};

template<>
struct NameIndexConcept < MultiAttribute<std::string> > 
{
    typedef std::map<std::string, size_t> IndexMap;

    bool lookup(const std::string &name, size_t &index) const {
        IndexMap::const_iterator iter = map_.find(name); 
        if(iter == map_.end()) {
            return false;
        }
        index = iter->second;
        return true;
    }

    bool add(const::std::string &name, size_t index) {
        bool added = false;
        IndexMap::iterator lb = map_.lower_bound(name); 
        if(lb == map_.end() || map_.key_comp()(name, lb->first)) {
            map_.insert(lb, IndexMap::value_type(name, index));
            added = true;
        }
        return added;
    }

  private:

    IndexMap map_;
};

} // namespace concepts
} // namespace avro

#endif