ATLAS Offline Software
Public Member Functions | Public Attributes | Private Member Functions | Private Attributes | List of all members
FakeBkgTools::FinalState Class Reference

#include <FakeBkgInternals.h>

Collaboration diagram for FakeBkgTools::FinalState:

Public Member Functions

 FinalState (size_t h)
 
 FinalState (size_t h, const unsigned nparticles, const std::string &strSelection, const std::string &strProc, std::string &error)
 
bool accept_selection (const FSBitset &tights, const FSBitset &charges) const
 Important: the accept() function doesn't check the selection (for speed reason), this has to be done separately. More...
 
bool accept_process (uint8_t nparticles, const FSBitset &reals, const FSBitset &tights) const
 
FSBitset retrieveCharges (const std::vector< FakeBkgTools::ParticleData > &particles) const
 
bool hasSS () const
 
bool hasOS () const
 
void setSS (bool set=true)
 
void setOS (bool set=true)
 
bool hasChargeRequirement () const
 
bool operator< (const FinalState &rhs) const
 

Public Attributes

FSBitset selection = 0
 

Private Member Functions

bool parseProcess (std::string process, std::string &error)
 
bool parseSelection (const unsigned short nparticles, std::string strSelection, std::string &error)
 

Private Attributes

size_t m_hash
 
uint32_t m_wmin = 0x0
 
uint32_t m_wmax =0x3FFFFFFF
 

Detailed Description

Definition at line 97 of file FakeBkgInternals.h.

Constructor & Destructor Documentation

◆ FinalState() [1/2]

FakeBkgTools::FinalState::FinalState ( size_t  h)
inline

Definition at line 101 of file FakeBkgInternals.h.

101 : m_hash(h) {} // for search by hash

◆ FinalState() [2/2]

FinalState::FinalState ( size_t  h,
const unsigned  nparticles,
const std::string &  strSelection,
const std::string &  strProc,
std::string &  error 
)

Definition at line 31 of file FakeBkgInternals.cxx.

31  : m_hash(h)
32 {
33  error.clear();
34  if(!parseProcess(strProc, error)) return;
35  if(!parseSelection(nparticles, strSelection, error)) return;
36 }

Member Function Documentation

◆ accept_process()

bool FakeBkgTools::FinalState::accept_process ( uint8_t  nparticles,
const FSBitset reals,
const FSBitset tights 
) const
inline

4 bits per number + extra bit for carry

Definition at line 112 of file FakeBkgInternals.h.

113  {
114  auto nT = tights.count();
115  auto nR = reals.count(), nRT = (reals&tights).count(), nRL = nR - nRT;
116  auto nF = nparticles - nR, nFT = nT - nRT, nFL = nF - nFT;
117  uint32_t w = nRT | (nRL<<5) | (nR<<10) | (nFT<<15) | (nFL<<20) | (nF<<25);
118  if((m_wmax-w)&0x21084210 || (w-m_wmin)&0x21084210) return false;
119  return true;
120  }

◆ accept_selection()

bool FakeBkgTools::FinalState::accept_selection ( const FSBitset tights,
const FSBitset charges 
) const
inline

Important: the accept() function doesn't check the selection (for speed reason), this has to be done separately.

SS tight

OS tight

Definition at line 104 of file FakeBkgInternals.h.

105  {
106  if(!selection[tights.to_ulong()]) return false;
107  auto nT = tights.count();
108  if(hasSS() && nT==2 && (tights&charges).count()==1) return false;
109  if(hasOS() && nT>=2 && ((tights&charges).count()%nT)==0) return false;
110  return true;
111  }

◆ hasChargeRequirement()

bool FakeBkgTools::FinalState::hasChargeRequirement ( ) const
inline

Definition at line 131 of file FakeBkgInternals.h.

131 { return hasOS() || hasSS(); }

◆ hasOS()

bool FakeBkgTools::FinalState::hasOS ( ) const
inline

Definition at line 128 of file FakeBkgInternals.h.

128 { return m_wmax & 0x40000000; }

◆ hasSS()

bool FakeBkgTools::FinalState::hasSS ( ) const
inline

Definition at line 127 of file FakeBkgInternals.h.

127 { return m_wmin & 0x40000000; }

◆ operator<()

bool FakeBkgTools::FinalState::operator< ( const FinalState rhs) const
inline

Definition at line 132 of file FakeBkgInternals.h.

132 { return m_hash < rhs.m_hash; }

◆ parseProcess()

bool FinalState::parseProcess ( std::string  process,
std::string &  error 
)
private

