#include "lineRegionPoints.hpp"
#include "fieldsDataBase.hpp"

pFlow::lineRegionPoints::lineRegionPoints
(
    const dictionary &dict, 
    fieldsDataBase &fieldsDataBase
)
:
    regionPoints(dict, fieldsDataBase),
    line_(dict.subDict("lineInfo")),
    centerPoints_("centerPoints"),
    volumes_("volumes"),
    selectedPoints_("selectedPoints")
{
    const auto& lDict = dict.subDict("lineInfo");
    uint32 nPoints = lDict.getValMax<uint32>("numPoints",2);
    realList raddi;

    if( lDict.containsDataEntry("radii"))
    {
        raddi = lDict.getVal<realList>("radii");
    }
    else
    {
        auto r = lDict.getVal<real>("radius");
        raddi = realList(nPoints, r);
    }

    if(raddi.size() != nPoints)
    {
        fatalErrorInFunction
            << "The number elements in of radii list should be equal to the number of points"<<endl;
            fatalExit;
    }
    
    sphereRegions_.resize(nPoints, sphere(realx3(0,0,0),1));
    centerPoints_.resize(nPoints);
    volumes_.resize(nPoints);
    selectedPoints_.resize(nPoints);
    real dt = 1.0/(nPoints-1);
    for(uint32 i = 0; i < nPoints; ++i)
    {
        centerPoints_[i] = line_.point(i*dt);
        sphereRegions_[i] = pFlow::sphere(centerPoints_[i], raddi[i]);
        volumes_[i] = sphereRegions_[i].volume();
        diameters_[i] = 2*sphereRegions_[i].radius();
    }
}

pFlow::span<const pFlow::uint32> pFlow::lineRegionPoints::indices(uint32 elem) const
{
    if(elem>=size())
    {
        fatalErrorInFunction
            << "The element index is out of range. elem: " << elem
            << " size: " << size() << endl;
        fatalExit;
    }
    
    return span<const uint32>(selectedPoints_[elem].data(), selectedPoints_[elem].size());
}

bool pFlow::lineRegionPoints::update()
{
    const auto points = database().updatePoints();
    for(auto& elem:selectedPoints_)
    {
        elem.clear();
    }

    for(uint32 i = 0; i < points.size(); ++i)
    {
        for(uint32 j = 0; j < sphereRegions_.size(); ++j)
        {
            if( sphereRegions_[j].isInside(points[i]))
            {
                selectedPoints_[j].push_back(i);
            }
        }
    }
    return true;
}