StEEmcPool/StEEmcIUPi0/StEEmcIUMixEvent.cxx

#include "TMath.h"
#include "StEEmcIUMixEvent.h"

ClassImp(StEEmcIUMixEvent);
ClassImp(StEEmcIUMixEventHead);
ClassImp(StEEmcIUMixEventClust);

// ----------------------------------------------------------------------------
StEEmcIUMixEvent::StEEmcIUMixEvent()
{
  for ( Int_t ii=0;ii<90;ii++ )
    {
      pedEEmcDSM_HT[ii]=1;
      pedEEmcDSM_TP[ii]=1;    
    }

  for ( Int_t ii=0;ii<10;ii++ )
    {
      pedEEmcDSM_TP[ii+00]  = (ii%2)? 3 : 5;
      pedEEmcDSM_TP[ii+20]  = (ii%2)? 3 : 5;
      pedEEmcDSM_TP[ii+30]  = (ii%2)? 3 : 5;
      pedEEmcDSM_TP[ii+50]  = (ii%2)? 3 : 5;
      pedEEmcDSM_TP[ii+60]  = (ii%2)? 3 : 5;
      pedEEmcDSM_TP[ii+80]  = (ii%2)? 3 : 5;
    }
  Clear();
}

// ----------------------------------------------------------------------------
void StEEmcIUMixEvent::setEvent( StMuEvent *event )
{

  mEventId               = event->eventId();
  mEventNumber           = event->eventNumber();
  mRunId                 = event->runId();
  mRunNumber             = event->runNumber();
  mEventInfo             = event->eventInfo();
  mRunInfo               = event->runInfo();
  mL0trigger             = event->l0Trigger();
  mMuTriggerIdCollection = event->triggerIdCollection();
  mMagneticField         = event->magneticField();
  mBbcTrigger            = event->bbcTriggerDetector();
  mEmcTrigger            = event->emcTriggerDetector();

  mPrimaryVertex         = event->primaryVertexPosition();

}

void StEEmcIUMixEventHead::setEvent( StMuEvent *event )
{

  mEventId               = event->eventId();
  mEventNumber           = event->eventNumber();
  mRunId                 = event->runId();
  mRunNumber             = event->runNumber();

  mNumberOfCandidates    = 0;

}

// ----------------------------------------------------------------------------
void StEEmcIUMixEvent::Clear(Option_t *o)
{
  nPairs=0;
  for ( Int_t i=0;i<MAX_PAIRS;i++ )
  {
      mMass[i]=0.;
      mPT[i]=0.;
      mEta[i]=-10.;
      mPhi[i]=-10.;
      mEEmcEta[i]=-10.;
      mZgg[i]=0.;
      mPhigg[i]=0.; 
      mEnergy[i]=0.;
      mEpre1[i]=0.;
      mEpre2[i]=0.;
      mEpost[i]=0.;
      mEsmdu[i]=0.;
      mEsmdv[i]=0.;
      mZvertex[i]=0.;

      mTower1[i]=0;
      mTower2[i]=0;
      mEnergy1[i]=0.;
      mEnergy2[i]=0.;

      mEnergyRatio[i]=0.;

      mNumberT[i]=0;
      mNumberR[i]=0;
      mNumberU[i]=0;
      mNumberV[i]=0;
  } 
  for ( Int_t ii=0;ii<720;ii++ ) {
    mADC[ii]=0.;
    mStat[ii]=0;
  }
    
    

  mTotalEnergyT=0.;
  mTotalEnergyP=0.;
  mTotalEnergyQ=0.;
  mTotalEnergyR=0.;
  mTotalEnergyU=0.;
  mTotalEnergyV=0.; 

}

