Add Vlads calibration

This commit is contained in:
Alexander Nozik 2022-03-14 11:21:21 +03:00
parent e8d8e1ea8a
commit 7f61146af9
2 changed files with 57 additions and 17 deletions

View File

@ -1,9 +1,9 @@
package ru.inr.mass.data.analysis package ru.inr.mass.data.analysis
import kotlinx.coroutines.coroutineScope import kotlinx.coroutines.coroutineScope
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.launch import kotlinx.coroutines.launch
import ru.inr.mass.data.api.NumassBlock import ru.inr.mass.data.api.NumassBlock
import ru.inr.mass.data.api.NumassEvent
import space.kscience.kmath.histogram.LongCounter import space.kscience.kmath.histogram.LongCounter
import kotlin.math.min import kotlin.math.min
@ -44,6 +44,17 @@ public suspend fun NumassBlock.amplitudeSpectrum(
return NumassAmplitudeSpectrum(map.mapValues { it.value.value.toULong() }) return NumassAmplitudeSpectrum(map.mapValues { it.value.value.toULong() })
} }
public suspend fun NumassBlock.energySpectrum(
extractor: NumassEventExtractor = NumassEventExtractor.EVENTS_ONLY,
calibration: (NumassEvent) -> Double,
): Map<Double, Long> {
val map = HashMap<Double, LongCounter>()
extractor.extract(this).collect { event ->
map.getOrPut(calibration(event)) { LongCounter() }.add(1L)
}
return map.mapValues { it.value.value }
}
/** /**
* Collect events from block in parallel * Collect events from block in parallel
*/ */

View File

@ -1,8 +1,10 @@
package ru.inr.mass.scripts package ru.inr.mass.scripts
import ru.inr.mass.data.analysis.NumassEventExtractor import ru.inr.mass.data.analysis.NumassEventExtractor
import ru.inr.mass.data.analysis.amplitudeSpectrum import ru.inr.mass.data.analysis.energySpectrum
import ru.inr.mass.data.api.NumassEvent
import ru.inr.mass.data.api.NumassPoint import ru.inr.mass.data.api.NumassPoint
import ru.inr.mass.data.api.channel
import ru.inr.mass.data.proto.NumassDirectorySet import ru.inr.mass.data.proto.NumassDirectorySet
import ru.inr.mass.models.* import ru.inr.mass.models.*
import ru.inr.mass.workspace.Numass import ru.inr.mass.workspace.Numass
@ -38,6 +40,30 @@ fun Spectrum.convolve(range: ClosedRange<Double>, function: (Double) -> Double):
}.value }.value
} }
/**
* E = A * ADC +B
* Channel A B
* 0 0.01453 1.3
* 2 0.01494 -4.332
* 3 0.01542 -5.183
* 4 0.01573 -2.115
* 5 0.0152 -3.808
* 6 0.0155 -3.015
* 7 0.01517 -0.5429
*/
val calibration: (NumassEvent) -> Double = {
when (it.channel) {
0 -> 0.01453 * it.amplitude + 1.3
2 -> 0.01494 * it.amplitude - 5.183
3 -> 0.01542 * it.amplitude - 5.183
4 -> 0.01573 * it.amplitude - 2.115
5 -> 0.0152 * it.amplitude - 3.808
6 -> 0.0155 * it.amplitude - 3.015
7 -> 0.01517 * it.amplitude - 0.5429
else -> error("Unrecognized channel ${it.channel}")
} * 1000.0
}
private val neutrinoSpectrum = NumassBeta.withFixedX(0.0) private val neutrinoSpectrum = NumassBeta.withFixedX(0.0)
private val args: Map<Symbol, Double> = mapOf( private val args: Map<Symbol, Double> = mapOf(
@ -65,20 +91,21 @@ suspend fun main() {
point point
} }
val spectrum: Map<Short, ULong> = points.values.first()!! val spectrum: Map<Double, Long> = points.values.first()!!
.amplitudeSpectrum(NumassEventExtractor.TQDC) .energySpectrum(NumassEventExtractor.TQDC, calibration)
.amplitudes.toSortedMap() .filter { it.key > 9000.0 }
.toSortedMap()
//the channel of spectrum peak position // //the channel of spectrum peak position
val argmax = spectrum.maxByOrNull { it.value }!!.key // val argmax = spectrum.maxByOrNull { it.value }!!.key
//
// convert channel to energy // // convert channel to energy
fun Short.toEnergy(): Double = toDouble() / argmax * gunEnergy // fun Short.toEnergy(): Double = toDouble() / argmax * gunEnergy
val norm = spectrum.values.sum().toDouble() val norm = spectrum.values.sum().toDouble()
val interpolated: PiecewisePolynomial<Double> = LinearInterpolator(DoubleField).interpolatePolynomials( val interpolated: PiecewisePolynomial<Double> = LinearInterpolator(DoubleField).interpolatePolynomials(
spectrum.keys.map { it.toEnergy() - gunEnergy }.asBuffer(), spectrum.keys.map { it - gunEnergy }.asBuffer(),
spectrum.values.map { it.toDouble() / norm }.asBuffer() spectrum.values.map { it.toDouble() / norm }.asBuffer()
) )
@ -93,7 +120,7 @@ suspend fun main() {
Plotly.plot { Plotly.plot {
scatter { scatter {
name = "gun" name = "gun"
x.numbers = spectrum.keys.map { it.toEnergy() } x.numbers = spectrum.keys
y.numbers = spectrum.values.map { it.toDouble() / norm } y.numbers = spectrum.values.map { it.toDouble() / norm }
} }
@ -101,15 +128,17 @@ suspend fun main() {
name = "convoluted" name = "convoluted"
x.buffer = 0.0..19000.0 step 100.0 x.buffer = 0.0..19000.0 step 100.0
y.numbers = x.doubles.map { model(it, args) } y.numbers = x.doubles.map { model(it, args) }
y.numbers = y.doubles.map { it / y.doubles.maxOrNull()!! } val yNorm = y.doubles.maxOrNull()!!
y.numbers = y.doubles.map { it / yNorm }
} }
scatter { scatter {
name = "tritium" name = "tritium"
val tritiumSpectrum = tritiumData.amplitudeSpectrum(NumassEventExtractor.TQDC).amplitudes.toSortedMap() val tritiumSpectrum = tritiumData.energySpectrum(NumassEventExtractor.TQDC, calibration).toSortedMap()
x.numbers = tritiumSpectrum.keys.map { it.toEnergy() } x.numbers = tritiumSpectrum.keys
y.numbers = tritiumSpectrum.values.map { it.toDouble()} y.numbers = tritiumSpectrum.values.map { it.toDouble() }
y.numbers = y.doubles.map { it / y.doubles.maxOrNull()!! } val yNorm = y.doubles.maxOrNull()!!
y.numbers = y.doubles.map { it / yNorm }
} }
}.makeFile() }.makeFile()
} }