TEveBoxSet* elliptic_boxset_cones(Float_t x=0, Float_t y=0, Float_t z=0, Int_t num=100, Bool_t register=kTRUE) { TEveManager::Create(); using namespace TMath; TEveManager::Create(); TEveStraightLineSet* lines = new TEveStraightLineSet("StraightLines"); lines->SetLineColor(kYellow); lines->SetLineWidth(2); TRandom r(0); TEveBoxSet* cones = new TEveBoxSet("EllipticConeSet"); cones->Reset(TEveBoxSet::kBT_EllipticCone, kTRUE, 64); cones->SetPickable(kTRUE); Float_t a = 40; // max distance between cones TEveVector dir, pos; Float_t theta, phi, height, rad; for (Int_t i=0; i<num; ++i) { theta = r.Uniform(0,TMath::Pi()); phi = r.Uniform (-TMath::Pi(), TMath::Pi()); height = r.Uniform(5, 15); rad = r.Uniform(3, 5); dir.Set(Cos(phi)*Cos(theta), Sin(phi)*Cos(theta), Sin(theta)); dir *= height; pos.Set(r.Uniform(-a,a), r.Uniform(-a, a), r.Uniform(-a, a)); cones->AddEllipticCone(pos, dir, rad, 0.5*rad, r.Uniform(0,360)); cones->DigitColor(r.Uniform(20, 255), r.Uniform(20, 255), r.Uniform(20, 255), r.Uniform(20, 255)); // draw axis line 30% longer than cone height TEveVector end = pos + dir*1.3f; lines->AddLine(pos.fX, pos.fY, pos.fZ, end.fX, end.fY, end.fZ); } // by default cone cap not drawn if (r.Integer(2)>0) cones->SetDrawConeCap(kTRUE); cones->RefitPlex(); TEveTrans& t = cones->RefMainTrans(); t.SetPos(x, y, z); gEve->AddElement(cones); gEve->AddElement(lines); gEve->Redraw3D(kTRUE); return cones; }
void jetcone() { TEveManager::Create(); using namespace TMath; TRandom r(0); // -- Set Constants Int_t nCones = 10; Int_t nTracks = 200; Float_t coneRadius = 0.4; Float_t length = 300.; // -- Define palette gStyle->SetPalette(1, 0); TEveRGBAPalette* pal = new TEveRGBAPalette(0, 500); // ----------------------------------------------------------------------- // -- Line sets // ----------------------------------------------------------------------- // -- Define cone center TEveStraightLineSet* axis = new TEveStraightLineSet("Cone Axis"); axis->SetLineColor(kGreen); axis->SetLineWidth(2); TEveStraightLineSet* tracksXYZ = new TEveStraightLineSet("StraightLinesXYZ"); tracksXYZ->SetLineColor(kRed); tracksXYZ->SetLineWidth(2); TEveStraightLineSet* tracksEtaPhi = new TEveStraightLineSet("StraightLinesEtaPhi"); tracksEtaPhi->SetLineColor(kYellow); tracksEtaPhi->SetLineWidth(2); TEveStraightLineSet* tracksSeedEtaPhi = new TEveStraightLineSet("StraightLinesEtaPhiSeed"); tracksSeedEtaPhi->SetLineColor(kBlue); tracksSeedEtaPhi->SetLineWidth(2); // ----------------------------------------------------------------------- // -- Draw track distribution in XYZ in TPC Volume +/-250, +/-250, +/-250 // ----------------------------------------------------------------------- for ( Int_t track=0; track < nTracks ; track++ ) { Float_t trackX = r.Uniform(-250.0, 250.0); Float_t trackY = r.Uniform(-250.0, 250.0); Float_t trackZ = r.Uniform(-250.0, 250.0); Float_t trackR = Sqrt(trackX*trackX + trackY*trackY + trackZ*trackZ); TEveVector trackDir(trackX/trackR, trackY/trackR ,trackZ/trackR); TEveVector trackEnd = trackDir * length; tracksXYZ->AddLine(0., 0., 0., trackEnd.fX, trackEnd.fY, trackEnd.fZ ); } // ----------------------------------------------------------------------- // -- Draw track distribution in eta phi in TPC Volume +/-0.9, {0, 2Pi} // ----------------------------------------------------------------------- for ( Int_t track=0; track < nTracks ; track++ ) { Float_t trackEta = r.Uniform(-0.9, 0.9); Float_t trackPhi = r.Uniform(0.0, TwoPi()); TEveVector trackDir( GetTEveVector(trackEta, trackPhi) ); TEveVector trackEnd = trackDir * length; if ( trackEta > coneRadius || trackEta < -coneRadius ) tracksEtaPhi->AddLine(0., 0., 0., trackEnd.fX, trackEnd.fY, trackEnd.fZ); else tracksSeedEtaPhi->AddLine(0., 0., 0., trackEnd.fX, trackEnd.fY, trackEnd.fZ); } // ----------------------------------------------------------------------- // -- Draw cones // ----------------------------------------------------------------------- for ( Int_t iter = 0; iter < nCones; ++iter ) { // -- Get Random ( eta ,phi ) Float_t coneEta = r.Uniform(-0.9, 0.9); Float_t conePhi = r.Uniform(0.0, TwoPi() ); // -- Primary vertx as origin TEveVector coneOrigin(0.0,0.0,0.0); // -- Get Cone Axis - axis line 10% longer than cone height TEveVector coneAxis ( GetTEveVector( coneEta, conePhi) ); coneAxis *= length * 1.1; axis->AddLine( 0., 0., 0., coneAxis.fX, coneAxis.fY, coneAxis.fZ ); // -- Draw jet cone TEveJetCone* jetCone = new TEveJetCone("JetCone"); jetCone->SetPickable(kTRUE); jetCone->SetCylinder( 250., 250. ); if ( (jetCone->AddCone( coneEta, conePhi, coneRadius ) ) != -1) gEve->AddElement( jetCone ); } // ----------------------------------------------------------------------- // -- Add cone axis gEve->AddElement(axis); // -- Add lines // gEve->AddElement(tracksXYZ); gEve->AddElement(tracksSeedEtaPhi); gEve->AddElement(tracksEtaPhi); // -- Load TPC geometry geomGentleTPC(); gEve->Redraw3D(kTRUE); return; }
void VizDB_scan_screen() { TEvePointSet *ps = 0; TEveStraightLineSet *ls = 0; //============================================================================ // Hits //============================================================================ ps = new TEvePointSet(); ps->SetMarkerColor(2); ps->SetMarkerSize(0.5); ps->SetMarkerStyle(2); gEve->InsertVizDBEntry("Hits", ps); ps = new TEvePointSet(); ps->SetMarkerColor(2); ps->SetMarkerSize(0.5); ps->SetMarkerStyle(2); gEve->InsertVizDBEntry("SIM Hits ITS", ps); ps = new TEvePointSet(); ps->SetMarkerColor(3); ps->SetMarkerSize(0.5); ps->SetMarkerStyle(2); gEve->InsertVizDBEntry("SIM Hits TPC", ps); ps = new TEvePointSet(); ps->SetMarkerColor(3); ps->SetMarkerSize(0.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("SIM Hits T0", ps); ps = new TEvePointSet(); ps->SetMarkerColor(2); ps->SetMarkerSize(0.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("SIM Hits FMD", ps); ps = new TEvePointSet(); ps->SetMarkerColor(2); ps->SetMarkerSize(.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("SIM Hits ACORDE", ps); ps = new TEvePointSet(); ps->SetMarkerColor(2); ps->SetMarkerSize(.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("SIM Hits EMCAL", ps); ps = new TEvePointSet(); ps->SetMarkerColor(2); ps->SetMarkerSize(.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("SIM Hits PMD", ps); ps = new TEvePointSet(); ps->SetMarkerColor(2); ps->SetMarkerSize(.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("SIM Hits TOF", ps); ps = new TEvePointSet(); ps->SetMarkerColor(7); ps->SetMarkerSize(.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("SIM Hits TRD", ps); ps = new TEvePointSet(); ps->SetMarkerColor(2); ps->SetMarkerSize(.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("SIM Hits VZERO", ps); //============================================================================ // Clusters //============================================================================ ps = new TEvePointSet(); ps->SetMarkerColor(2); ps->SetMarkerSize(0.5); ps->SetMarkerStyle(2); gEve->InsertVizDBEntry("Clusters", ps); ps = new TEvePointSet(); ps->SetMarkerColor(5); ps->SetMarkerSize(0.2); ps->SetMarkerStyle(2); gEve->InsertVizDBEntry("REC Clusters ITS", ps); ps = new TEvePointSet(); ps->SetMarkerColor(4); ps->SetMarkerSize(0.2); ps->SetMarkerStyle(2); gEve->InsertVizDBEntry("REC Clusters TPC", ps); ps = new TEvePointSet(); ps->SetMarkerColor(7); ps->SetMarkerSize(0.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("REC Clusters TRD", ps); ps = new TEvePointSet(); ps->SetMarkerColor(kOrange); ps->SetMarkerSize(0.5); ps->SetMarkerStyle(4); gEve->InsertVizDBEntry("REC Clusters TOF", ps); ps = new TEvePointSet(); ps->SetMarkerColor(4); ps->SetMarkerSize(0.2); ps->SetMarkerStyle(2); gEve->InsertVizDBEntry("REC Clusters HMPID", ps); ps = new TEvePointSet(); ps->SetMarkerColor(4); ps->SetMarkerSize(0.5); ps->SetMarkerStyle(2); gEve->InsertVizDBEntry("REC Clusters PHOS", ps); //============================================================================ // Primary vertex //============================================================================ // Combined vertex ls = new TEveStraightLineSet; ls->SetMarkerStyle(2); ls->SetMarkerColor(7); ls->SetLineColor(7); ls->SetLineWidth(3); gEve->InsertVizDBEntry("REC PVTX", ls); ls = new TEveStraightLineSet; ls->SetMarkerStyle(2); ls->SetMarkerColor(7); ls->SetLineColor(7); ls->SetLineWidth(1); gEve->InsertVizDBEntry("REC PVTX Ellipse", ls); ls = new TEveStraightLineSet; ls->SetMarkerStyle(2); ls->SetMarkerColor(7); ls->SetLineColor(7); ls->SetLineWidth(1); gEve->InsertVizDBEntry("REC PVTX Box", ls); // SPD vertex ls = new TEveStraightLineSet; ls->SetMarkerStyle(2); ls->SetMarkerColor(6); ls->SetLineColor(6); ls->SetLineWidth(3); gEve->InsertVizDBEntry("REC PVTX SPD", ls); ls = new TEveStraightLineSet; ls->SetMarkerStyle(2); ls->SetMarkerColor(6); ls->SetLineColor(6); ls->SetLineWidth(1); gEve->InsertVizDBEntry("REC PVTX Ellipse SPD", ls); ls = new TEveStraightLineSet; ls->SetMarkerStyle(2); ls->SetMarkerColor(6); ls->SetLineColor(6); ls->SetLineWidth(1); gEve->InsertVizDBEntry("REC PVTX Box SPD", ls); // TPC vertex ls = new TEveStraightLineSet; ls->SetMarkerStyle(2); ls->SetMarkerColor(5); ls->SetLineColor(5); ls->SetLineWidth(3); gEve->InsertVizDBEntry("REC PVTX TPC", ls); ls = new TEveStraightLineSet; ls->SetMarkerStyle(2); ls->SetMarkerColor(5); ls->SetLineColor(5); ls->SetLineWidth(1); gEve->InsertVizDBEntry("REC PVTX Ellipse TPC", ls); ls = new TEveStraightLineSet; ls->SetMarkerStyle(2); ls->SetMarkerColor(5); ls->SetLineColor(5); ls->SetLineWidth(1); gEve->InsertVizDBEntry("REC PVTX Box TPC", ls); //Tracks tl = new TEveTrackList("ESD Tracks"); tl->SetLineStyle(6); tl->SetMainColor(1); tl->SetLineWidth(1); gEve->InsertVizDBEntry("REC Tracks",tl); tl = new TEveTrackList("ESD Tracks MI"); tl->SetLineStyle(6); tl->SetMainColor(1); tl->SetLineWidth(1); gEve->InsertVizDBEntry("REC Tracks MI",tl); TEveElementList* el = new TEveElementList("ESD Tracks by category"); TEveTrackList *tltemp[7]; tltemp[0] = new TEveTrackList("Sigma < 3"); tltemp[0]->SetLineStyle(6); tltemp[0]->SetLineColor(1); tltemp[0]->SetLineWidth(1); el->AddElement(tltemp[0]); tltemp[1] = new TEveTrackList("3 < Sigma < 5"); tltemp[1]->SetLineStyle(6); tltemp[1]->SetLineColor(1); tltemp[1]->SetLineWidth(1); el->AddElement(tltemp[1]); tltemp[2] = new TEveTrackList("5 < Sigma"); tltemp[2]->SetLineStyle(6); tltemp[2]->SetLineColor(1); tltemp[2]->SetLineWidth(1); el->AddElement(tltemp[2]); tltemp[3] = new TEveTrackList("no ITS refit; Sigma < 5"); tltemp[3]->SetLineStyle(6); tltemp[3]->SetLineColor(1); tltemp[3]->SetLineWidth(1); el->AddElement(tltemp[3]); tltemp[4] = new TEveTrackList("no ITS refit; Sigma > 5"); tltemp[4]->SetLineStyle(6); tltemp[4]->SetLineColor(1); tltemp[4]->SetLineWidth(1); el->AddElement(tltemp[4]); tltemp[5] = new TEveTrackList("no TPC refit"); tltemp[5]->SetLineStyle(6); tltemp[5]->SetLineColor(1); tltemp[5]->SetLineWidth(1); el->AddElement(tltemp[5]); tltemp[6] = new TEveTrackList("ITS stand-alone"); tltemp[6]->SetLineStyle(6); tltemp[6]->SetLineColor(1); tltemp[6]->SetLineWidth(1); el->AddElement(tltemp[6]); el->SetVizTag("ESD Tracks by category"); gEve->AddElement(el); TEveElementList* el = new TEveElementList("ESD Tracks by anal cuts"); TEveTrackList *tlac[2]; tlac[0] = new TEveTrackList("Passed"); tlac[0]->SetLineStyle(6); tlac[0]->SetMainColor(1); tlac[0]->SetLineWidth(1); el->AddElement(tlac[0]); tlac[1] = new TEveTrackList("Rejected"); tlac[1]->SetLineStyle(6); tlac[1]->SetMainColor(1); tlac[1]->SetLineWidth(1); el->AddElement(tlac[1]); el->SetVizTag("ESD Tracks by anal cut"); gEve->AddElement(el); TEveElementList* el = new TEveElementList("ESD Tracklets SPD"); TEveTrackList *tlac[2]; tlac[0] = new TEveTrackList("Good"); tlac[0]->SetLineStyle(6); tlac[0]->SetMainColor(1); tlac[0]->SetLineWidth(1); el->AddElement(tlac[0]); tlac[1] = new TEveTrackList("Bad"); tlac[1]->SetLineStyle(6); tlac[1]->SetMainColor(1); tlac[1]->SetLineWidth(1); el->AddElement(tlac[1]); el->SetVizTag("ESD Tracklets SPD"); gEve->AddElement(el); }