Skip to content

feat: ALERT pre- and post-KF AI PID inference #1368

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
baltzell merged 12 commits into from
Aug 4, 2026
Merged

feat: ALERT pre- and post-KF AI PID inference #1368

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
ac52fe2
feat: added implementation for post-KF PID model
Jul 30, 2026
23f0b70
fix: updated PrePID model to give a prediction even for tracks withou…
Jul 30, 2026
414053d
feat: updated RecoBankWriter to write the ALERT::ai:pid bank
Jul 30, 2026
4cb2d38
fix/feat: updated the feature extraction for the prepid model and add…
Jul 30, 2026
d619a0d
Merge branch 'development' into alert-ai-pid
skuditha Jul 31, 2026
700fd3c
fixes based on Mathieu's comments
skuditha Jul 31, 2026
7f24276
Merge branch 'development' into alert-ai-pid
skuditha Jul 31, 2026
a99c907
Merge branch 'development' into alert-ai-pid
baltzell Jul 31, 2026
8d54afd
Merge branch 'development' into alert-ai-pid
baltzell Aug 1, 2026
871b9b5
Merge branch 'development' into alert-ai-pid
skuditha Aug 3, 2026
f3b41d5
Merge branch 'development' into alert-ai-pid
skuditha Aug 4, 2026
bca662c
added guard for appendPIDBank
Aug 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
feat: updated RecoBankWriter to write the ALERT::ai:pid bank
  • Loading branch information
Uditha Weerasinghe
Uditha Weerasinghe committed Jul 30, 2026
commit 414053d11ac317f8dd8d45a9ea4234645e313b8f
36 changes: 28 additions & 8 deletions reconstruction/alert/src/main/java/org/jlab/rec/alert/banks/RecoBankWriter.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,9 @@
import java.util.List;
import org.jlab.io.base.DataBank;
import org.jlab.io.base.DataEvent;
import org.jlab.rec.alert.AIPID.PIDResult;
import org.jlab.rec.alert.AIPID.PrePIDResult;
import org.jlab.rec.alert.projections.TrackProjection;
//import org.jlab.rec.alert.AIpid.PIDResult;

import ai.djl.util.Pair;

Expand All @@ -17,7 +18,7 @@
* @author Whit Armstrong
*/
public class RecoBankWriter {

/**
* Writes the bank of track projections.
*
Expand Down Expand Up @@ -54,12 +55,12 @@ public static DataBank fillProjectionsBank(DataEvent event, ArrayList<TrackProje
}
return bank;
}

/**
* Appends the alert match banks to an event.
*
* @param event the {@link DataEvent} in which to append the banks
* @param projections the {@link ArrayList} of {@link TrackProjection} containing the
* @param projections the {@link ArrayList} of {@link TrackProjection} containing the
* track projections info to be added
*
* @return 0 if it worked, 1 if it failed
Expand Down Expand Up @@ -91,17 +92,16 @@ public int appendTrackMatchingAIBank(DataEvent event, ArrayList<Pair<Integer, In

return 0;
}
public int appendPrePIDBank(DataEvent event, ArrayList<org.jlab.rec.alert.AIPID.PrePIDResult> results) {

public int appendPrePIDBank(DataEvent event, ArrayList<PrePIDResult> results) {

DataBank bank = event.createBank("ALERT::ai:prepid", results.size());
if (bank == null) {
System.err.println("COULD NOT CREATE A ALERT::ai:prepid BANK!!!!!!");
return 1;
}

for (int i = 0; i < results.size(); i++) {
org.jlab.rec.alert.AIPID.PrePIDResult r = results.get(i);
PrePIDResult r = results.get(i);
bank.setInt("trackid", i, r.trackid);
bank.setInt("clusterid", i, r.clusterid);
bank.setInt("prepid", i, r.prepid);
Expand All @@ -111,7 +111,27 @@ public int appendPrePIDBank(DataEvent event, ArrayList<org.jlab.rec.alert.AIPID.
bank.setFloat("p47", i, r.p47);
bank.setFloat("p49", i, r.p49);
}
event.appendBank(bank);
return 0;
}

public int appendPIDBank(DataEvent event, ArrayList<PIDResult> results) {
DataBank bank = event.createBank("ALERT::ai:pid", results.size());
if (bank == null) {
System.err.println("COULD NOT CREATE A ALERT::ai:pid BANK!!!!!!");
return 1;
}
for (int i = 0; i < results.size(); i++) {
PIDResult r = results.get(i);
bank.setInt("trackid", i, r.trackid);
bank.setInt("clusterid", i, r.clusterid);
bank.setInt("pid", i, r.pid);
bank.setFloat("prob_2212", i, r.p2212);
bank.setFloat("prob_45", i, r.p45);
bank.setFloat("prob_46", i, r.p46);
bank.setFloat("prob_47", i, r.p47);
bank.setFloat("prob_49", i, r.p49);
}
event.appendBank(bank);
return 0;
}
Expand Down