[no commit message]

numass-main/src/main/java/inr/numass/generators/EventChainGenerator.java

@@ -23,40 +23,40 @@ import java.util.List;
 import java.util.Map;
 
 /**
- *
+ * Utility class for pile-up simulation
  * @author Darksnake
  */
 public final class EventChainGenerator {
 
-    private final static double us = 1e-6;
+    private final static double us = 1e-6;//microsecond
 
     private double blockStartTime = 0;
 
     private final List<NMEvent> generatedChain = new ArrayList<>();
-    private final EventGenerator generator;
+    private final NMEventGenerator generator;
     private final double length;
     private final List<NMEvent> pileupChain = new ArrayList<>();
 
     private final List<NMEvent> registredChain = new ArrayList<>();
 
     public EventChainGenerator(double cr, double length) {
-        generator = new EventGenerator(cr);
+        generator = new NMEventGenerator(cr);
         this.length = length;
         
         run();
     }
 
     public EventChainGenerator(double cr, double length, RawNMPoint source, int minChanel, int maxChanel) {
-        generator = new EventGenerator(cr);
+        generator = new NMEventGenerator(cr);
-        this.generator.loadFromPoint(source, minChanel, maxChanel);
+        this.generator.loadSpectrum(source, minChanel, maxChanel);
         this.length = length;
         
         run();
     }
     
     public EventChainGenerator(double cr, double length, Map<Double,Double> spectrum, int minChanel, int maxChanel) {
-        generator = new EventGenerator(cr);
+        generator = new NMEventGenerator(cr);
-        this.generator.loadFromSpectrum(spectrum, minChanel, maxChanel);
+        this.generator.loadSpectrum(spectrum, minChanel, maxChanel);
         this.length = length;
         
         run();

numass-main/src/main/java/inr/numass/generators/NMEventGenerator.java

@@ -16,6 +16,7 @@
 package inr.numass.generators;
 
 import inr.numass.storage.NMEvent;
+import inr.numass.storage.NMPoint;
 import inr.numass.storage.RawNMPoint;
 import java.util.ArrayList;
 import java.util.List;
@@ -26,12 +27,12 @@ import org.apache.commons.math3.random.EmpiricalDistribution;
 import org.apache.commons.math3.random.JDKRandomGenerator;
 import org.apache.commons.math3.random.RandomGenerator;
 
-
 /**
+ * A generator for Numass events with given energy spectrum
  *
  * @author Darksnake
  */
-public class EventGenerator {
+public class NMEventGenerator {
 
     double cr;
 //    UnivariateFunction signalShape;
@@ -39,34 +40,32 @@ public class EventGenerator {
 
     private final RandomGenerator generator;
 
-    public EventGenerator(double cr) {
+    public NMEventGenerator(double cr) {
         this.cr = cr;
         generator = new JDKRandomGenerator();
     }
-    
+
-    public void loadFromPoint(RawNMPoint point,int minChanel, int maxChanel){
+    public void loadSpectrum(RawNMPoint point, int minChanel, int maxChanel) {
         List<Short> shorts = new ArrayList<>();
-        for (NMEvent event : point.getEvents()) {
+        point.getEvents().stream()
-            if((event.getChanel()>minChanel)&&(event.getChanel()<maxChanel)) {
+                .filter((event) -> ((event.getChanel() > minChanel) && (event.getChanel() < maxChanel)))
-                shorts.add(event.getChanel());
+                .forEach((event) -> shorts.add(event.getChanel()));
-            }
-        }
         double[] doubles = new double[shorts.size()];
-        
+
         for (int i = 0; i < shorts.size(); i++) {
             doubles[i] = shorts.get(i);
         }
-        
+
         EmpiricalDistribution d = new EmpiricalDistribution();
         d.load(doubles);
-        
+
         distribution = d;
     }
-    
+
-    public void loadFromSpectrum(Map<Double,Double> spectrum, int minChanel, int maxChanel){
+    public void loadSpectrum(Map<Double, Double> spectrum, int minChanel, int maxChanel) {
         assert minChanel > 0;
         assert maxChanel < RawNMPoint.MAX_CHANEL;
-        
+
         double[] chanels = new double[spectrum.size()];
         double[] values = new double[spectrum.size()];
         int i = 0;
@@ -78,15 +77,29 @@ public class EventGenerator {
         distribution = new EnumeratedRealDistribution(chanels, values);
     }
 
-    public NMEvent nextEvent(NMEvent prev) {
+    public void loadSpectrum(NMPoint point, int minChanel, int maxChanel) {
-        //пока без канала
+        assert minChanel > 0;
-        short chanel = 1600;
+        assert maxChanel < RawNMPoint.MAX_CHANEL;
-        
+
-        if(distribution!=null){
+        double[] chanels = new double[RawNMPoint.MAX_CHANEL];
-            chanel = (short) distribution.sample();
+        double[] values = new double[RawNMPoint.MAX_CHANEL];
+        for (int i = 0; i < RawNMPoint.MAX_CHANEL; i++) {
+            chanels[i] = i;
+            values[i] = point.getCountInChanel(i);
+            i++;
         }
-        
+        distribution = new EnumeratedRealDistribution(chanels, values);
-        
+    }
+
+    public NMEvent nextEvent(NMEvent prev) {
+        short chanel;
+
+        if (distribution != null) {
+            chanel = (short) distribution.sample();
+        } else {
+            chanel = 1600;
+        }
+
         return new NMEvent(chanel, prev.getTime() + nextExp(1 / cr));
     }
 

numass-main/src/main/java/inr/numass/utils/PileUpSimulator.java

@@ -0,0 +1,28 @@
+/*
+ * 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.utils;
+
+import inr.numass.generators.NMEventGenerator;
+import inr.numass.storage.NMEvent;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.function.Function;
+
+/**
+ *
+ * @author Alexander Nozik <altavir@gmail.com>
+ */
+public class PileUpSimulator {
+
+    private NMEventGenerator generator;
+    private final List<NMEvent> generated = new ArrayList<>();
+    private final List<NMEvent> pileup = new ArrayList<>();
+    private final List<NMEvent> registred = new ArrayList<>();
+
+    private Function<Double, Double> firstEventSurvivalProb;
+    private Function<Double, Double> secondEventSurvivalProb;
+    
+}