#include <WTAObject.h>

Inheritance diagram for WTAJet:

Collaboration diagram for WTAJet:

Public Member Functions
	WTAJet (pt_t pt=0, eta_t eta=0, phi_t phi=0, m_t m=0, int idx=-99, tech_t jet_dr=R_PAR)
const WTATrigObj &	GetSeed () const
void	MergeConstituent (WTATrigObj &tower)
void	MergeConstituent (WTATrigObj *tower)
void	MergeWTAJet (WTAJet &targetjet)
void	PopOutLastConstituent ()
const std::vector< WTATrigObj > &	GetConstituentList () const
tobn_t	GetConstituentCount () const
unsigned int	GetRingN (const WTATrigObj &tower) const
void	CreateERingInfo ()
void	PrintERingInfo () const
const WTA4JetERingInfo &	GetERingInfo () const
void	SetERingInfo (const WTA4JetERingInfo &ering_info)
pt_t	pt () const
void	pt (pt_t pt)
eta_t	eta () const
void	eta (eta_t eta)
phi_t	phi () const
void	phi (phi_t phi)
m_t	m () const
void	m (m_t m)
int	idx () const
void	idx (int idx)
phi_t	d_phi_MPI_PI (const WTATrigObj &o2) const
eta_t	d_eta (const WTATrigObj &o2) const
bool	IsAssocdR (WTATrigObj &tower, tech_t dr)
WTATrigObj	operator+ (const WTATrigObj &obj)

Private Attributes
WTATrigObj	m_Seed
std::vector< WTATrigObj >	m_ConstituentList
tech_t	m_jet_dr
unsigned int	m_ring_n
WTA4JetERingInfo	m_ERingInfo
pt_t	m_pt
eta_t	m_eta
phi_t	m_phi
m_t	m_m
int	m_idx

Detailed Description

Definition at line 150 of file WTAObject.h.

Constructor & Destructor Documentation

◆ WTAJet()

WTAJet::WTAJet	(	pt_t	pt = 0,
		eta_t	eta = 0,
		phi_t	phi = 0,
		m_t	m = 0,
		int	idx = -99,
		tech_t	jet_dr = R_PAR )

inline

Definition at line 152 of file WTAObject.h.

                                                                                                        :
        WTATrigObj(pt, eta, phi, m, idx) {
            m_Seed = WTATrigObj(pt, eta, phi, m, idx);
            m_jet_dr = jet_dr;
            m_ConstituentList.clear();
            m_ConstituentList.push_back(m_Seed);
            #ifdef FLOATING_POINT_SIMULATION
                m_ring_n = static_cast<unsigned int>(jet_dr / 0.1);
            #else
                m_ring_n = jet_dr;
            #endif
        };

Member Function Documentation

◆ CreateERingInfo()

void WTAJet::CreateERingInfo ( )

inline

Definition at line 240 of file WTAObject.h.

                                   {
    // This function is ideally called after all jet clustering is done
 
    // Check if the constituent list is empty
    if(m_ConstituentList.empty()){
        std::cerr << "Error: Constituent list is empty. Cannot create ERingInfo." << std::endl;
        return;
    }
 
    // Reset the ERingInfo
    WTA4JetERingInfo ering_info;
    m_ERingInfo = ering_info;
 
    for(const auto& tower:m_ConstituentList){
        unsigned int ring_n = this->GetRingN(tower);
        m_ERingInfo.total_TobN = m_ERingInfo.total_TobN + 1;
        switch(ring_n){
            case 0:
                m_ERingInfo.ring0_Et = m_ERingInfo.ring0_Et + tower.pt();
                m_ERingInfo.ring0_TobN = m_ERingInfo.ring0_TobN + 1;
                break;
            case 1:
                m_ERingInfo.ring1_Et = m_ERingInfo.ring1_Et + tower.pt();
                m_ERingInfo.ring1_TobN = m_ERingInfo.ring1_TobN + 1;
                break;
            case 2:
                m_ERingInfo.ring2_Et = m_ERingInfo.ring2_Et + tower.pt();
                m_ERingInfo.ring2_TobN = m_ERingInfo.ring2_TobN + 1;
                break;
            case 3:
                m_ERingInfo.ring3_Et = m_ERingInfo.ring3_Et + tower.pt();
                m_ERingInfo.ring3_TobN = m_ERingInfo.ring3_TobN + 1;
                break;
            case 4:
                m_ERingInfo.ring4_Et = m_ERingInfo.ring4_Et + tower.pt();
                m_ERingInfo.ring4_TobN = m_ERingInfo.ring4_TobN + 1;
                break;
            default:
                break;
        }
    }
}

