HEPTutorial/MyMuon.h at master · monttj/HEPTutorial · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
/*
 * MyMuon.h
 *
 *  Created on: Feb 1, 2012
 *      Author: csander
 */

#ifndef MYMUON_H_
#define MYMUON_H_

#include <TLorentzVector.h>

class MyMuon: public TLorentzVector {

   public:

      MyMuon();
      MyMuon(double px, double py, double pz, double e) {
         SetPxPyPzE(px, py, pz, e);
      }
      ;
      virtual ~MyMuon();

      void SetIsolation(double x) {
         isolation = x;
      }
      ;

      void SetCharge(int q) {
         charge = q;
      }
      ;

      const double GetIsolation() {
         return isolation;
      }
      ;

      const bool IsIsolated(double relcut = 0.05) {
         return ((isolation / Pt()) < relcut);
         //return (isolation < 1.);
      }
      ;
      const int GetCharge() {
         return charge;
      }
      ;

   private:

      double isolation;
      int charge;

};

#endif /* MYMUON_H_ */