Line data    Source code

       1              : /*
       2              :  * Copyright (c) 2013-2015:  G-CSC, Goethe University Frankfurt
       3              :  * Author: Andreas Vogel
       4              :  * 
       5              :  * This file is part of UG4.
       6              :  * 
       7              :  * UG4 is free software: you can redistribute it and/or modify it under the
       8              :  * terms of the GNU Lesser General Public License version 3 (as published by the
       9              :  * Free Software Foundation) with the following additional attribution
      10              :  * requirements (according to LGPL/GPL v3 §7):
      11              :  * 
      12              :  * (1) The following notice must be displayed in the Appropriate Legal Notices
      13              :  * of covered and combined works: "Based on UG4 (www.ug4.org/license)".
      14              :  * 
      15              :  * (2) The following notice must be displayed at a prominent place in the
      16              :  * terminal output of covered works: "Based on UG4 (www.ug4.org/license)".
      17              :  * 
      18              :  * (3) The following bibliography is recommended for citation and must be
      19              :  * preserved in all covered files:
      20              :  * "Reiter, S., Vogel, A., Heppner, I., Rupp, M., and Wittum, G. A massively
      21              :  *   parallel geometric multigrid solver on hierarchically distributed grids.
      22              :  *   Computing and visualization in science 16, 4 (2013), 151-164"
      23              :  * "Vogel, A., Reiter, S., Rupp, M., Nägel, A., and Wittum, G. UG4 -- a novel
      24              :  *   flexible software system for simulating pde based models on high performance
      25              :  *   computers. Computing and visualization in science 16, 4 (2013), 165-179"
      26              :  * 
      27              :  * This program is distributed in the hope that it will be useful,
      28              :  * but WITHOUT ANY WARRANTY; without even the implied warranty of
      29              :  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
      30              :  * GNU Lesser General Public License for more details.
      31              :  */
      32              : 
      33              : #ifndef __H__UG__LIB_DISC__DOF_MANAGER__DOF_COUNT__
      34              : #define __H__UG__LIB_DISC__DOF_MANAGER__DOF_COUNT__
      35              : 
      36              : #include "lib_grid/tools/surface_view.h"
      37              : #include "lib_grid/parallelization/distributed_grid.h"
      38              : #include "dof_distribution_info.h"
      39              : 
      40              : namespace ug{
      41              : 
      42            0 : class DoFCount : public DoFDistributionInfoProvider
      43              : {
      44              :         public:
      45              :                 static const byte ES_MAX = (ES_V_MASTER | ES_H_MASTER | ES_H_SLAVE | ES_V_SLAVE);
      46              :                 static const byte SS_MAX = SurfaceView::ALL;
      47              : 
      48              :                 static const int ALL_FCT = -1;
      49              :                 static const int ALL_SUBSET = -1;
      50              :                 static const byte ALL_ES = ES_MAX + 1;
      51              :                 static const byte ALL_SS = SS_MAX + 1;
      52              :                 static const byte UNIQUE_ES = ES_MAX + 2;
      53              :                 static const byte UNIQUE_SS = SS_MAX + 2;
      54              : 
      55              :         protected:
      56            0 :                 struct Cnt{
      57              : 
      58              :                         Cnt();
      59              :                         void add(uint64 num, SurfaceView::SurfaceState ss, byte is);
      60              :                         void collect_values(std::vector<uint64>& vNum) const;
      61              :                         void set_values(const std::vector<uint64>& vNum, size_t& cnt);
      62              : 
      63            0 :                         struct PCnt {
      64              : 
      65              :                                 PCnt();
      66              : 
      67              :                                 // dofs exactly matching this state
      68              :                                 uint64 num(byte is) const;
      69              : 
      70              :                                 // dofs that contain this state
      71              :                                 uint64 num_contains(byte is) const;
      72              : 
      73              :                                 void collect_values(std::vector<uint64>& vNum) const;
      74              :                                 void set_values(const std::vector<uint64>& vNum, size_t& cnt);
      75              : 
      76              :                                 std::vector<uint64> vNumIS;
      77              :                         };
      78              : 
      79              :                         uint64 num(SurfaceView::SurfaceState ss, byte is) const;
      80              :                         uint64 num_contains(SurfaceView::SurfaceState ss, byte is) const;
      81              : 
      82              :                         std::vector<PCnt> vNumSS;
      83              :                 };
      84              : 
      85              :         public:
      86            0 :                 DoFCount() {};
      87              :                 DoFCount(const GridLevel& gl, ConstSmartPtr<DoFDistributionInfo> spDDInfo);
      88              : 
      89              :                 /// sums values over all procs (reduced to 'proc', allreduce for -1)
      90              :                 void sum_values_over_procs(int proc = -1);
      91              :                 void collect_values(std::vector<uint64>& vNum) const;
      92              :                 void set_values(const std::vector<uint64>& vNum);
      93              : 
      94              :                 void add(int fct, int si, SurfaceView::SurfaceState ss, byte is, uint64 numDoF);
      95              : 
      96              :                 const GridLevel& grid_level() const {return m_gridLevel;}
      97              : 
      98              :                 uint64 num(int fct, int si, SurfaceView::SurfaceState ss, byte is) const;
      99              :                 uint64 num_contains(int fct, int si, SurfaceView::SurfaceState ss, byte is) const;
     100              : 
     101              :         protected:
     102              :                 GridLevel m_gridLevel;
     103              :                 std::vector<std::vector<Cnt> > vvCmpSubset;
     104              : };
     105              : 
     106              : } // end namespace ug
     107              : 
     108              : #endif /* __H__UG__LIB_DISC__DOF_MANAGER__DOF_COUNT__ */