[no commit message]

This commit is contained in:
Alexander Nozik 2016-07-27 20:18:42 +03:00
parent fbcc8eef94
commit 23e2997b0f
14 changed files with 181 additions and 54 deletions

View File

@ -14,6 +14,7 @@ dependencies {
compile project(':dataforge-fx') compile project(':dataforge-fx')
compile project(':dataforge-plots') compile project(':dataforge-plots')
compile project(':numass-storage') compile project(':numass-storage')
compile project(':dataforge-grind')
} }
task workbench(dependsOn: classes, type: JavaExec){ task workbench(dependsOn: classes, type: JavaExec){

View File

@ -42,6 +42,8 @@ import static java.util.Locale.setDefault;
import inr.numass.utils.TritiumUtils; import inr.numass.utils.TritiumUtils;
import inr.numass.data.SpectrumDataAdapter; import inr.numass.data.SpectrumDataAdapter;
import hep.dataforge.io.FittingIOUtils import hep.dataforge.io.FittingIOUtils
import hep.dataforge.meta.Meta
import hep.dataforge.grind.GrindMetaBuilder
/** /**
* *
@ -50,9 +52,16 @@ import hep.dataforge.io.FittingIOUtils
setDefault(Locale.US); setDefault(Locale.US);
ModularSpectrum beta = new ModularSpectrum(new BetaSpectrum(), 8.3e-5, 13990d, 18600d); //ParametricFunction beta = new BetaSpectrum();
//ParametricFunction beta = new SterileNeutrinoSpectrum(); //ModularSpectrum beta = new ModularSpectrum(new BetaSpectrum(), 8.3e-5, 13990d, 18600d);
//beta.setCaching(false)
Meta cfg = new GrindMetaBuilder().meta{
resolution(width: 8.3e-5)
}.build();
ParametricFunction beta = new SterileNeutrinoSpectrum(cfg);
NBkgSpectrum spectrum = new NBkgSpectrum(beta); NBkgSpectrum spectrum = new NBkgSpectrum(beta);
XYModel model = new XYModel(spectrum, new SpectrumDataAdapter()); XYModel model = new XYModel(spectrum, new SpectrumDataAdapter());
@ -65,10 +74,10 @@ allPars.setPar("E0", 18574.94, 1.4);
allPars.setPar("mnu2", 0d, 1d); allPars.setPar("mnu2", 0d, 1d);
allPars.setPar("msterile2", 1000d * 1000d,0); allPars.setPar("msterile2", 1000d * 1000d,0);
allPars.setPar("U2", 0.0, 1e-4, -1d, 1d); allPars.setPar("U2", 0.0, 1e-4, -1d, 1d);
allPars.setPar("X", 0.04000, 0.01, 0d, Double.POSITIVE_INFINITY); allPars.setPar("X", 0.04, 0.01, 0d, Double.POSITIVE_INFINITY);
allPars.setPar("trap", 1.634, 0.01,0d, Double.POSITIVE_INFINITY); allPars.setPar("trap", 1.634, 0.01,0d, Double.POSITIVE_INFINITY);
FittingIOUtils.printSpectrum(GlobalContext.out(), spectrum, allPars, 14000.0, 18600.0, 400); FittingIOUtils.printSpectrum(GlobalContext.out(), spectrum, allPars, 14000, 18600.0, 400);
//SpectrumGenerator generator = new SpectrumGenerator(model, allPars, 12316); //SpectrumGenerator generator = new SpectrumGenerator(model, allPars, 12316);
// //

View File

@ -34,10 +34,6 @@ import java.io.PrintWriter;
*/ */
public class NumassContext extends Context { public class NumassContext extends Context {
public static UnivariateIntegrator defaultIntegrator = new GaussRuleIntegrator(300);
public static UnivariateIntegrator highDensityIntegrator = new GaussRuleIntegrator(500);
public NumassContext(Context parent, Meta config) { public NumassContext(Context parent, Meta config) {
super(parent, "numass", config); super(parent, "numass", config);
init(); init();

View File

@ -0,0 +1,42 @@
/*
* To change this license header, choose License Headers in Project Properties.
* To change this template file, choose Tools | Templates
* and open the template in the editor.
*/
package inr.numass;
import hep.dataforge.maths.integration.GaussRuleIntegrator;
import hep.dataforge.maths.integration.UnivariateIntegrator;
/**
*
* @author Alexander Nozik
*/
public class NumassIntegrator {
private static UnivariateIntegrator fastInterator;
private static UnivariateIntegrator defaultIntegrator;
private static UnivariateIntegrator highDensityIntegrator;
public static UnivariateIntegrator getFastInterator() {
if (fastInterator == null) {
fastInterator = new GaussRuleIntegrator(100);
}
return fastInterator;
}
public static UnivariateIntegrator getDefaultIntegrator() {
if (defaultIntegrator == null) {
defaultIntegrator = new GaussRuleIntegrator(300);
}
return defaultIntegrator;
}
public static UnivariateIntegrator getHighDensityIntegrator() {
if (highDensityIntegrator == null) {
highDensityIntegrator = new GaussRuleIntegrator(500);
}
return highDensityIntegrator;
}
}

View File

@ -40,6 +40,7 @@ import hep.dataforge.tables.MapPoint;
import hep.dataforge.tables.Table; import hep.dataforge.tables.Table;
import hep.dataforge.tables.XYAdapter; import hep.dataforge.tables.XYAdapter;
import hep.dataforge.values.NamedValueSet; import hep.dataforge.values.NamedValueSet;
import inr.numass.NumassIntegrator;
import inr.numass.NumassContext; import inr.numass.NumassContext;
import inr.numass.models.ExperimentalVariableLossSpectrum; import inr.numass.models.ExperimentalVariableLossSpectrum;
import inr.numass.models.LossCalculator; import inr.numass.models.LossCalculator;
@ -79,7 +80,7 @@ public class ShowLossSpectrumAction extends OneToOneAction<FitState, FitState> {
new MetaBuilder("plot") new MetaBuilder("plot")
.setValue("plotTitle", "Differential scattering crossection for " + name) .setValue("plotTitle", "Differential scattering crossection for " + name)
); );
switch (input.getModel().meta().getString("name","")) { switch (input.getModel().meta().getString("name", "")) {
case "scatter-variable": case "scatter-variable":
scatterFunction = LossCalculator.getSingleScatterFunction(pars); scatterFunction = LossCalculator.getSingleScatterFunction(pars);
calculateRatio = true; calculateRatio = true;
@ -178,7 +179,7 @@ public class ShowLossSpectrumAction extends OneToOneAction<FitState, FitState> {
} }
public static double calcultateIonRatio(NamedValueSet set, double threshold) { public static double calcultateIonRatio(NamedValueSet set, double threshold) {
UnivariateIntegrator integrator = NumassContext.highDensityIntegrator; UnivariateIntegrator integrator = NumassIntegrator.getHighDensityIntegrator();
UnivariateFunction integrand = LossCalculator.getSingleScatterFunction(set); UnivariateFunction integrand = LossCalculator.getSingleScatterFunction(set);
return 1d - integrator.integrate(integrand, 5d, threshold); return 1d - integrator.integrate(integrand, 5d, threshold);
} }

View File

@ -7,6 +7,7 @@ package inr.numass.models;
import hep.dataforge.fitting.parametric.ParametricFunction; import hep.dataforge.fitting.parametric.ParametricFunction;
import hep.dataforge.values.NamedValueSet; import hep.dataforge.values.NamedValueSet;
import inr.numass.NumassIntegrator;
import inr.numass.NumassContext; import inr.numass.NumassContext;
import inr.numass.utils.TritiumUtils; import inr.numass.utils.TritiumUtils;
import org.apache.commons.math3.analysis.UnivariateFunction; import org.apache.commons.math3.analysis.UnivariateFunction;
@ -21,7 +22,7 @@ public class CustomNBkgSpectrum extends NBkgSpectrum {
public static CustomNBkgSpectrum tritiumBkgSpectrum(ParametricFunction source, double amplitude){ public static CustomNBkgSpectrum tritiumBkgSpectrum(ParametricFunction source, double amplitude){
UnivariateFunction differentialBkgFunction = TritiumUtils.tritiumBackgroundFunction(amplitude); UnivariateFunction differentialBkgFunction = TritiumUtils.tritiumBackgroundFunction(amplitude);
UnivariateFunction integralBkgFunction = UnivariateFunction integralBkgFunction =
(x) -> NumassContext.defaultIntegrator (x) -> NumassIntegrator.getDefaultIntegrator()
.integrate(differentialBkgFunction, x, 18580d); .integrate(differentialBkgFunction, x, 18580d);
return new CustomNBkgSpectrum(source, integralBkgFunction); return new CustomNBkgSpectrum(source, integralBkgFunction);
} }

View File

@ -19,14 +19,12 @@ import hep.dataforge.exceptions.NotDefinedException;
import hep.dataforge.fitting.parametric.AbstractParametricFunction; import hep.dataforge.fitting.parametric.AbstractParametricFunction;
import hep.dataforge.maths.integration.UnivariateIntegrator; import hep.dataforge.maths.integration.UnivariateIntegrator;
import hep.dataforge.values.NamedValueSet; import hep.dataforge.values.NamedValueSet;
import inr.numass.NumassContext; import inr.numass.NumassIntegrator;
import static java.lang.Math.abs;
import static java.lang.Math.exp; import static java.lang.Math.exp;
import static java.lang.Math.sqrt; import static java.lang.Math.sqrt;
import org.apache.commons.math3.analysis.UnivariateFunction; import org.apache.commons.math3.analysis.UnivariateFunction;
import static java.lang.Math.abs; import static java.lang.Math.abs;
import static java.lang.Math.abs; import static java.lang.Math.abs;
import static java.lang.Math.abs;
/** /**
* *
@ -37,11 +35,10 @@ public class GunSpectrum extends AbstractParametricFunction {
private static final String[] list = {"pos", "resA", "sigma"}; private static final String[] list = {"pos", "resA", "sigma"};
private final double cutoff = 4d; private final double cutoff = 4d;
protected final UnivariateIntegrator integrator; protected final UnivariateIntegrator integrator;
public GunSpectrum() { public GunSpectrum() {
super(list); super(list);
integrator = NumassContext.defaultIntegrator; integrator = NumassIntegrator.getDefaultIntegrator();
} }
@Override @Override
@ -50,8 +47,10 @@ public class GunSpectrum extends AbstractParametricFunction {
final double sigma = set.getDouble("sigma"); final double sigma = set.getDouble("sigma");
final double resA = set.getDouble("resA"); final double resA = set.getDouble("resA");
if(sigma == 0) throw new NotDefinedException(); if (sigma == 0) {
throw new NotDefinedException();
}
UnivariateFunction integrand; UnivariateFunction integrand;
switch (parName) { switch (parName) {
case "pos": case "pos":
@ -139,13 +138,13 @@ public class GunSpectrum extends AbstractParametricFunction {
final double pos = set.getDouble("pos"); final double pos = set.getDouble("pos");
final double sigma = set.getDouble("sigma"); final double sigma = set.getDouble("sigma");
final double resA = set.getDouble("resA"); final double resA = set.getDouble("resA");
if (sigma <1e-5 ) { if (sigma < 1e-5) {
return transmissionValueFast(U, pos, resA); return transmissionValueFast(U, pos, resA);
} }
UnivariateFunction integrand = (double E) -> transmissionValueFast(U, E, resA) * getGauss(E, pos, sigma); UnivariateFunction integrand = (double E) -> transmissionValueFast(U, E, resA) * getGauss(E, pos, sigma);
if (pos + cutoff * sigma < U) { if (pos + cutoff * sigma < U) {
return 0; return 0;
} else if (pos - cutoff * sigma > U * (1 + resA)) { } else if (pos - cutoff * sigma > U * (1 + resA)) {

View File

@ -46,7 +46,7 @@ public class LossCalculator {
private static final LossCalculator instance = new LossCalculator(); private static final LossCalculator instance = new LossCalculator();
private static final UnivariateIntegrator integrator = new GaussRuleIntegrator(100); private static final UnivariateIntegrator integrator = new GaussRuleIntegrator(100);
// private static final UnivariateIntegrator tailIntegrator = new GaussRuleIntegrator(50);
public static UnivariateFunction getSingleScatterFunction() { public static UnivariateFunction getSingleScatterFunction() {
final double A1 = 0.204; final double A1 = 0.204;
@ -282,6 +282,7 @@ public class LossCalculator {
* @return * @return
*/ */
public List<Double> getLossProbabilities(double X) { public List<Double> getLossProbabilities(double X) {
List<Double> res = new ArrayList<>(); List<Double> res = new ArrayList<>();
double prob; double prob;
if (X > 0) { if (X > 0) {

View File

@ -15,6 +15,7 @@
*/ */
package inr.numass.models; package inr.numass.models;
import inr.numass.NumassIntegrator;
import inr.numass.NumassContext; import inr.numass.NumassContext;
import org.apache.commons.math3.analysis.BivariateFunction; import org.apache.commons.math3.analysis.BivariateFunction;
import org.apache.commons.math3.analysis.UnivariateFunction; import org.apache.commons.math3.analysis.UnivariateFunction;
@ -38,7 +39,7 @@ class LossResConvolution implements BivariateFunction {
public double value(final double Ein, final double U) { public double value(final double Ein, final double U) {
UnivariateFunction integrand = (double Eout) -> loss.value(Ein, Eout) * res.value(Eout, U); UnivariateFunction integrand = (double Eout) -> loss.value(Ein, Eout) * res.value(Eout, U);
//Энергия в принципе не может быть больше начальной и меньше напряжения //Энергия в принципе не может быть больше начальной и меньше напряжения
return NumassContext.defaultIntegrator.integrate(integrand, U, Ein); return NumassIntegrator.getDefaultIntegrator().integrate(integrand, U, Ein);
} }
} }

View File

@ -18,6 +18,7 @@ package inr.numass.models;
import hep.dataforge.fitting.parametric.AbstractParametricFunction; import hep.dataforge.fitting.parametric.AbstractParametricFunction;
import hep.dataforge.fitting.parametric.ParametricFunction; import hep.dataforge.fitting.parametric.ParametricFunction;
import hep.dataforge.values.NamedValueSet; import hep.dataforge.values.NamedValueSet;
import inr.numass.NumassIntegrator;
import inr.numass.NumassContext; import inr.numass.NumassContext;
import org.apache.commons.math3.analysis.BivariateFunction; import org.apache.commons.math3.analysis.BivariateFunction;
import org.apache.commons.math3.analysis.UnivariateFunction; import org.apache.commons.math3.analysis.UnivariateFunction;
@ -58,7 +59,7 @@ class TransmissionConvolution extends AbstractParametricFunction {
} }
return trans.value(E, U) * spectrum.derivValue(parName, E, set); return trans.value(E, U) * spectrum.derivValue(parName, E, set);
}; };
return NumassContext.defaultIntegrator.integrate(integrand, Math.max(U, min), max + 1d); return NumassIntegrator.getDefaultIntegrator().integrate(integrand, Math.max(U, min), max + 1d);
} }
@Override @Override
@ -80,6 +81,6 @@ class TransmissionConvolution extends AbstractParametricFunction {
} }
return trans.value(E, U) * spectrum.value(E, set); return trans.value(E, U) * spectrum.value(E, set);
}; };
return NumassContext.defaultIntegrator.integrate(integrand, Math.max(U, min), max + 1d); return NumassIntegrator.getDefaultIntegrator().integrate(integrand, Math.max(U, min), max + 1d);
} }
} }

