Skip to content

pad_map Module

This module contains a utility class for retrieving pad information using either a pad number or HardwareID

PadData dataclass

Dataclass for storing AT-TPC pad information

Attributes:

Name Type Description
x float

The pad x-coordinates
y float

The pad y-coordinates
gain float

The relative pad gain
time_offset float

The pad time offset due to GET electronics
scale float

The pad scale (big pad or small pad)
Source code in src/spyral/core/pad_map.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
@dataclass
class PadData:
    """Dataclass for storing AT-TPC pad information

    Attributes
    ----------
    x: float
        The pad x-coordinates
    y: float
        The pad y-coordinates
    gain: float
        The relative pad gain
    time_offset: float
        The pad time offset due to GET electronics
    scale: float
        The pad scale (big pad or small pad)
    """

    x: float = 0.0
    y: float = 0.0
    gain: float = 1.0
    time_offset: float = 0.0
    scale: float = 0.0

PadMap

A map of pad number to PadData

Parameters:

Name Type Description Default
params PadParameters

Pad map configuration parameters

 required

Attributes:

Name Type Description
map dict[int, PadData]

The forward map (pad number -> PadData)

Methods:

Name Description
PadMap

Construct the PadMap
get_pad_data

Get the PadData for a given pad. Returns None if the pad does not exist
is_beam_pad

Check if a pad is a beam pad (returns True if beam, False otherwise)
Source code in src/spyral/core/pad_map.py 
 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
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
class PadMap:
    """A map of pad number to PadData

    Parameters
    ----------
    params: PadParameters
        Pad map configuration parameters

    Attributes
    ----------
    map: dict[int, PadData]
        The forward map (pad number -> PadData)

    Methods
    -------
    PadMap(params)
        Construct the PadMap
    get_pad_data(pad_number)
        Get the PadData for a given pad. Returns None if the pad does not exist
    is_beam_pad(pad_number)
        Check if a pad is a beam pad (returns True if beam, False otherwise)
    """

    def __init__(self, params: PadParameters):
        self.map: dict[int, PadData] = {}
        self.is_valid = False
        self.load(params)

    def load(self, params: PadParameters):
        """Load the map data

        Parameters
        ----------
        params: PadParameters
            Paths to map files
        """
        # Defaults are here
        directory = resources.files("spyral.data")

        # Geometry
        if params.pad_geometry_path == DEFAULT_MAP:
            geom_handle = directory.joinpath("padxy.csv")
            with resources.as_file(geom_handle) as geopath:
                geofile = open(geopath, "r")
                geofile.readline()  # Remove header
                lines = geofile.readlines()
                for pad_number, line in enumerate(lines):
                    entries = line.split(",")
                    self.map[pad_number] = PadData(
                        x=float(entries[0]), y=float(entries[1])
                    )
                geofile.close()
        else:
            with open(params.pad_geometry_path, "r") as geofile:
                geofile.readline()  # Remove header
                lines = geofile.readlines()
                for pad_number, line in enumerate(lines):
                    entries = line.split(",")
                    self.map[pad_number] = PadData(
                        x=float(entries[0]), y=float(entries[1])
                    )

        # Time
        if params.pad_time_path == DEFAULT_MAP:
            time_handle = directory.joinpath("pad_time_correction.csv")
            with resources.as_file(time_handle) as timepath:
                timefile = open(timepath, "r")
                timefile.readline()
                lines = timefile.readlines()
                for pad_number, line in enumerate(lines):
                    entries = line.split(",")
                    self.map[pad_number].time_offset = float(entries[0])
                timefile.close()
        else:
            with open(params.pad_time_path, "r") as timefile:
                timefile.readline()
                lines = timefile.readlines()
                for pad_number, line in enumerate(lines):
                    entries = line.split(",")
                    self.map[pad_number].time_offset = float(entries[0])

        # Scale
        if params.pad_scale_path == DEFAULT_MAP:
            scale_handle = directory.joinpath("pad_scale.csv")
            with resources.as_file(scale_handle) as scalepath:
                scalefile = open(scalepath, "r")
                scalefile.readline()
                lines = scalefile.readlines()
                for pad_number, line in enumerate(lines):
                    entries = line.split(",")
                    self.map[pad_number].scale = float(entries[0])
                scalefile.close()
        else:
            with open(params.pad_scale_path, "r") as scalefile:
                scalefile.readline()
                lines = scalefile.readlines()
                for pad_number, line in enumerate(lines):
                    entries = line.split(",")
                    self.map[pad_number].scale = float(entries[0])

        self.is_valid = True

    def get_pad_data(self, pad_number: int) -> PadData | None:
        """Get the PadData associated with a pad number

        Returns None if the pad number is invalid

        Parameters
        ----------
        pad_number: int
            A pad number

        Returns
        -------
        PadData | None
            The associated PadData, or None if the pad number is invalid

        """
        if (pad_number == INVALID_PAD_ID) or pad_number not in self.map.keys():
            return None

        return self.map[pad_number]

    def is_beam_pad(self, pad_id: int) -> bool:
        """Check if a pad is a Beam Pad (TM)

        Parameters
        ----------
        pad_id: int
            The pad number to check

        Returns
        -------
        bool
            True if Beam Pad, False otherwise
        """
        return pad_id in LEGACY_BEAM_PADS

