StFstFastSimMaker.cxx

#include "StFstFastSimMaker.h"

#include "St_base/StMessMgr.h"

#include "StEvent/StEvent.h"
#include "StEvent/StRnDHit.h"
#include "StEvent/StFstHit.h"
#include "StEvent/StFstHitCollection.h"
#include "StEvent/StRnDHitCollection.h"

#include "tables/St_g2t_fts_hit_Table.h"
#include "tables/St_g2t_track_Table.h"

#include "StarClassLibrary/StThreeVectorF.hh"
#include "StarGenerator/UTIL/StarRandom.h"

#include "TCanvas.h"
#include "TCernLib.h"
#include "TH2F.h"
#include "TLine.h"
#include "TRandom3.h"
#include "TString.h"
#include "TVector2.h"
#include "TVector3.h"

#include <array>
#define _USE_MATH_DEFINES
#include <cmath>
#include <map>
#include <algorithm>


// lets not polute the global scope
namespace FstGlobal{
        // converts the position error to a cov mat
    StMatrixF Hack1to6(const StHit *stHit);
        // used to convert between uniform and 1 sigma error
    constexpr float SQRT12 = sqrt(12.0);

    // Disk segmentation
    //
    float RMIN[] = {0.95 * 4.3, 0.95 * 4.3, 0.95 * 4.3, 0.95 * 5.0, 0.95 * 5.0, 0.95 * 5.0};
    float RMAX[] = {1.05 * 15.0, 1.05 * 25.0, 1.05 * 25.0, 1.05 * 28.0, 1.05 * 28.0, 1.05 * 28.0};

    //NEXT IS only for disk ARRAY 456 with the radius from 5 to 28.
    float RSegment[] = {5., 7.875, 10.75, 13.625, 16.5, 19.375, 22.25, 25.125, 28.};

    // controls some extra output
    const bool verbose = false;

        // key type for lookup tables on disk r-phi strip
        typedef std::tuple<int, int, int> FstKeyTriple;


}

StFstFastSimMaker::StFstFastSimMaker(const Char_t *name)
        : StMaker{name},
    mNumR{8},
    mNumPHI{128},
    mNumSEC{12},
    mInEff{0},
    mHist{false},
        mGEANTPassthrough{false},
    mQAFileName(0),
    hTrutHitYXDisk(0),
    hTrutHitRDisk(0),
    hTrutHitRShower{0},
    hTrutHitPhiDisk(0),
    hTrutHitPhiZ(0),
    hRecoHitYXDisk(0),
    hRecoHitRDisk(0),
    hRecoHitPhiDisk(0),
    hRecoHitPhiZ(0),
    hGlobalDRDisk(0),
    hGlobalZ(0),
    h2GlobalXY(0),
    h2GlobalSmearedXY(0),
    h2GlobalDeltaXY(0),
    h3GlobalDeltaXYDisk(0),
    h3GlobalDeltaXYR(0) { }

