コード例 #1
0
//----------------------------------------------------------------------------
void snl_fei::RecordCollection::
setOwners_lowestSharing(fei::SharedIDs<int>& sharedIDs)
{
  fei::SharedIDs<int>::map_type::iterator
    s_beg = sharedIDs.getSharedIDs().begin(),
    s_end = sharedIDs.getSharedIDs().end(),
    s_it;

  std::vector<int>& owningProcs = sharedIDs.getOwningProcs();

  int i=0;
  for(i=0, s_it = s_beg; s_it != s_end; ++i, ++s_it) {
    int sh_id = s_it->first;
    fei::Record<int>* record = getRecordWithID(sh_id);
    if (record == NULL) continue;

    int proc = owningProcs[i];

    if (debugOutput_) {
      *dbgOut_ << "#   setting ID " << (int)(record->getID())
               << "'s owner to proc " << proc << FEI_ENDL;
    }

    record->setOwnerProc(proc);
  }
}
コード例 #2
0
void set_shared_ids(MPI_Comm comm,
                    const snl_fei::RecordCollection& records,
                    fei::SharedIDs<int>& sharedIDs,
                    int lowest_global_id, int highest_global_id)
{
    sharedIDs.getSharedIDs().clear();
    sharedIDs.getOwningProcs().clear();

    int numProcs = fei::numProcs(comm);
    if (numProcs < 2) return;
    int myProc = fei::localProc(comm);

    fei::LinearDecomposition<int> lindecomp(myProc,numProcs,
                                            lowest_global_id, highest_global_id);

    //Fill a CommMap (procs_to_shared_ids) that maps other procs to ids which we
    //have, but which are not in our portion of the linear decomposition.
    fei::CommMap<int>::Type procs_to_shared_ids;
    copy_remotelyowned_ids_into_CommMap(myProc, lindecomp, records, procs_to_shared_ids);

    //Do a global-exchange where we send those ids to procs that own them in the
    //linear decomposition.
    //And receive IDs in our portion of the linear decomposition from other procs
    //that hold them.
    fei::CommMap<int>::Type procs_to_owned_ids;
    fei::exchangeCommMapData<int>(comm, procs_to_shared_ids, procs_to_owned_ids);

    //transpose procs_to_owned_ids:
    fei::CommMap<int>::Type owned_ids_to_procs;

    fei::CommMap<int>::Type::iterator
    o_iter = procs_to_owned_ids.begin(), o_end = procs_to_owned_ids.end();

    for(; o_iter != o_end; ++o_iter) {
        int proc = o_iter->first;
        std::vector<int>& ids = o_iter->second;
        for(size_t i=0; i<ids.size(); ++i) {
            addItemsToCommMap(ids[i], 1, &proc, owned_ids_to_procs);
            if (records.getRecordWithID(ids[i]) != NULL) {
                addItemsToCommMap(ids[i], 1, &myProc, owned_ids_to_procs);
            }
        }
    }

    fei::CommMap<int>::Type procs_to_owned_ids_and_sharing_procs;

    for(o_iter=procs_to_owned_ids.begin(); o_iter!=procs_to_owned_ids.end(); ++o_iter) {
        int proc = o_iter->first;
        std::vector<int>& ids = o_iter->second;
        for(size_t i=0; i<ids.size(); ++i) {
            std::vector<int>& sharing_procs = owned_ids_to_procs[ids[i]];
            int num_sharing_procs = sharing_procs.size();
            if (num_sharing_procs > 1) {
                addItemsToCommMap(proc, 1, &ids[i], procs_to_owned_ids_and_sharing_procs, false);
                addItemsToCommMap(proc, 1, &num_sharing_procs, procs_to_owned_ids_and_sharing_procs, false);
                addItemsToCommMap(proc, num_sharing_procs, &sharing_procs[0], procs_to_owned_ids_and_sharing_procs, false);
            }
        }
    }

    fei::CommMap<int>::Type procs_to_shared_ids_and_sharing_procs;
    fei::exchangeCommMapData<int>(comm, procs_to_owned_ids_and_sharing_procs,
                                  procs_to_shared_ids_and_sharing_procs);

    copy_into_shared_ids(procs_to_owned_ids_and_sharing_procs, records, sharedIDs);
    copy_into_shared_ids(procs_to_shared_ids_and_sharing_procs, records, sharedIDs);
}