43 #include "Thyra_DefaultSpmdVectorSpace.hpp"
44 #include "Thyra_DefaultSpmdMultiVector.hpp"
45 #include "Thyra_DefaultSpmdVector.hpp"
46 #include "Thyra_DetachedVectorView.hpp"
47 #include "Thyra_DetachedMultiVectorView.hpp"
48 #include "Thyra_VectorSpaceFactoryBase.hpp"
49 #include "Thyra_ProductVectorSpaceBase.hpp"
58 #include "Epetra_Map.h"
59 #include "Epetra_Comm.h"
60 #include "Epetra_SerialComm.h"
62 # include "Epetra_MpiComm.h"
64 #include "Epetra_MultiVector.h"
65 #include "Epetra_Vector.h"
76 unwrapSingleProductVectorSpace(
77 const Teuchos::RCP<
const Thyra::VectorSpaceBase<double> > &vs_in
81 using Teuchos::rcp_dynamic_cast;
82 using Thyra::ProductVectorSpaceBase;
83 const RCP<const ProductVectorSpaceBase<double> > pvs =
84 rcp_dynamic_cast<
const ProductVectorSpaceBase<double> >(vs_in);
87 return pvs->getBlock(0);
105 using Teuchos::rcp_dynamic_cast;
106 using Teuchos::set_extra_data;
109 serialEpetraComm = rcp_dynamic_cast<const Epetra_SerialComm>(epetraComm);
110 if( serialEpetraComm.
get() ) {
113 set_extra_data( serialEpetraComm,
"serialEpetraComm", Teuchos::inOutArg(serialComm) );
120 mpiEpetraComm = rcp_dynamic_cast<const Epetra_MpiComm>(epetraComm);
121 if( mpiEpetraComm.
get() ) {
123 rawMpiComm = Teuchos::opaqueWrapper(mpiEpetraComm->Comm());
124 set_extra_data( mpiEpetraComm,
"mpiEpetraComm", Teuchos::inOutArg(rawMpiComm) );
126 mpiComm =
rcp(
new Teuchos::MpiComm<Ordinal>(rawMpiComm));
133 return Teuchos::null;
147 !epetra_map.
get(), std::invalid_argument,
148 "create_VectorSpace::initialize(...): Error!" );
149 #endif // TEUCHOS_DEBUG
154 Teuchos::set_extra_data(epetra_map,
"epetra_map", inoutArg(comm));
155 const Ordinal localSubDim = epetra_map->NumMyElements();
158 defaultSpmdVectorSpace<double>(
159 comm, localSubDim, epetra_map->NumGlobalElements64(),
160 !epetra_map->DistributedGlobal());
170 Teuchos::set_extra_data( epetra_map,
"epetra_map", inoutArg(vs) );
178 const RCP<
const VectorSpaceBase<double> > &parentSpace,
182 using Teuchos::rcp_dynamic_cast;
188 return parentSpace->smallVecSpcFcty()->createVecSpc(dim);
195 const RCP<
const VectorSpaceBase<double> > &space_in
202 space = Teuchos::rcp_dynamic_cast<
const SpmdVectorSpaceBase<double> >(
209 return Teuchos::null;
212 epetra_v->ExtractView( &localValues );
216 new DefaultSpmdVector<double>(
218 Teuchos::arcp(localValues,0,epetra_v->Map().NumMyElements(),
false),
222 Teuchos::set_extra_data<RCP<Epetra_Vector> >( epetra_v,
"Epetra_Vector",
223 Teuchos::inOutArg(v) );
231 const RCP<
const VectorSpaceBase<double> > &space_in
238 space = Teuchos::rcp_dynamic_cast<
const SpmdVectorSpaceBase<double> >(
245 return Teuchos::null;
248 epetra_v->ExtractView( &localValues );
252 new DefaultSpmdVector<double>(
254 Teuchos::arcp(localValues,0,epetra_v->Map().NumMyElements(),
false),
258 Teuchos::set_extra_data<RCP<const Epetra_Vector> >( epetra_v,
"Epetra_Vector",
259 Teuchos::inOutArg(v) );
267 const RCP<
const VectorSpaceBase<double> > &range_in,
268 const RCP<
const VectorSpaceBase<double> > &domain_in
271 using Teuchos::rcp_dynamic_cast;
276 Teuchos::rcp_dynamic_cast<
const SpmdVectorSpaceBase<double> >(
277 unwrapSingleProductVectorSpace(range_in),
281 Teuchos::rcp_dynamic_cast<
const ScalarProdVectorSpaceBase<double> >(
282 unwrapSingleProductVectorSpace(domain_in),
289 if (!epetra_mv.
get() )
290 return Teuchos::null;
292 domain = rcp_dynamic_cast<
const ScalarProdVectorSpaceBase<double> >(
297 double *localValues;
int leadingDim;
298 if( epetra_mv->ConstantStride() ) {
299 epetra_mv->ExtractView( &localValues, &leadingDim );
307 new DefaultSpmdMultiVector<double>(
310 Teuchos::arcp(localValues,0,leadingDim*epetra_mv->NumVectors(),
false),
314 Teuchos::set_extra_data<RCP<Epetra_MultiVector> >(
315 epetra_mv,
"Epetra_MultiVector", Teuchos::inOutArg(mv) );
323 const RCP<
const VectorSpaceBase<double> > &range_in,
324 const RCP<
const VectorSpaceBase<double> > &domain_in
327 using Teuchos::rcp_dynamic_cast;
332 Teuchos::rcp_dynamic_cast<
const SpmdVectorSpaceBase<double> >(
333 unwrapSingleProductVectorSpace(range_in),
337 Teuchos::rcp_dynamic_cast<
const ScalarProdVectorSpaceBase<double> >(
338 unwrapSingleProductVectorSpace(domain_in),
345 if (!epetra_mv.
get())
346 return Teuchos::null;
348 domain = rcp_dynamic_cast<
const ScalarProdVectorSpaceBase<double> >(
353 double *localValues;
int leadingDim;
354 if( epetra_mv->ConstantStride() ) {
355 epetra_mv->ExtractView( &localValues, &leadingDim );
363 new DefaultSpmdMultiVector<double>(
366 Teuchos::arcp(localValues,0,leadingDim*epetra_mv->NumVectors(),
false),
370 Teuchos::set_extra_data<RCP<const Epetra_MultiVector> >(
371 epetra_mv,
"Epetra_MultiVector", Teuchos::inOutArg(mv) );
381 using Teuchos::ptrFromRef;
382 using Teuchos::ptr_dynamic_cast;
385 using Teuchos::MpiComm;
388 const Ptr<const Teuchos::Comm<Ordinal> > comm = Teuchos::ptrFromRef(comm_in);
390 const Ptr<const SerialComm<Ordinal> > serialComm =
391 ptr_dynamic_cast<
const SerialComm<Ordinal> >(comm);
397 const Ptr<const MpiComm<Ordinal> > mpiComm =
398 ptr_dynamic_cast<
const MpiComm<Ordinal> >(comm);
402 "SPMD std::vector space has a communicator that is "
403 "neither a serial comm nor an MPI comm");
406 epetraComm =
rcp(
new Epetra_MpiComm(*mpiComm->getRawMpiComm()()));
409 epetraComm =
rcp(
new Epetra_SerialComm());
415 "SPMD std::vector space has a communicator that is "
416 "neither a serial comm nor an MPI comm");
418 epetraComm =
rcp(
new Epetra_SerialComm());
423 "null communicator created");
432 const VectorSpaceBase<double>& vs_in,
436 using Teuchos::rcpFromRef;
437 using Teuchos::rcpFromPtr;
438 using Teuchos::rcp_dynamic_cast;
439 using Teuchos::ptrFromRef;
440 using Teuchos::ptr_dynamic_cast;
442 const Ptr<const VectorSpaceBase<double> > vs_ptr = ptrFromRef(vs_in);
444 const Ptr<const SpmdVectorSpaceBase<double> > spmd_vs =
445 ptr_dynamic_cast<
const SpmdVectorSpaceBase<double> >(vs_ptr);
447 const Ptr<const ProductVectorSpaceBase<double> > &prod_vs =
448 ptr_dynamic_cast<
const ProductVectorSpaceBase<double> >(vs_ptr);
451 "Error, the concrete VectorSpaceBase object of type "
453 " SpmdVectorSpaceBase or the ProductVectorSpaceBase interfaces!" );
455 const int numBlocks = (
nonnull(prod_vs) ? prod_vs->numBlocks() : 1);
459 Array<RCP<const SpmdVectorSpaceBase<double> > > spmd_vs_blocks;
461 for (
int block_i = 0; block_i < numBlocks; ++block_i) {
463 rcp_dynamic_cast<
const SpmdVectorSpaceBase<double> >(
464 prod_vs->getBlock(block_i),
true);
465 spmd_vs_blocks.push_back(spmd_vs_i);
469 spmd_vs_blocks.push_back(rcpFromPtr(spmd_vs));
474 int myLocalElements = 0;
475 for (
int block_i = 0; block_i < numBlocks; ++block_i) {
476 myLocalElements += spmd_vs_blocks[block_i]->localSubDim();
484 for (
int block_i = 0; block_i < numBlocks; ++block_i) {
486 const int lowGIDInBlock = spmd_vs_i->localOffset();
487 const int numLocalElementsInBlock = spmd_vs_i->localSubDim();
488 for (
int i=0; i < numLocalElementsInBlock; ++i, ++count) {
489 myGIDs[count] = blockOffset + lowGIDInBlock + i;
491 blockOffset += spmd_vs_i->dim();
494 const int globalDim = vs_in.dim();
497 new Epetra_Map(globalDim, myLocalElements, myGIDs.
getRawPtr(), 0, *comm));
505 const RCP<
const VectorSpaceBase<double>>& vs,
512 const Ptr<const RCP<const Epetra_Map> >
513 epetra_map_ptr = Teuchos::get_optional_extra_data<RCP<const Epetra_Map> >(
518 return *epetra_map_ptr;
523 "Error! No RCP Epetra_Map attached to the input vector space RCP, "
524 "and the input comm RCP is null.\n");
531 const Epetra_Map &map,
532 const RCP<VectorBase<double> > &v
535 using Teuchos::get_optional_extra_data;
539 THYRA_ASSERT_VEC_SPACES(
540 "Thyra::get_Epetra_Vector(map,v)", *epetra_vs, *v->space() );
546 const Ptr<const RCP<Epetra_Vector> >
547 epetra_v_ptr = get_optional_extra_data<RCP<Epetra_Vector> >(
552 return *epetra_v_ptr;
560 const VectorSpaceBase<double> &vs = *v->range();
561 const SpmdVectorSpaceBase<double> *mpi_vs
562 =
dynamic_cast<const SpmdVectorSpaceBase<double>*
>(&vs);
563 const Ordinal localOffset = ( mpi_vs ? mpi_vs->localOffset() : 0 );
564 const Ordinal localSubDim = ( mpi_vs ? mpi_vs->localSubDim() : vs.dim() );
580 new DetachedVectorView<double>(
582 ,Range1D(localOffset,localOffset+localSubDim-1)
593 ,const_cast<double*>(emvv->values())
601 Teuchos::set_extra_data( emvv,
"emvv", Teuchos::inOutArg(epetra_v),
611 const RCP<VectorBase<double> > &v,
619 const Ptr<const RCP<Epetra_Vector> >
620 epetra_v_ptr = Teuchos::get_optional_extra_data<RCP<Epetra_Vector> >(
625 return *epetra_v_ptr;
630 "Error! No RCP Epetra_Vector attached to the input vector RCP, "
631 "and the input map RCP is null.\n");
638 const Epetra_Map &map,
639 const RCP<
const VectorBase<double> > &v
642 using Teuchos::get_optional_extra_data;
646 THYRA_ASSERT_VEC_SPACES(
647 "Thyra::get_Epetra_Vector(map,v)", *epetra_vs, *v->space() );
653 const Ptr<const RCP<const Epetra_Vector> >
654 epetra_v_ptr = get_optional_extra_data<RCP<const Epetra_Vector> >(
659 return *epetra_v_ptr;
660 const Ptr<const RCP<Epetra_Vector> >
661 epetra_nonconst_v_ptr = get_optional_extra_data<RCP<Epetra_Vector> >(
665 if(!
is_null(epetra_nonconst_v_ptr))
666 return *epetra_nonconst_v_ptr;
670 const VectorSpaceBase<double> &vs = *v->range();
671 const SpmdVectorSpaceBase<double> *mpi_vs
672 =
dynamic_cast<const SpmdVectorSpaceBase<double>*
>(&vs);
673 const Ordinal localOffset = ( mpi_vs ? mpi_vs->localOffset() : 0 );
674 const Ordinal localSubDim = ( mpi_vs ? mpi_vs->localSubDim() : vs.dim() );
679 new ConstDetachedVectorView<double>(
681 ,Range1D(localOffset,localOffset+localSubDim-1)
692 ,const_cast<double*>(evv->values())
700 Teuchos::set_extra_data( evv,
"evv", Teuchos::inOutArg(epetra_v),
710 const RCP<
const VectorBase<double> > &v,
718 const Ptr<const RCP<const Epetra_Vector> >
719 epetra_v_ptr = Teuchos::get_optional_extra_data<RCP<const Epetra_Vector> >(
724 return *epetra_v_ptr;
729 "Error! No RCP to Epetra_Vector attached to the input vector RCP, "
730 "and the input map RCP is null.\n");
737 const Epetra_Map &map,
738 const RCP<MultiVectorBase<double> > &mv
741 using Teuchos::get_optional_extra_data;
745 THYRA_ASSERT_VEC_SPACES(
746 "Thyra::get_Epetra_MultiVector(map,mv)", *epetra_vs, *mv->range() );
752 const Ptr<const RCP<Epetra_MultiVector> >
753 epetra_mv_ptr = get_optional_extra_data<RCP<Epetra_MultiVector> >(
754 mv,
"Epetra_MultiVector");
758 return *epetra_mv_ptr;
766 const VectorSpaceBase<double> &vs = *mv->range();
767 const SpmdVectorSpaceBase<double> *mpi_vs
768 =
dynamic_cast<const SpmdVectorSpaceBase<double>*
>(&vs);
769 const Ordinal localOffset = ( mpi_vs ? mpi_vs->localOffset() : 0 );
770 const Ordinal localSubDim = ( mpi_vs ? mpi_vs->localSubDim() : vs.dim() );
783 new DetachedMultiVectorView<double>(
785 ,Range1D(localOffset,localOffset+localSubDim-1)
792 new Epetra_MultiVector(
795 ,const_cast<double*>(emmvv->values())
806 Teuchos::set_extra_data( emmvv,
"emmvv", Teuchos::inOutArg(epetra_mv),
809 Teuchos::set_extra_data( mv,
"mv", Teuchos::inOutArg(epetra_mv) );
818 const RCP<MultiVectorBase<double> > &mv,
826 const Ptr<const RCP<Epetra_MultiVector> >
827 epetra_mv_ptr = Teuchos::get_optional_extra_data<RCP<Epetra_MultiVector> >(
828 mv,
"Epetra_MultiVector");
832 return *epetra_mv_ptr;
837 "Error! No RCP to Epetra_MultiVector attached to the input vector RCP, "
838 "and the input map RCP is null.\n");
845 const Epetra_Map &map,
846 const RCP<
const MultiVectorBase<double> > &mv
849 using Teuchos::get_optional_extra_data;
855 THYRA_ASSERT_VEC_SPACES(
856 "Thyra::get_Epetra_MultiVector(map,mv)",
857 *epetra_vs, *mv->range() );
864 const Ptr<const RCP<const Epetra_MultiVector> >
866 = get_optional_extra_data<RCP<const Epetra_MultiVector> >(
867 mv,
"Epetra_MultiVector" );
871 return *epetra_mv_ptr;
877 const VectorSpaceBase<double> &vs = *mv->range();
878 const SpmdVectorSpaceBase<double> *mpi_vs
879 =
dynamic_cast<const SpmdVectorSpaceBase<double>*
>(&vs);
880 const Ordinal localOffset = ( mpi_vs ? mpi_vs->localOffset() : 0 );
881 const Ordinal localSubDim = ( mpi_vs ? mpi_vs->localSubDim() : vs.dim() );
886 new ConstDetachedMultiVectorView<double>(
888 ,Range1D(localOffset,localOffset+localSubDim-1)
896 new Epetra_MultiVector(
899 ,const_cast<double*>(emvv->values())
910 Teuchos::set_extra_data( emvv,
"emvv", Teuchos::inOutArg(epetra_mv),
913 Teuchos::set_extra_data( mv,
"mv", Teuchos::inOutArg(epetra_mv) );
923 const RCP<
const MultiVectorBase<double> > &mv,
931 const Ptr<const RCP<const Epetra_MultiVector> >
932 epetra_mv_ptr = Teuchos::get_optional_extra_data<RCP<const Epetra_MultiVector> >(
933 mv,
"Epetra_MultiVector");
937 return *epetra_mv_ptr;
942 "Error! No RCP to Epetra_MultiVector attached to the input vector RCP, "
943 "and the input map RCP is null.\n");
950 const Epetra_Map &map,
951 MultiVectorBase<double> &mv
954 using Teuchos::rcpWithEmbeddedObj;
955 using Teuchos::ptrFromRef;
956 using Teuchos::ptr_dynamic_cast;
957 using Teuchos::outArg;
959 Ptr<SpmdMultiVectorBase<double> > mvSpmdMv =
960 ptr_dynamic_cast<SpmdMultiVectorBase<double> >(ptrFromRef(mv));
962 ArrayRCP<double> mvData;
965 mvSpmdMv->getNonconstLocalData(outArg(mvData), outArg(mvLeadingDim));
968 return rcpWithEmbeddedObj(
969 new Epetra_MultiVector(
970 ::
View, map, mvData.getRawPtr(), mvLeadingDim, mv.domain()->dim()
979 const Epetra_Map &map,
980 const MultiVectorBase<double> &mv
983 using Teuchos::rcpWithEmbeddedObj;
984 using Teuchos::ptrFromRef;
985 using Teuchos::ptr_dynamic_cast;
986 using Teuchos::outArg;
988 Ptr<const SpmdMultiVectorBase<double> > mvSpmdMv =
989 ptr_dynamic_cast<
const SpmdMultiVectorBase<double> >(ptrFromRef(mv));
991 ArrayRCP<const double> mvData;
994 mvSpmdMv->getLocalData(outArg(mvData), outArg(mvLeadingDim));
997 return rcpWithEmbeddedObj(
998 new Epetra_MultiVector(
999 ::
View, map, const_cast<double*>(mvData.getRawPtr()), mvLeadingDim, mv.domain()->dim()