fix an issue in DCHBPostClusterAI (#589)

* fix an issue in DCHBPostClusterAI.java to save all necessary clusters in HB cluster bank

* add clusters in tracks from AI-assisted tracking into cluster list

Author: tongtongcao

reconstruction/dc/src/main/java/org/jlab/service/dc/DCHBPostClusterAI.java

@@ -30,7 +30,7 @@
 
 /**
  *
- * @author ziegler
+ * @author ziegler, tongtong
  */
 public class DCHBPostClusterAI extends DCEngine {
 
@@ -71,7 +71,7 @@ public boolean processDataEvent(DataEvent event) {
         //AI
         List<Track> trkcands = null;
         List<Cross> crosses = null;
-        List<FittedCluster> clusters = null;
+        List<FittedCluster> clusters = new ArrayList<>();
         List<Segment> segments = null;
         List<FittedHit> fhits = null;
 
@@ -103,7 +103,6 @@ public boolean processDataEvent(DataEvent event) {
                 LOGGER.log(Level.FINE, "Pass Cross"+c.printInfo());
         }
         if (crosses.isEmpty()) {
-            clusters = new ArrayList<>();
             for(Segment seg : segments) {
                 clusters.add(seg.get_fittedCluster());
             }
@@ -126,14 +125,18 @@ public boolean processDataEvent(DataEvent event) {
             dcSwim, true);
 
         // track found
-        clusters = new ArrayList<>();
         int trkId = 1;
         if (trkcands.size() > 0) {
             // remove overlaps
             trkcandFinder.removeOverlappingTracks(trkcands);
             for (Track trk : trkcands) {
                 trk.setIsAITrack(true);
 
+                for (Cross c : trk) {
+                    clusters.add(c.get_Segment1().get_fittedCluster());
+                    clusters.add(c.get_Segment2().get_fittedCluster());
+                }
+                
                 // reset the id
                 trk.set_Id(trkId);
                 trkcandFinder.matchHits(trk.getStateVecs(),
@@ -312,6 +315,7 @@ public boolean processDataEvent(DataEvent event) {
         if (trkcands.isEmpty()) {
             event.appendBanks(
                     writer.fillHBHitsBank(event, fhits),    
+                    writer.fillHBClustersBank(event, clusters),
                     writer.fillHBSegmentsBank(event, segments),
                     writer.fillHBCrossesBank(event, crosses));
         }