needs to be kept consistent with format used by acceptCombination() and parseSelection(); said format is the following: uint32_t w = nRT | (nRL<<5) | (nR<<10) | (nFT<<15) | (nFL<<20) | (nF<<25); // 4 bits per number + extra bit for carry

Replace the 4-bits digit at position 'offset' by the new value (nmin/nmax):

clear carry bits

clear carry bits

Definition at line 38 of file FakeBkgInternals.cxx.

39 {
40  strProc.erase(std::remove_if(strProc.begin(), strProc.end(),
41  [](char c){ return std::isspace(c); }), strProc.end());
42  std::stringstream ss(strProc);
43  std::string token;
44  std::regex rx0("([<>]=?)([0-9])([FR])(?:\\[(L|!?T)\\])?");
45  std::regex rx1("=?([0-9])(-[0-9])?([FR])(?:\\[(L|!?T)\\])?");
46  std::smatch sm;
47  bool empty = true;
48  m_wmin = 0x0;
49  m_wmax = 0x3FFFFFFF;
50  while(std::getline(ss,token,','))
51  {
52  uint8_t nmin = 0, nmax = 0x7F;
53  empty = false;
54  if(std::regex_match(token, sm, rx0))
55  {
56  bool strict = (sm[1].str()[1] != '=');
57  if(sm[1].str()[0]=='>') nmin = sm[2].str()[0] - '0' + strict*1;
58  else
59  {
60  nmax = sm[2].str()[0] - '0';
61  if(strict && !nmax)
62  {
63  error = "number of leptons constrained to be negative in the 'process' argument: \"" + strProc + "\"";
64  }
65  nmax -= strict*1;
66  }
67  }
68  else if(std::regex_match(token, sm, rx1))
69  {
70  nmin = sm[1].str()[0] - '0';
71  if(sm[2].length()) nmax = sm[2].str()[1] - '0';
72  else nmax = nmin;
73  }
74  else
75  {
76  error = "unable to understand the specified 'process' argument: \"" + strProc + "\"";
77  return false;
78  }
81  unsigned char offset = ((sm[3]=="F")? 15 : 0);
82  if(sm[4] == "!T") offset += 5;
83  else if(sm[4] == "L") offset += 10;
84  nmin = std::max(nmin, uint8_t((m_wmin>>offset)&0xF));
85  nmax = std::min(nmax, uint8_t((m_wmax>>offset)&0xF));
87  m_wmin = (m_wmin&~(0xF<<offset)) | (nmin<<offset);
88  m_wmax = (m_wmax&~(0xF<<offset)) | (nmax<<offset);
89  if(nmin > nmax)
90  {
91  error = "unexpected error (nmin>nmax) while parsing the specified 'process' argument: \"" + strProc + "\"";
92  return false;
93  }
94  }
95  if(empty)
96  {
97  error = "the 'process' argument is empty";
98  return false;
99  }
100  m_wmin &= ~0x21084210;
101  m_wmax &= ~0x21084210;
102  return true;
103 }

◆ parseSelection()

bool FinalState::parseSelection ( const unsigned short  nparticles,
std::string  strSelection,
std::string &  error 
)
private

Note: it is important that parseProcess() is called before this function!

replace all "!T" by "A"

Partial check that the selection and process arguments are consistent

Definition at line 105 of file FakeBkgInternals.cxx.

