color code individual "pretracks" modified: src/GenFind/tests/hough_transform2.cxx

src/GenFind/tests/hough_transform2.cxx

@@ -28,6 +28,8 @@ void hough_transform2(
   TH2F * hxz   = new TH2F("hxz",   "hxz",   100, -1000, 1000, 100, -1000, 1000);
   TH2F * hyz   = new TH2F("hyz",   "hyz",   100, -1000, 1000, 100, -1000, 1000);
   TH2F * hxy2  = new TH2F("hxy2",  "hxy2",   50, -100 ,  100, 50,  -100,  100);
+  TH2F * htrack2   = new TH2F("htrack2",   "ht2",   50, -100, 100, 50, -1000, 1000);
+  TH2F * htrack1   = new TH2F("htrack1",   "ht1",   100, -1000, 1000, 100, -1000, 1000);
   TH2F * hpeaky   = new TH2F("hpeakxy",   "hxy",   100, -1000, 1000, 100, -1000, 1000);
   TH2F * hpeaky2  = new TH2F("hpeakxy2",  "hxy2",   50, -100 ,  100, 50,  -100,  100);
   TH2F * huv   = new TH2F("huv",   "huv",   100,   -0.12, 0.12  , 100,   -0.12, 0.12  );
@@ -100,17 +102,39 @@ void hough_transform2(
 
 
   std::vector<TH1F*> hists;
+  std::vector<TH1F*> alltrack;
+  std::vector<TH1F*> vertextrack;
   for( auto atrack : pretrack_hits ) {
+    TH1F* singletrack = (TH1F*)htrack1->Clone();
+    singletrack->Reset();
+    singletrack->SetLineColor(2+hists.size());
+    TH1F* singlevertextrack = (TH1F*)htrack2->Clone();
+    singlevertextrack->Reset();
+    singlevertextrack->SetLineColor(2+hists.size());
     TH1F* fPhi2 = (TH1F*)ht->fPhi->Clone();
     fPhi2->Reset();
     fPhi2->SetLineColor(2+hists.size());
     for( auto thit : atrack ) {
+      singletrack->Fill( std::get<0>(thit).X(), std::get<0>(thit).Y());
+      singlevertextrack->Fill( std::get<0>(thit).X(), std::get<0>(thit).Y());
       hpeaky->Fill( std::get<0>(thit).X(), std::get<0>(thit).Y());
       hpeaky2->Fill(std::get<0>(thit).X(), std::get<0>(thit).Y());
       fPhi2->Fill(TMath::ATan(std::get<1>(thit).Y()/std::get<1>(thit).X())/degree);
     }
     hists.push_back(fPhi2);
+    alltrack.push_back(singletrack);
+    vertextrack.push_back(singlevertextrack);
   }
+//  auto trackone = pretrack_hits.at(0);
+//  for( auto thit : trackone ) {
+//    htrack1->Fill( std::get<0>(thit).X(), std::get<0>(thit).Y());
+//  }
+//  auto tracktwo = pretrack_hits.at(1);
+//  for( auto thit : tracktwo ) {
+//    htrack2->Fill( std::get<0>(thit).X(), std::get<0>(thit).Y());
+//  }
+
+
 
   for(auto t: ht->fHTRoots) {
     htheta->Fill(std::get<2>(t));
@@ -121,14 +145,23 @@ void hough_transform2(
   c->Divide(2,2);
   c->cd(1);
   hxy->Draw("box");
-  hpeaky->SetLineColor(2);
-  hpeaky->Draw("box,same");
+  for(auto ahist : alltrack){
+    ahist->Draw("box,same");
+  }
+  //htrack1->SetLineColor(2);
+  //htrack1->Draw("box,same");
+  //htrack2->SetLineColor(3);
+  //htrack2->Draw("box,same");
+  //hpeaky->Draw("box,same");
   c->cd(2);
   huv->Draw("box");
   c->cd(3);
   hxy2->Draw("box");
-  hpeaky2->SetLineColor(2);
-  hpeaky2->Draw("box,same");
+  for(auto ahist : vertextrack){
+    ahist->Draw("box,same");
+  }
+  //hpeaky2->SetLineColor(2);
+  //hpeaky2->Draw("box,same");
   c->cd(4);
   hphi->Draw();
   //hrphi->Draw("lego2");