void EventDriven::computeJacobianfx(OneStepIntegrator& osi,
integer *sizeOfX,
doublereal *time,
doublereal *x,
doublereal *jacob)
{
assert((osi.getType() == OSI::LSODAROSI) &&
"EventDriven::computeJacobianfx(osi, ...), not yet implemented for a one step integrator of type " + osi.getType());
LsodarOSI& lsodar = static_cast<LsodarOSI&>(osi);
// Remark A: according to DLSODAR doc, each call to jacobian is
// preceded by a call to f with the same arguments NEQ, T, and Y.
// Thus to gain some efficiency, intermediate quantities shared by
// both calculations may be saved in class members? fill in xWork
// vector (ie all the x of the ds of this osi) with x fillXWork(x);
// -> copy
// Maybe this step is not necessary? because of
// remark A above
// Compute the jacobian of the vector field according to x for the
// current ds
double t = *time;
lsodar.computeJacobianRhs(t, *_DSG0);
// Save jacobianX values from dynamical system into current jacob
// (in-out parameter)
unsigned int i = 0;
unsigned pos = 0;
DynamicalSystemsGraph::VIterator dsi, dsend;
SP::DynamicalSystemsGraph osiDSGraph = lsodar.dynamicalSystemsGraph();
for (std11::tie(dsi, dsend) = osiDSGraph->vertices(); dsi != dsend; ++dsi)
{
if (!(lsodar.checkOSI(dsi))) continue;
DynamicalSystem& ds = *(osiDSGraph->bundle(*dsi));
Type::Siconos dsType = Type::value(ds);
if (dsType == Type::LagrangianDS || dsType == Type::LagrangianLinearTIDS)
{
LagrangianDS& lds = static_cast<LagrangianDS&>(ds);
BlockMatrix& jacotmp = *lds.jacobianRhsx();
for (unsigned int j = 0; j < lds.n(); ++j)
{
for (unsigned int k = 0; k < lds.dimension(); ++k)
jacob[i++] = jacotmp(k, j);
}
}
else if(dsType == Type::FirstOrderNonLinearDS || dsType == Type::FirstOrderLinearDS
|| dsType == Type::FirstOrderLinearTIDS)
{
SimpleMatrix& jacotmp = static_cast<SimpleMatrix&>(*(ds.jacobianRhsx())); // Pointer link !
pos += jacotmp.copyData(&jacob[pos]);
}
else
{
RuntimeException::selfThrow("EventDriven::computeJacobianfx, type of DynamicalSystem not yet supported.");
}
}
}
void EventDriven::initOSIs()
{
for (OSIIterator itosi = _allOSI->begin(); itosi != _allOSI->end(); ++itosi)
{
// Initialize the acceleration like for NewMarkAlphaScheme
if ((*itosi)->getType() == OSI::NEWMARKALPHAOSI)
{
SP::NewMarkAlphaOSI osi_NewMark = std11::static_pointer_cast<NewMarkAlphaOSI>(*itosi);
DynamicalSystemsGraph::VIterator dsi, dsend;
SP::DynamicalSystemsGraph osiDSGraph = (*itosi)->dynamicalSystemsGraph();
for (std11::tie(dsi, dsend) = osiDSGraph->vertices(); dsi != dsend; ++dsi)
{
if (!(*itosi)->checkOSI(dsi)) continue;
SP::DynamicalSystem ds = osiDSGraph->bundle(*dsi);
if ((Type::value(*ds) == Type::LagrangianDS) || (Type::value(*ds) == Type::LagrangianLinearTIDS))
{
SP::LagrangianDS d = std11::static_pointer_cast<LagrangianDS>(ds);
*(d->workspace(DynamicalSystem::acce_like)) = *(d->acceleration()); // set a0 = ddotq0
// Allocate the memory to stock coefficients of the polynomial for the dense output
d->allocateWorkMatrix(LagrangianDS::coeffs_denseoutput, ds->dimension(), (osi_NewMark->getOrderDenseOutput() + 1));
}
}
}
}
}
std::vector<SP::DynamicalSystem> dynamicalSystems(SP::DynamicalSystemsGraph dsg)
{
std::vector<SP::DynamicalSystem> r = std::vector<SP::DynamicalSystem>();
DynamicalSystemsGraph::VIterator vi, viend;
for (boost::tie(vi, viend) = dsg->vertices(); vi != viend; ++vi)
{
r.push_back(dsg->bundle(*vi));
};
return r;
};
void EventDriven::initOSIRhs()
{
// === initialization for OneStepIntegrators ===
OSI::TYPES osiType = (*_allOSI->begin())->getType();
for (OSIIterator itosi = _allOSI->begin(); itosi != _allOSI->end(); ++itosi)
{
//Check whether OSIs used are of the same type
if ((*itosi)->getType() != osiType)
RuntimeException::selfThrow("OSIs used must be of the same type");
// perform the initialization
DynamicalSystemsGraph::VIterator dsi, dsend;
SP::DynamicalSystemsGraph osiDSGraph = (*itosi)->dynamicalSystemsGraph();
for (std11::tie(dsi, dsend) = osiDSGraph->vertices(); dsi != dsend; ++dsi)
{
if (!(*itosi)->checkOSI(dsi)) continue;
SP::DynamicalSystem ds = osiDSGraph->bundle(*dsi);
// Initialize right-hand side
ds->initRhs(startingTime());
}
}
}
void Simulation::initialize(SP::Model m, bool withOSI)
{
// === Connection with the model ===
assert(m && "Simulation::initialize(model) - model = NULL.");
_model = std11::weak_ptr<Model>(m);
_T = m->finalT();
// === Events manager initialization ===
_eventsManager->initialize(_T);
_tinit = _eventsManager->startingTime();
//===
if (withOSI)
{
if (numberOfOSI() == 0)
RuntimeException::selfThrow("Simulation::initialize No OSI !");
// === OneStepIntegrators initialization ===
for (OSIIterator itosi = _allOSI->begin();
itosi != _allOSI->end(); ++itosi)
{
for (DSIterator itds = (*itosi)->dynamicalSystems()->begin();
itds != (*itosi)->dynamicalSystems()->end();
++itds)
{
(*itds)->initialize(model()->t0(),
(*itosi)->getSizeMem());
addInOSIMap(*itds, *itosi);
}
(*itosi)->setSimulationPtr(shared_from_this());
(*itosi)->initialize();
}
}
// This is the default
_levelMinForInput = LEVELMAX;
_levelMaxForInput = 0;
_levelMinForOutput = LEVELMAX;
_levelMaxForOutput = 0;
computeLevelsForInputAndOutput();
// Loop over all DS in the graph, to reset NS part of each DS.
// Note FP : this was formerly done in inter->initialize call with local levels values
// but I think it's ok (better?) to do it with the simulation levels values.
DynamicalSystemsGraph::VIterator dsi, dsend;
SP::DynamicalSystemsGraph DSG = model()->nonSmoothDynamicalSystem()->topology()->dSG(0);
for (std11::tie(dsi, dsend) = DSG->vertices(); dsi != dsend; ++dsi)
{
//assert(_levelMinForInput <= _levelMaxForInput);
for (unsigned int k = _levelMinForInput ; k < _levelMaxForInput + 1; k++)
{
DSG->bundle(*dsi)->initializeNonSmoothInput(k);
}
}
InteractionsGraph::VIterator ui, uiend;
SP::InteractionsGraph indexSet0 = model()->nonSmoothDynamicalSystem()->topology()->indexSet0();
for (std11::tie(ui, uiend) = indexSet0->vertices(); ui != uiend; ++ui)
{
Interaction& inter = *indexSet0->bundle(*ui);
inter.initialize(_tinit, indexSet0->properties(*ui));
}
// Initialize OneStepNSProblem(s). Depends on the type of simulation.
// Warning FP : must be done in any case, even if the interactions set
// is empty.
initOSNS();
// Process events at time _tinit. Useful to save values in memories
// for example. Warning: can not be called during
// eventsManager->initialize, because it needs the initialization of
// OSI, OSNS ...
_eventsManager->preUpdate(*this);
_tend = _eventsManager->nextTime();
// Set Model current time (warning: current time of the model
// corresponds to the time of the next event to be treated).
model()->setCurrentTime(nextTime());
// End of initialize:
// - all OSI and OSNS (ie DS and Interactions) states are computed
// - for time _tinit and saved into memories.
// - Sensors or related objects are updated for t=_tinit.
// - current time of the model is equal to t1, time of the first
// - event after _tinit.
// - currentEvent of the simu. corresponds to _tinit and nextEvent
// - to _tend.
// If _printStat is true, open output file.
if (_printStat)
{
statOut.open("simulationStat.dat", std::ios::out | std::ios::trunc);
if (!statOut.is_open())
SiconosVectorException::selfThrow("writing error : Fail to open file simulationStat.dat ");
statOut << "============================================" <<std::endl;
statOut << " Siconos Simulation of type " << Type::name(*this) << "." <<std::endl;
statOut <<std::endl;
statOut << "The tolerance parameter is equal to: " << _tolerance <<std::endl;
statOut <<std::endl <<std::endl;
}
}
void EventDriven::computef(OneStepIntegrator& osi, integer * sizeOfX, doublereal * time, doublereal * x, doublereal * xdot)
{
// computeF is supposed to fill xdot in, using the definition of the
// dynamical systems belonging to the osi
// Check osi type: only lsodar is allowed.
assert((osi.getType() == OSI::LSODAROSI) && "EventDriven::computef(osi, ...), not yet implemented for a one step integrator of type " + osi.getType());
LsodarOSI& lsodar = static_cast<LsodarOSI&>(osi);
// fill in xWork vector (ie all the x of the ds of this osi) with x
lsodar.fillXWork(sizeOfX, x);
double t = *time;
// Update Jacobian matrices at all interactions
InteractionsGraph::VIterator ui, uiend;
for (std11::tie(ui, uiend) = _indexSet0->vertices(); ui != uiend; ++ui)
{
Interaction& inter = *_indexSet0->bundle(*ui);
inter.relation()->computeJach(t, inter, _indexSet0->properties(*ui));
}
// solve a LCP at "acceleration" level if required
if (!_allNSProblems->empty())
{
if (((*_allNSProblems)[SICONOS_OSNSP_ED_SMOOTH_ACC]->hasInteractions()))
{
// Update the state of the DS
(*_allNSProblems)[SICONOS_OSNSP_ED_SMOOTH_ACC]->compute(t);
_nsds->updateInput(t,2); // Necessary to compute DS state below
}
// Compute the right-hand side ( xdot = f + r in DS) for all the
//ds, with the new value of input. lsodar->computeRhs(t);
}
// update the DS of the OSI.
lsodar.computeRhs(t, *_DSG0);
// for the DS state, ie the ones computed by lsodar (x above)
// Update Index sets? No !!
// Get the required value, ie xdot for output.
unsigned pos = 0;
DynamicalSystemsGraph::VIterator dsi, dsend;
SP::DynamicalSystemsGraph osiDSGraph = lsodar.dynamicalSystemsGraph();
for (std11::tie(dsi, dsend) = osiDSGraph->vertices(); dsi != dsend; ++dsi)
{
if (!(lsodar.checkOSI(dsi))) continue;
DynamicalSystem& ds = *(osiDSGraph->bundle(*dsi));
Type::Siconos dsType = Type::value(ds);
if (dsType == Type::LagrangianDS || dsType == Type::LagrangianLinearTIDS)
{
LagrangianDS& LDS = static_cast<LagrangianDS&>(ds);
SiconosVector& qDotTmp = *LDS.velocity();
SiconosVector& qDotDotTmp = *LDS.acceleration();
pos += qDotTmp.copyData(&xdot[pos]);
pos += qDotDotTmp.copyData(&xdot[pos]);
}
else
{
SiconosVector& xtmp2 = ds.getRhs(); // Pointer link !
pos += xtmp2.copyData(&xdot[pos]);
}
}
}