StFmsHitMaker.cxx

Go to the documentation of this file.

// ROOT
#include "TFile.h"
#include "TH1.h"
#include "TH2.h"
#include "TTree.h"


#include "StMessMgr.h"
#include "StEventTypes.h"
#include "StMuDSTMaker/COMMON/StMuTypes.hh"

#include "StFmsDbMaker/StFmsDbMaker.h"
#include "StFmsCollection.h"
#include "StFmsHit.h"
#include "StFmsHitMaker.h"
#include "StTriggerData2009.h"

// PSU-FMS package
#include "StFmsPointMaker/StFmsClusterFitter.h"

#include <iostream>
#include <fstream>
#include <stdlib.h>
using namespace std;

ClassImp(StFmsHitMaker);

StFmsHitMaker::StFmsHitMaker(const char* name) : StMaker(name){
    mFmsDbMaker = NULL;
    mFmsCollection = NULL;
    mMuFmsColl = NULL;
    LOG_INFO << "StFmsHitMaker initializing...."<<endm;
    LOG_DEBUG << "StFmsHitMaker::constructor." << endm;
}

StFmsHitMaker::~StFmsHitMaker(){
  
  LOG_INFO << "StFmsHitMaker::destructor." << endm;
}

void StFmsHitMaker::Clear(Option_t* option){
  LOG_DEBUG << "StFmsHitMaker::Clear()" << endm;
  StMaker::Clear(option);
}

int StFmsHitMaker::Init() {
  
    LOG_INFO<<"StFmsHitMaker::Init() "<<endm;

    return StMaker::Init();

}

int StFmsHitMaker::InitRun(Int_t runNumber) {
  
    LOG_INFO << "StFmsHitMaker::InitRun --run# changed to " << runNumber << endm;
    mFmsDbMaker = gStFmsDbMaker;
    mCurrentRunNumber = runNumber; //called by maker's method :InitRun(run); when the run# changes

    return kStOK;
}

