TridPageSlice.cxx

Go to the documentation of this file.

#include <TGeoManager.h>

#include "TridPageSlice.h"
#include "TridModelSlice.h"
#include "TridModelMaker.h"
#include "TridControl.h"
#include "TridGLFrame.h"
#include "TridFlatGLFrame.h"

#include "Midad/Base/PageProxy.h"
#include "Midad/Base/Mint.h"

#include "JobControl/JobC.h"
#include "MessageService/MsgService.h"

using namespace std;
using namespace SigC;

CVSID("$Id: TridPageSlice.cxx,v 1.4 2006/06/04 20:37:40 tagg Exp $");

static PageProxy<TridPageSlice> gsTridPageSliceProxy("TridSlice");

const double kWidth = 10;
const double kHeight = 3;


TridPageSlice::TridPageSlice() : TridPage()
{
  // Default view.
  fViewModesSupported = kView2D | kView3D;
  fViewMode = kView2D;
  fWindowName = "TridSlice";
  fFullPOV.Set(kWidth*0.5, 0, -kHeight*0.5, 9, 90, 0 );  
  fAutoPOV.Set(kWidth*0.5, 0, -kHeight*0.5, 9, 90, 0 );  
}

TridPageSlice::~TridPageSlice()
{
}


TObject* TridPageSlice::Init(Mint* mint, PageDisplay* pageDisplay, GuiBox& box)
{
  MSG("TriD",Msg::kDebug) << "TridPageSlice: Init()\n";
  return TridPage::Init(mint, pageDisplay, box);
}

TridGLFrame* TridPageSlice::CreateNewGLFrame( TridPage::ViewMode_t mode )
{
  TridPOV min(-kWidth   ,0, -kHeight,        0, 0,-180);
  TridPOV max(kWidth*2  ,0, +kHeight*2,   1000,90, 180);
  
  if(mode==kView2D) {
    MSG("TriD",Msg::kDebug) << "Changing to 2D" << endl;
    return new TridFlatGLFrame(this,*fGuiBox,fTridControl,min,max,1.5);
  };

  MSG("TriD",Msg::kDebug) << "Changing to 3D" << endl;
  return new TridGLFrame(this,*fGuiBox,fTridControl,min,max);
}


void TridPageSlice::CreateModels() 
{
  fModels.Clear();
  fTridControl->ClearPicked();
  fTridControl->ClearSelected();

  TridModelMaker maker;
  maker.Prepare(&(fMint->GetJobC().Mom));
  maker.CreateSliceModels(&(fMint->GetJobC().Mom),fModels);
}

TVector3 TridPageSlice::SliceColor(int c)
{
  // what color should slice c of n be?
  float f = (c%5)/4.;
  return GetColor(f);
}

void TridPageSlice::CreateSketches()
{  
  fModels.ClearAssociations();

  TridModel* basemodel;

  // First, loop through to get the start/stop times.
  double tlow = 1e19;
  double thigh = -1e19;
  Int_t  nslices = 0;

  TridModelItr itr1 = fModels.GetIterator();
  while( (basemodel = itr1.Next()) ) {    
    // Make a new sketch for this model.
    TridModelSlice* slice = dynamic_cast<TridModelSlice*>(basemodel);
    if(slice) {
      nslices++;
      cout << nslices << " " << slice->GetEarliestTime() << " " << slice->GetLatestTime() << endl;
      if(slice->GetEarliestTime()<tlow) tlow = slice->GetEarliestTime();
      if(slice->GetLatestTime()>thigh)  thigh = slice->GetLatestTime();     
    }
  }

  if(nslices==0) return;
  if(tlow>1) {
    MSG("TriD",Msg::kWarning) << "Strange slice times. tlow=" << tlow << " thigh=" << thigh << std::endl;
  }

  thigh = tlow + 10.0e-6; // 10 us.

  // Set Z limits
  double zlow=0;
  double zhigh=120; // plane 120 
  if(fContext.GetDetector()==Detector::kFar) {
    zhigh = 294;
  }

  int icolor = 0;

  // Now loop through again to draw.
  TridModelItr itr2 = fModels.GetIterator();
  while( (basemodel = itr2.Next()) ) {    
    // Make a new sketch for this model.
    TridModelSlice* slice = dynamic_cast<TridModelSlice*>(basemodel);

    if(slice) {
      //       cout << "Drawing slice " << slice->GetSlice() << endl;
      //       cout << "  Hits: " << slice->fHits.size() << endl; 
      //       cout << "  Time start: " << slice->GetEarliestTime()-tlow << endl;
      //       cout << "  Time start: " << slice->GetLatestTime()  -tlow << endl;
      //       cout << "  Color: " << icolor << endl;
      
      for(UInt_t i=0; i< slice->fHits.size(); i++) {
        double z = slice->fHits[i].fPlane - zlow;
        double t = slice->fHits[i].fTime - tlow;
        double x = (t) /(thigh-tlow) * kWidth;
        double y = (z) /(zhigh-zlow) * kHeight;
        double w = (50e-9) /(thigh-tlow) * kWidth;   // duration of a hit
        double h = (2)      /(zhigh-zlow) * kHeight;

        cout << "   Hit: " << z << "\t" << t 
          << "\t" << x << "\t" << y << endl;
        double depth = 0.1;
        
        TVector3 pos(x+w*0.5,0,-(y+h*0.5));
        TridSketch* sk = new TridSketchBox(pos,
                                           kv_x, w*0.5,
                                           kv_z, h*0.5,
                                           kv_y, depth/2.);
        sk->SetTime(t);
        sk->SetColor(SliceColor(icolor));
        fGLFrame->AddSketch(sk);
        fModels.AssociateModel(slice,sk->GetId());
      }
      
    }
    icolor++;
  }
}

