Skip to content

Commit

Permalink
Updating StFcsTriggerSimMake to include StMuFcsCollection hit informa…
Browse files Browse the repository at this point in the history 
…tion for trigger reconstruction in simulation. This includes suggestions made by Dima
  • Loading branch information
@mrosales2016
Manuel A. Rosales Aguilar authored and mrosales2016 committed Jan 14, 2024
1 parent 08d209b commit a4099cf
Show file tree
Hide file tree
Showing 2 changed files with 142 additions and 46 deletions.
178 changes: 132 additions & 46 deletions StRoot/StSpinPool/StFcsTriggerSimMaker/StFcsTriggerSimMaker.cxx
View file
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,16 @@

#include "StRoot/RTS/src/TRG_FCS/fcs_trg_base.h"

#include "StMaker.h"
#include "StChain.h"
#include "StMuDSTMaker/COMMON/StMuDst.h"
#include "StMuDSTMaker/COMMON/StMuEvent.h"
#include "StMuDSTMaker/COMMON/StMuFcsCollection.h"
#include "StMuDSTMaker/COMMON/StMuFcsHit.h"
#include "StMuDSTMaker/COMMON/StMuTypes.hh"
#include "StMuDSTMaker/COMMON/StMuEvent.h"
#include "StMuDSTMaker/COMMON/StMuFcsCollection.h"

