DqPtrsEntities.h

// -*-c++-*-
//----------------------------------------------------------------------------
//  XC program; finite element analysis code
//  for structural analysis and design.
//
//  Copyright (C)  Luis C. Pérez Tato
//
//  This program derives from OpenSees <http://opensees.berkeley.edu>
//  developed by the  «Pacific earthquake engineering research center».
//
//  Except for the restrictions that may arise from the copyright
//  of the original program (see copyright_opensees.txt)
//  XC is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or 
//  (at your option) any later version.
//
//  This software is distributed in the hope that it will be useful, but 
//  WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details. 
//
//
// You should have received a copy of the GNU General Public License 
// along with this program.
// If not, see <http://www.gnu.org/licenses/>.
//----------------------------------------------------------------------------
//DqPtrsEntities.h
//deque de pointers (se emplear en la clase Set).


#ifndef DQPTRSENTITIES_H
#define DQPTRSENTITIES_H

#include "DqPtrs.h"
#include "utility/geom/pos_vec/Pos3d.h"
#include "utility/geom/pos_vec/Vector3d.h"
#include "utility/geom/d2/Polygon3d.h"
#include "utility/geom/d1/Segment3d.h"
#include "utility/geom/d3/BND3d.h"
#include "boost/icl/interval_map.hpp"
#include "utility/utils/misc_utils/colormod.h"

class GeomObj3d;
class BND3d;

namespace XC {

template <class T>
class DqPtrsEntities: public DqPtrs<T>
  {
  public:
    typedef DqPtrs<T> dq_ptr;
    typedef typename dq_ptr::const_iterator const_iterator;
    typedef typename dq_ptr::iterator iterator;
    
  public:
    DqPtrsEntities(CommandEntity *owr= nullptr)
      : DqPtrs<T>(owr) {}
    DqPtrsEntities(const DqPtrs<T> &other)
      : DqPtrs<T>(other) {}
    explicit DqPtrsEntities(const std::deque<T *> &ts)
      : DqPtrs<T>(ts) {}
    explicit DqPtrsEntities(const std::set<const T *> &ts)
      : DqPtrs<T>(ts) {}

    DqPtrsEntities<T> &operator-=(const DqPtrsEntities<T> &);
    DqPtrsEntities<T> &operator*=(const DqPtrsEntities<T> &);

    void remove(const DqPtrsEntities<T> &other);
    void intersect(const DqPtrsEntities<T> &other);
    bool remove(Element *);
    bool remove(Node *);

