void FirstOrderType1R::initComponents(Interaction& inter, VectorOfBlockVectors& DSlink, VectorOfVectors& workV, VectorOfSMatrices& workM)
{
// Check if an Interaction is connected to the Relation.
unsigned int sizeY = inter.getSizeOfY();
unsigned int sizeDS = inter.getSizeOfDS();
unsigned int sizeZ = DSlink[FirstOrderR::z]->size();
workV.resize(FirstOrderR::workVecSize);
workV[FirstOrderR::vec_z].reset(new SiconosVector(sizeZ));
workV[FirstOrderR::vec_x].reset(new SiconosVector(sizeDS));
workV[FirstOrderR::vec_r].reset(new SiconosVector(sizeDS));
workM.resize(FirstOrderR::mat_workMatSize);
if (!_C)
workM[FirstOrderR::mat_C].reset(new SimpleMatrix(sizeY, sizeDS));
if (!_D)
workM[FirstOrderR::mat_D].reset(new SimpleMatrix(sizeY, sizeY));
if (!_F)
workM[FirstOrderR::mat_F].reset(new SimpleMatrix(sizeY, sizeZ));
if (!_B)
workM[FirstOrderR::mat_B].reset(new SimpleMatrix(sizeDS, sizeY));
}
TEST(vectorTest, getNumCols) {
VectorOfVectors<int>* i = new VectorOfVectors<int>(5, 5, 0);
VectorOfVectors<double>* d = new VectorOfVectors<int>(5, 10, 0);
VectorOfVectors<std::string>* s = new VectorOfVectors<std::string>(5, 19, 0);
i->insert(2, 2, 7);
d->insert(4, 4, 3.145);
s->insert(0, 0, "Hello");
EXPECT_EQ(5, i->getNumCols());
EXPECT_EQ(10, d->getNumCols());
EXPECT_EQ(19, s->getNumCols());
delete i;
delete d;
delete s;
}
TEST(vectorTest, access) {
VectorOfVectors<int>* i = new VectorOfVectors<int>(5, 5, 0);
VectorOfVectors<double>* d = new VectorOfVectors<int>(5, 5, 0);
VectorOfVectors<std::string>* s = new VectorOfVectors<std::string>(5, 5, "World");
i->insert(2, 2, 7);
d->insert(4, 4, 3.145);
s->insert(0, 0, "Hello");
EXPECT_EQ(7, i->access(2, 2));
EXPECT_EQ(3.145, d->access(4, 4));
EXPECT_EQ("World", s->access(2, 2));
delete i;
delete d;
delete s;
}
TEST(vectorTest, print) {
VectorOfVectors<int>* i = new VectorOfVectors<int>(5, 5, 0);
VectorOfVectors<double>* d = new VectorOfVectors<int>(5, 5, 0);
VectorOfVectors<std::string>* s = new VectorOfVectors<std::string>(5, 5, "o");
i->insert(2, 2, 7);
d->insert(4, 4, 3.145);
s->insert(0, 0, "Hello");
i->print();
std::cout << std::endl;
d->print();
std::cout << std::endl;
s->print();
delete i;
delete d;
delete s;
}
Test(vectorTest, remove) {
VectorOfVectors<int>* i = new VectorOfVectors<int>(5, 5, 0);
VectorOfVectors<double>* d = new VectorOfVectors<int>(5, 5, 0);
VectorOfVectors<std::string>* s = new VectorOfVectors<std::string>(5, 5, "o");
i->insert(2, 2, 7);
d->insert(4, 4, 3.145);
s->insert(0, 0, "Hello");
i->insert(3, 3, 9);
i->remove(2, 2);
d->remove(4, 4);
s->remove(0, 0);
s->insert(0, 0, "y");
EXPECT_EQ(0, i->access(2, 2));
EXPECT_EQ(0, d->access(4, 4));
EXPECT_EQ("y", s->access(0, 0));
EXPECT_EQ(9, i->insert(3, 3));
delete i;
delete d;
delete s;
}
void FirstOrderR::initialize(Interaction& inter, VectorOfBlockVectors& DSlink, VectorOfVectors& workV, VectorOfSMatrices& workM)
{
workV.resize(FirstOrderR::workVecSize);
workM.resize(FirstOrderR::mat_workMatSize);
initComponents(inter, DSlink, workV, workM);
}
void FirstOrderLinearR::initComponents(Interaction& inter, VectorOfBlockVectors& DSlink, VectorOfVectors& workV, VectorOfSMatrices& workM)
{
// Note: do not call FirstOrderR::initialize to avoid jacobianH and jacobianG allocation.
// get interesting size
unsigned int sizeY = inter.getSizeOfY();
unsigned int sizeX = inter.getSizeOfDS();
unsigned int sizeZ = DSlink[FirstOrderR::z]->size();
// Update workV (copy of DS variables)
workV.resize(FirstOrderR::workVecSize);
workV[FirstOrderR::vec_z].reset(new SiconosVector(sizeZ));
workM.resize(FirstOrderR::mat_workMatSize);
if (!_C && _pluginJachx->fPtr)
workM[FirstOrderR::mat_C].reset(new SimpleMatrix(sizeY, sizeX));
if (!_D && _pluginJachlambda->fPtr)
workM[FirstOrderR::mat_D].reset(new SimpleMatrix(sizeY, sizeY));
if (!_B && _pluginJacglambda->fPtr)
workM[FirstOrderR::mat_B].reset(new SimpleMatrix(sizeX, sizeY));
if (!_F && _pluginf->fPtr)
{
workM[FirstOrderR::mat_F].reset(new SimpleMatrix(sizeY, sizeZ));
}
if (!_e && _plugine->fPtr)
{
workV[FirstOrderR::e].reset(new SiconosVector(sizeY));
}
// Check if various operators sizes are consistent.
// Reference: interaction.
if (_C)
{
if (_C->size(0) == 0)
_C->resize(sizeX, sizeY);
else
assert((_C->size(0) == sizeY && _C->size(1) == sizeX) && "FirstOrderLinearR::initialize , inconsistent size between C and Interaction.");
}
if (_B)
{
if (_B->size(0) == 0)
_B->resize(sizeY, sizeX);
else
assert((_B->size(1) == sizeY && _B->size(0) == sizeX) && "FirstOrderLinearR::initialize , inconsistent size between B and interaction.");
}
// C and B are the minimum inputs. The others may remain null.
if (_D)
{
if (_D->size(0) == 0)
_D->resize(sizeY, sizeY);
else
assert((_D->size(0) == sizeY || _D->size(1) == sizeY) && "FirstOrderLinearR::initialize , inconsistent size between C and D.");
}
if (_F)
{
if (_F->size(0) == 0)
_F->resize(sizeY, sizeZ);
else
assert(((_F->size(0) == sizeY) && (_F->size(1) == sizeZ)) && "FirstOrderLinearR::initialize , inconsistent size between C and F.");
}
if (_e)
{
if (_e->size() == 0)
_e->resize(sizeY);
else
assert(_e->size() == sizeY && "FirstOrderLinearR::initialize , inconsistent size between C and e.");
}
}