View File

@ -21,6 +21,7 @@ import hep.dataforge.fitting.parametric.ParametricFunction;
import hep.dataforge.maths.integration.UnivariateIntegrator; import hep.dataforge.maths.integration.UnivariateIntegrator;
import hep.dataforge.values.NamedValueSet; import hep.dataforge.values.NamedValueSet;
import hep.dataforge.values.ValueProvider; import hep.dataforge.values.ValueProvider;
import inr.numass.NumassIntegrator;
import inr.numass.NumassContext; import inr.numass.NumassContext;
import java.util.List; import java.util.List;
import org.apache.commons.math3.analysis.BivariateFunction; import org.apache.commons.math3.analysis.BivariateFunction;
@ -101,9 +102,9 @@ public class VariableLossSpectrum extends AbstractParametricFunction {
UnivariateFunction integrand = (double x) -> transmission.value(x, set) * lossFunction.value(x, U - shift); UnivariateFunction integrand = (double x) -> transmission.value(x, set) * lossFunction.value(x, U - shift);
UnivariateIntegrator integrator; UnivariateIntegrator integrator;
if (eMax - U > 150) { if (eMax - U > 150) {
integrator = NumassContext.highDensityIntegrator; integrator = NumassIntegrator.getHighDensityIntegrator();
} else { } else {
integrator = NumassContext.defaultIntegrator; integrator = NumassIntegrator.getDefaultIntegrator();
} }
return noLossProb * transmission.value(U - shift, set) + integrator.integrate(integrand, U, eMax); return noLossProb * transmission.value(U - shift, set) + integrator.integrate(integrand, U, eMax);
} }