get_pad_data(pad_number)

Get the PadData associated with a pad number

Returns None if the pad number is invalid

Parameters:

Name Type Description Default
pad_number int

A pad number

 required

Returns:

Type Description
PadData | None

The associated PadData, or None if the pad number is invalid
Source code in src/spyral/core/pad_map.py 
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
def get_pad_data(self, pad_number: int) -> PadData | None:
    """Get the PadData associated with a pad number

    Returns None if the pad number is invalid

    Parameters
    ----------
    pad_number: int
        A pad number

    Returns
    -------
    PadData | None
        The associated PadData, or None if the pad number is invalid

    """
    if (pad_number == INVALID_PAD_ID) or pad_number not in self.map.keys():
        return None

    return self.map[pad_number]

is_beam_pad(pad_id)

Check if a pad is a Beam Pad (TM)

Parameters:

Name Type Description Default
pad_id int

The pad number to check

 required

Returns:

Type Description
bool

True if Beam Pad, False otherwise
Source code in src/spyral/core/pad_map.py 
157
158
159
160
161
162
163
164
165
166
167
168
169
170
def is_beam_pad(self, pad_id: int) -> bool:
    """Check if a pad is a Beam Pad (TM)

    Parameters
    ----------
    pad_id: int
        The pad number to check

    Returns
    -------
    bool
        True if Beam Pad, False otherwise
    """
    return pad_id in LEGACY_BEAM_PADS

load(params)

Load the map data

Parameters:

Name Type Description Default
params PadParameters

Paths to map files

 required
Source code in src/spyral/core/pad_map.py 
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
def load(self, params: PadParameters):
    """Load the map data

    Parameters
    ----------
    params: PadParameters
        Paths to map files
    """
    # Defaults are here
    directory = resources.files("spyral.data")

    # Geometry
    if params.pad_geometry_path == DEFAULT_MAP:
        geom_handle = directory.joinpath("padxy.csv")
        with resources.as_file(geom_handle) as geopath:
            geofile = open(geopath, "r")
            geofile.readline()  # Remove header
            lines = geofile.readlines()
            for pad_number, line in enumerate(lines):
                entries = line.split(",")
                self.map[pad_number] = PadData(
                    x=float(entries[0]), y=float(entries[1])
                )
            geofile.close()
    else:
        with open(params.pad_geometry_path, "r") as geofile:
            geofile.readline()  # Remove header
            lines = geofile.readlines()
            for pad_number, line in enumerate(lines):
                entries = line.split(",")
                self.map[pad_number] = PadData(
                    x=float(entries[0]), y=float(entries[1])
                )

    # Time
    if params.pad_time_path == DEFAULT_MAP:
        time_handle = directory.joinpath("pad_time_correction.csv")
        with resources.as_file(time_handle) as timepath:
            timefile = open(timepath, "r")
            timefile.readline()
            lines = timefile.readlines()
            for pad_number, line in enumerate(lines):
                entries = line.split(",")
                self.map[pad_number].time_offset = float(entries[0])
            timefile.close()
    else:
        with open(params.pad_time_path, "r") as timefile:
            timefile.readline()
            lines = timefile.readlines()
            for pad_number, line in enumerate(lines):
                entries = line.split(",")
                self.map[pad_number].time_offset = float(entries[0])

    # Scale
    if params.pad_scale_path == DEFAULT_MAP:
        scale_handle = directory.joinpath("pad_scale.csv")
        with resources.as_file(scale_handle) as scalepath:
            scalefile = open(scalepath, "r")
            scalefile.readline()
            lines = scalefile.readlines()
            for pad_number, line in enumerate(lines):
                entries = line.split(",")
                self.map[pad_number].scale = float(entries[0])
            scalefile.close()
    else:
        with open(params.pad_scale_path, "r") as scalefile:
            scalefile.readline()
            lines = scalefile.readlines()
            for pad_number, line in enumerate(lines):
                entries = line.split(",")
                self.map[pad_number].scale = float(entries[0])

    self.is_valid = True