◆ d_eta()

eta_t WTATrigObj::d_eta ( const WTATrigObj & o2 ) const

inlineinherited

Definition at line 46 of file WTAObject.h.

46{eta_t tmp = m_eta - o2.eta(); return tmp;}

WTATrigObj::m_eta

eta_t m_eta

Definition WTAObject.h:67

DeMoUpdate.tmp

str tmp

Definition DeMoUpdate.py:1167

python.changerun.o2

o2

Definition changerun.py:43

◆ d_phi_MPI_PI()

phi_t WTATrigObj::d_phi_MPI_PI ( const WTATrigObj & o2 ) const

inlineinherited

Definition at line 90 of file WTAObject.h.

{
    phi_t tmp;
    if (m_phi >= o2.phi()){
        tmp = m_phi - o2.phi();
    } else {
        tmp = o2.phi() - m_phi;
    }
 
    // Wrap difference to [-32, +31]
    if (tmp < -HALF_PHI_LEN) tmp = tmp + PHI_LEN;
    else if (tmp >= HALF_PHI_LEN) tmp = tmp - PHI_LEN;
 
    return tmp;
}

◆ eta() [1/2]

eta_t WTATrigObj::eta ( ) const

inlineinherited

Definition at line 33 of file WTAObject.h.

33{return m_eta;};

◆ eta() [2/2]

void WTATrigObj::eta ( eta_t eta )

inlineinherited

Definition at line 34 of file WTAObject.h.

34{m_eta = eta;};

◆ GetConstituentCount()

tobn_t WTAJet::GetConstituentCount ( ) const

inline

Definition at line 171 of file WTAObject.h.

171{return static_cast<tobn_t>(m_ConstituentList.size());}; // Match the type

◆ GetConstituentList()

const std::vector< WTATrigObj > & WTAJet::GetConstituentList ( ) const

inline

Definition at line 170 of file WTAObject.h.

170{return m_ConstituentList;};

◆ GetERingInfo()

const WTA4JetERingInfo & WTAJet::GetERingInfo ( ) const

inline

Definition at line 176 of file WTAObject.h.

176{return m_ERingInfo;};

◆ GetRingN()

unsigned int WTAJet::GetRingN ( const WTATrigObj & tower ) const

inline

Definition at line 231 of file WTAObject.h.

                                                                  {
    eta_t d_eta = this->eta() > tower.eta() ? this->d_eta(tower) : tower.d_eta(*this); // ensure positive d_eta d_phi
    phi_t raw_dphi = abs(int32_t(this->phi()) - int32_t(tower.phi()));
    phi_t d_phi = std::min(raw_dphi, PHI_LEN - raw_dphi);
    unsigned int ring_n = std::max(d_eta, static_cast<eta_t>(d_phi)); // No .raw() in long long int
    return ring_n;
}

◆ GetSeed()

const WTATrigObj & WTAJet::GetSeed ( ) const

inline

Definition at line 165 of file WTAObject.h.

165{return m_Seed;};

◆ idx() [1/2]

int WTATrigObj::idx ( ) const

inlineinherited

Definition at line 39 of file WTAObject.h.

39{return m_idx;};

WTATrigObj::m_idx

int m_idx

Definition WTAObject.h:70

◆ idx() [2/2]

void WTATrigObj::idx ( int idx )

inlineinherited

Definition at line 40 of file WTAObject.h.

40{m_idx = idx;};

◆ IsAssocdR()

bool WTATrigObj::IsAssocdR	(	WTATrigObj &	tower,
		tech_t	dr )

inlineinherited