int StFstFastSimMaker::Init() {

        if(mHist){
                fOut = new TFile(mQAFileName.Data(), "RECREATE");
                AddHist(hTrutHitYXDisk = new TH3F("hTrutHitYXDisk", "Global hits before segmentation", 151, -75.5, 75.5, 151, -75.5, 75.5, 10, 0, 10));
                AddHist(hTrutHitRDisk = new TH2F("hTrutHitRDisk", "Global hits before segmentation", 400, 0, 40, 10, 0, 10));
                AddHist(hTrutHitRShower[0] = new TH2F("hTrutHitRShower_4", "Global hits before segmentation", 400, 0, 40, 20, -10, 10));
                AddHist(hTrutHitRShower[1] = new TH2F("hTrutHitRShower_5", "Global hits before segmentation", 400, 0, 40, 20, -10, 10));
                AddHist(hTrutHitRShower[2] = new TH2F("hTrutHitRShower_6", "Global hits before segmentation", 400, 0, 40, 20, -10, 10));
                AddHist(hTrutHitPhiDisk = new TH2F("hTrutHitPhiDisk", "Global hits before segmentation", 360, 0, 360, 10, 0, 10));
                AddHist(hTrutHitPhiZ = new TH2F("hTrutHitPhiZ", "Global hits before segmentation", 360, 0, 360, 6000, 0, 600));
                AddHist(hRecoHitYXDisk = new TH3F("hRecoHitYXDisk", "Global hits after segmentation", 151, -75.5, 75.5, 151, -75.5, 75.5, 10, 0, 10));
                AddHist(hRecoHitRDisk = new TH2F("hRecoHitRDisk", "Global hits after segmentation", 400, 0, 40, 10, 0, 10));
                AddHist(hRecoHitPhiDisk = new TH2F("hRecoHitPhiDisk", "Global hits after segmentation", 360, 0, 360, 10, 0, 10));
                AddHist(hRecoHitPhiZ = new TH2F("hRecoHitPhiZ", "Global hits after segmentation", 360, 0, 360, 6000, 0, 600));
                AddHist(hGlobalDRDisk = new TH2F("hGlobalDRDisk", "; Reco. r - MC r [cm]; Events;", 1000, -50, 50, 10, 0, 10));
                AddHist(hGlobalZ = new TH1F("hGlobalZ", "; Z [cm]; Events;", 6000, 0, 600));
                AddHist(h3GlobalDeltaXYR = new TH3F("h3GlobalDeltaXYR", ";globalDeltaX; globalDeltaY; R", 300, -0.3, 0.3, 300, -3, 3, 100, 0, 30));
                AddHist(h2GlobalXY = new TH2F("h2GlobalXY", ";globalX; globalY", 1510, -75.5, 75.5, 1510, -75.5, 75.5));
                AddHist(h2GlobalSmearedXY = new TH2F("h2GlobalSmearedXY", ";globalSmearedX; globalSmearedY", 1510, -75.5, 75.5, 1510, -75.5, 75.5));
                AddHist(h2GlobalDeltaXY = new TH2F("h2GlobalDeltaXY", ";globalDeltaX; globalDeltaY", 151, -75.5, 75.5, 151, -75.5, 75.5));
                AddHist(h3GlobalDeltaXYDisk = new TH3F("h3GlobalDeltaXYDisk", ";globalDeltaX; globalDeltaY; Disk", 151, -75.5, 75.5, 151, -75.5, 75.5, 10, 0, 10));
        }
        return StMaker::Init();
}

void StFstFastSimMaker::SetDisk(const int i, const float rmn, const float rmx) {
    FstGlobal::RMIN[i] = rmn;
    FstGlobal::RMAX[i] = rmx;
}

Int_t StFstFastSimMaker::Make() {
        LOG_DEBUG << "StFstFastSimMaker::Make" << endm;

        // Get the existing StEvent, or add one if it doesn't exist.
        StEvent *event = static_cast<StEvent *>(GetDataSet("StEvent"));
        if (!event) {
                event = new StEvent;
                AddData(event);
                LOG_DEBUG << "Creating StEvent" << endm;
        }

        if (0 == event->rndHitCollection()) {
                event->setRnDHitCollection(new StRnDHitCollection());
                LOG_DEBUG << "Creating StRnDHitCollection for FTS" << endm;
        }

        // Get pointer to an existing StFstHitCollection if any
        StFstHitCollection *fstHitCollection = event->fstHitCollection();
        // If no fst hit collection, create one
        if (!fstHitCollection) {
                fstHitCollection = new StFstHitCollection();
                event->setFstHitCollection(fstHitCollection);
                LOG_DEBUG << "Make() - Added new StFstHitCollection to this StEvent" << endm;
        }

        // Digitize GEANT FTS hits
        FillSilicon(event);

        return kStOk;
}

/* Fill an event with StFtsHits. */
/* This should fill StFtsStrip for realistic simulator and let clustering fill StFtsHit */
/* For now skipping StFtsStrip and clustering, and fill StFtsHits directly here*/