namespace {
enum {kMaxNS=2, kMaxDet=3, kMaxDep=24, kMaxCh=32, kMaxEcalDep=24, kMaxHcalDep=8, kMaxPresDep=4, kMaxLink2=2};
uint32_t fcs_trg_sim_adc[kMaxNS][kMaxDet][kMaxDep][kMaxCh] ;
Expand Down Expand Up @@ -184,17 +194,17 @@ int StFcsTriggerSimMaker::InitRun(int runNumber){
mTree->Branch("trg",&mTrg,"trg/I");
}
if(mQaHistFilename){
mQaHistFile=new TFile(mQaHistFilename,"RECREATE");
mTrgRate = new TH1F("FcsTrgRate","FcsTrgRate",mNTRG+1,0,mNTRG+1);
mQaHistFile=new TFile(mQaHistFilename,"RECREATE");
mTrgRate = new TH1F("FcsTrgRate","FcsTrgRate",mNTRG+1,0,mNTRG+1);
}

//Write Text event file & gainfile
FILE* gainfile=0;
FILE* gainfile2=0;
if(mFilename){
mFile = fopen(mFilename,"w");
gainfile=fopen("fcs_et_gain.txt","w");
gainfile2=fopen("fcs_et_gain2.txt","w");
mFile = fopen(mFilename,"w");
gainfile=fopen("fcs_et_gain.txt","w");
gainfile2=fopen("fcs_et_gain2.txt","w");
}

//Fill ETgain, GainCorr and Pedestal
Expand Down Expand Up @@ -266,11 +276,27 @@ int StFcsTriggerSimMaker::Finish(){
}

int StFcsTriggerSimMaker::Make(){
StEvent* event = (StEvent*)GetInputDS("StEvent");
if(!event) {LOG_ERROR << "StFcsTriggerSimMaker::Make did not find StEvent"<<endm; return kStErr;}
StEvent* event = nullptr;
event = (StEvent*)GetInputDS("StEvent");
if(!event) {LOG_INFO << "StFcsTriggerSimMaker::Make did not find StEvent"<<endm;}
mFcsColl = event->fcsCollection();
if(!mFcsColl) {LOG_ERROR << "StFcsTriggerSimMaker::Make did not find StEvent->StFcsCollection"<<endm; return kStErr;}
if(!mFcsColl) {LOG_INFO << "StFcsTriggerSimMaker::Make did not find StEvent->StFcsCollection"<<endm;}

StMuEvent* muevent = nullptr;
if((!event)||(!mFcsColl)){

LOG_INFO<<"No StEvent info available for StFcsTriggerSimMaker. "<< endm;

muevent = StMuDst::event();
mMuFcsColl = StMuDst::muFcsCollection();
if(muevent && mMuFcsColl){
LOG_INFO <<"Proceeding with StMuDst info to be used with StfcsTriggerSimMaker."<< endm;
}else{
LOG_ERROR << "StFcsTriggerSimMaker::Make did not find StEvent and MuEvent." << endm;
return kStErr;
}
}

mTrgSim->start_event();

//Feed ADC
Expand All @@ -279,43 +305,50 @@ int StFcsTriggerSimMaker::Make(){
memset(fcs_trg_sim_adc,0,sizeof(fcs_trg_sim_adc));
int n=0;
for(int det=0; det<=kFcsNDet; det++){
StSPtrVecFcsHit& hits = mFcsColl->hits(det);
int ns = mFcsDb->northSouth(det);
int ehp = mFcsDb->ecalHcalPres(det);
int nh = mFcsColl->numberOfHits(det);
for(int i=0; i<nh; i++){
StFcsHit* hit=hits[i];
unsigned short dep = hit->dep();
unsigned short ch = hit->channel();
//printf("ns=%1d ehp=%1d dep=%2d ch=%2d adc=%4d sim=\n",ns,ehp,dep,ch,hit->adc(0),mSimMode);
if(ehp<0 || ch>=32) continue;
fcs_trg_sim_adc[ns][ehp][dep][ch] = hit->adc(0);
if(mSimMode==0){
int ntb=hit->nTimeBin();
for(int t=0; t<ntb; t++){
int tb = hit->timebin(t);
if(tb>=mTrgTimebin-3 && tb<=mTrgTimebin+4){
data[tb-mTrgTimebin+3] = hit->adc(t);
//printf("tb=%3d i=%2d adc=%4d\n",tb,tb-mTrgTimebin+3,hit->adc(t));
}
}
mTrgSim->fill_event(ehp,ns,dep,ch,data,8) ;
}else{
data[1] = hit->adc(0)-1; //removing 1 to add at tb6
data[6] = 1; //add this so tb6>tb7
mTrgSim->fill_event(ehp,ns,dep,ch,data,8) ;
}
if(mFile) fprintf(mFile,"%2d %2d %2d %2d %5d\n",ns,ehp,dep,ch,hit->adc(0));
n++;
}

int ns = mFcsDb->northSouth(det);
int ehp = mFcsDb->ecalHcalPres(det);

if((event)&&(mFcsColl)){
StSPtrVecFcsHit& hits = mFcsColl->hits(det);

int nh = mFcsColl->numberOfHits(det);

for(int i=0; i<nh; i++){
StFcsHit* hit=hits[i];
unsigned short ch = hit->channel();

if(ehp<0 || ch>=32) continue;

feedADC(hit, ns, ehp, data);
n++;
}

}else if((muevent)&&(mMuFcsColl)){ //Use StMuDst info instead of StEvent for Trigger Reconstruction

int nh = mMuFcsColl->numberOfHits(det);
int det_hit_index = mMuFcsColl->indexOfFirstHit(det);

for(int i=0; i<nh; i++){

int hit_index = i + det_hit_index;
StMuFcsHit* hit = mMuFcsColl->getHit(hit_index);
unsigned short ch = hit->channel();

if(ehp<0 || ch>=32) continue;

feedADC(hit, ns, ehp, data);
n++;
}
}
}
if(mFile) fprintf(mFile,"%2d %2d %2d %2d %5d\n",-1,0,0,0,0);
LOG_INFO << Form("StFcsTriggerSimMaker feeded %d hits",n) << endm;;

//Run Trigger Simulation
// uint16_t dsm_out = fcs_trg_run(mTrgSelect, mDebug);
uint32_t dsm_out = mTrgSim->end_event();

//QA Tree
mFlt=0;
StarPrimaryMaker* pmkr= static_cast<StarPrimaryMaker*>(GetMaker("PrimaryMaker"));
Expand All @@ -327,7 +360,7 @@ int StFcsTriggerSimMaker::Make(){
}
mTcu=dsm_out;
if(mQaTreeFile) mTree->Fill();

//Results
LOG_INFO << Form("Output to TCU = 0x%08x Filter=0x%08x",mTcu,mFlt)<<endm;

Expand Down Expand Up @@ -364,17 +397,17 @@ int StFcsTriggerSimMaker::Make(){

if(mTrgRate) mTrgRate->Fill(mNTRG);
NTRG[mNTRG]++;
for(int i=0; i<mNTRG; i++){
if(trg[i]) {
if(mTrgRate) mTrgRate->Fill(i);
NTRG[i]++;
}
for(int i=0; i<mNTRG; i++){
if(trg[i]) {
if(mTrgRate) mTrgRate->Fill(i);
NTRG[i]++;
}
}

LOG_INFO << "Triggers = ";
for(int i=0; i<mNTRG; i++){ if(trg[i]) LOG_INFO << ctrg[i] << " ";}
LOG_INFO << endm;

return kStOK;
}

Expand Down Expand Up @@ -656,3 +689,56 @@ void StFcsTriggerSimMaker::readThresholdFile(){
void StFcsTriggerSimMaker::readThresholdDb(){
//to be implemented before run22 online DB moves to offline
}

void StFcsTriggerSimMaker::feedADC(StFcsHit* hit, int ns, int ehp, uint16_t data_array[]){

unsigned short dep = hit->dep();
unsigned short ch = hit->channel();

// printf("ns=%1d ehp=%1d dep=%2d ch=%2d adc=%4d sim=%d\n",ns,ehp,dep,ch,hit->adc(0),mSimMode);
fcs_trg_sim_adc[ns][ehp][dep][ch] = hit->adc(0);
if(mSimMode==0){
int ntb=hit->nTimeBin();
for(int t=0; t<ntb; t++){
int tb = hit->timebin(t);
if(tb>=mTrgTimebin-3 && tb<=mTrgTimebin+4){
data_array[tb-mTrgTimebin+3] = hit->adc(t);
// printf("tb=%3d i=%2d adc=%4d\n",tb,tb-mTrgTimebin+3,hit->adc(t));
}
}
mTrgSim->fill_event(ehp,ns,dep,ch,data_array,8) ;
}else{
data_array[1] = hit->adc(0)-1; //removing 1 to add at tb6
data_array[6] = 1; //add this so tb6>tb7
mTrgSim->fill_event(ehp,ns,dep,ch,data_array,8) ;
}
if(mFile) fprintf(mFile,"%2d %2d %2d %2d %5d\n",ns,ehp,dep,ch,hit->adc(0));

}

void StFcsTriggerSimMaker::feedADC(StMuFcsHit* hit, int ns, int ehp, uint16_t data_array[]){

unsigned short dep = hit->dep();
unsigned short ch = hit->channel();

// printf("ns=%1d ehp=%1d dep=%2d ch=%2d adc=%4d sim=%d\n",ns,ehp,dep,ch,hit->adc(0),mSimMode);
fcs_trg_sim_adc[ns][ehp][dep][ch] = hit->adc(0);
if(mSimMode==0){
int ntb=hit->nTimeBin();
for(int t=0; t<ntb; t++){
int tb = hit->timebin(t);
if(tb>=mTrgTimebin-3 && tb<=mTrgTimebin+4){
data_array[tb-mTrgTimebin+3] = hit->adc(t);
// printf("tb=%3d i=%2d adc=%4d\n",tb,tb-mTrgTimebin+3,hit->adc(t));
}
}
mTrgSim->fill_event(ehp,ns,dep,ch,data_array,8) ;
}else{
data_array[1] = hit->adc(0)-1; //removing 1 to add at tb6
data_array[6] = 1; //add this so tb6>tb7
mTrgSim->fill_event(ehp,ns,dep,ch,data_array,8) ;
}
if(mFile) fprintf(mFile,"%2d %2d %2d %2d %5d\n",ns,ehp,dep,ch,hit->adc(0));

}

10 changes: 10 additions & 0 deletions StRoot/StSpinPool/StFcsTriggerSimMaker/StFcsTriggerSimMaker.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,15 +8,19 @@
#define STAR_StFcsTriggerSimMaker_HH

#include "StMaker.h"
#include <stdint.h>

class StFcsDb;
class StFcsCollection;
class StMuFcsCollection;
class TFile;
class fcs_trg_base;
class link_t;
class geom_t;
class TTree;
class TH1F;
class StFcsHit;
class StMuFcsHit;

class StFcsTriggerSimMaker : public StMaker{
public:
Expand All @@ -42,13 +46,19 @@ class StFcsTriggerSimMaker : public StMaker{
void setEtGain(float v) {mEtFactor=v;}

fcs_trg_base* getTriggerEmu() {return mTrgSim;}

void feedADC(StFcsHit* hit, int ns, int ehp, uint16_t data_array[]);
void feedADC(StMuFcsHit* hit, int ns, int ehp, uint16_t data_array[]);


//this is for just running stage2 from macro
void runStage2(link_t ecal[], link_t hcal[], link_t pres[], geom_t &geo, link_t output[]);

private:
StFcsDb* mFcsDb=0;
StFcsCollection* mFcsColl=0;
StMuFcsCollection* mMuFcsColl = 0;

int mTrgSelect=0;
int mDebug=0;
char* mFilename=0;
Expand Down

0 comments on commit a4099cf

Please sign in to comment.