portelli/LatAnalyze
1
0
Fork 0
mirror of https://github.com/aportelli/LatAnalyze.git synced 2025-04-11 03:20:46 +01:00
Code Releases Activity

Removed coth model and changed save option name

Browse Source
This commit is contained in:
Andrew Zhen Ning Yong 2019-11-28 15:49:28 +00:00
parent 161c06292e
commit c3d0f93e1a
1 changed files with 17 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
physics/2pt-fit.cpp
View File

@ -16,7 +16,7 @@ int main(int argc, char *argv[])
// parse arguments ///////////////////////////////////////////////////////// // parse arguments /////////////////////////////////////////////////////////
OptParser opt; OptParser opt;
bool parsed, doPlot, doHeatmap, doCorr, fold; bool parsed, doPlot, doHeatmap, doCorr, fold;
string corrFileName, model, outFileName, outFmt, save; string corrFileName, model, outFileName, outFmt, savePlot;
Index ti, tf, shift, nPar, thinning; Index ti, tf, shift, nPar, thinning;
double svdTol; double svdTol;
Minimizer::Verbosity verbosity; Minimizer::Verbosity verbosity;
@ -30,7 +30,7 @@ int main(int argc, char *argv[])
opt.addOption("s", "shift" , OptParser::OptType::value , true, opt.addOption("s", "shift" , OptParser::OptType::value , true,
"time variable shift", "0"); "time variable shift", "0");
opt.addOption("m", "model" , OptParser::OptType::value , true, opt.addOption("m", "model" , OptParser::OptType::value , true,
"fit model (exp|exp2|exp3|sinh|cosh|cosh2|cosh3|coth|explin|const|<interpreter code>)", "cosh"); "fit model (exp|exp2|exp3|sinh|cosh|cosh2|cosh3|explin|const|<interpreter code>)", "cosh");
opt.addOption("" , "nPar" , OptParser::OptType::value , true, opt.addOption("" , "nPar" , OptParser::OptType::value , true,
"number of model parameters for custom models " "number of model parameters for custom models "
"(-1 if irrelevant)", "-1"); "(-1 if irrelevant)", "-1");
@ -48,7 +48,7 @@ int main(int argc, char *argv[])
"show the fit plot"); "show the fit plot");
opt.addOption("h", "heatmap" , OptParser::OptType::trigger, true, opt.addOption("h", "heatmap" , OptParser::OptType::trigger, true,
"show the fit correlation heatmap"); "show the fit correlation heatmap");
opt.addOption("s", "save", OptParser::OptType::value, true, opt.addOption("s", "save-plot", OptParser::OptType::value, true,
"saves the source and .pdf", ""); "saves the source and .pdf", "");
opt.addOption("", "help" , OptParser::OptType::trigger, true, opt.addOption("", "help" , OptParser::OptType::trigger, true,
"show this help message and exit"); "show this help message and exit");
@ -73,7 +73,7 @@ int main(int argc, char *argv[])
fold = opt.gotOption("fold"); fold = opt.gotOption("fold");
doPlot = opt.gotOption("p"); doPlot = opt.gotOption("p");
doHeatmap = opt.gotOption("h"); doHeatmap = opt.gotOption("h");
save = opt.optionValue("s"); savePlot = opt.optionValue("s");
switch (opt.optionValue<unsigned int>("v")) switch (opt.optionValue<unsigned int>("v"))
{ {
case 0: case 0:
@ -120,7 +120,7 @@ int main(int argc, char *argv[])
// make model ////////////////////////////////////////////////////////////// // make model //////////////////////////////////////////////////////////////
DoubleModel mod; DoubleModel mod;
bool sinhModel = false, coshModel = false, cotanhModel = false, linearModel = false, constModel = false; bool sinhModel = false, coshModel = false, linearModel = false, constModel = false;
if ((model == "exp") or (model == "exp1")) if ((model == "exp") or (model == "exp1"))
{ {
@ -186,18 +186,6 @@ int main(int argc, char *argv[])
+ p[5]*(exp(-p[2]*x[0])+exp(-p[4]*(nt-x[0]))); + p[5]*(exp(-p[2]*x[0])+exp(-p[4]*(nt-x[0])));
}, 1, nPar); }, 1, nPar);
} }
else if (model == "coth")
{
cotanhModel = true;
nPar = 2;
mod.setFunction([nt](const double *x, const double *p)
{
// cout << "Using effective pion mass, am = 0.1892" << endl;
// kaon physical mass:0.28856
// pion physical mass: 0.08046
return p[0] + p[1]/tanh(0.28856*(nt/2-x[0])); // 0.1892 is the meson eff mass in lattice unit from 2pt fit; 0.08046 is the meson mass at physical point
}, 1, nPar);
}
else if (model == "explin") else if (model == "explin")
{ {
linearModel = true; linearModel = true;
@ -271,20 +259,11 @@ int main(int argc, char *argv[])
init(0) = data.y(nt/4, 0)[central]; init(0) = data.y(nt/4, 0)[central];
} }
else if(cotanhModel)
{
// init(0) = 0.5*(data.y(0, 0)[central]+ data.y(nt-1, 0)[central]);
// init(1) = 0.5*(data.y(0, 0)[central]- data.y(nt-1, 0)[central]);
init(0) = -0.1677;
init(1) = 0.02526;
cout << init(0) << endl;
}
else else
{ {
init(0) = log(data.y(nt/4, 0)[central]/data.y(nt/4 + 1, 0)[central]); init(0) = log(data.y(nt/4, 0)[central]/data.y(nt/4 + 1, 0)[central]);
init(1) = data.y(nt/4, 0)[central]/(exp(-init(0)*nt/4)); init(1) = data.y(nt/4, 0)[central]/(exp(-init(0)*nt/4));
// cout << init(0) << "\t" << init(1) << endl;
} }
for (Index p = 2; p < nPar; p += 2) for (Index p = 2; p < nPar; p += 2)
{ {
@ -306,27 +285,15 @@ int main(int argc, char *argv[])
locMin.setLowLimit(p, -10*fabs(init(0))); locMin.setLowLimit(p, -10*fabs(init(0)));
globMin.setHighLimit(p, 10*fabs(init(0))); globMin.setHighLimit(p, 10*fabs(init(0)));
} }
else if(cotanhModel)
{
// globMin.setLowLimit(p,-0.5);
// locMin.setLowLimit(p, -0.3);
// globMin.setHighLimit(p,1);
globMin.setLowLimit(p,-1);
locMin.setLowLimit(p, -0.3);
globMin.setHighLimit(p,0);
globMin.setLowLimit(p+1,-0.12);
locMin.setLowLimit(p+1, -0.1);
globMin.setHighLimit(p+1,0.9);
}
else else
{ {
globMin.setLowLimit(p, 0.); globMin.setLowLimit(p, 0.);
locMin.setLowLimit(p, 0.); // locMin.setLowLimit(p, 0.);
globMin.setHighLimit(p, 10.*init(p)); globMin.setHighLimit(p, 10.*init(p));
} }
if(!constModel) if(!constModel)
{ {
locMin.setLowLimit(p+1, -1);
globMin.setLowLimit(p + 1, -10.*fabs(init(p + 1))); globMin.setLowLimit(p + 1, -10.*fabs(init(p + 1)));
globMin.setHighLimit(p + 1, 10.*fabs(init(p + 1))); globMin.setHighLimit(p + 1, 10.*fabs(init(p + 1)));
} }
@ -375,10 +342,10 @@ int main(int argc, char *argv[])
p << Color("rgb 'blue'") << PlotFunction(fit.getModel(), 0, nt - 1); p << Color("rgb 'blue'") << PlotFunction(fit.getModel(), 0, nt - 1);
p << Color("rgb 'red'") << PlotData(data.getData()); p << Color("rgb 'red'") << PlotData(data.getData());
p.display(); p.display();
if(save != "") if(savePlot != "")
{ {
cout << "Saving plot and source code to " << save << endl; cout << "Saving plot and source code to " << save << endl;
p.save(save); p.save(savePlot);
} }
// effective mass plot ////////////////////////////////////////////////////// // effective mass plot //////////////////////////////////////////////////////
if (!constModel) if (!constModel)
@ -393,7 +360,7 @@ int main(int argc, char *argv[])
fitErr = fit.variance().cwiseSqrt(); fitErr = fit.variance().cwiseSqrt();
e0 = fit[central](0); e0 = fit[central](0);
e0Err = fitErr(0); e0Err = fitErr(0);
if (coshModel) if (coshModel or sinhModel)
{ {
FOR_STAT_ARRAY(effMass, s) FOR_STAT_ARRAY(effMass, s)
{ {
@ -432,6 +399,12 @@ int main(int argc, char *argv[])
p << Color("rgb 'red'") << PlotData(effMassT, effMass); p << Color("rgb 'red'") << PlotData(effMassT, effMass);
p << Caption("Effective Mass"); p << Caption("Effective Mass");
p.display(); p.display();
if(savePlot != "")
{
string savename = savePlot + "_effMass";
cout << "Saving effective mass plot and source code to " << savename << endl;
p.save(savename);
}
} }
} }