void read_init(char *infile, ELEMS *p1, ELEMS *p2,double *m,int *origin) {
char c_origin[24];
FILE *fp;
fp=fopen(infile,"r");
fscanf(fp,"%lf",&m[0]);
fscanf(fp,"%lf %lf %lf %lf %lf %lf %lf",&m[1],&p1->a,&p1->e,&p1->i,&p1->aper,&p1->lasc,&p1->mean_an);
fscanf(fp,"%lf %lf %lf %lf %lf %lf %lf",&m[2],&p2->a,&p2->e,&p2->i,&p2->aper,&p2->lasc,&p2->mean_an);
fscanf(fp,"%s",c_origin);
if (memcmp(sLower(c_origin),"ba",2) == 0)
*origin = 0;
else if (memcmp(sLower(c_origin),"bo",2) == 0)
*origin = 1;
else {
fprintf(stderr,"ERROR: Unknown coordinate system. Options are barycentric or bodycentric.\n");
exit(1);
}
/* Convert to cgs, rads */
m[0] *= MSUN;
m[1] *= MJUP;
m[2] *= MJUP;
p1->a *= AUCM;
p1->aper *= M_PI/180;
p1->i *= M_PI/180;
p1->lasc *= M_PI/180;
p2->a *= AUCM;
p2->aper *= M_PI/180;
p2->i *= M_PI/180;
p2->lasc *= M_PI/180;
}
void PushPairInto
( DistMultiVec<Real>& s,
DistMultiVec<Real>& z,
const DistMultiVec<Real>& w,
const DistMultiVec<Int>& orders,
const DistMultiVec<Int>& firstInds,
Real wMaxNormLimit, Int cutoff )
{
DEBUG_ONLY(CSE cse("soc::PushPairInto"))
DistMultiVec<Real> sLower(s.Comm()), zLower(z.Comm());
soc::LowerNorms( s, sLower, orders, firstInds, cutoff );
soc::LowerNorms( z, zLower, orders, firstInds, cutoff );
const int localHeight = s.LocalHeight();
for( Int iLoc=0; iLoc<localHeight; ++iLoc )
{
const Int i = s.GlobalRow(iLoc);
const Real w0 = w.GetLocal(iLoc,0);
if( i == firstInds.GetLocal(iLoc,0) && w0 > wMaxNormLimit )
{
// TODO: Switch to a non-adhoc modification
s.UpdateLocal( iLoc, 0, Real(1)/wMaxNormLimit );
z.UpdateLocal( iLoc, 0, Real(1)/wMaxNormLimit );
}
}
}
void PushPairInto
( ElementalMatrix<Real>& sPre,
ElementalMatrix<Real>& zPre,
const ElementalMatrix<Real>& wPre,
const ElementalMatrix<Int>& ordersPre,
const ElementalMatrix<Int>& firstIndsPre,
Real wMaxNormLimit, Int cutoff )
{
DEBUG_ONLY(CSE cse("soc::PushPairInto"))
AssertSameGrids( sPre, zPre, wPre, ordersPre, firstIndsPre );
ElementalProxyCtrl ctrl;
ctrl.colConstrain = true;
ctrl.colAlign = 0;
DistMatrixReadWriteProxy<Real,Real,VC,STAR>
sProx( sPre, ctrl ),
zProx( zPre, ctrl );
DistMatrixReadProxy<Real,Real,VC,STAR>
wProx( wPre, ctrl );
DistMatrixReadProxy<Int,Int,VC,STAR>
ordersProx( ordersPre, ctrl ),
firstIndsProx( firstIndsPre, ctrl );
auto& s = sProx.Get();
auto& z = zProx.Get();
auto& w = wProx.GetLocked();
auto& orders = ordersProx.GetLocked();
auto& firstInds = firstIndsProx.GetLocked();
DistMatrix<Real,VC,STAR> sLower(s.Grid()), zLower(z.Grid());
soc::LowerNorms( s, sLower, orders, firstInds, cutoff );
soc::LowerNorms( z, zLower, orders, firstInds, cutoff );
const Int localHeight = s.LocalHeight();
for( Int iLoc=0; iLoc<localHeight; ++iLoc )
{
const Int i = s.GlobalRow(iLoc);
const Real w0 = w.GetLocal(iLoc,0);
if( i == firstInds.GetLocal(iLoc,0) && w0 > wMaxNormLimit )
{
// TODO: Switch to a non-adhoc modification
s.UpdateLocal( iLoc, 0, Real(1)/wMaxNormLimit );
z.UpdateLocal( iLoc, 0, Real(1)/wMaxNormLimit );
}
}
}