void StFstFastSimMaker::FillSilicon(StEvent *event) {

        StRnDHitCollection *fsicollection = event->rndHitCollection();

        const int MAXR = mNumR;
        const int MAXPHI = mNumPHI * mNumSEC;

        if ( mGEANTPassthrough ){
                LOG_INFO << "FST Hits using GEANT xyz directly (no raster etc.)" << endm;
        }
        
        // maps for hit and energy for each disk's r-phi strip
        std::map< FstGlobal::FstKeyTriple, StRnDHit* > hitMap;
        std::map< FstGlobal::FstKeyTriple, double > energySum;
        std::map< FstGlobal::FstKeyTriple, double > energyMax;

        // Read the g2t table
        St_g2t_fts_hit *hitTable = static_cast<St_g2t_fts_hit *>(GetDataSet("g2t_fsi_hit"));
        if (!hitTable) {
                LOG_INFO << "g2t_fsi_hit table is empty" << endm;
                return; // Nothing to do
        }

        const Int_t nHits = hitTable->GetNRows();
        LOG_DEBUG << "g2t_fsi_hit table has " << nHits << " hits" << endm;
        const g2t_fts_hit_st *hit = hitTable->GetTable();
        
        StPtrVecRnDHit hits;

        // track table
        St_g2t_track *trkTable = static_cast<St_g2t_track *>(GetDataSet("g2t_track"));
        if (!trkTable) {
                LOG_INFO << "g2t_track table is empty" << endm;
                return; // Nothing to do
        }

        const Int_t nTrks = trkTable->GetNRows();
        
        LOG_DEBUG << "g2t_track table has " << nTrks << " tracks" << endm;
        
        const g2t_track_st *trk = trkTable->GetTable();

        
        int count = 0;
        for (Int_t i = 0; i < nHits; ++i) {
                
                hit = (g2t_fts_hit_st *)hitTable->At(i);
                
                // skip bad hits
                if (!hit) 
                        continue;


                int volume_id = hit->volume_id;
                LOG_DEBUG << "volume_id = " << volume_id << endm;
                int disk = volume_id / 1000;         // disk id
                int wedge = (volume_id % 1000) / 10; // wedge id
                int sensor = volume_id % 10;         // sensor id
                
                // used as an index for various arrays
                size_t disk_index = disk - 1;

                LOG_DEBUG << "disk = " << disk << ", wedge = " << wedge << ", sensor = " << sensor << endm;

                // skip non-FST hits
                if ( disk > 6 ) continue;


                double energy = hit->de;
                int track = hit->track_p;

                trk = (g2t_track_st *)trkTable->At(track);
                int isShower = false;
                if (trk)
                        isShower = trk->is_shower;

                // This z-offset is used to shift the hits
                // to the center of the FST where the tracking planes are defined
                const double z_delta = 1.755;
                // hit coordinates
                double x = hit->x[0];
                double y = hit->x[1];
                double z = hit->x[2] + z_delta;

                if (z > 200)
                        continue; // skip large disks

                double r = sqrt(x * x + y * y);
                double p = atan2(y, x);

                // wrap an angle between 0 and 2pi
                auto wrapAngle = [&]( double angle ) {
                        angle = fmod( angle, 2.0 * M_PI );
                        if ( angle < 0 )
                                angle += 2.0 * M_PI;
                        return angle;
                };

                p = wrapAngle( p );

                LOG_DEBUG << "r = " << r << " p=" << p << endm;
                LOG_DEBUG << "RMIN = " << FstGlobal::RMIN[disk_index] << " RMAX= " << FstGlobal::RMAX[disk_index] << endm;

                // Cuts made on the r value to ensure it is within limits
                if (r < FstGlobal::RMIN[disk_index] || r > FstGlobal::RMAX[disk_index])
                        continue;

                // Strip numbers on rastered value
                int r_index = floor(MAXR * (r - FstGlobal::RMIN[disk_index]) / (FstGlobal::RMAX[disk_index] - FstGlobal::RMIN[disk_index]));
                for (int ii = 0; ii < MAXR; ii++)
                        if (r > FstGlobal::RSegment[ii] && r <= FstGlobal::RSegment[ii + 1])
                                r_index = ii;
                
                // Phi number
                int phi_index = int(MAXPHI * p / 2.0 / M_PI);

                if (r_index >= 8)
                        continue;

                if (MAXR)
                        assert(r_index < MAXR);
                if (MAXPHI)
                        assert(phi_index < MAXPHI);

                StRnDHit *fsihit = nullptr;

                // key of this disk's r & phi strip
                auto threeKey = std::tie( disk_index, r_index, phi_index );
                
                if (hitMap.count( threeKey ) == 0) { // New hit

                        if (FstGlobal::verbose){
                                LOG_INFO << Form("NEW d=%1d xyz=%8.4f %8.4f %8.4f r=%8.4f phi=%8.4f iR=%2d iPhi=%4d dE=%8.4f[MeV] truth=%d",
                                                disk, x, y, z, r, p, r_index, phi_index, energy * 1000.0, track)
                                        << endm;
                        }

                        count++;
                        fsihit = new StRnDHit();
                        fsihit->setDetectorId(kFtsId);
                        fsihit->setLayer(disk);
                        fsihit->setLadder(wedge);
                        fsihit->setWafer(sensor);
                        //
                        // Set position and position error based on radius-constant bins
                        //
                        double p0 = (phi_index + 0.5) * 2.0 * M_PI / double(MAXPHI);
                        double dp = 2.0 * M_PI / double(MAXPHI) / FstGlobal::SQRT12;
                        
                        // ONLY valid for the disk array 456, no difference for each disk
                        double r0 = (FstGlobal::RSegment[r_index] + FstGlobal::RSegment[r_index + 1]) * 0.5;
                        double dr = FstGlobal::RSegment[r_index + 1] - FstGlobal::RSegment[r_index];
                        
                        double x0 = r0 * cos(p0);
                        double y0 = r0 * sin(p0);
                        assert(TMath::Abs(x0) + TMath::Abs(y0) > 0);
                        double dz = 0.03 / FstGlobal::SQRT12;
                        double er = dr / FstGlobal::SQRT12;
                        fsihit->setPosition(StThreeVectorF(x0, y0, z));
                        // pass the GEANT hits through without modification
                        if ( mGEANTPassthrough ){
                                fsihit->setPosition(StThreeVectorF(x, y, z));
                        }
                        
                        fsihit->setPositionError(StThreeVectorF(er, dp, dz));
                        // set covariance matrix
                        fsihit->setErrorMatrix(&FstGlobal::Hack1to6(fsihit)[0][0]);

                        fsihit->setCharge(energy);
                        fsihit->setIdTruth(track, 100);
                        hits.push_back(fsihit);
                        
                        hitMap[ threeKey ] = fsihit;
                        energySum[ threeKey ] = energy;
                        energyMax[ threeKey ] = energy;

                        if (FstGlobal::verbose){
                                LOG_INFO << Form("NEW d=%1d xyz=%8.4f %8.4f %8.4f ", disk, x, y, z) << endm;
                                LOG_INFO << Form("smeared xyz=%8.4f %8.4f %8.4f ", fsihit->position().x(), fsihit->position().y(), fsihit->position().z()) << endm;
                        }

                        if(mHist){
                                TVector2 hitpos_mc(x, y);
                                TVector2 hitpos_rc(fsihit->position().x(), fsihit->position().y());

                                hTrutHitYXDisk->Fill(x, y, disk);
                                hTrutHitRDisk->Fill(hitpos_mc.Mod(), disk);
                                
                                if (disk == 4)
                                        hTrutHitRShower[0]->Fill(hitpos_mc.Mod(), isShower);
                                if (disk == 5)
                                        hTrutHitRShower[1]->Fill(hitpos_mc.Mod(), isShower);
                                if (disk == 6)
                                        hTrutHitRShower[2]->Fill(hitpos_mc.Mod(), isShower);

                                hTrutHitPhiDisk->Fill(hitpos_mc.Phi() * 180.0 / TMath::Pi(), disk);
                                hTrutHitPhiZ->Fill(hitpos_mc.Phi() * 180.0 / TMath::Pi(), z);
                                hRecoHitYXDisk->Fill(fsihit->position().x(), fsihit->position().y(), disk);
                                hRecoHitRDisk->Fill(hitpos_rc.Mod(), disk);
                                hRecoHitPhiDisk->Fill(hitpos_rc.Phi() * 180.0 / TMath::Pi(), disk);
                                hRecoHitPhiZ->Fill(hitpos_rc.Phi() * 180.0 / TMath::Pi(), z);
                                hGlobalDRDisk->Fill(hitpos_rc.Mod() - hitpos_mc.Mod(), disk);
                                hGlobalZ->Fill(fsihit->position().z());

                                // cout << "CHECK : " << fsihit->position().x()-x << " |||  "<<  fsihit->position().y()-y << endl;
                                h2GlobalXY->Fill(x, y);
                                h2GlobalSmearedXY->Fill(fsihit->position().x(), fsihit->position().y());
                                h2GlobalDeltaXY->Fill(fsihit->position().x() - x, fsihit->position().y() - y);
                                h3GlobalDeltaXYDisk->Fill(fsihit->position().x() - x, fsihit->position().y() - y, disk);

                                h3GlobalDeltaXYR->Fill(fsihit->position().x() - x, fsihit->position().y() - y, sqrt(pow(fsihit->position().x(), 2) + pow(fsihit->position().y(), 2)));
                        }
                }
                else { // Hit on this strip already exists, adding energy to old hit
                        // get hit from the map
                        fsihit = hitMap[ threeKey ] ;
                        fsihit->setCharge(fsihit->charge() + energy);

                        // Add energy to running sum
                        energySum[ threeKey ] = energySum[ threeKey ] + energy;

                        if (energy> energyMax[ threeKey ])
                                energyMax[ threeKey ] = energy;

                        // keep idtruth but dilute it...
                        track = fsihit->idTruth();

                        fsihit->setIdTruth(track, 100 * (energyMax[ threeKey ] / energySum[ threeKey ]));
                }
        } // loop on hits
        int nfsihit = hits.size();

        StarRandom &rand = StarRandom::Instance();
        LOG_INFO << "FST Fast simulator is using mInEff = " << mInEff << endm;
        // NOW run back through the hits and add them if they pass an efficiency roll
        for (int i = 0; i < nfsihit; i++) {
                double rnd_save = rand.flat();
                if (rnd_save > mInEff || mGEANTPassthrough){
                        fsicollection->addHit(hits[i]);

                        StFstHit *fHit = new StFstHit(hits[i]->position(), hits[i]->positionError(), 0, hits[i]->charge(), 0);
                        fHit->setIdTruth(hits[i]->idTruth());
                        float r = sqrt(hits[i]->position().x() * hits[i]->position().x() + hits[i]->position().y() * hits[i]->position().y());
                        float phi = atan2(hits[i]->position().y(), hits[i]->position().x());
                        fHit->setLocalPosition( r, phi, hits[i]->position().z() );
                        fHit->setDiskWedgeSensor(hits[i]->layer(), hits[i]->ladder(), hits[i]->wafer());
                        event->fstHitCollection()->addHit(fHit);
                } else {
                }
        }
        if (FstGlobal::verbose) {
                LOG_DEBUG << Form("Found %d/%d g2t hits in %d cells, created %d hits with ADC>0 and put %d into StFstHitCollection", count, nHits, nfsihit, fsicollection->numberOfHits(), event->fstHitCollection()->numberOfHits()) << endm;
        }

}
//