Definition at line 115 of file WTAObject.h.

                                                             {
    assert(tower.eta() >= 0 && "Input tower.eta() must be non-negative");
    assert(tower.phi() >= 0 && "Input tower.phi() must be non-negative");
    assert(dr >= 0 && "Input dr must be non-negative");
    eta_t d_eta = this->eta() > tower.eta() ? this->d_eta(tower) : tower.d_eta(*this); // ensure positive d_eta d_phi
    phi_t raw_dphi = abs(int32_t(this->phi()) - int32_t(tower.phi()));
    phi_t d_phi = std::min(raw_dphi, PHI_LEN - raw_dphi);
    bool assoc = false;
    for (unsigned int i = 0; i <= dr; i++){
        unsigned int max_j = sqrt(static_cast<double>(dr*dr - i*i)); // Need to be careful with the sqrt
        for(unsigned int j = 0; j <= max_j; j++){
            if(d_eta <= i && d_phi <= j){
                assoc = true;
                break;
            }
        }
    }
    return assoc;
}

◆ m() [1/2]

m_t WTATrigObj::m ( ) const

inlineinherited

Definition at line 37 of file WTAObject.h.

37{return m_m;};

WTATrigObj::m_m

m_t m_m

Definition WTAObject.h:69

◆ m() [2/2]

void WTATrigObj::m ( m_t m )

inlineinherited

Definition at line 38 of file WTAObject.h.

38{m_m = m;};

◆ MergeConstituent() [1/2]

void WTAJet::MergeConstituent ( WTATrigObj & tower )

inline

Definition at line 187 of file WTAObject.h.

{
    this->pt(this->pt() + tower.pt());// Scalar sum, WTA
    m_ConstituentList.push_back(tower);
}

◆ MergeConstituent() [2/2]

void WTAJet::MergeConstituent ( WTATrigObj * tower )

inline

Definition at line 193 of file WTAObject.h.

{
    this->pt(this->pt() + tower->pt());// Scalar sum, WTA
    m_ConstituentList.push_back(*tower);
}

◆ MergeWTAJet()

void WTAJet::MergeWTAJet ( WTAJet & targetjet )

inline

Definition at line 199 of file WTAObject.h.

{
    WTATrigObj tmptower(targetjet.pt(), targetjet.eta(), targetjet.phi(), targetjet.m());
    this->pt(this->pt() + tmptower.pt()); // Scalar sum, WTA
    std::vector<WTATrigObj> tmp_const_list = targetjet.GetConstituentList();
    m_ConstituentList.insert(m_ConstituentList.end(), tmp_const_list.begin(), tmp_const_list.end());
}

◆ operator+()

WTATrigObj WTATrigObj::operator+ ( const WTATrigObj & obj )

inlineinherited

Definition at line 56 of file WTAObject.h.

                                                     {
            WTATrigObj new_obj;
            new_obj.m_pt = this->pt() + obj.pt(); // Current default is the scalar sum, WTA
            new_obj.m_eta = this->eta();
            new_obj.m_phi = this->phi();
            new_obj.m_m = this->m();
            return new_obj;
        };

◆ phi() [1/2]

phi_t WTATrigObj::phi ( ) const

inlineinherited

Definition at line 35 of file WTAObject.h.

35{return m_phi;};

◆ phi() [2/2]

void WTATrigObj::phi ( phi_t phi )

inlineinherited

Definition at line 36 of file WTAObject.h.

36{m_phi = phi;};

◆ PopOutLastConstituent()

void WTAJet::PopOutLastConstituent ( )

inline

Definition at line 207 of file WTAObject.h.

{
    WTATrigObj lastconst = m_ConstituentList.back(); //std::vector::back() returns the last element
    this->pt(this->pt() - lastconst.pt()); // Scalar subtraction, WTA
    m_ConstituentList.pop_back(); // Popback the last element
}

◆ PrintERingInfo()

void WTAJet::PrintERingInfo ( ) const

inline

