93 lines
3.6 KiB
Groovy
93 lines
3.6 KiB
Groovy
/*
|
||
* Copyright 2015 Alexander Nozik.
|
||
*
|
||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
* you may not use this file except in compliance with the License.
|
||
* You may obtain a copy of the License at
|
||
*
|
||
* http://www.apache.org/licenses/LICENSE-2.0
|
||
*
|
||
* Unless required by applicable law or agreed to in writing, software
|
||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
* See the License for the specific language governing permissions and
|
||
* limitations under the License.
|
||
*/
|
||
package inr.numass.scripts;
|
||
|
||
import hep.dataforge.context.GlobalContext;
|
||
import static hep.dataforge.context.GlobalContext.out;
|
||
import hep.dataforge.data.ListDataSet;
|
||
import hep.dataforge.datafitter.FitManager;
|
||
import hep.dataforge.datafitter.FitState;
|
||
import hep.dataforge.datafitter.FitTask;
|
||
import hep.dataforge.datafitter.ParamSet;
|
||
import hep.dataforge.datafitter.models.XYModel;
|
||
import hep.dataforge.exceptions.NamingException;
|
||
import inr.numass.data.SpectrumDataAdapter;
|
||
import inr.numass.data.SpectrumGenerator;
|
||
import inr.numass.models.ModularTritiumSpectrum;
|
||
import inr.numass.models.NBkgSpectrum;
|
||
import inr.numass.utils.DataModelUtils;
|
||
import java.io.FileNotFoundException;
|
||
import java.util.Locale;
|
||
import static java.util.Locale.setDefault;
|
||
|
||
/**
|
||
*
|
||
* @author Darksnake
|
||
*/
|
||
|
||
setDefault(Locale.US);
|
||
GlobalContext global = GlobalContext.instance();
|
||
// global.loadModule(new MINUITModule());
|
||
|
||
FitManager fm = new FitManager();
|
||
|
||
ModularTritiumSpectrum beta = new ModularTritiumSpectrum(9e-5, 14390d, 19001d, null);
|
||
beta.setCaching(false);
|
||
|
||
NBkgSpectrum spectrum = new NBkgSpectrum(beta);
|
||
XYModel model = new XYModel("tritium", spectrum, new SpectrumDataAdapter());
|
||
|
||
ParamSet allPars = new ParamSet();
|
||
|
||
allPars.setParValue("N", 3e5);
|
||
//значение 6е-6 соответствует полной интенстивности 6е7 распадов в секунду
|
||
//Проблема была в переполнении счетчика событий в генераторе. Заменил на long. Возможно стоит поставить туда число с плавающей точкой
|
||
allPars.setParError("N", 6);
|
||
allPars.setParDomain("N", 0d, Double.POSITIVE_INFINITY);
|
||
allPars.setParValue("bkg", 2d);
|
||
allPars.setParError("bkg", 0.03);
|
||
allPars.setParValue("E0", 18575.0);
|
||
allPars.setParError("E0", 2);
|
||
allPars.setParValue("mnu2", 0d);
|
||
allPars.setParError("mnu2", 1d);
|
||
allPars.setParValue("msterile2", 1000 * 1000);
|
||
allPars.setParValue("U2", 0);
|
||
allPars.setParError("U2", 1e-4);
|
||
allPars.setParDomain("U2", -1d, 1d);
|
||
allPars.setParValue("X", 0);
|
||
allPars.setParError("X", 0.01);
|
||
allPars.setParDomain("X", 0d, Double.POSITIVE_INFINITY);
|
||
allPars.setParValue("trap", 0);
|
||
allPars.setParError("trap", 0.01d);
|
||
allPars.setParDomain("trap", 0d, Double.POSITIVE_INFINITY);
|
||
|
||
// PrintNamed.printSpectrum(GlobalContext.out(), spectrum, allPars, 0.0, 18700.0, 600);
|
||
//String fileName = "d:\\PlayGround\\merge\\scans.out";
|
||
// String configName = "d:\\PlayGround\\SCAN.CFG";
|
||
// ListDataSet config = OldDataReader.readConfig(configName);
|
||
SpectrumGenerator generator = new SpectrumGenerator(model, allPars, 12316);
|
||
|
||
ListDataSet data = generator.generateData(DataModelUtils.getUniformSpectrumConfiguration(13500d, 18200, 1e6, 60));
|
||
|
||
// data = data.filter("X", Value.of(15510.0), Value.of(18610.0));
|
||
// allPars.setParValue("X", 0.4);
|
||
FitState state = fm.buildState(data, model, allPars);
|
||
|
||
FitState res = fm.runTask(state, "QOW", FitTask.TASK_RUN, "N", "bkg", "E0", "U2");
|
||
|
||
res.print(out());
|
||
|