View File

@ -13,7 +13,12 @@ import static java.lang.Math.exp;
import static java.lang.Math.sqrt; import static java.lang.Math.sqrt;
/** /**
* A bi-function for beta-spectrum calculation taking energy and final state as input. * A bi-function for beta-spectrum calculation taking energy and final state as
* input.
* <p>
* dissertation p.33
* </p>
*
* @author Alexander Nozik <altavir@gmail.com> * @author Alexander Nozik <altavir@gmail.com>
*/ */
public class NumassBeta extends AbstractParametricBiFunction { public class NumassBeta extends AbstractParametricBiFunction {
@ -25,6 +30,16 @@ public class NumassBeta extends AbstractParametricBiFunction {
super(list); super(list);
} }
/**
* Beta spectrum derivative
*
* @param n parameter number
* @param E0
* @param mnu2
* @param E
* @return
* @throws NotDefinedException
*/
private double derivRoot(int n, double E0, double mnu2, double E) throws NotDefinedException { private double derivRoot(int n, double E0, double mnu2, double E) throws NotDefinedException {
double D = E0 - E;//E0-E double D = E0 - E;//E0-E
double res; double res;
@ -71,12 +86,13 @@ public class NumassBeta extends AbstractParametricBiFunction {
/** /**
* Derivative of spectrum with sterile neutrinos * Derivative of spectrum with sterile neutrinos
*
* @param name * @param name
* @param E * @param E
* @param E0 * @param E0
* @param pars * @param pars
* @return * @return
* @throws NotDefinedException * @throws NotDefinedException
*/ */
private double derivRootsterile(String name, double E, double E0, NamedValueSet pars) throws NotDefinedException { private double derivRootsterile(String name, double E, double E0, NamedValueSet pars) throws NotDefinedException {
double mnu2 = getParameter("mnu2", pars); double mnu2 = getParameter("mnu2", pars);
@ -104,11 +120,14 @@ public class NumassBeta extends AbstractParametricBiFunction {
} }
/**
* The part independent of neutrino mass. Includes global normalization
* constant, momentum and Fermi correction
*
* @param E
* @return
*/
private double factor(double E) { private double factor(double E) {
return K * pfactor(E);
}
private double pfactor(double E) {
double me = 0.511006E6; double me = 0.511006E6;
double Etot = E + me; double Etot = E + me;
double pe = sqrt(E * (E + 2d * me)); double pe = sqrt(E * (E + 2d * me));
@ -118,7 +137,7 @@ public class NumassBeta extends AbstractParametricBiFunction {
double Fn = y / abs(1d - exp(-y)); double Fn = y / abs(1d - exp(-y));
double Fermi = Fn * (1.002037 - 0.001427 * ve); double Fermi = Fn * (1.002037 - 0.001427 * ve);
double res = Fermi * pe * Etot; double res = Fermi * pe * Etot;
return res; return K * res;
} }
@Override @Override
@ -127,11 +146,12 @@ public class NumassBeta extends AbstractParametricBiFunction {
} }
/** /**
* Bare beta spectrum with Mintz negative mass correction * Bare beta spectrum with Mainz negative mass correction
*
* @param E0 * @param E0
* @param mnu2 * @param mnu2
* @param E * @param E
* @return * @return
*/ */
private double root(double E0, double mnu2, double E) { private double root(double E0, double mnu2, double E) {
/*чистый бета-спектр*/ /*чистый бета-спектр*/
@ -155,10 +175,11 @@ public class NumassBeta extends AbstractParametricBiFunction {
/** /**
* beta-spectrum with sterile neutrinos * beta-spectrum with sterile neutrinos
*
* @param E * @param E
* @param E0 * @param E0
* @param pars * @param pars
* @return * @return
*/ */
private double rootsterile(double E, double E0, NamedValueSet pars) { private double rootsterile(double E, double E0, NamedValueSet pars) {
double mnu2 = getParameter("mnu2", pars); double mnu2 = getParameter("mnu2", pars);
@ -189,7 +210,7 @@ public class NumassBeta extends AbstractParametricBiFunction {
@Override @Override
public double derivValue(String parName, double fs, double eIn, NamedValueSet pars) { public double derivValue(String parName, double fs, double eIn, NamedValueSet pars) {
double E0 = getParameter("E0", pars); double E0 = getParameter("E0", pars);
return derivRootsterile(parName,eIn, E0 - fs, pars); return derivRootsterile(parName, eIn, E0 - fs, pars);
} }
@Override @Override

View File

@ -6,6 +6,7 @@
package inr.numass.models.sterile; package inr.numass.models.sterile;
import hep.dataforge.fitting.parametric.AbstractParametricBiFunction; import hep.dataforge.fitting.parametric.AbstractParametricBiFunction;
import hep.dataforge.meta.Meta;
import hep.dataforge.values.NamedValueSet; import hep.dataforge.values.NamedValueSet;
import inr.numass.models.LossCalculator; import inr.numass.models.LossCalculator;
import org.apache.commons.math3.analysis.BivariateFunction; import org.apache.commons.math3.analysis.BivariateFunction;
@ -21,16 +22,21 @@ public class NumassTransmission extends AbstractParametricBiFunction {
private final LossCalculator calculator; private final LossCalculator calculator;
private final BivariateFunction trapFunc; private final BivariateFunction trapFunc;
public NumassTransmission() { // private BicubicInterpolatingFunction cache;
// private double cachedX;
// private Meta cacheMeta;
public NumassTransmission(Meta meta) {
super(list); super(list);
this.calculator = LossCalculator.instance(); this.calculator = LossCalculator.instance();
trapFunc = LossCalculator.getTrapFunction(); trapFunc = getTrapFunction(meta.getNodeOrEmpty("trapping"));
// if (meta.hasNode("cache")) {
// cacheMeta = meta.getNode("cache");
// }
} }
public NumassTransmission(BivariateFunction trapFunc) { private BivariateFunction getTrapFunction(Meta meta) {
super(list); return LossCalculator.getTrapFunction();
this.calculator = LossCalculator.instance();
this.trapFunc = trapFunc;
} }
@Override @Override
@ -52,16 +58,34 @@ public class NumassTransmission extends AbstractParametricBiFunction {
public static double getX(double eIn, NamedValueSet set) { public static double getX(double eIn, NamedValueSet set) {
//From our article //From our article
return set.getDouble("X")*Math.log(eIn / ION_POTENTIAL) * eIn * ION_POTENTIAL / 1.9580741410115568e6; return set.getDouble("X") * Math.log(eIn / ION_POTENTIAL) * eIn * ION_POTENTIAL / 1.9580741410115568e6;
} }
public static double p0(double eIn, NamedValueSet set) { public static double p0(double eIn, NamedValueSet set) {
return LossCalculator.instance().getLossProbability(0, getX(eIn, set)); return LossCalculator.instance().getLossProbability(0, getX(eIn, set));
} }
// private synchronized void setupCache(double X) {
// if (this.cachedX != X) {
// double cacheLo = cacheMeta.getDouble("lo", 14000);
// double cacheHi = cacheMeta.getDouble("hi", 18575);
// int numPoints = cacheMeta.getInt("numPoints", 1000);
// double[] eIns = GridCalculator.getUniformUnivariateGrid(cacheLo, cacheHi, numPoints);
// double[] eOuts = GridCalculator.getUniformUnivariateGrid(cacheLo, cacheHi, numPoints);
// double[][] vals = MathUtils.calculateFunction(calculator.getTotalLossBivariateFunction(X), eOuts, eOuts);
// this.cachedX = X;
// this.cache = new BicubicInterpolator().interpolate(eIns, eOuts, vals);
// }
// }
@Override @Override
public double value(double eIn, double eOut, NamedValueSet set) { public double value(double eIn, double eOut, NamedValueSet set) {
return calculator.getTotalLossValue(getX(eIn, set), eIn, eOut) + getParameter("trap", set) * trapFunc.value(eIn, eOut); //calculate X taking into account its energy dependence
double X = getX(eIn, set);
// loss part
double loss = calculator.getTotalLossValue(X, eIn, eOut);
//trapping part
double trap = getParameter("trap", set) * trapFunc.value(eIn, eOut);
return loss + trap;
} }
} }

View File

@ -7,11 +7,14 @@ package inr.numass.models.sterile;
import hep.dataforge.context.Context; import hep.dataforge.context.Context;
import hep.dataforge.context.GlobalContext; import hep.dataforge.context.GlobalContext;
import hep.dataforge.description.NodeDef;
import hep.dataforge.description.ValueDef;
import hep.dataforge.exceptions.NotDefinedException; import hep.dataforge.exceptions.NotDefinedException;
import hep.dataforge.fitting.parametric.AbstractParametricFunction; import hep.dataforge.fitting.parametric.AbstractParametricFunction;
import hep.dataforge.fitting.parametric.ParametricBiFunction; import hep.dataforge.fitting.parametric.ParametricBiFunction;
import hep.dataforge.meta.Meta; import hep.dataforge.meta.Meta;
import hep.dataforge.values.NamedValueSet; import hep.dataforge.values.NamedValueSet;
import inr.numass.NumassIntegrator;
import inr.numass.NumassContext; import inr.numass.NumassContext;
import inr.numass.models.FSS; import inr.numass.models.FSS;
import java.util.stream.DoubleStream; import java.util.stream.DoubleStream;
@ -28,6 +31,10 @@ import org.apache.commons.math3.random.SynchronizedRandomGenerator;
* *
* @author Alexander Nozik * @author Alexander Nozik
*/ */
@NodeDef(name = "resolution")
@NodeDef(name = "transmission")
@ValueDef(name = "fssFile")
public class SterileNeutrinoSpectrum extends AbstractParametricFunction { public class SterileNeutrinoSpectrum extends AbstractParametricFunction {
private static final String[] list = {"X", "trap", "E0", "mnu2", "msterile2", "U2"}; private static final String[] list = {"X", "trap", "E0", "mnu2", "msterile2", "U2"};
@ -49,6 +56,8 @@ public class SterileNeutrinoSpectrum extends AbstractParametricFunction {
* variables:Eout,U; parameters: "X", "trap" * variables:Eout,U; parameters: "X", "trap"
*/ */
private final ParametricBiFunction resolution; private final ParametricBiFunction resolution;
private boolean useMC;
public SterileNeutrinoSpectrum(Context context, Meta configuration) { public SterileNeutrinoSpectrum(Context context, Meta configuration) {
super(list); super(list);
@ -58,10 +67,15 @@ public class SterileNeutrinoSpectrum extends AbstractParametricFunction {
fssDistribution = new EnumeratedRealDistribution(rnd, fss.getEs(), fss.getPs()); fssDistribution = new EnumeratedRealDistribution(rnd, fss.getEs(), fss.getPs());
} }
transmission = new NumassTransmission(); transmission = new NumassTransmission(configuration.getNodeOrEmpty("transmission"));
resolution = new NumassResolution(configuration.getNode("resolution", Meta.empty())); resolution = new NumassResolution(configuration.getNode("resolution", Meta.empty()));
this.useMC = configuration.getBoolean("useMC",false);
} }
public SterileNeutrinoSpectrum(Meta configuration) {
this(GlobalContext.instance(),configuration);
}
public SterileNeutrinoSpectrum() { public SterileNeutrinoSpectrum() {
this(GlobalContext.instance(), Meta.empty()); this(GlobalContext.instance(), Meta.empty());
} }
@ -96,7 +110,7 @@ public class SterileNeutrinoSpectrum extends AbstractParametricFunction {
} }
private boolean useDirect() { private boolean useDirect() {
return true; return !useMC;
} }
@Override @Override
@ -104,6 +118,12 @@ public class SterileNeutrinoSpectrum extends AbstractParametricFunction {
return source.providesDeriv(name) && transmission.providesDeriv(name) && resolution.providesDeriv(name); return source.providesDeriv(name) && transmission.providesDeriv(name) && resolution.providesDeriv(name);
} }
/**
* Random E generator
* @param a
* @param b
* @return
*/
private double rndE(double a, double b) { private double rndE(double a, double b) {
return rnd.nextDouble() * (b - a) + a; return rnd.nextDouble() * (b - a) + a;
} }
@ -157,6 +177,15 @@ public class SterileNeutrinoSpectrum extends AbstractParametricFunction {
return Math.pow(eMax - u, 2d) / 2d * sum / num; return Math.pow(eMax - u, 2d) / 2d * sum / num;
} }
/**
* Direct Gauss-Legandre integration
* @param u
* @param sourceFunction
* @param transmissionFunction
* @param resolutionFunction
* @param set
* @return
*/
private double integrateDirect( private double integrateDirect(
double u, double u,
ParametricBiFunction sourceFunction, ParametricBiFunction sourceFunction,
@ -175,7 +204,7 @@ public class SterileNeutrinoSpectrum extends AbstractParametricFunction {
fsSource = (eIn) -> { fsSource = (eIn) -> {
double res = 0; double res = 0;
for (int i = 0; i < fss.size(); i++) { for (int i = 0; i < fss.size(); i++) {
res += fss.getP(i) * sourceFunction.value(fss.getE(i), u, set); res += fss.getP(i) * sourceFunction.value(fss.getE(i), eIn, set);
} }
return res; return res;
}; };
@ -185,7 +214,7 @@ public class SterileNeutrinoSpectrum extends AbstractParametricFunction {
BivariateFunction transRes = (eIn, usp) -> { BivariateFunction transRes = (eIn, usp) -> {
UnivariateFunction integrand = (eOut) -> transmissionFunction.value(eIn, eOut, set) * resolutionFunction.value(eOut, usp, set); UnivariateFunction integrand = (eOut) -> transmissionFunction.value(eIn, eOut, set) * resolutionFunction.value(eOut, usp, set);
return NumassContext.defaultIntegrator.integrate(integrand, usp, eIn); return NumassIntegrator.getDefaultIntegrator().integrate(integrand, usp, eIn);
}; };
UnivariateFunction integrand = (eIn) -> { UnivariateFunction integrand = (eIn) -> {
@ -193,7 +222,7 @@ public class SterileNeutrinoSpectrum extends AbstractParametricFunction {
return fsSource.value(eIn) * (p0 * resolutionFunction.value(eIn, u, set) + transRes.value(eIn, u)); return fsSource.value(eIn) * (p0 * resolutionFunction.value(eIn, u, set) + transRes.value(eIn, u));
}; };
return NumassContext.defaultIntegrator.integrate(integrand, u, eMax); return NumassIntegrator.getDefaultIntegrator().integrate(integrand, u, eMax);
} }
} }