    T *searchName(const std::string &nmb);
    T *findTag(const size_t &);
    const T *findTag(const size_t &) const;
    T *getNearest(const Pos3d &p);
    Pos3d getCentroid(void) const;
    const T *getNearest(const Pos3d &p) const;
    GEOM_FT getDistanceTo(const Pos3d &p) const;
    bool isCloserThan(const Pos3d &, const GEOM_FT &) const;
    bool isCloserThan(const Segment3d &, const GEOM_FT &) const;
    bool isCloserThan(const GeomObj::list_Pos3d &, const GEOM_FT &) const;
    bool isCloserThan(const Polygon3d &, const GEOM_FT &) const;
    DqPtrsEntities<T> pickEntitiesInside(const GeomObj3d &, const double &tol= 0.0) const;
    BND3d Bnd(void) const;
  };
 
template <class T>
T *DqPtrsEntities<T>::searchName(const std::string &nmb)
  {
    for(const_iterator i= this->begin();i!=this->end();i++)
      if((*i)->getName()==nmb) return *i;
    return nullptr;
  }
  
template <class T>
T *DqPtrsEntities<T>::findTag(const size_t &tag)
  {
    for(const_iterator i= this->begin();i!=this->end();i++)
      if((*i)->getTag()==tag) return *i;
    return nullptr;
  }
  
template <class T>
const T *DqPtrsEntities<T>::findTag(const size_t &tag) const
  {
    for(const_iterator i= this->begin();i!=this->end();i++)
      if((*i)->getTag()==tag) return *i;
    return nullptr;
  }
  
template <class T>
Pos3d DqPtrsEntities<T>::getCentroid(void) const
  {
    Pos3d retval;
    if(!this->empty())
      {
    const size_t sz= this->size();
    const_iterator bg= this->begin();
    if(sz<2)
      {
        const Pos3d pos= (*bg)->getCentroid();
        retval= pos;
      }
        else
      {
        const_iterator i= bg;
        const Pos3d pos= (*i)->getCentroid();
            Vector3d vpos_center_of_mass(pos.VectorPos());
            i++;
            for(; i != this->end(); i++)
          {
            const Pos3d pos= (*i)->getCentroid();
                vpos_center_of_mass= vpos_center_of_mass + pos.VectorPos();
          }
            vpos_center_of_mass= vpos_center_of_mass * (1.0/sz);
            retval+= vpos_center_of_mass;
      }
      }
    else
      std::cerr << Color::red << this->getClassName() << "::" << __FUNCTION__
        << "; set is empty, so it has no centroid."
        << Color::def << std::endl;
    return retval;
  }
  
template <class T>
T *DqPtrsEntities<T>::getNearest(const Pos3d &p)
  {
    T *retval= nullptr;
    if(!this->empty())
      {
        const_iterator i= this->begin();
        double d2= (*i)->getSquaredDistanceTo(p);
        retval= *i; i++;
        double tmp;
        for(;i!=this->end();i++)
          {
            tmp= (*i)->getSquaredDistanceTo(p);
            if(tmp<d2)
              {
                d2= tmp;
                retval= *i;
              }
          }
      }
    return retval;
  }

template <class T>
const T *DqPtrsEntities<T>::getNearest(const Pos3d &p) const
  {
    const T *retval= nullptr;
    if(!this->empty())
      {
        const_iterator i= this->begin();
        double d2= (*i)->getSquaredDistanceTo(p);
        retval= *i; i++;
        double tmp;
        for(;i!=this->end();i++)
          {
            tmp= (*i)->getSquaredDistanceTo(p);
            if(tmp<d2)
              {
                d2= tmp;
                retval= *i;
              }
          }
      }
    return retval;
  }

template <class T>
GEOM_FT DqPtrsEntities<T>::getDistanceTo(const Pos3d &p) const
  {
    GEOM_FT retval= std::numeric_limits<GEOM_FT>::quiet_NaN();
    if(!this->empty())
      {
    const T *nearest= this->getNearest(p);
    if(nearest)
      retval= nearest->getDistanceTo(p);
    else
      {
        std::cerr  << Color::red << this->getClassName() << "::" << __FUNCTION__
               << "; something went wrong, can't compute distance."
               << Color::def << std::endl;
      }
      }
    else
      {
    std::cerr  << Color::red << this->getClassName() << "::" << __FUNCTION__
           << "; this set is empty, so there is no distance."
           << Color::def << std::endl;
      }
    return retval;  
  }

template <class T>
bool DqPtrsEntities<T>::isCloserThan(const Pos3d &p, const GEOM_FT &d) const
  {
    bool retval= false;
    const GEOM_FT dist= this->getDistanceTo(p);
    if(!std::isnan(dist))
      retval= (this->getDistanceTo(p)<=d);
    return retval;
  }

template <class T>
bool DqPtrsEntities<T>::isCloserThan(const Segment3d &s, const GEOM_FT &d) const
  {
    const Pos3d &p1= s.getFromPoint();
    const GEOM_FT d1= this->getDistanceTo(p1);
    const Pos3d &p2= s.getToPoint();
    const GEOM_FT d2= this->getDistanceTo(p2);
    return ((d1<=d) && (d2<=d)); 
  }


template <class T>
bool DqPtrsEntities<T>::isCloserThan(const GeomObj::list_Pos3d &vertices, const GEOM_FT &d) const
  {
    bool retval= false;
    if(!vertices.empty())
      {
    GeomObj::list_Pos3d::const_iterator i= vertices.begin();
        const Pos3d &pi= *i;
    retval= this->isCloserThan(pi, d);
    if(retval) // The first one is OK.
      {
        i++;
        for(;i!=vertices.end();i++)
          {
        const Pos3d &pj= *i;
        retval= this->isCloserThan(pj, d);
        if(!retval)
          break;
          }
      }
      }
    return retval;
  }
  
template <class T>
bool DqPtrsEntities<T>::isCloserThan(const Polygon3d &plg, const GEOM_FT &d) const
  {
    const GeomObj::list_Pos3d vertices= plg.getVertexList();
    return this->isCloserThan(vertices, d);
  }
  
template <class T>
DqPtrsEntities<T> DqPtrsEntities<T>::pickEntitiesInside(const GeomObj3d &geomObj, const double &tol) const
  {
    DqPtrsEntities<T> retval;
    for(const_iterator i= this->begin();i!= this->end();i++)
      {
        T *t= (*i);
        assert(t);
    if(t->In(geomObj,tol))
      retval.push_back(t);
      }
    return retval;
  }

  
template <class T>
BND3d DqPtrsEntities<T>::Bnd(void) const
  {
    BND3d retval;
    if(!this->empty())
      {
    const_iterator i= this->begin();
        const T *t= (*i);
        assert(t);
        retval= t->Bnd();
    i++;
    for(;i!= this->end();i++)
      {
        const T *t= (*i);
        assert(t);
        retval+= t->Bnd();
      }
      }
    return retval;
  }

template <class T>
void DqPtrsEntities<T>::remove(const DqPtrsEntities<T> &other)
  {
    for(const_iterator i= other.begin();i!= other.end();i++)
      {
        const T *t= (*i);
    iterator j= find(this->begin(),this->end(),t);
    if(j!=this->end()) //Found.
      this->erase(j);
      }
  }

template <class T>
void DqPtrsEntities<T>::intersect(const DqPtrsEntities<T> &other)
  {
    for(const_iterator i= other.begin();i!= other.end();i++)
      {
        const T *t= (*i);
    iterator j= find(this->begin(),this->end(),t);
    if(j==this->end()) //Not found
      this->erase(j);
      }
  }

// (remove means set the corresponding pointer to null).
template <class T>
bool DqPtrsEntities<T>::remove(Element *ePtr)
  {
    bool retval= false;
    if(ePtr)
      {
    for(const_iterator i= this->begin();i!= this->end();i++)
      {
        T *t= (*i);
        const bool tmp= t->remove(ePtr);
        if(tmp)
          retval= true;
      }
      }
    return retval;
  }
  
// (remove means set the corresponding pointer to null).
template <class T>
bool DqPtrsEntities<T>::remove(Node *nPtr)
  {
    bool retval= false;
    if(nPtr)
      {
    for(const_iterator i= this->begin();i!= this->end();i++)
      {
        T *t= (*i);
        const bool tmp= t->remove(nPtr);
        if(tmp)
          retval= true;
      }
      }
    return retval;
  }  

template <class T>
DqPtrsEntities<T> &DqPtrsEntities<T>::operator-=(const DqPtrsEntities &other)
  {
    remove(other);
    return *this;
  }

template <class T>
DqPtrsEntities<T> &DqPtrsEntities<T>::operator*=(const DqPtrsEntities &other)
  {
    intersect(other);
    return *this;
  }

 
template <class T>
DqPtrsEntities<T> operator+(const DqPtrsEntities<T> &a,const DqPtrsEntities<T> &b)
  {
    DqPtrsEntities<T> retval(a);
    retval+=b;
    return retval;
  }

template <class T>
DqPtrsEntities<T> operator-(const DqPtrsEntities<T> &a,const DqPtrsEntities<T> &b)
  {
    DqPtrsEntities<T> retval;
    for(typename DqPtrsEntities<T>::const_iterator i= a.begin();i!= a.end();i++)
      {
        const T *t= (*i);
    const typename DqPtrsEntities<T>::const_iterator j= find(b.begin(),b.end(),t);
    if(j==b.end()) //Not found in b.
      retval.push_back(t);
      }
    return retval;
  }

template <class T>
DqPtrsEntities<T> operator*(const DqPtrsEntities<T> &a,const DqPtrsEntities<T> &b)
  {
    DqPtrsEntities<T> retval;
    for(typename DqPtrsEntities<T>::const_iterator i= a.begin();i!= a.end();i++)
      {
        const T *t= (*i);
    const typename DqPtrsEntities<T>::const_iterator j= find(b.begin(),b.end(),t);
    if(j!=b.end()) //Found also in b.
      retval.push_back(t);
      }
    return retval;
  }
  
template <class T>
class EntitiesShadows
  {
  public:
    typedef std::set<T*> shadow_makers;
    typedef boost::icl::interval_map<double, shadow_makers> shadow_interval_map;
    typedef typename shadow_interval_map::iterator interval_iterator;
    typedef typename shadow_interval_map::const_iterator interval_const_iterator;
    typedef boost::icl::interval<double> shadow_interval;
  protected:    
    shadow_interval_map x_shadows; // "shadows" of the objects in the x axis.
    shadow_interval_map y_shadows; // "shadows" of the objects in the y axis.
    shadow_interval_map z_shadows; // "shadows" of the objects in the z axis.
  public:
    EntitiesShadows(const DqPtrsEntities<T> &);

