diff --git a/src/GenFind/tests/hough_transform2.cxx b/src/GenFind/tests/hough_transform2.cxx index 64bac92f952b6f91e824ff48db3832fc2f156065..e273ac2bad17aaa0f3cfb326d09446ebc6be7824 100644 --- a/src/GenFind/tests/hough_transform2.cxx +++ b/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");