50 #ifndef _ZOLTAN2_BASICIDENTIFIERADAPTER_HPP_ 51 #define _ZOLTAN2_BASICIDENTIFIERADAPTER_HPP_ 82 template <
typename User>
111 std::vector<const scalar_t *> &
weights, std::vector<int> &weightStrides);
125 if (idx < 0 || idx >= weights_.size()) {
126 std::ostringstream emsg;
127 emsg << __FILE__ <<
":" << __LINE__
128 <<
" Invalid weight index " << idx << std::endl;
129 throw std::runtime_error(emsg.str());
132 weights_[idx].getStridedList(length, weights, stride);
138 const gno_t *idList_;
139 ArrayRCP<StridedData<lno_t, scalar_t> > weights_;
146 template <
typename User>
149 std::vector<const scalar_t *> &
weights, std::vector<int> &weightStrides):
150 numIds_(numIds), idList_(idPtr), weights_()
153 size_t numWeights = weights.
size();
156 weights_ = arcp(
new input_t [numWeights], 0, numWeights,
true);
159 for (
size_t i=0; i < numWeights; i++){
160 int stride = weightStrides.size() ? weightStrides[i] : 1;
161 ArrayRCP<const scalar_t> wgtV(weights[i], 0, stride*numIds,
false);
162 weights_[i] = input_t(wgtV, stride);
BasicIdentifierAdapter(lno_t numIds, const gno_t *idPtr, std::vector< const scalar_t * > &weights, std::vector< int > &weightStrides)
Constructor.
IdentifierAdapter defines the interface for identifiers.
void getIDsView(const gno_t *&Ids) const
Provide a pointer to this process' identifiers.
lno_t size() const
Return the length of the strided array.
Defines the IdentifierAdapter interface.
InputTraits< User >::scalar_t scalar_t
This class represents a collection of global Identifiers and their associated weights, if any.
IdentifierAdapter< User > base_adapter_t
The StridedData class manages lists of weights or coordinates.
InputTraits< User >::lno_t lno_t
InputTraits< User >::part_t part_t
int getNumWeightsPerID() const
Returns the number of weights per object. Number of weights per object should be zero or greater...
void getWeightsView(const scalar_t *&weights, int &stride, int idx) const
Provide pointer to a weight array with stride.
InputTraits< User >::gno_t gno_t
InputTraits< User >::node_t node_t
size_t getLocalNumIDs() const
Returns the number of objects on this process.
This file defines the StridedData class.