escript  Revision_
Distribution.h
Go to the documentation of this file.
1 
2 /*****************************************************************************
3 *
4 * Copyright (c) 2003-2020 by The University of Queensland
5 * http://www.uq.edu.au
6 *
7 * Primary Business: Queensland, Australia
8 * Licensed under the Apache License, version 2.0
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Development until 2012 by Earth Systems Science Computational Center (ESSCC)
12 * Development 2012-2013 by School of Earth Sciences
13 * Development from 2014-2017 by Centre for Geoscience Computing (GeoComp)
14 * Development from 2019 by School of Earth and Environmental Sciences
15 **
16 *****************************************************************************/
17 
18 #ifndef __ESCRIPT_DISTRIBUTION_H__
19 #define __ESCRIPT_DISTRIBUTION_H__
20 
21 #include <escript/DataTypes.h>
22 
23 namespace escript {
24 
25 struct Distribution;
26 typedef boost::shared_ptr<Distribution> Distribution_ptr;
27 typedef boost::shared_ptr<const Distribution> const_Distribution_ptr;
28 
32 struct Distribution
33 {
34  Distribution(JMPI mpiInfo, const DataTypes::IndexVector& firstComponent,
36  mpi_info(mpiInfo)
37  {
38  first_component.resize(mpi_info->size + 1);
39  for (int i = 0; i < mpi_info->size+1; ++i)
40  first_component[i] = m * firstComponent[i] + b;
41  }
42 
44  {
45  return first_component[mpi_info->rank];
46  }
47 
49  {
50  return first_component[mpi_info->rank+1];
51  }
52 
54  {
56  }
57 
59  {
61  }
62 
64  {
65  return first_component[0];
66  }
67 
69  {
70  return first_component[mpi_info->size];
71  }
72 
74  JMPI mpi_info;
75 };
76 
77 } // namespace escript
78 
79 #endif // __ESCRIPT_DISTRIBUTION_H__
80 
escript::Distribution::getGlobalNumComponents
DataTypes::dim_t getGlobalNumComponents() const
Definition: Distribution.h:67
escript::Distribution_ptr
boost::shared_ptr< Distribution > Distribution_ptr
Definition: Distribution.h:39
escript::Distribution::getMyNumComponents
DataTypes::dim_t getMyNumComponents() const
Definition: Distribution.h:72
escript::Distribution::first_component
DataTypes::IndexVector first_component
Definition: Distribution.h:87
escript::Distribution::Distribution
Distribution(JMPI mpiInfo, const DataTypes::IndexVector &firstComponent, DataTypes::index_t m=1, DataTypes::index_t b=0)
Definition: Distribution.h:48
escript::Distribution::getMinGlobalComponents
DataTypes::dim_t getMinGlobalComponents() const
Definition: Distribution.h:77
escript::DataTypes::dim_t
index_t dim_t
Definition: DataTypes.h:93
escript::JMPI
boost::shared_ptr< JMPI_ > JMPI
Definition: EsysMPI.h:72
escript::Distribution::getFirstComponent
DataTypes::index_t getFirstComponent() const
Definition: Distribution.h:57
escript::const_Distribution_ptr
boost::shared_ptr< const Distribution > const_Distribution_ptr
Definition: Distribution.h:41
escript::Distribution::getLastComponent
DataTypes::index_t getLastComponent() const
Definition: Distribution.h:62
escript::Distribution::getMaxGlobalComponents
DataTypes::dim_t getMaxGlobalComponents() const
Definition: Distribution.h:82
escript::Distribution::mpi_info
JMPI mpi_info
Definition: Distribution.h:88
escript::DataTypes::index_t
int index_t
type for array/matrix indices used both globally and on each rank
Definition: DataTypes.h:88
escript
Definition: AbstractContinuousDomain.cpp:23
escript::DataTypes::IndexVector
std::vector< index_t > IndexVector
Definition: DataTypes.h:91