/**
 * \file   ElementStatus.cpp
 * \author Karsten Rink
 * \date   2012-12-18
 * \brief  Implementation of the ElementStatus class.
 *
 * \copyright
 * Copyright (c) 2012-2015, OpenGeoSys Community (http://www.opengeosys.org)
 *            Distributed under a Modified BSD License.
 *              See accompanying file LICENSE.txt or
 *              http://www.opengeosys.org/project/license
 *
 */

#include "ElementStatus.h"

#include "Mesh.h"
#include "MeshLib/Node.h"
#include "Elements/Element.h"

namespace MeshLib {

ElementStatus::ElementStatus(Mesh const* const mesh, bool hasAnyInactive)
    : _mesh(mesh),
      _element_status(mesh->getNElements(), true),
      _hasAnyInactive(hasAnyInactive)
{
	const std::vector<MeshLib::Node*>& nodes(_mesh->getNodes());
	for (auto node : nodes)
		_active_nodes.push_back(node->getNElements());
}

ElementStatus::ElementStatus(Mesh const* const mesh,
                             std::vector<int> const& vec_inactive_matIDs)
    : ElementStatus(mesh, !vec_inactive_matIDs.empty())
{
	auto materialIds = mesh->getProperties().getPropertyVector<int>("MaterialIDs");
	if (materialIds) {
		for (auto material_id : vec_inactive_matIDs) {
			for (auto e : _mesh->getElements()) {
				if ((*materialIds)[e->getID()] == material_id) {
					setElementStatus(e->getID(), false);
				}
			}
		}
	}

	_vec_active_eles.reserve(getNActiveElements());
	const std::size_t nElems (_mesh->getNElements());
	for (std::size_t i=0; i<nElems; ++i)
		if (_element_status[i])
			_vec_active_eles.push_back(const_cast<MeshLib::Element*>(_mesh->getElement(i)));

	_vec_active_nodes.reserve(this->getNActiveNodes());
	const std::size_t nNodes (_mesh->getNNodes());
	for (std::size_t i=0; i<nNodes; ++i)
		if (_active_nodes[i]>0)
			_vec_active_nodes.push_back(const_cast<MeshLib::Node*>(_mesh->getNode(i)));

	DBUG(
	    "Deactivated %d materials and resulting active %d nodes and %d "
	    "elements",
	    vec_inactive_matIDs.size(),
	    _vec_active_nodes.size(),
	    _vec_active_eles.size());
}

std::vector<MeshLib::Element*> const& ElementStatus::getActiveElements() const
{
	if (_hasAnyInactive)
		return _vec_active_eles;
	else
		return _mesh->getElements();
}

std::vector<MeshLib::Node*> const& ElementStatus::getActiveNodes() const
{
	if (_hasAnyInactive)
		return _vec_active_nodes;
	else
		return _mesh->getNodes();
}

std::vector<MeshLib::Element*> ElementStatus::getActiveElementsAtNode(std::size_t node_id) const
{
	const std::size_t nActiveElements (_active_nodes[node_id]);
	std::vector<MeshLib::Element*> active_elements;
	active_elements.reserve(nActiveElements);
	for (auto elem : _mesh->getNode(node_id)->getElements())
	{
		if (active_elements.size() == nActiveElements)
			return active_elements;
		if (_element_status[elem->getID()])
			active_elements.push_back(elem);
	}
	return active_elements;
}

std::size_t ElementStatus::getNActiveNodes() const
{
	return _active_nodes.size() - std::count(_active_nodes.cbegin(), _active_nodes.cend(), 0);
}

std::size_t ElementStatus::getNActiveElements() const
{
	return static_cast<std::size_t>(
	    std::count(_element_status.cbegin(), _element_status.cend(), true));
}

void ElementStatus::setElementStatus(std::size_t i, bool status)
{
	if (_element_status[i] != status)
	{
		const int change = (status) ? 1 : -1;
		_element_status[i] = status;
		const unsigned nElemNodes (_mesh->getElement(i)->getNBaseNodes());
		MeshLib::Node const*const*const nodes = _mesh->getElement(i)->getNodes();
		for (unsigned j=0; j<nElemNodes; ++j)
		{
			assert(_active_nodes[j] < 255);  // if one node has >255 connected
			                                 // elements the data type is too
			                                 // small
			_active_nodes[nodes[j]->getID()] += change;
		}
	}
}

}