    void add(T *);
    void remove(T *);
    
    std::set<T *> getNeighbors(const Pos3d &pMin, const Pos3d &pMax) const;
  };
  
template <class T>
EntitiesShadows<T>::EntitiesShadows(const DqPtrsEntities<T> &entities)
  {
    typedef typename DqPtrsEntities<T>::const_iterator const_iterator;
    for(const_iterator i= entities.begin();i!= entities.end();i++)
      {
        T *t= (*i);
        assert(t);
    this->add(t);
      }
  }
  
template <class T>
void EntitiesShadows<T>::add(T *t)
  {
    shadow_makers sm;
    sm.insert(t);
    BND3d tmp= t->Bnd();
    const Pos3d pMin= tmp.getPMin();
    const Pos3d pMax= tmp.getPMax();
    auto shadow_interval_x= shadow_interval::closed(pMin.x(), pMax.x());
    auto shadow_interval_y= shadow_interval::closed(pMin.y(), pMax.y());
    auto shadow_interval_z= shadow_interval::closed(pMin.z(), pMax.z());
    x_shadows.add(std::make_pair(shadow_interval_x, sm));
    y_shadows.add(std::make_pair(shadow_interval_y, sm));
    z_shadows.add(std::make_pair(shadow_interval_z, sm));
  }
  
template <class T>
void EntitiesShadows<T>::remove(T *t)
  {
    shadow_makers sm;
    sm.insert(t);
    BND3d tmp= t->Bnd();
    const Pos3d pMin= tmp.getPMin();
    const Pos3d pMax= tmp.getPMax();
    const double margin= 1.0; // To be sure that the interval is completely removed.
    auto shadow_interval_x= shadow_interval::closed(pMin.x()-margin, pMax.x()+margin);
    auto shadow_interval_y= shadow_interval::closed(pMin.y()-margin, pMax.y()+margin);
    auto shadow_interval_z= shadow_interval::closed(pMin.z()-margin, pMax.z()+margin);
    x_shadows-= std::make_pair(shadow_interval_x, sm);
    y_shadows-= std::make_pair(shadow_interval_y, sm);
    z_shadows-= std::make_pair(shadow_interval_z, sm);
  }

template <class T>
std::set<T *> EntitiesShadows<T>::getNeighbors(const Pos3d &pMin, const Pos3d &pMax) const
  {
    std::set<T *> retval;
    auto x_shadow= shadow_interval::closed(pMin.x(), pMax.x());
    shadow_interval_map x_neighbors= x_shadows & x_shadow;
    if(!x_neighbors.empty())
      {
    auto y_shadow= shadow_interval::closed(pMin.y(), pMax.y());
    shadow_interval_map y_neighbors= y_shadows & y_shadow;
    if(!y_neighbors.empty())
      {
        auto z_shadow= shadow_interval::closed(pMin.z(), pMax.z());
        shadow_interval_map z_neighbors= z_shadows & z_shadow;
        if(!z_neighbors.empty())
          {
            std::set<T *> z_set;
            for(interval_const_iterator k= z_neighbors.begin();k!=z_neighbors.end();k++)
                  { z_set.insert((*k).second.begin(), (*k).second.end()); }
            std::set<T *> y_set;
                for(interval_const_iterator j= y_neighbors.begin();j!=y_neighbors.end();j++)
          { y_set.insert((*j).second.begin(), (*j).second.end()); }
            std::set<T *> x_set;
                for(interval_const_iterator i= x_neighbors.begin();i!=x_neighbors.end();i++)
          { x_set.insert((*i).second.begin(), (*i).second.end()); }
        // Intersection of the three sets.
        std::set<T *> yz_set;
                set_intersection(y_set.begin(), y_set.end(), z_set.begin(), z_set.end(), std::inserter(yz_set, yz_set.begin()));
                set_intersection(x_set.begin(), x_set.end(), yz_set.begin(), yz_set.end(), std::inserter(retval, retval.begin()));
          }
      }
      }
    return retval;
  }


} //end of XC namespace

#endif