TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL( SpmdLocalDataAccess,
getLocalSubVectorView_empty_p0, Scalar )
{
typedef typename ScalarTraits<Scalar>::magnitudeType ScalarMag;
const RCP<const DefaultSpmdVectorSpace<Scalar> > vs =
createZeroEleProcVS<Scalar>(g_localDim);
TEST_ASSERT(!vs->isLocallyReplicated());
const RCP<const Teuchos::Comm<Ordinal> > comm = vs->getComm();
const int procRank = comm->getRank();
PRINT_VAR(procRank);
const int numProcs = comm->getSize();
PRINT_VAR(numProcs);
const RCP<VectorBase<Scalar> > v = createMember<Scalar>(vs);
const Scalar val = as<Scalar>(1.5);
PRINT_VAR(val);
assign<Scalar>(v.ptr(), val);
out << "*** Test that we get the view correctly including an empty view on p0 ...\n";
RTOpPack::ConstSubVectorView<Scalar> lsv =
getLocalSubVectorView<Scalar>(v);
if (procRank == 0) {
TEST_EQUALITY_CONST(lsv.globalOffset(), 0);
TEST_EQUALITY_CONST(lsv.subDim(), 0);
TEST_EQUALITY_CONST(lsv.values(), null);
}
else {
TEST_EQUALITY(lsv.globalOffset(), as<Ordinal>((procRank-1)*g_localDim));
TEST_EQUALITY(lsv.subDim(), g_localDim);
}
TEST_EQUALITY_CONST(lsv.stride(), 1);
for (int k = 0; k < lsv.subDim(); ++k) {
TEST_EQUALITY(lsv[k], val);
}
}
RCP<const VectorBase<Scalar> >
DefaultSpmdVectorSpace<Scalar>::createMemberView(
const RTOpPack::ConstSubVectorView<Scalar> &raw_v
) const
{
#ifdef TEUCHOS_DEBUG
TEUCHOS_TEST_FOR_EXCEPT( localSubDim_ != raw_v.subDim() );
#endif
return Teuchos::rcp(
new DefaultSpmdVector<Scalar>(
weakSelfPtr_.create_strong(),
Teuchos::arcp(const_cast<Scalar*>(raw_v.values().get()),0,raw_v.subDim(),false),
raw_v.stride()
)
);
}
void VectorDefaultBase<Scalar>::acquireDetachedMultiVectorViewImpl(
const Range1D &rowRng,
const Range1D &colRng,
RTOpPack::ConstSubMultiVectorView<Scalar> *sub_mv
) const
{
#ifdef THYRA_VECTOR_VERBOSE_TO_ERROR_OUT
THYRA_VECTOR_VERBOSE_OUT_STATEMENT;
*dbgout << "\nThyra::VectorDefaultBase<"
<<Teuchos::ScalarTraits<Scalar>::name()
<<">::acquireDetachedMultiVectorViewImpl() const called!\n";
#endif
#ifdef TEUCHOS_DEBUG
TEUCHOS_TEST_FOR_EXCEPT(sub_mv==NULL);
#endif
validateColRng(colRng);
RTOpPack::ConstSubVectorView<Scalar> sv;
this->acquireDetachedView(rowRng,&sv);
#ifdef TEUCHOS_DEBUG
TEUCHOS_TEST_FOR_EXCEPT( sv.stride() != 1 ); // Can't handle non-unit stride yet but we could
#endif
sub_mv->initialize( sv.globalOffset(), sv.subDim(), 0, 1, sv.values(), sv.subDim() );
}