Definition at line 283 of file WTAObject.h.

                                         {
    std::cout << "WTAJet ERing Info:" << std::endl;
    std::cout << " Raw Jet Et: " << this->pt() << " , Constituent Count: " << this->GetConstituentCount() << std::endl;
    std::cout << "  Total TobN: " << m_ERingInfo.total_TobN << std::endl;
    std::cout << "  Ring 0: Et = " << m_ERingInfo.ring0_Et << ", TobN = " << m_ERingInfo.ring0_TobN << std::endl;
    std::cout << "  Ring 1: Et = " << m_ERingInfo.ring1_Et << ", TobN = " << m_ERingInfo.ring1_TobN << std::endl;
    std::cout << "  Ring 2: Et = " << m_ERingInfo.ring2_Et << ", TobN = " << m_ERingInfo.ring2_TobN << std::endl;
    std::cout << "  Ring 3: Et = " << m_ERingInfo.ring3_Et << ", TobN = " << m_ERingInfo.ring3_TobN << std::endl;
    std::cout << "  Ring 4: Et = " << m_ERingInfo.ring4_Et << ", TobN = " << m_ERingInfo.ring4_TobN << std::endl;
}

◆ pt() [1/2]

pt_t WTATrigObj::pt ( ) const

inlineinherited

Definition at line 31 of file WTAObject.h.

31{return m_pt;};

◆ pt() [2/2]

void WTATrigObj::pt ( pt_t pt )

inlineinherited

Definition at line 32 of file WTAObject.h.

32{m_pt = pt;};

◆ SetERingInfo()

void WTAJet::SetERingInfo ( const WTA4JetERingInfo & ering_info )

inline

Definition at line 177 of file WTAObject.h.

177{m_ERingInfo = ering_info;}; // When parsing the ref jet, ERingInfo does not get created automatically

Member Data Documentation

◆ m_ConstituentList

std::vector<WTATrigObj> WTAJet::m_ConstituentList

private

Definition at line 181 of file WTAObject.h.

◆ m_ERingInfo

WTA4JetERingInfo WTAJet::m_ERingInfo

private

Definition at line 184 of file WTAObject.h.

◆ m_eta

eta_t WTATrigObj::m_eta

privateinherited

Definition at line 67 of file WTAObject.h.

◆ m_idx

int WTATrigObj::m_idx

privateinherited

Definition at line 70 of file WTAObject.h.

◆ m_jet_dr

tech_t WTAJet::m_jet_dr

private

Definition at line 182 of file WTAObject.h.

◆ m_m

m_t WTATrigObj::m_m

privateinherited

Definition at line 69 of file WTAObject.h.

◆ m_phi

phi_t WTATrigObj::m_phi

privateinherited

Definition at line 68 of file WTAObject.h.

◆ m_pt

pt_t WTATrigObj::m_pt

privateinherited

Definition at line 66 of file WTAObject.h.

◆ m_ring_n

unsigned int WTAJet::m_ring_n

private

Definition at line 183 of file WTAObject.h.

◆ m_Seed

WTATrigObj WTAJet::m_Seed

private

Definition at line 180 of file WTAObject.h.

The documentation for this class was generated from the following file:

WTAObject.h

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ WTAJet()

Member Function Documentation

◆ CreateERingInfo()

◆ d_eta()

◆ d_phi_MPI_PI()

◆ eta() [1/2]

◆ eta() [2/2]

◆ GetConstituentCount()

◆ GetConstituentList()

◆ GetERingInfo()

◆ GetRingN()

◆ GetSeed()

◆ idx() [1/2]

◆ idx() [2/2]

◆ IsAssocdR()

◆ m() [1/2]

◆ m() [2/2]

◆ MergeConstituent() [1/2]

◆ MergeConstituent() [2/2]

◆ MergeWTAJet()

◆ operator+()

◆ phi() [1/2]

◆ phi() [2/2]

◆ PopOutLastConstituent()

◆ PrintERingInfo()

◆ pt() [1/2]

◆ pt() [2/2]

◆ SetERingInfo()

Member Data Documentation

◆ m_ConstituentList

◆ m_ERingInfo

◆ m_eta

◆ m_idx

◆ m_jet_dr

◆ m_m

◆ m_phi

◆ m_pt

◆ m_ring_n

◆ m_Seed