void
Space::_trycommit(const Choice& c, unsigned int a) {
if (a >= c.alternatives())
throw SpaceIllegalAlternative("Space::commit");
if (failed())
return;
if (Brancher* b = brancher(c._id)) {
// There is a matching brancher
if (b->commit(*this,c,a) == ES_FAILED)
fail();
}
}
NGL*
Space::ngl(const Choice& c, unsigned int a) {
if (a >= c.alternatives())
throw SpaceIllegalAlternative("Space::ngl");
if (failed())
return NULL;
if (Brancher* b = brancher(c._id)) {
// There is a matching brancher
return b->ngl(*this,c,a);
} else {
return NULL;
}
}
void brancher(Game *&parent, int player){
player %= PLAYERS; //This player moves now
vector<Position *> validMoves = findValidMoves(parent);
for(int i=0; i<(int)validMoves.size(); i++){
Position *move = validMoves[i];
Game *t = new Game;
t->gameState = parent->gameState;
t->parent = parent;
t->gameState[move->r][move->c] = player; //make move
parent->children.push_back(t);
if(checkState(t))//checks if game over (is there a point in continuing?)
brancher(t, player + 1);
}
}
/** Advance the walkers nblocks*nsteps timesteps.
* @param nblocks number of blocks
* @param nsteps number of steps
* @param tau the timestep
*
* For each timestep:
* <ul>
* <li> Move all the particles of a walker.
* <li> Calculate the properties for the new walker configuration.
* <li> Accept/reject the new configuration.
* <li> Accumulate the estimators.
* <li> Update the trial energy \f$ E_T. \f$
* <li> Branch the population of walkers (birth/death algorithm).
* </ul>
* For each block:
* <ul>
* <li> Flush the estimators and print to file.
* <li> Update the estimate of the local energy.
* <li> (Optional) Print the ensemble of walker configurations.
* </ul>
* Default mode: Print the ensemble of walker configurations
* at the end of the run.
*/
bool DMCWOS::run() {
//create a distance table for one walker
//DistanceTable::create(1);
//deltaR.resize(W.getTotalNum());
//drift.resize(W.getTotalNum());
//if(put(qmc_node)){
// extract the WOS potential and reset the parameters for DMC
wos_ref = dynamic_cast<WOSPotential*>(H.getHamiltonian("wos"));
if(wos_ref) {
cout << wos_ref->m_runs << '\t' << wos_ref->m_norm << endl;
wos_ref->set_mrun(wos_ref->dmc_runs);
cout << "Using " << wos_ref->m_runs << " runners/walker for DMC" << endl;
H.setTau(Tau);
Tau_var = Tau;
} else {
H.setTau(0.0);
Tau_var = 0.0;
}
MolecuFixedNodeBranch<RealType> brancher(Tau,W.getActiveWalkers());
brancher.put(qmc_node,LogOut);
if(BranchInfo != "default") brancher.read(BranchInfo);
//set the data members to start a new run
// getReady();
int PopIndex, E_TIndex;
Estimators.resetReportSettings(RootName);
AcceptIndex = Estimators.addColumn("AcceptRatio");
PopIndex = Estimators.addColumn("Population");
E_TIndex = Estimators.addColumn("E_T");
Estimators.reportHeader();
MCWalkerConfiguration::iterator it = W.begin();
MCWalkerConfiguration::iterator it_end = W.end();
while(it != it_end) {
(*it)->Properties(WEIGHT) = 1.0;
(*it)->Properties(MULTIPLICITY) = 1.0;
++it;
}
/*if VMC/DMC directly preceded DMC (Counter > 0) then
use the average value of the energy estimator for
the reference energy of the brancher*/
if(Counter) {
RealType e_ref = W.getLocalEnergy();
LOGMSG("Overwriting the reference energy by the local energy " << e_ref)
brancher.setEguess(e_ref);
}
IndexType block = 0;
Pooma::Clock timer;
int Population = W.getActiveWalkers();
int tPopulation = W.getActiveWalkers();
RealType Eest = brancher.E_T;
IndexType nAcceptTot = 0;
IndexType nRejectTot = 0;
do {
IndexType step = 0;
timer.start();
do {
Population = W.getActiveWalkers();
advanceWalkerByWalker(brancher);
step++; CurrentStep++;
Estimators.accumulate(W);
//E_T = brancher.update(Population,Eest);
Eest = brancher.update(Population,Eest);
brancher.branch(CurrentStep,W);
} while(step<nSteps);
timer.stop();
nAcceptTot += nAccept;
nRejectTot += nReject;
Estimators.flush();
Estimators.setColumn(PopIndex,static_cast<double>(Population));
Estimators.setColumn(E_TIndex,Eest);
Estimators.setColumn(AcceptIndex,
static_cast<double>(nAccept)/static_cast<double>(nAccept+nReject));
Estimators.report(CurrentStep);
LogOut->getStream() << "Block " << block << " " << timer.cpu_time()
<< " " << Population << endl;
Eest = Estimators.average(0);
nAccept = 0; nReject = 0;
block++;
if(pStride) {
//create an output engine: could accumulate the configurations
HDFWalkerOutput WO(RootName);
WO.get(W);
brancher.write(WO.getGroupID());
}
W.reset();
} while(block<nBlocks);
LogOut->getStream()
<< "ratio = " << static_cast<double>(nAcceptTot)/static_cast<double>(nAcceptTot+nRejectTot)
<< endl;
if(!pStride) {
//create an output engine: could accumulate the configurations
HDFWalkerOutput WO(RootName);
WO.get(W);
brancher.write(WO.getGroupID());
}
Estimators.finalize();
return true;
//} else
// return false;
}
int statement(){
symbol_t id;
if(sym->type==IF){
nextSym();
brancher();
}else if(sym->type==WHILE){
nextSym();
looper();
}else if(sym->type==LBRACE){
nextSym();
statementlist();
if(sym->type!=RBRACE){
msg(ERR, "missing \'}\'", line);
ERROR_STATUS = 1;
}
nextSym();
}else if(sym->type==ID){
id = copySym(sym);
nextSym();
if(sym->type==LPAREN){
nextSym();
funccall(id);
}else if(sym->type==ASN){
nextSym();
assign(id);
}else{
msg(ERR, "expect a \'=\' or \'(\'", line);
ERROR_STATUS = 1;
}
mfree(id);
if(sym->type!=SEMIC){
msg(ERR, "missing \';\'", line);
ERROR_STATUS = 1;
}
nextSym();
}else if(sym->type==PRINTF){
nextSym();
printfcall();
if(sym->type!=SEMIC){
msg(ERR, "missing \';\'", line);
ERROR_STATUS = 1;
}
nextSym();
}else if(sym->type==SCANF){
nextSym();
scanfcall();
if(sym->type!=SEMIC){
msg(ERR, "missing \';\'", line);
ERROR_STATUS = 1;
}
nextSym();
}else if(sym->type==SWITCH){
nextSym();
switcher();
}else if(sym->type==RETURN){
nextSym();
returns();
if(sym->type!=SEMIC){
msg(ERR, "missing \';\'", line);
ERROR_STATUS = 1;
}
nextSym();
}else return 0;// a 0 statement
return 1;
}
bool DMCParticleByParticle::run() {
//add columns
IndexType PopIndex = Estimators->addColumn("Population");
IndexType EtrialIndex = Estimators->addColumn("E_T");
//write the header
Estimators->reportHeader();
MolecuFixedNodeBranch<RealType> brancher(Tau,W.getActiveWalkers());
//initialize parameters for fixed-node branching
brancher.put(qmcNode,LogOut);
if(BranchInfo != "default") brancher.read(BranchInfo);
else {
/*if VMC/DMC directly preceded DMC (Counter > 0) then
use the average value of the energy estimator for
the reference energy of the brancher*/
if(Counter) {
RealType e_ref = W.getLocalEnergy();
LOGMSG("Overwriting the reference energy by the local energy " << e_ref)
brancher.setEguess(e_ref);
}
}
MCWalkerConfiguration::iterator it(W.begin());
MCWalkerConfiguration::iterator it_end(W.end());
while(it!=it_end) {
(*it)->Properties(WEIGHT) = 1.0;
(*it)->Properties(MULTIPLICITY) = 1.0;
++it;
}
//going to add routines to calculate how much we need
bool require_register = W.createAuxDataSet();
int iwalker=0;
it = W.begin();
it_end = W.end();
if(require_register) {
while(it != it_end) {
W.DataSet[iwalker]->rewind();
W.registerData(**it,*(W.DataSet[iwalker]));
Psi.registerData(W,*(W.DataSet[iwalker]));
++it;++iwalker;
}
}
Estimators->reset();
IndexType block = 0;
Pooma::Clock timer;
int Population = W.getActiveWalkers();
int tPopulation = W.getActiveWalkers();
RealType Eest = brancher.E_T;
RealType E_T = Eest;
RealType oneovertau = 1.0/Tau;
RealType oneover2tau = 0.5*oneovertau;
RealType g = sqrt(Tau);
MCWalkerConfiguration::PropertyContainer_t Properties;
ParticleSet::ParticlePos_t deltaR(W.getTotalNum());
ParticleSet::ParticleGradient_t G(W.getTotalNum()), dG(W.getTotalNum());
ParticleSet::ParticleLaplacian_t L(W.getTotalNum()), dL(W.getTotalNum());
IndexType accstep=0;
IndexType nAcceptTot = 0;
IndexType nRejectTot = 0;
IndexType nat = W.getTotalNum();
int ncross = 0;
do {
IndexType step = 0;
timer.start();
nAccept = 0; nReject=0;
IndexType nAllRejected = 0;
do {
Population = W.getActiveWalkers();
it = W.begin();
it_end = W.end();
iwalker=0;
while(it != it_end) {
MCWalkerConfiguration::WalkerData_t& w_buffer = *(W.DataSet[iwalker]);
(*it)->Properties(WEIGHT) = 1.0;
(*it)->Properties(MULTIPLICITY) = 1.0;
//save old local energy
ValueType eold((*it)->Properties(LOCALENERGY));
ValueType emixed(eold);
W.R = (*it)->R;
w_buffer.rewind();
W.copyFromBuffer(w_buffer);
Psi.copyFromBuffer(W,w_buffer);
ValueType psi_old((*it)->Properties(SIGN));
ValueType psi(psi_old);
//create a 3N-Dimensional Gaussian with variance=1
makeGaussRandom(deltaR);
bool notcrossed(true);
int nAcceptTemp(0);
int nRejectTemp(0);
int iat=0;
while(notcrossed && iat<nat){
PosType dr(g*deltaR[iat]+(*it)->Drift[iat]);
PosType newpos(W.makeMove(iat,dr));
RealType ratio(Psi.ratio(W,iat,dG,dL));
if(ratio < 0.0) {//node is crossed, stop here
notcrossed = false;
} else {
G = W.G+dG;
RealType logGf = -0.5*dot(deltaR[iat],deltaR[iat]);
ValueType vsq = Dot(G,G);
ValueType scale = ((-1.0+sqrt(1.0+2.0*Tau*vsq))/vsq);
dr = (*it)->R[iat]-newpos-scale*G[iat];
//dr = (*it)->R[iat]-newpos-Tau*G[iat];
RealType logGb = -oneover2tau*dot(dr,dr);
//RealType ratio2 = pow(ratio,2)
RealType prob = std::min(1.0,pow(ratio,2)*exp(logGb-logGf));
if(Random() < prob) {
++nAcceptTemp;
W.acceptMove(iat);
Psi.update2(W,iat);
W.G = G;
W.L += dL;
// (*it)->Drift = Tau*G;
(*it)->Drift = scale*G;
} else {
++nRejectTemp;
Psi.restore(iat);
}
}
++iat;
}
if(notcrossed) {
if(nAcceptTemp) {//need to overwrite the walker properties
w_buffer.rewind();
W.copyToBuffer(w_buffer);
psi = Psi.evaluate(W,w_buffer);
(*it)->R = W.R;
(*it)->Properties(AGE) = 0;
//This is not so useful: allow overflow/underflow
(*it)->Properties(LOGPSI) = log(fabs(psi));
(*it)->Properties(SIGN) = psi;
(*it)->Properties(LOCALENERGY) = H.evaluate(W);
H.copy((*it)->getEnergyBase());
(*it)->Properties(LOCALPOTENTIAL) = H.getLocalPotential();
emixed += (*it)->Properties(LOCALENERGY);
} else {
//WARNMSG("All the particle moves are rejected.")
(*it)->Properties(AGE)++;
++nAllRejected;
emixed += eold;
}
ValueType M = brancher.branchGF(Tau,emixed*0.5,0.0);
// if((*it)->Properties(AGE) > 3.0) M = min(0.5,M);
//persistent configurations
if((*it)->Properties(AGE) > 1.9) M = std::min(0.5,M);
if((*it)->Properties(AGE) > 0.9) M = std::min(1.0,M);
(*it)->Properties(WEIGHT) = M;
(*it)->Properties(MULTIPLICITY) = M + Random();
nAccept += nAcceptTemp;
nReject += nRejectTemp;
} else {//set the weight and multiplicity to zero
(*it)->Properties(WEIGHT) = 0.0;
(*it)->Properties(MULTIPLICITY) = 0.0;
nReject += W.getTotalNum();//not make sense
}
++it; ++iwalker;
}
++step;++accstep;
Estimators->accumulate(W);
Eest = brancher.update(Population,Eest);
//E_T = brancher.update(Population,Eest);
brancher.branch(accstep,W);
} while(step<nSteps);
//WARNMSG("The number of a complete rejectoin " << nAllRejected)
timer.stop();
nAcceptTot += nAccept;
nRejectTot += nReject;
Estimators->flush();
Estimators->setColumn(PopIndex,static_cast<RealType>(Population));
Estimators->setColumn(EtrialIndex,Eest); //E_T);
Estimators->setColumn(AcceptIndex,
static_cast<RealType>(nAccept)/static_cast<RealType>(nAccept+nReject));
Estimators->report(accstep);
Eest = Estimators->average(0);
LogOut->getStream() << "Block " << block << " " << timer.cpu_time() << " Fixed_configs "
<< static_cast<RealType>(nAllRejected)/static_cast<RealType>(step*W.getActiveWalkers()) << endl;
if(pStride) {
//create an output engine
HDFWalkerOutput WO(RootName);
WO.get(W);
brancher.write(WO.getGroupID());
}
nAccept = 0; nReject = 0;
block++;
} while(block<nBlocks);
LogOut->getStream()
<< "Ratio = "
<< static_cast<double>(nAcceptTot)/static_cast<double>(nAcceptTot+nRejectTot)
<< endl;
if(!pStride) {
//create an output engine
HDFWalkerOutput WO(RootName);
WO.get(W);
brancher.write(WO.getGroupID());
}
Estimators->finalize();
return true;
}
