92 typedef Xpetra::MultiVector<typename Teuchos::ScalarTraits<Scalar>::coordinateType,LO,GO,NO> dMV;
93 typedef Xpetra::BlockedMultiVector<typename Teuchos::ScalarTraits<Scalar>::coordinateType,LO,GO,NO> dBV;
95 GetOStream(
Runtime0) <<
"Transferring (blocked) coordinates" << std::endl;
97 const size_t numSubFactories = NumFactories();
98 std::vector<RCP<const Map> > subBlockMaps(numSubFactories);
99 std::vector<RCP<dMV> > subBlockCoords(numSubFactories);
101 if (coarseLevel.
IsAvailable(
"Coordinates",
this)) {
102 GetOStream(
Runtime0) <<
"Reusing coordinates" << std::endl;
107 for(
size_t i=0; i<numSubFactories; i++) {
108 GetOStream(
Runtime1) <<
"Generating Coordinates for block " << i <<
"/"<<numSubFactories <<std::endl;
109 const RCP<const FactoryBase>& myFactory = subFactories_[i];
110 myFactory->CallBuild(coarseLevel);
111 subBlockCoords[i] = coarseLevel.
Get<RCP<dMV> >(
"Coordinates", myFactory.get());
112 subBlockMaps[i] = subBlockCoords[i]->getMap();
116 RCP<const BlockedMap> coarseCoordMapBlocked;
122 RCP<const BlockedMap> coarseMap = Get< RCP<const BlockedMap> >(coarseLevel,
"CoarseMap");
123 bool thyraMode = coarseMap->getThyraMode();
125 ArrayView<const GO> elementAList = coarseMap->getFullMap()->getLocalElementList();
128 if (rcp_dynamic_cast<const StridedMap>(coarseMap->getMap(0, thyraMode)) != Teuchos::null)
129 blkSize = rcp_dynamic_cast<const StridedMap>(coarseMap->getMap(0, thyraMode))->getFixedBlockSize();
131 for(
size_t i=1; i<numSubFactories; i++) {
133 if (rcp_dynamic_cast<const StridedMap>(coarseMap->getMap(i, thyraMode)) != Teuchos::null)
134 otherBlkSize = rcp_dynamic_cast<const StridedMap>(coarseMap->getMap(i, thyraMode))->getFixedBlockSize();
135 TEUCHOS_TEST_FOR_EXCEPTION(otherBlkSize != blkSize,
Exceptions::RuntimeError,
"BlockedCoordinatesTransferFactory: Subblocks have different Block sizes. This is not yet supported.");
138 GO indexBase = coarseMap->getFullMap()->getIndexBase();
139 size_t numElements = elementAList.size() / blkSize;
140 Array<GO> elementList(numElements);
143 for (LO i = 0; i < Teuchos::as<LO>(numElements); i++)
144 elementList[i] = (elementAList[i*blkSize]-indexBase)/blkSize + indexBase;
146 RCP<const Map> coarseCoordMap = MapFactory::Build(coarseMap->getFullMap()->lib(),
147 Teuchos::OrdinalTraits<Xpetra::global_size_t>::invalid(), elementList, indexBase, coarseMap->getFullMap()->getComm());
149 coarseCoordMapBlocked = rcp(
new BlockedMap(coarseCoordMap, subBlockMaps, thyraMode));
153 RCP<dBV> bcoarseCoords = rcp(
new dBV(coarseCoordMapBlocked,subBlockCoords));
156 RCP<dMV> coarseCoords = bcoarseCoords->Merge();
157 Set<RCP<dMV> >(coarseLevel,
"Coordinates", coarseCoords);