blockGeometryStatistics2D.h

Go to the documentation of this file.

/*  This file is part of the OpenLB library
 *
 *  Copyright (C) 2011, 2014 Mathias J. Krause, Simon Zimny
 *  E-mail contact: info@openlb.net
 *  The most recent release of OpenLB can be downloaded at
 *  <http://www.openlb.net/>
 *
 *  This program 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 2
 *  of the License, or (at your option) any later version.
 *
 *  This program 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, write to the Free
 *  Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 *  Boston, MA  02110-1301, USA.
 */
 
#ifndef BLOCK_GEOMETRY_STATISTICS_2D_H
#define BLOCK_GEOMETRY_STATISTICS_2D_H
 
#include <vector>
#include <map>
#include <string>
 
#include "io/ostreamManager.h"
 
// All OpenLB code is contained in this namespace.
namespace olb {
 
 
template<typename T, unsigned D>
class BlockGeometry;
 
template<typename T>
class BlockGeometryStatistics2D {
private:
  BlockGeometry<T,2>* _blockGeometry{};
  bool _statisticsUpdateNeeded{};
  int _nX{}, _nY{};
  T _h{};
 
  int _nMaterials;
  std::map<int, int> _material2n;
  std::map<int, std::vector<int> > _material2min;
  std::map<int, std::vector<int> > _material2max;
 
  mutable OstreamManager clout;
 
  const BlockGeometryStatistics2D<T>* const_this;
 
public:
  BlockGeometryStatistics2D(BlockGeometry<T,2>* blockGeometry);
 
  bool& getStatisticsStatus();
  bool const & getStatisticsStatus() const;
  std::map<int, int> getMaterial2n();
  std::map<int, int> getMaterial2n() const;
 
  void update(bool verbose=true);
 
  int getNmaterials();
  int getNmaterials() const;
  int getNvoxel(int material);
  int getNvoxel(int material) const ;
  int getNvoxel();
  int getNvoxel() const;
  std::vector<int> getMinLatticeR(int material);
  std::vector<int> getMinLatticeR(int material) const;
  std::vector<int> getMaxLatticeR(int material);
  std::vector<int> getMaxLatticeR(int material) const;
  std::vector<T> getMinPhysR(int material) const;
  std::vector<T> getMaxPhysR(int material) const;
  std::vector<T> getLatticeExtend(int material);
  std::vector<T> getLatticeExtend(int material) const;
  std::vector<T> getPhysExtend(int material);
  std::vector<T> getPhysExtend(int material) const;
  std::vector<T> getPhysRadius(int material);
  std::vector<T> getPhysRadius(int material) const;
  std::vector<T> getCenterPhysR(int material);
  std::vector<T> getCenterPhysR(int material) const;
  std::vector<int> getType(const int* input) const;
  std::vector<int> getType(int iX, int iY) const;
 
  std::vector<int> computeNormal(int iX, int iY);
  std::vector<int> computeNormal(int iX, int iY) const;
  std::vector<T> computeNormal (int material);
  std::vector<T> computeNormal (int material) const;
  std::vector<int> computeDiscreteNormal (int material, T maxNorm = 1.1);
  std::vector<int> computeDiscreteNormal (int material, T maxNorm = 1.1) const;
 
  // Returns true if at position (iX,iY,iZ) and in a neighbourhood of size (offsetX,offsetY,offsetZ) only voxels of the given material are found
  bool check(int material, int iX, int iY, unsigned offsetX, unsigned offsetY);
  bool check(int material, int iX, int iY, unsigned offsetX, unsigned offsetY) const;
 
  // Returns true and a position (iX,iY,iZ) if there is a neighbourhood of size (offsetX,offsetY,offsetZ) around (iX,iY,iZ) with only voxels of the given material
  bool find(int material, unsigned offsetX, unsigned offsetY, int& iX, int& iY);
  bool find(int material, unsigned offsetX, unsigned offsetY, int& iX, int& iY) const;
 
  void print();
  void print() const;
 
private:
 
  void takeStatistics(int iX, int iY);
};
 
} // namespace olb
 
#endif