106 {
108  const unsigned nc = (1 << nparticles);
109  selection.reset();
110  strSelection.erase(std::remove_if(strSelection.begin(), strSelection.end(),
111  [](char c){ return std::isspace(c); }), strSelection.end());
112  std::stringstream ss(strSelection);
113 
114  std::string word = "(?:!?T){" + std::to_string(nparticles) + "}(\\+(?:!?T){" + std::to_string(nparticles) + "})*";
115 
116  if(std::regex_match(strSelection, std::regex(word)))
117  {
118  while(std::getline(ss, word, '+'))
119  {
120  if(!word.length()) continue;
121  std::size_t pos;
122  while((pos=word.find("!T")) != std::string::npos) word.replace(pos, 2, 1, 'A');
123  std::reverse(word.begin(), word.end());
124  std::bitset<FakeBkgTools::maxParticles()> combination(word, 0, std::string::npos, 'A', 'T');
125  selection.set(combination.to_ulong());
126  }
127  return true;
128  }
129 
130  std::regex rx0("([<>]=?)([0-9])(!?T)");
131  std::regex rx1("=?([0-9])(-[0-9])?(!?T)");
132  std::regex rx2("(OS|SS)");
133  std::smatch sm;
134  for(unsigned i=0;i<nc;++i) selection.set(i);
135  bool empty = true;
136  uint8_t maxT=15, maxL=15, minT=0, minL=0;
137  while(std::getline(ss, word, ','))
138  {
139  uint8_t nmin = 0, nmax = 0x7F;
140  if(std::regex_match(word, sm, rx0))
141  {
142  uint8_t o = (sm[1].length()<=1)? 1 : 0;
143  if(sm[1].str()[0]=='>') nmin = sm[2].str()[0] - '0' + o;
144  else
145  {
146  nmax = sm[2].str()[0] - '0';
147  if(o && !nmax)
148  {
149  error = "number of leptons constrained to be negative in the 'selection' argument: \"" + strSelection + "\"";
150  selection.reset();
151  return false;
152  }
153  nmax -= o;
154  }
155  }
156  else if(std::regex_match(word, sm, rx1))
157  {
158  nmin = sm[1].str()[0] - '0';
159  if(sm[2].length()) nmax = sm[2].str()[1] - '0';
160  else nmax = nmin;
161  }
162  else if(std::regex_match(word, sm, rx2))
163  {
165  {
166  error = "both OS and SS requirements were specifed in the 'selection' argument: \"" + strSelection + "\"; this is not allowed";
167  selection.reset();
168  return false;
169  }
170  if(sm[0].str()[0]=='S') setSS();
171  else setOS();
172  continue;
173  }
174  else
175  {
176  error = "unable to understand the specified 'selection' argument: \"" + strSelection + "\"";
177  selection.reset();
178  return false;
179  }
180  if(nmin > nmax)
181  {
182  error = "unexpected error (nmin>nmax) while parsing the specified 'selection' argument: \"" + strSelection + "\"";
183  selection.reset();
184  return false;
185  }
186  bool count_loose = sm[sm.size()-1] == "!T";
187  if(count_loose)
188  {
189  minL = std::max(minL, nmin);
190  maxL = std::min(maxL, nmax);
191  }
192  else
193  {
194  minT = std::max(minT, nmin);
195  maxT = std::min(maxT, nmax);
196  }
197  for(unsigned i=0;i<nc;++i)
198  {
199  auto n = FSBitset(i).count();
200  if(count_loose) n = nparticles - n;
201  if(n<nmin || n>nmax) selection.reset(i);
202  }
203  empty = false;
204  }
205  if(empty)
206  {
207  error = "the 'selection' argument is empty";
208  selection.reset();
209  return false;
210  }
211 
213  if((((m_wmin+(m_wmin>>15))&0x1F) > maxT) // minRT + minFT > maxT
214  || (((m_wmax+(m_wmax>>15))&0x1F) < minT) // maxRT + maxFT < minT
215  || ((((m_wmin>>5)+(m_wmin>>20))&0x1F) > maxL) // minRL + minFL > maxL
216  || ((((m_wmax>>5)+(m_wmax>>20))&0x1F) < minL) // maxRL + maxFL < maxL
217  || ((((m_wmin>>10)+(m_wmin>>25))&0x1F) > maxL+maxT) // minR + minF > maxL + maxT
218  ||((((m_wmax>>10)+(m_wmax>>25))&0x1F) < minL+minT)) // maxR + maxF < minL + minT
219  {
220  error = "the specified 'selection' and 'process' arguments are not consistent";
221  selection.reset();
222  return false;
223  }
224 
225  return true;
226 }

◆ retrieveCharges()

FSBitset FakeBkgTools::FinalState::retrieveCharges ( const std::vector< FakeBkgTools::ParticleData > &  particles) const
inline

Definition at line 121 of file FakeBkgInternals.h.

122  {
123  FSBitset charges{0};
124  for(unsigned i=0;i<particles.size();++i) charges.set(i, particles[i].charge>0);
125  return charges;
126  }

◆ setOS()

void FakeBkgTools::FinalState::setOS ( bool  set = true)
inline

Definition at line 130 of file FakeBkgInternals.h.

130 { m_wmax = set? (m_wmax|0x40000000) : m_wmax&~0x40000000; }

◆ setSS()

void FakeBkgTools::FinalState::setSS ( bool  set = true)
inline

Definition at line 129 of file FakeBkgInternals.h.

129 { m_wmin = set? (m_wmin|0x40000000) : m_wmin&~0x40000000; }

Member Data Documentation

◆ m_hash

size_t FakeBkgTools::FinalState::m_hash
private

Definition at line 134 of file FakeBkgInternals.h.