void TridPageSlice::CreateScenery()
{  
  TridSketch::SetDefaultColor(fTridControl->GetForegroundColor());
  TridSketch::StartScenery();
      // Put a black quad behind everything so mouse movement works better.

  
  TridSketch* backplane = new TridSketchPlane
    ( TVector3(kWidth* 0.5, -1.0, kHeight*0.5),
      kv_x, kWidth  * 4,
      kv_z, kHeight * 4 
      );
  backplane->SetColor(fTridControl->GetBackgroundColor());
  fGLFrame->AddSketch(backplane);
  

  TridSketch* l;

  l = new TridSketchLine( TVector3(0, -0.1, 0),
                          TVector3(0, -0.1, -kHeight) );
  fGLFrame->AddSketch(l);

  l = new TridSketchLine( TVector3(0,      -0.1, 0),
                          TVector3(kWidth, -0.1, 0) );
  fGLFrame->AddSketch(l);

  TridSketchText* txt;
  for(int i=0;i<=10;i+=2) {
    l = new TridSketchLine( TVector3((double)i,      -0.1, 0),
                            TVector3((double)i,      -0.1, 0.1)
                            );
    fGLFrame->AddSketch(l);

    txt = new TridSketchText( TVector3((double)i, -0.1, 0.5), -kv_z, kv_y, 
                              Form("%d us",i) );
    txt->SetScale(0.2);
    txt->SetThickness(0.005);
    fGLFrame->AddSketch(txt);
  }
  
  l = new TridSketchLine( TVector3(   0, -0.1, 0 ),
                          TVector3(-0.2, -0.1, 0 ) );
  fGLFrame->AddSketch(l);

  txt = new TridSketchText( TVector3(-0.5, -0.1, 0), -kv_z, kv_y, "Plane 0");
  txt->SetScale(0.2);
  txt->SetThickness(0.005);
  fGLFrame->AddSketch(txt);

  l = new TridSketchLine( TVector3(   0, -0.1, -kHeight ),
                          TVector3(-0.2, -0.1, -kHeight ) );
  fGLFrame->AddSketch(l);

  txt = new TridSketchText( TVector3(-1.0, -0.1, -kHeight), -kv_z, kv_y, 
                            (fContext.GetDetector()==Detector::kFar)?("Plane 295"):("Plane 120"));
  txt->SetScale(0.2);
  txt->SetThickness(0.005);
  fGLFrame->AddSketch(txt);
  

  TridSketch::StopScenery();
}


void TridPageSlice::ModifySketches()
{
  // Go through all models. Modify their sketches to match the current
  // transparency and color maps.

  fColorHistogram->Reset();
  fTransHistogram->Reset();

  int nsketches =0;

  TridSketchListItr itr = fGLFrame->GetIterator();
  while( TridSketch* sketch = itr.Next() ) {
    int sketchnum = sketch->GetId();
    TridModel* model = fModels.GetModelFromSketch(sketchnum);

    if((sketch)&&(model)) {
      nsketches++;
 
      Double_t trans_x = fTridControl->GetRangedModelTrans(model);
      Double_t color_x = fTridControl->GetRangedModelColor(model);
      fColorHistogram->Fill(color_x);
      fTransHistogram->Fill(trans_x);
      sketch->SetTransparency(trans_x);

      // set color by number:
      sketch->SetColor(SliceColor(model->GetSlice()));

     // If picked, shine as white.
     if(fTridControl->IsPicked(model))
       sketch->SetColor(1,1,1);
     
     // Ditto selected.
     if(fTridControl->IsSelected(model))
       sketch->SetColor(1,1,1);
         

    } else {
      if(!(sketch->IsScenery())) {
        MSG("TriD",Msg::kError) << "Cannot find sketch or model "
                                << " ModelAdd:"     << model 
                                << " Model ID:"     << ((model) ? (model->GetId()) : 0)
                                << " ModelKey:"     << ((model) ? (model->GetSortKey()) : 0)
                                << " SketchNumber:" << sketchnum
                                << " SketchAddr: " << sketch << endl;
        //if(model) model->Print();
      }
    }
  }
}

---