|
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
57
58
59
60
61
62
63
64
65
66
|
#ifndef Fireworks_Tracks_FWMagField_h
#define Fireworks_Tracks_FWMagField_h
// -*- C++ -*-
//
// Simplified model of the CMS detector magnetic field
//
#include "TEveTrackPropagator.h"
class TH1F;
namespace edm {
class EventBase;
}
class FWMagField : public TEveMagField {
Float_t GetFieldMag() const;
public:
enum ESource { kNone, kEvent, kUser };
FWMagField();
~FWMagField() override;
// get field values
TEveVector GetField(Float_t x, Float_t y, Float_t z) const override;
Float_t GetMaxFieldMag() const override;
// auto/user behaviour
void setUserField(float b) { m_userField = b; }
float getUserField() const { return m_userField; }
void setSource(ESource x) { m_source = x; }
ESource getSource() const { return m_source; }
// field model
void setReverseState(bool state) { m_reverse = state; }
bool isReverse() const { return m_reverse; }
void setSimpleModel(bool simpleModel) { m_simpleModel = simpleModel; }
bool isSimpleModel() const { return m_simpleModel; }
// field estimate
void guessFieldIsOn(bool guess) const;
void guessField(float estimate) const;
void resetFieldEstimate() const;
void checkFieldInfo(const edm::EventBase*);
void setFFFieldMag(float);
FWMagField(const FWMagField&) = delete; // stop default
const FWMagField& operator=(const FWMagField&) = delete; // stop default
private:
ESource m_source;
float m_userField;
float m_eventField;
bool m_reverse;
bool m_simpleModel;
// runtime estimate , have to be mutable becuse of GetField() is const
mutable TH1F* m_guessValHist;
mutable int m_numberOfFieldIsOnEstimates;
mutable int m_numberOfFieldEstimates;
mutable bool m_updateFieldEstimate;
mutable float m_guessedField;
};
#endif
|