◆ m_wmax

uint32_t FakeBkgTools::FinalState::m_wmax =0x3FFFFFFF
private

Definition at line 135 of file FakeBkgInternals.h.

◆ m_wmin

uint32_t FakeBkgTools::FinalState::m_wmin = 0x0
private

Definition at line 135 of file FakeBkgInternals.h.

◆ selection

FSBitset FakeBkgTools::FinalState::selection = 0

Definition at line 100 of file FakeBkgInternals.h.


The documentation for this class was generated from the following files:
FakeBkgTools::FinalState::hasChargeRequirement
bool hasChargeRequirement() const
Definition: FakeBkgInternals.h:131
FakeBkgTools::FinalState::hasSS
bool hasSS() const
Definition: FakeBkgInternals.h:127
FakeBkgTools::FinalState::selection
FSBitset selection
Definition: FakeBkgInternals.h:100
max
#define max(a, b)
Definition: cfImp.cxx:41
PowhegControl_ttHplus_NLO.ss
ss
Definition: PowhegControl_ttHplus_NLO.py:83
xAOD::uint8_t
uint8_t
Definition: Muon_v1.cxx:557
xAOD::uint32_t
setEventNumber uint32_t
Definition: EventInfo_v1.cxx:127
FakeBkgTools::FinalState::parseProcess
bool parseProcess(std::string process, std::string &error)
Definition: FakeBkgInternals.cxx:38
FakeBkgTools::FinalState::parseSelection
bool parseSelection(const unsigned short nparticles, std::string strSelection, std::string &error)
Definition: FakeBkgInternals.cxx:105
empty
bool empty(TH1 *h)
Definition: computils.cxx:294
XMLtoHeader.count
count
Definition: XMLtoHeader.py:85
FakeBkgTools::FinalState::setOS
void setOS(bool set=true)
Definition: FakeBkgInternals.h:130
DeMoUpdate.reverse
reverse
Definition: DeMoUpdate.py:563
PrepareReferenceFile.regex
regex
Definition: PrepareReferenceFile.py:43
FakeBkgTools::FinalState::m_wmin
uint32_t m_wmin
Definition: FakeBkgInternals.h:135
FakeBkgTools::FinalState::m_hash
size_t m_hash
Definition: FakeBkgInternals.h:134
lumiFormat.i
int i
Definition: lumiFormat.py:85
h
beamspotman.n
n
Definition: beamspotman.py:731
diffPoolFiles.strict
strict
Definition: diffPoolFiles.py:68
FakeBkgTools::FinalState::m_wmax
uint32_t m_wmax
Definition: FakeBkgInternals.h:135
CxxUtils::set
constexpr std::enable_if_t< is_bitmask_v< E >, E & > set(E &lhs, E rhs)
Convenience function to set bits in a class enum bitmask.
Definition: bitmask.h:232
min
#define min(a, b)
Definition: cfImp.cxx:40
WriteCellNoiseToCool.nF
nF
Definition: WriteCellNoiseToCool.py:541
ActsTrk::to_string
std::string to_string(const DetectorType &type)
Definition: GeometryDefs.h:34
FakeBkgTools::maxParticles
constexpr uint8_t maxParticles()
Definition: FakeBkgInternals.h:93
python.LumiBlobConversion.pos
pos
Definition: LumiBlobConversion.py:18
h
python.CaloScaleNoiseConfig.str
str
Definition: CaloScaleNoiseConfig.py:78
LArG4FSStartPointFilter.particles
list particles
Definition: LArG4FSStartPointFilter.py:84
convertTimingResiduals.offset
offset
Definition: convertTimingResiduals.py:71
FakeBkgTools::FinalState::hasOS
bool hasOS() const
Definition: FakeBkgInternals.h:128
python.IoTestsLib.w
def w
Definition: IoTestsLib.py:200
FakeBkgTools::FSBitset
std::bitset< maxCombinations()> FSBitset
Definition: FakeBkgInternals.h:95
error
Definition: IImpactPoint3dEstimator.h:70
python.compressB64.c
def c
Definition: compressB64.py:93
length
double length(const pvec &v)
Definition: FPGATrackSimLLPDoubletHoughTransformTool.cxx:26
nmax
const int nmax(200)
plotBeamSpotMon.nc
int nc
Definition: plotBeamSpotMon.py:83
HLTNavDetails::rx1
const boost::regex rx1("_v[0-9]+$")
FakeBkgTools::FinalState::setSS
void setSS(bool set=true)
Definition: FakeBkgInternals.h:129