int StFmsHitMaker::Make(){

    LOG_DEBUG<<"StFmsHitMaker::Make start"<<endm;
    int flag = 0;
    StTriggerData* triggerData = 0;

    //first try to get StTriggerData from StTriggerDataMaker (works for proudction) and create StFmsCollection
    TObjectSet *os = (TObjectSet*)GetDataSet("StTriggerData");
    if (os) {
        triggerData = (StTriggerData*)os->GetObject();
        if(triggerData){
            flag=1;
            // mFmsCollection = new StFmsCollection();
            LOG_DEBUG<<"StFmsHitMaker::Make Found StTriggerData from StTriggerDataMaker"<<endm;
        }
    }
  
    //2nd try to get StTriggerData from StEvent
    //but once FMS data is killed in StEvent, this will not work and all you see is empty data
    StEvent* stEvent = (StEvent*) GetInputDS("StEvent");
    //StEvent* stEvent = (StEvent*) GetInputDS("StEvent");
    if(flag==0){
        if(stEvent){
            triggerData = stEvent->triggerData();
            if(triggerData) {
                flag=2;
                LOG_DEBUG<<"StFmsHitMaker::Make Found StTriggerData from StEvent"<<endm;
            }
            else{
                mFmsCollection = stEvent->fmsCollection();
                if(mFmsCollection){
                    flag=3;
                    LOG_DEBUG<<"StFmsHitMaker::Make Found StFmsCollection from StEvent"<<endm;
                }
            }
        } //found StEvent
    }
    LOG_DEBUG<<"before checking MuDst, flag is: "<<flag<<endl;
  
    //3rd try to get StTriggerData from StMuEvent, works for produced data (.MuDst.root) --Yuxi
    StMuDst* muDst = (StMuDst*)GetInputDS("MuDst");
    if(flag==0){
        if(muDst){
        mCurrentRunNumber = muDst->event()->runNumber();
            triggerData = (StTriggerData*)StMuDst::event()->triggerData();
            if(triggerData){
                flag = 4; //Yuxi
                LOG_INFO<<"StFmsHitMaker::Make Found StFmsTriggerData in MuDst"<<endm;
            }
            else LOG_ERROR << "Finally, no StFmsTriggerData in MuDst " <<endm;
        }
    }
   
 
    LOG_DEBUG<<"after checking MuDst, flag is: "<<flag<<endm;
    //after this step triggerData is pointing to StTriggerData block of StEvent

    if(flag>0){
        mFmsCollection = new StFmsCollection();
        //create StFmsHit and add it to StFmsCollection
        for(unsigned short crt=1; crt<=4; crt++){
            for(unsigned short slot=1; slot<=16; slot++){
                for(unsigned short ch=0; ch<32; ch++){
                    unsigned short adc=0;
                    unsigned short tdc=0;
                    if(flag<=4){ //wont work when flag=3
                            adc=triggerData->fmsADC(crt,slot-1,ch);
                            tdc=triggerData->fmsTDC(crt,slot-1,ch);
                    }
                
                    if(adc>0 || tdc>0){
                    //  LOG_INFO<<"adc of crt "<<crt<<", slot "<<slot<<", channel "<<ch<<" is: "<<adc<<endm;
                    //  LOG_INFO<<"tdc=====================================================is: "<<tdc<<endm;
                            StFmsHit* hit = new StFmsHit();
                            if(!hit){
                                LOG_ERROR <<"Failed to create FMS hit, skip this hit."<<endm;
                                continue;
                            }
                        hit->setDetectorId(0);
                        hit->setChannel(0);
                        hit->setQtCrate(crt);
                        hit->setQtSlot(slot);
                        hit->setQtChannel(ch);
                        hit->setAdc(adc);
                        hit->setTdc(tdc);
                        hit->setEnergy(0.0);
                        mFmsCollection->addHit(hit);
                        if(GetDebug()>0) hit->print();
                    }
                }
                }   
            }


  if(flag>0){
    for(unsigned int i=0; i<mFmsCollection->numberOfHits(); i++){
      int d,c;
      StFmsHit* fmsHit = (mFmsCollection->hits())[i];
      int crt   =fmsHit->qtCrate();
      int slot  =fmsHit->qtSlot();
      int ch    =fmsHit->qtChannel();
      float adc =fmsHit->adc();
      mFmsDbMaker->getReverseMap(crt,slot,ch,&d,&c);
      float e=0.0;
      if(d>0 || c>0){
  float g1=mFmsDbMaker->getGain(d,c);
  float g2=mFmsDbMaker->getGainCorrection(d,c);
  e       =adc*g1*g2;
      }
      fmsHit->setDetectorId(d);
      fmsHit->setChannel(c);
      fmsHit->setEnergy(e);
      if(GetDebug()>0) fmsHit->print();
    }
  }

        LOG_INFO<<"StFmsHitMaker::Make got "<<mFmsCollection->numberOfHits()<<" hits in StFmsCollection"<<endm;
    

                        

    }//flag, received fms data from mudst, triggerdata, etc.

/*   
    //flag = 0; //yuxi debug
    if(flag==0) { //read hits created during production
        if(muDst){
            mMuFmsColl = StMuDst::muFmsCollection();
            if(mMuFmsColl){
                int nhits = mMuFmsColl->numberOfHits();
                LOG_DEBUG<<"StFmsHitMaker::Make Found "<<nhits<<" hits in MuDst"<<endm;
                cout<<"StFmsHitMaker::Make Found "<<nhits<<" hits in MuDst"<<endl;
                for(int i=0; i<nhits; i++){
                    StMuFmsHit* mHit = mMuFmsColl->getHit(i);
                    if(mHit && GetDebug()>0) mHit->print();
                    mHit->print();
                }
                flag=5;
            }
        }
    }
*/
    if(stEvent) {
        //Adding StFmsCollection to StEvent
        LOG_DEBUG<<"StFmsHitMaker::Make Adding StFmsCollection as FMSCOLLECTION"<<endm;
        stEvent->setFmsCollection(mFmsCollection);
    }

    return kStOk;
}

int StFmsHitMaker::Finish(){
  
  LOG_INFO << "StFmsHitMaker::Finish() " << endm;
  return kStOk;
}