int StFstFastSimMaker::Finish() {
        if(mHist){
                fOut->cd();
                hTrutHitYXDisk->Write();
                hTrutHitRDisk->Write();
                hTrutHitRShower[0]->Write();
                hTrutHitRShower[1]->Write();
                hTrutHitRShower[2]->Write();
                hTrutHitPhiDisk->Write();
                hTrutHitPhiZ->Write();
                hRecoHitYXDisk->Write();
                hRecoHitRDisk->Write();
                hRecoHitPhiDisk->Write();
                hRecoHitPhiZ->Write();
                hGlobalDRDisk->Write();
                hGlobalZ->Write();
                h3GlobalDeltaXYR->Write();
                h2GlobalXY->Write();
                h2GlobalSmearedXY->Write();
                h2GlobalDeltaXY->Write();
                h3GlobalDeltaXYDisk->Write();
                fOut->Close();
        }
        return kStOK;
}

//_____________________________________________________________________________
StMatrixF FstGlobal::Hack1to6(const StHit *stHit) {
        //   X = R*cos(Fi), Y=R*sin(Fi), Z = z
        //   dX/dR  = (    cos(Fi)  ,sin(Fi),0)
        //   dX/dFi = (-R*sin(Fi), R*cos(Fi),0)
        //   dX/dZ  = (         0,         0,1)

        auto hiPos = stHit->position();
        auto hiErr = stHit->positionError();
        double Rxy = sqrt(hiPos[0] * hiPos[0] + hiPos[1] * hiPos[1]);
        double cosFi = hiPos[0] / Rxy;
        double sinFi = hiPos[1] / Rxy;
        double T[3][3] = {{cosFi, -Rxy * sinFi, 0}, {sinFi, Rxy * cosFi, 0}, {0, 0, 1}};
        double Ginp[6] = {hiErr[0] * hiErr[0], 0, hiErr[1] * hiErr[1], 0, 0, hiErr[2] * hiErr[2]};
        double Gout[6];

        TCL::trasat(T[0], Ginp, Gout, 3, 3);
        StMatrixF mtxF(3, 3);

        for (int i = 0, li = 0; i < 3; li += ++i) {
                for (int j = 0; j <= i; j++) {
                        mtxF[i][j] = Gout[li + j];
                        mtxF[j][i] = mtxF[i][j];
                }
        }

        return mtxF;
}