local_dof_lists_cache.hpp

// Copyright (C) 2011-2012 by the Bem++ Authors
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#ifndef bempp_local_dof_lists_cache_hpp
#define bempp_local_dof_lists_cache_hpp

#include "../common/common.hpp"
#include "../common/shared_ptr.hpp"
#include "../common/types.hpp"

#include <tbb/concurrent_unordered_map.h>
#include <vector>
#include <iostream>

namespace Bempp
{

template <typename BasisFunctionType> class Space;
template <typename BasisFunctionType>
struct LocalDofLists
{
    typedef int DofIndex;
    std::vector<DofIndex> originalIndices;
    std::vector<int> elementIndices;
    std::vector<std::vector<LocalDofIndex> > localDofIndices;
    std::vector<std::vector<BasisFunctionType> > localDofWeights;
    std::vector<std::vector<int> > arrayIndices;
};

template <typename BasisFunctionType>
class LocalDofListsCache
{
public:
    LocalDofListsCache(const Space<BasisFunctionType>& space,
                       const std::vector<unsigned int>& p2o,
                       bool indexWithGlobalDofs);
    ~LocalDofListsCache();

    shared_ptr<const LocalDofLists<BasisFunctionType> > get(
        int start, int indexCount);

private:
    void findLocalDofs(
        int start,
        int indexCount,
        std::vector<typename LocalDofLists<BasisFunctionType>::DofIndex>& originalIndices,
        std::vector<int>& elementIndices,
        std::vector<std::vector<LocalDofIndex> >& localDofIndices,
        std::vector<std::vector<BasisFunctionType> >& localDofWeights,
        std::vector<std::vector<int> >& arrayIndices) const;

    void findLocalDofs(
        int index,
        std::vector<typename LocalDofLists<BasisFunctionType>::DofIndex>& originalIndices,
        std::vector<int>& elementIndices,
        std::vector<std::vector<LocalDofIndex> >& localDofIndices,
        std::vector<std::vector<BasisFunctionType> >& localDofWeights,
        std::vector<std::vector<int> >& arrayIndices) const;

private:
    const Space<BasisFunctionType>& m_space;
    const std::vector<unsigned int>& m_p2o;
    bool m_indexWithGlobalDofs;

    typedef tbb::concurrent_unordered_map<std::pair<int, int>,
        const LocalDofLists<BasisFunctionType>*> LocalDofListsMap;
    LocalDofListsMap m_map;
};

} // namespace Bempp

#endif