Revision 5babd949c69f75d753463915963a9ac024849fc9 authored by Axel Naumann on 09 February 2015, 08:59:47 UTC, committed by Axel Naumann on 09 February 2015, 08:59:47 UTC
1 parent 041b8bd
QpRandomDriver.cxx
#include <stdlib.h>
#include "Riostream.h"
#include "TQpDataDens.h"
#include "TQpVar.h"
#include "TQpProbDens.h"
#include "TGondzioSolver.h"
Bool_t SolutionMatches(TQpVar *vars,TQpVar *soln,TQpVar *temp,Double_t tol);
int main(int argc,char *argv[])
{
Int_t n1 = 5, m1 = 2, m2 = 2;
if (argc >= 4) {
n1 = atoi(argv[1]);
m1 = atoi(argv[2]);
m2 = atoi(argv[3]);
} else {
std::cout << std::endl
<< " Usage: QpRandomDriver n my mz " << std::endl
<< " where n = # primal variables, " << std::endl
<< " my = # equality constraints, " << std::endl
<< " mz = # inequality constraints " << std::endl << std::endl;
return 1;
}
Int_t nnzQ = (Int_t) .20*(n1*n1);
Int_t nnzA = (Int_t) .15*(m1*n1);
Int_t nnzC = (Int_t) .10*(m2*n1);
if (nnzQ < 3*n1) nnzQ = 3*n1;
if (nnzA < 3*m1) nnzA = 3*m1;
if (nnzC < 3*m2) nnzC = 2*m2;
TQpProbDens *qp = new TQpProbDens(n1,m1,m2);
TQpDataDens *prob;
TQpVar *soln;
qp->MakeRandomData(prob,soln,0,0,0);
// qp->MakeRandomData(prob,soln,nnzQ,nnzA,nnzC);
TQpVar *vars = qp->MakeVariables(prob);
TQpResidual *resid = qp->MakeResiduals(prob);
TGondzioSolver *s = new TGondzioSolver(qp,prob);
const Int_t status = s->Solve(prob,vars,resid);
delete s;
if (status == 0) {
std::cout.precision(4);
std::cout << std::endl << "Computed solution:\n\n" <<std::endl <<std::endl;
vars->fX.Print();
TQpVar *temp = qp->MakeVariables(prob);
std::cout << std::endl << "Checking the solution...";
if (SolutionMatches(vars,soln,temp,1e-4)) {
std::cout << "The solution appears to be correct." <<std::endl;
} else {
std::cout << std::endl << "The solution may be wrong "
"(or the generated problem may be ill conditioned.)" <<std::endl;
}
delete temp;
} else {
std::cout << "Could not solve this problem." <<std::endl;
}
delete vars;
delete soln;
delete prob;
delete qp;
return status;
}
Bool_t SolutionMatches(TQpVar *vars,TQpVar *soln,TQpVar *temp,double tol)
{
temp = vars;
// Only x is significant
temp->fX -= soln->fX;
if ((temp->fX).NormInf()/(1+(soln->fX).NormInf()) < tol)
return kTRUE;
else
return kFALSE;
}
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...