#include "TQpProbDens.h"
#include "TMatrixD.h"
ClassImp(TQpProbDens)
TQpProbDens::TQpProbDens(Int_t nx,Int_t my,Int_t mz) :
TQpProbBase(nx,my,mz)
{
R__ASSERT(nx-my-mz > 0);
}
TQpProbDens::TQpProbDens(const TQpProbDens &another) : TQpProbBase(another)
{
*this = another;
}
TQpDataBase *TQpProbDens::MakeData(Double_t *c,
Double_t *Q,
Double_t *xlo,Bool_t *ixlo,
Double_t *xup,Bool_t *ixup,
Double_t *A, Double_t *bA,
Double_t *C,
Double_t *clo,Bool_t *iclo,
Double_t *cup,Bool_t *icup)
{
TVectorD vc ; vc .Use(fNx,c);
TMatrixDSym mQ ; mQ .Use(fNx,Q);
TVectorD vxlo; vxlo.Use(fNx,xlo);
TVectorD vxup; vxup.Use(fNx,xup);
TMatrixD mA ;
TVectorD vbA ;
if (fMy > 0) {
mA .Use(fMy,fNx,A);
vbA .Use(fMy,bA);
}
TMatrixD mC ;
TVectorD vclo;
TVectorD vcup;
if (fMz > 0) {
mC .Use(fMz,fNx,C);
vclo.Use(fMz,clo);
vcup.Use(fMz,cup);
}
TVectorD vixlo(fNx);
TVectorD vixup(fNx);
for (Int_t ix = 0; ix < fNx; ix++) {
vixlo[ix] = (ixlo[ix]) ? 1.0 : 0.0;
vixup[ix] = (ixup[ix]) ? 1.0 : 0.0;
}
TVectorD viclo(fMz);
TVectorD vicup(fMz);
for (Int_t ic = 0; ic < fMz; ic++) {
viclo[ic] = (iclo[ic]) ? 1.0 : 0.0;
vicup[ic] = (icup[ic]) ? 1.0 : 0.0;
}
TQpDataDens *data = new TQpDataDens(vc,mQ,vxlo,vixlo,vxup,vixup,mA,vbA,mC,vclo,
viclo,vcup,vicup);
return data;
}
TQpDataBase *TQpProbDens::MakeData(TVectorD &c,
TMatrixDBase &Q_in,
TVectorD &xlo, TVectorD &ixlo,
TVectorD &xup, TVectorD &ixup,
TMatrixDBase &A_in,TVectorD &bA,
TMatrixDBase &C_in,
TVectorD &clo, TVectorD &iclo,
TVectorD &cup, TVectorD &icup)
{
TMatrixDSym &mQ = (TMatrixDSym &) Q_in;
TMatrixD &mA = (TMatrixD &) A_in;
TMatrixD &mC = (TMatrixD &) C_in;
R__ASSERT(mQ.GetNrows() == fNx && mQ.GetNcols() == fNx);
if (fMy > 0) R__ASSERT(mA.GetNrows() == fMy && mA.GetNcols() == fNx);
else R__ASSERT(mA.GetNrows() == fMy);
if (fMz > 0) R__ASSERT(mC.GetNrows() == fMz && mC.GetNcols() == fNx);
else R__ASSERT(mC.GetNrows() == fMz);
R__ASSERT(c.GetNrows() == fNx);
R__ASSERT(xlo.GetNrows() == fNx);
R__ASSERT(ixlo.GetNrows() == fNx);
R__ASSERT(xup.GetNrows() == fNx);
R__ASSERT(ixup.GetNrows() == fNx);
R__ASSERT(bA.GetNrows() == fMy);
R__ASSERT(clo.GetNrows() == fMz);
R__ASSERT(iclo.GetNrows() == fMz);
R__ASSERT(cup.GetNrows() == fMz);
R__ASSERT(icup.GetNrows() == fMz);
TQpDataDens *data = new TQpDataDens(c,mQ,xlo,ixlo,xup,ixup,mA,bA,mC,clo,iclo,cup,icup);
return data;
}
TQpResidual* TQpProbDens::MakeResiduals(const TQpDataBase *data_in)
{
TQpDataDens *data = (TQpDataDens *) data_in;
return new TQpResidual(fNx,fMy,fMz,data->fXloIndex,data->fXupIndex,data->fCloIndex,data->fCupIndex);
}
TQpVar* TQpProbDens::MakeVariables(const TQpDataBase *data_in)
{
TQpDataDens *data = (TQpDataDens *) data_in;
return new TQpVar(fNx,fMy,fMz,data->fXloIndex,data->fXupIndex,data->fCloIndex,data->fCupIndex);
}
TQpLinSolverBase* TQpProbDens::MakeLinSys(const TQpDataBase *data_in)
{
TQpDataDens *data = (TQpDataDens *) data_in;
return new TQpLinSolverDens(this,data);
}
void TQpProbDens::JoinRHS(TVectorD &rhs,TVectorD &rhs1_in,TVectorD &rhs2_in,TVectorD &rhs3_in)
{
rhs.SetSub(0,rhs1_in);
if (fMy > 0) rhs.SetSub(fNx, rhs2_in);
if (fMz > 0) rhs.SetSub(fNx+fMy,rhs3_in);
}
void TQpProbDens::SeparateVars(TVectorD &x_in,TVectorD &y_in,TVectorD &z_in,TVectorD &vars_in)
{
x_in = vars_in.GetSub(0,fNx-1);
if (fMy > 0) y_in = vars_in.GetSub(fNx, fNx+fMy-1);
if (fMz > 0) z_in = vars_in.GetSub(fNx+fMy,fNx+fMy+fMz-1);
}
void TQpProbDens::MakeRandomData(TQpDataDens *&data,TQpVar *&soln,Int_t ,Int_t ,Int_t )
{
data = new TQpDataDens(fNx,fMy,fMz);
soln = this->MakeVariables(data);
data->DataRandom(soln->fX,soln->fY,soln->fZ,soln->fS);
}
TQpProbDens &TQpProbDens::operator=(const TQpProbDens &source)
{
if (this != &source) {
TQpProbBase::operator=(source);
}
return *this;
}
ROOT page - Class index - Class Hierarchy - Top of the page
This page has been automatically generated. If you have any comments or suggestions about the page layout send a mail to ROOT support, or contact the developers with any questions or problems regarding ROOT.