// ----------------------------------------------------------------------------
void StEEmcIUMixEvent::addPair ( StEEmcIUPair  p ) { 

    if ( nPairs >= MAX_PAIRS ) return;  
    mMass[nPairs]   = p.mass();
    mPT[nPairs]     = p.pt();
    mEta[nPairs]    = p.momentum().Eta();
    mPhi[nPairs]    = p.momentum().Phi(); 
    mZgg[nPairs]    = p.zgg();
    mPhigg[nPairs]  = p.phigg(); 
    mEnergy[nPairs] = p.energy();
    //calculate detector Eta
    float Rxy=TMath::Sqrt(p.vertex().x()*p.vertex().x()+p.vertex().y()*p.vertex().y());
    float hHeight=p.pt()*(270.0-p.vertex().Z())/p.pz()+Rxy;
    float etatheta=TMath::ATan(hHeight/270.0);
    //printf("accept pz=%f\n",pair.pz());
    float mideta=tan(etatheta/2.0);
    float eemceta=-log(mideta);
    mEEmcEta[nPairs]=eemceta;
    Float_t esmdu=0.;
    Float_t esmdv=0.; 
    esmdu += p.point(0).cluster(0).energy();
    esmdu += p.point(1).cluster(0).energy();
    esmdv += p.point(0).cluster(1).energy();
    esmdv += p.point(1).cluster(1).energy();
    Float_t epre1 = 0.;
    epre1 += p.point(0).energy(1);
    epre1 += p.point(1).energy(1);
    Float_t epre2 = 0.;
    epre2 += p.point(0).energy(2);
    epre2 += p.point(1).energy(2);
    Float_t epost = 0.;
    epost += p.point(0).energy(3);
    epost += p.point(1).energy(3);

    mEpre1[nPairs]=epre1;
    mEpre2[nPairs]=epre2;
    mEpost[nPairs]=epost;
    mEsmdu[nPairs]=esmdu;
    mEsmdv[nPairs]=esmdv; 

    mZvertex[nPairs]=p.vertex().Z(); 

    mTower1[nPairs] = p.point(0).tower(0).index();
    mEnergy1[nPairs]= p.point(0).energy();

    mTower2[nPairs] = p.point(1).tower(0).index();
    mEnergy2[nPairs] = p.point(1).energy(); 


    nPairs++; 

}

// ----------------------------------------------------------------------------
Float_t StEEmcIUMixEvent::sum3x3(Int_t index)
{
  Float_t sum=this->mADC[index];
  Int_t ieta = index%12;
  Int_t iphi = index/12;
  for ( Int_t jeta=ieta-1;jeta<=ieta+1;jeta++ ) {
    if ( jeta<0||jeta>11 ) continue;
    for ( Int_t jphi=iphi-1;jphi<=iphi+1;jphi++ ) {
      Int_t myeta=jeta;
      Int_t myphi=(jphi+60)%60;
      if ( myeta==ieta&&myphi==iphi ) continue;
      Float_t myEt=this->mADC[ myeta + 12*myphi ];
      sum += myEt;
    }
  }
  return sum;
}
Float_t StEEmcIUMixEvent::sum3x3()
{
  Float_t max=0.;
  Int_t index = 0;
  for ( Int_t ii=0;ii<720;ii++ )
    if ( this->mADC[ii] > max ) { 
      index=ii;
      max=this->mADC[ii];
    }
  return sum3x3(index);    
}
Float_t StEEmcIUMixEvent::htdsm()
{
  Float_t max=0.;
  for ( Int_t ii=0;ii<90;ii++ ) {
    Float_t dsm=mEmcTrigger.highTowerEndcap(ii);
    dsm-=pedEEmcDSM_HT[ii];
    if ( dsm>max ) max=dsm;
  }
  return max;
}
Float_t StEEmcIUMixEvent::tpdsm()
{
  Float_t max=0.;
  for ( Int_t ii=0;ii<90;ii++ ) {
    Float_t dsm=mEmcTrigger.patchEndcap(ii);
    dsm-=pedEEmcDSM_TP[ii];
    if ( dsm>max ) max=dsm;
  }
  return max;
}
Int_t StEEmcIUMixEvent::npi0()
{
  Int_t count=0;
  for ( Int_t ii=0;ii<TMath::Min(5,nPairs);ii++ )
    {
      if ( mMass[ii]>0.08 && mMass[ii]<0.18 ) count++;
    }
  return count;
}
Int_t StEEmcIUMixEvent::deta(Int_t id)
{
  if ( id<0||id>=nPairs)
    return -1;
  Int_t del = ( mTower1[id]%12 - mTower2[id]%12 );
  return TMath::Abs(del);
}
Int_t StEEmcIUMixEvent::dphi(Int_t id)
{
  if ( id<0||id>=nPairs)
    return -1;
  Int_t del = ( mTower1[id]/12 - mTower2[id]/12 + 60 ) % 60;
  return TMath::Abs(del);
}


// ----------------------------------------------------------------------------
void StEEmcIUMixEventHead::Clear(Option_t *opts)
{
  mEventId=0;
  mEventNumber=0;
  mRunId=0;
  mRunNumber=0;
  mNumberOfCandidates=0;
}

// ----------------------------------------------------------------------------
void StEEmcIUMixEventClust::Clear(Option_t *opts)
{
  index=-1;
  eTower=0.;
  ePre1=0.;
  ePre2=0.;
  ePost=0.;
  eSmd=0.;
  etaTower=0.;
  etaPre1=0.;
  etaPre2=0.;
  etaPost=0.;
  etaSmd=0.;
  nTower=0;
  nPre1=0;
  nPre2=0;
  nPost=0;
}

---