int score_train(int loop) {
int method=loop;
if(nmethods) {
if(loop>=nmethods) return 0;
method=methods[loop];
}
switch(method) {
case 0:
mavg();
break;
case 1:
uavg();
break;
case 2:
usertime();
break;
case 3:
usertimemovie();
break;
case 4:
movietime();
break;
case 5:
movietimeuser();
break;
case 6:
useravgmovie();
break;
case 7:
usercntmovie();
break;
case 8:
movieavguser();
break;
case 9:
moviecntuser();
break;
case 10:
timecorr2();
break;
case 11:
globalavg();
break;
default:
return 0;
}
return 1;
}
void tet_hp_cns::minvrt() {
int i,j,k,n,tind,msgn,sgn,sind,v0;
Array<FLT,2> spokemass;
int last_phase, mp_phase;
Array<double,1> lcl(NV), lclug(NV),lclres(NV),uavg(NV);
Array<TinyVector<double,MXGP>,2> P(NV,NV);
Array<TinyVector<double,MXGP>,1> u1d(NV),res1d(NV),temp1d(NV);
Array<TinyVector<double,MXTM>,1> ucoef(NV),rcoef(NV),tcoef(NV);
if (basis::tet(log2p).p > 2) {
*gbl->log << "cns minvrt only works for p = 1 and 2" << endl;
exit(4);
}
/* LOOP THROUGH EDGES */
if (basis::tet(log2p).em > 0) {
for(int eind = 0; eind<nseg;++eind) {
/* SUBTRACT SIDE CONTRIBUTIONS TO VERTICES */
for (k=0; k <basis::tet(log2p).em; ++k) {
for (i=0; i<2; ++i) {
v0 = seg(eind).pnt(i);
for(n=0;n<NV;++n)
gbl->res.v(v0,n) -= basis::tet(log2p).sfmv(i,k)*gbl->res.e(eind,k,n);
}
}
}
}
gbl->res.v(Range(0,npnt-1),Range::all()) *= gbl->vprcn(Range(0,npnt-1),Range::all())*basis::tet(log2p).vdiag;
/* LOOP THROUGH VERTICES */
for(int i=0;i<npnt;++i){
for(int n = 0; n < NV; ++n)
lclres(n) = gbl->res.v(i,n);
if(gbl->preconditioner == 0 || gbl->preconditioner == 1) {
for(int n = 0; n < NV; ++n)
lclug(n) = ug.v(i,n);
switch_variables(lclug,lclres);
for(int j=0;j<NV;++j){
FLT lcl0 = lclres(j);
for(int k=0;k<j;++k){
lcl0 -= gbl->vpreconditioner(i,j,k)*lclres(k);
}
lclres(j) = lcl0/gbl->vpreconditioner(i,j,j);
}
}
else {
int info,ipiv[NV];
Array<double,2> P(NV,NV);
for(int j=0;j<NV;++j)
for(int k=0;k<NV;++k)
P(j,k) = gbl->vpreconditioner(i,j,k);
GETRF(NV, NV, P.data(), NV, ipiv, info);
if (info != 0) {
*gbl->log << "DGETRF FAILED FOR CNS MINVRT" << std::endl;
sim::abort(__LINE__,__FILE__,gbl->log);
}
char trans[] = "T";
GETRS(trans,NV,1,P.data(),NV,ipiv,lclres.data(),NV,info);
}
for(int n = 0; n < NV; ++n)
gbl->res.v(i,n) = lclres(n);
}
for(last_phase = false, mp_phase = 0; !last_phase; ++mp_phase) {
pc0load(mp_phase,gbl->res.v.data());
pmsgpass(boundary::all_phased,mp_phase,boundary::symmetric);
last_phase = true;
last_phase &= pc0wait_rcv(mp_phase,gbl->res.v.data());
}
/* APPLY VERTEX DIRICHLET B.C.'S */
for(i=0;i<nfbd;++i)
hp_fbdry(i)->vdirichlet();
for(i=0;i<nebd;++i)
hp_ebdry(i)->vdirichlet3d();
for(i=0;i<nvbd;++i)
hp_vbdry(i)->vdirichlet3d();
if(basis::tet(log2p).em == 0) return;
/* LOOP THROUGH SIDES */
for(int sind=0;sind<nseg;++sind) {
for(int n = 0; n < NV; ++n)
lclres(n) = gbl->res.e(sind,0,n);
Array<FLT,2> P(NV,NV);
for(int j=0;j<NV;++j){
for(int k=0;k<NV;++k){
P(j,k) = gbl->epreconditioner(sind,j,k);
//P(j,k) = 0.5*(gbl->vpreconditioner(seg(sind).pnt(0),j,k)+gbl->vpreconditioner(seg(sind).pnt(1),j,k));
}
}
if(gbl->preconditioner == 0 || gbl->preconditioner == 1) {
for(int n = 0; n < NV; ++n)
uavg(n) = 0.5*(ug.v(seg(sind).pnt(0),n)+ug.v(seg(sind).pnt(1),n));
switch_variables(uavg,lclres);
for(int j=0;j<NV;++j){
FLT lcl0 = lclres(j);
for(int k=0;k<j;++k){
lcl0 -= P(j,k)*lclres(k);
}
lclres(j) = lcl0/P(j,j);
}
}
else {
int info,ipiv[NV];
GETRF(NV, NV, P.data(), NV, ipiv, info);
if (info != 0) {
*gbl->log << "DGETRF FAILED FOR CNS MINVRT EDGE" << std::endl;
sim::abort(__LINE__,__FILE__,gbl->log);
}
char trans[] = "T";
GETRS(trans,NV,1,P.data(),NV,ipiv,lclres.data(),NV,info);
}
for(int n = 0; n < NV; ++n)
gbl->res.e(sind,0,n) = lclres(n);
}
/* REMOVE VERTEX CONTRIBUTION FROM SIDE MODES */
/* SOLVE FOR SIDE MODES */
/* PART 1 REMOVE VERTEX CONTRIBUTIONS */
for(tind=0;tind<ntet;++tind) {
for(i=0;i<4;++i) {
v0 = tet(tind).pnt(i);
for(n=0;n<NV;++n)
uht(n)(i) = gbl->res.v(v0,n)*gbl->iprcn(tind,n);
}
/* edges */
for(i=0;i<6;++i) {
sind = tet(tind).seg(i);
sgn = tet(tind).sgn(i);
for(j=0;j<4;++j) {
msgn = 1;
for(k=0;k<basis::tet(log2p).em;++k) {
for(n=0;n<NV;++n)
gbl->res.e(sind,k,n) -= msgn*basis::tet(log2p).vfms(j,4+k+i*basis::tet(log2p).em)*uht(n)(j);
msgn *= sgn;
}
}
}
}
basis::tet(log2p).ediag(0) = 100.0;//for fast convergence
//basis::tet(log2p).ediag(0) = 48.0; //for accuracy mass lumped edge modes
gbl->res.e(Range(0,nseg-1),0,Range::all()) *= gbl->eprcn(Range(0,nseg-1),Range::all())*basis::tet(log2p).ediag(0);
for(last_phase = false, mp_phase = 0; !last_phase; ++mp_phase) {
sc0load(mp_phase,gbl->res.e.data(),0,0,gbl->res.e.extent(secondDim));
smsgpass(boundary::all_phased,mp_phase,boundary::symmetric);
last_phase = true;
last_phase &= sc0wait_rcv(mp_phase,gbl->res.e.data(),0,0,gbl->res.e.extent(secondDim));
}
/* APPLY DIRCHLET B.C.S TO MODE */
for(int i=0;i<nfbd;++i)
hp_fbdry(i)->edirichlet();
for (int i=0;i<nebd;++i)
hp_ebdry(i)->edirichlet3d();
return;
}
