Calibration of the pixel position for a Pilatus detector

This tutorial summarizes the work done by Frederic Sulzman during his internship at ESRF during the summer 2015 entilted “Calibration for geometric distortion in multi- modules pixel detectors”.

The overall strategy is very similar to “CCD calibration” tutorial with some specificities due to the modular nature of the detector.

  1. Image preprocessing
  2. Peak picking
  3. Grid assignment
  4. Displacement fitting
  5. Reconstruction of the pixel position
  6. Saving into a detector definition file
  7. Validation of the geometry with a 2D integration

Each module being made by lithographic processes, the error within a module will be assumeed to be constant. We will use the name “displacement of the module” to describe the rigide movement of the module.

This tutorial uses data from the Pilatus3 2M CdTe from the ID15 beam line of the ESRF. They provided not only the intersnip subject but also the couple of images uses to calibrate the detector.

This detector contains 48 half-modules, each bound to a single CdTe monocrystal sensor and is designed for high energy X-ray radiation detection. Due to the construction procedure, these half-modules could show a misalignment within the detector plane. While the manufacturer (Dectris) garanties a precision within a pixel (172µm), the miss-alignment of certain modules can be seen while calibrating Debye-Scherrer ring using refereance sample. So the aim of this work is to provide a detector description with a better precision better than the original detector.

This work will be performed on the image of a grid available: http://www.silx.org/pub/pyFAI/detector_calibration/Pilatus2MCdTe_ID15_grid_plus_sample_0004.cbf and the scattering of ceria (CeO2) at 72.1keV available here. http://www.silx.org/pub/pyFAI/detector_calibration/Pilatus2MCdTe_ID15_CeO2_72100eV_800mm_0000.cbf

It is a good exercise to calibrate all rings of the later image using the pyFAI-calib2 tool. A calibration close to perfection is needed to visualize the module miss-alignement we aim at correcting.

In [1]:
%matplotlib notebook
In [2]:
#many imports which will be used all along the notebook
import time
start_time = time.time()
import os
import pyFAI
import fabio
import glob
import numpy
from numpy.lib.stride_tricks import as_strided
from math import sin, cos, sqrt
from scipy.ndimage import convolve, binary_dilation
from scipy.spatial import distance_matrix
from scipy.optimize import minimize
from matplotlib.pyplot import subplots
from pyFAI.ext.bilinear import Bilinear
from pyFAI.ext.watershed import InverseWatershed
from silx.resources import ExternalResources
from pyFAI.azimuthalIntegrator import AzimuthalIntegrator

# A couple of compound dtypes ...
dt = numpy.dtype([('y', numpy.float64),
                  ('x', numpy.float64),
                  ('i', numpy.int64),
                 ])
dl = numpy.dtype([('y', numpy.float64),
                  ('x', numpy.float64),
                  ('i', numpy.int64),
                  ('Y', numpy.int64),
                  ('X', numpy.int64),
                 ])

print("Using pyFAI verison: ", pyFAI.version)
WARNING:silx.opencl.common:Unable to import pyOpenCl. Please install it from: http://pypi.python.org/pypi/pyopencl
Using pyFAI verison:  0.20.0-dev0
In [3]:
#Nota: Configure here your proxy if you are behind a firewall
#os.environ["http_proxy"] = "http://proxy.comany.com:3128"
In [4]:
downloader = ExternalResources("detector_calibration", "http://www.silx.org/pub/pyFAI/detector_calibration/")
ring_file = downloader.getfile("Pilatus2MCdTe_ID15_CeO2_72100eV_800mm_0000.cbf")
print(ring_file)
grid_file = downloader.getfile("Pilatus2MCdTe_ID15_grid_plus_sample_0004.cbf")
print(grid_file)
/tmp/detector_calibration_testdata_kieffer/Pilatus2MCdTe_ID15_CeO2_72100eV_800mm_0000.cbf
/tmp/detector_calibration_testdata_kieffer/Pilatus2MCdTe_ID15_grid_plus_sample_0004.cbf
In [5]:
rings = fabio.open(ring_file).data
img = fabio.open(grid_file).data
fig,ax = subplots(1,2, figsize=(10,5))
ax[0].imshow(img.clip(0,1000), interpolation="bilinear")
ax[0].set_title("grid")
ax[1].imshow(numpy.arcsinh(rings), interpolation="bilinear")
ax[1].set_title("rings")
Data type cannot be displayed: application/javascript
Out[5]:
Text(0.5, 1.0, 'rings')

Image processing

There are 3 pre-processing steps which are needed.

  1. Define for each module a unique identifier which will be used later on during the fitting procedure
  2. Define the proper mask: each module is the assembly of 4x2 sub-modules and there are (3) interpolated pixels between each sub-module, such “unreliable pixels should be masked out as well
  3. Correct the grid image by the smoothed image to have a constant background.
  4. Convolve the raw image with a typical hole shape to allow a precise spotting of the hole center.
In [6]:
# This is the default detector as definied in pyFAI according to the specification provided by Dectris:
pilatus = pyFAI.detector_factory("Pilatus_2m_CdTe")
print(pilatus)

mask1 = pilatus.mask
module_size = pilatus.MODULE_SIZE
module_gap = pilatus.MODULE_GAP
submodule_size = (96,60)
Detector Pilatus CdTe 2M         PixelSize= 1.720e-04, 1.720e-04 m
In [7]:
#1 + 2 Calculation of the module_id and the interpolated-mask:
mid = numpy.zeros(pilatus.shape, dtype=int)
mask2 = numpy.zeros(pilatus.shape, dtype=int)
idx = 1
for i in range(8):
    y_start = i*(module_gap[0] + module_size[0])
    y_stop = y_start + module_size[0]
    for j in range(3):
        x_start = j*(module_gap[1] + module_size[1])
        x_stop = x_start + module_size[1]
        mid[y_start:y_stop,x_start: x_start+module_size[1]//2] = idx
        idx+=1
        mid[y_start:y_stop,x_start+module_size[1]//2: x_stop] = idx
        idx+=1
        mask2[y_start+submodule_size[0]-1:y_start+submodule_size[0]+2,
              x_start:x_stop] = 1
        for k in range(1,8):
            mask2[y_start:y_stop,
              x_start+k*(submodule_size[1]+1)-1:x_start+k*(submodule_size[1]+1)+2] = 1
In [8]:
#Extra masking
mask0 = img<0
#Those pixel are miss-behaving... they are the hot pixels next to the beam-stop
mask0[915:922,793:800] = 1
mask0[817:820,747:750] = 1
In [9]:
fig,ax = subplots(1,3, figsize=(10,4))
ax[0].imshow(mid, interpolation="bilinear")
ax[0].set_title("Module Id")

ax[1].imshow(mask2+mask1+mask0, interpolation="bilinear")
ax[1].set_title("Combined mask")

nimg = img.astype(float)
nimg[numpy.where(mask0+mask1+mask2)] = numpy.nan


ax[2].imshow(nimg)#, interpolation="bilinear")
ax[2].set_title("Nan masked image")
Data type cannot be displayed: application/javascript
Out[9]:
Text(0.5, 1.0, 'Nan masked image')
In [10]:
# The Nan-masked image contains now only valid values (and Nan elsewhere). We will make a large median filter to
# build up a smooth image without gaps.
#
# This function is backported from future version of numpy ... it allows to expose a winbowed view
# to perform the nanmedian-filter

def sliding_window_view(x, shape, subok=False, readonly=True):
    """
    Creates sliding window views of the N dimensional array with the given window
    shape. Window slides across each dimension of `x` and extract subsets of `x`
    at any window position.
    Parameters
    ----------
    x : array_like
        Array to create sliding window views of.
    shape : sequence of int
        The shape of the window. Must have same length as the number of input array dimensions.
    subok : bool, optional
        If True, then sub-classes will be passed-through, otherwise the returned
        array will be forced to be a base-class array (default).
    readonly : bool, optional
        If set to True, the returned array will always be readonly view.
        Otherwise it will return writable copies(see Notes).
    Returns
    -------
    view : ndarray
        Sliding window views (or copies) of `x`. view.shape = x.shape - shape + 1
    See also
    --------
    as_strided: Create a view into the array with the given shape and strides.
    broadcast_to: broadcast an array to a given shape.
    Notes
    -----
    ``sliding_window_view`` create sliding window views of the N dimensions array
    with the given window shape and its implementation based on ``as_strided``.
    Please note that if readonly set to True, views are returned, not copies
    of array. In this case, write operations could be unpredictable, so the returned
    views are readonly. Bear in mind that returned copies (readonly=False) will
    take more memory than the original array, due to overlapping windows.
    For some cases there may be more efficient approaches to calculate transformations
    across multi-dimensional arrays, for instance `scipy.signal.fftconvolve`, where combining
    the iterating step with the calculation itself while storing partial results can result
    in significant speedups.
    Examples
    --------
    >>> i, j = np.ogrid[:3,:4]
    >>> x = 10*i + j
    >>> shape = (2,2)
    >>> np.lib.stride_tricks.sliding_window_view(x, shape)
    array([[[[ 0,  1],
             [10, 11]],
            [[ 1,  2],
             [11, 12]],
            [[ 2,  3],
             [12, 13]]],
           [[[10, 11],
             [20, 21]],
            [[11, 12],
             [21, 22]],
            [[12, 13],
             [22, 23]]]])
    """
    np = numpy
    # first convert input to array, possibly keeping subclass
    x = np.array(x, copy=False, subok=subok)

    try:
        shape = np.array(shape, np.int)
    except:
        raise TypeError('`shape` must be a sequence of integer')
    else:
        if shape.ndim > 1:
            raise ValueError('`shape` must be one-dimensional sequence of integer')
        if len(x.shape) != len(shape):
            raise ValueError("`shape` length doesn't match with input array dimensions")
        if np.any(shape <= 0):
            raise ValueError('`shape` cannot contain non-positive value')

    o = np.array(x.shape) - shape  + 1 # output shape
    if np.any(o <= 0):
        raise ValueError('window shape cannot larger than input array shape')

    if type(readonly) != bool:
        raise TypeError('readonly must be a boolean')

    strides = x.strides
    view_strides = strides

    view_shape = np.concatenate((o, shape), axis=0)
    view_strides = np.concatenate((view_strides, strides), axis=0)
    view = as_strided(x, view_shape, view_strides, subok=subok, writeable=not readonly)

    if not readonly:
        return view.copy()
    else:
        return view
In [11]:
%%time
#Calculate a background image using a large median filter ... takes a while
shape = (19,11)
print(nimg.shape)
padded = numpy.pad(nimg, tuple((i//2,) for i in shape), mode="edge")
print(padded.shape)
background = numpy.nanmedian(sliding_window_view(padded,shape), axis = (-2,-1))
print(background.shape)
fig,ax = subplots()
ax.imshow(background)
ax.set_title("Background image")
(1679, 1475)
(1697, 1485)
(1679, 1475)
Data type cannot be displayed: application/javascript
CPU times: user 22.6 s, sys: 2.44 s, total: 25.1 s
Wall time: 25.1 s
Out[11]:
Text(0.5, 1.0, 'Background image')
In [12]:
fig,ax = subplots(1,2, figsize=(9,5))

normalized = (nimg/background)

low = numpy.nanmin(normalized)
high = numpy.nanmax(normalized)
print(low, high)
normalized[numpy.isnan(normalized)] = 0
normalized /= high

ax[0].imshow(normalized)
ax[0].set_title("Normalized image")

ax[1].hist(normalized.ravel(), 100, range=(0,1))
ax[1].set_title("Histogram of intensities in normalized image")
Data type cannot be displayed: application/javascript
0.0 17.728813559322035
Out[12]:
Text(0.5, 1.0, 'Histogram of intensities in normalized image')

For a precise measurement of the peak position, one trick is to convolve the image with a pattern which looks like a hole of the grid.

In [13]:
#print the profile of the normalized image: the center is difficult to measure due to the small size of the hole.
fig,ax = subplots(2)
ax[0].plot(normalized[:,545])
ax[1].plot(normalized[536,:])
Data type cannot be displayed: application/javascript
Out[13]:
[<matplotlib.lines.Line2D at 0x7fed27ff5310>]
In [14]:
#Definition of the convolution kernel
ksize = 5
y,x = numpy.ogrid[-(ksize-1)//2:ksize//2+1,-(ksize-1)//2:ksize//2+1]
d = numpy.sqrt(y*y+x*x)

#Fade out curve definition
fadeout = lambda x: 1/(1+numpy.exp(5*(x-2.5)))

kernel = fadeout(d)
mini=kernel.sum()
print(mini)

fig,ax = subplots(1,3)
ax[0].imshow(d)
ax[0].set_title("Distance array")

ax[1].plot(numpy.linspace(0,5,100),fadeout(numpy.linspace(0,5,100)))
ax[1].set_title("fade-out curve")

ax[2].imshow(kernel)
ax[2].set_title("Convolution kernel")

19.63857792789662
Data type cannot be displayed: application/javascript
Out[14]:
Text(0.5, 1.0, 'Convolution kernel')
In [15]:
my_smooth = convolve(normalized, kernel, mode="constant", cval=0)/mini
print(my_smooth.shape)
fig,ax = subplots(1,2)
ax[0].imshow(normalized.clip(0,1))
ax[0].set_ylim(1050,1100)
ax[0].set_xlim(300,350)
ax[1].imshow(my_smooth.clip(0,1))
ax[1].set_ylim(1050,1100)
ax[1].set_xlim(300,350)
numpy.where(my_smooth == my_smooth.max())
(1679, 1475)
Data type cannot be displayed: application/javascript
Out[15]:
(array([1065]), array([338]))
In [16]:
#mask out all pixels too close to any masked position

all_masks = numpy.logical_or(numpy.logical_or(mask0,mask1),mask2)
print(all_masks.sum())
big_mask = binary_dilation(all_masks, iterations=ksize//2+1+1)
print(big_mask.sum())
smooth2 = my_smooth.copy()
smooth2[big_mask] = 0
fig,ax = subplots()
ax.imshow(smooth2)
335009
782371
Data type cannot be displayed: application/javascript
Out[16]:
<matplotlib.image.AxesImage at 0x7fed27f742e0>
In [17]:
#Display the profile of the smoothed image: the center is easy to measure thanks to the smoothness of the signal
fig,ax = subplots(2)
ax[0].plot(my_smooth[:,545])
ax[1].plot(my_smooth[536,:])
Data type cannot be displayed: application/javascript
Out[17]:
[<matplotlib.lines.Line2D at 0x7fed27e6d250>]

Peak picking

We use the watershed module from pyFAI to retrieve all peak positions. Those regions are sieved out respectively for:

  • their size, it should be larger than the kernel itself
  • the peaks too close to masked regions are removed
  • the intensity of the peak
In [18]:
iw = InverseWatershed(my_smooth)
iw.init()
iw.merge_singleton()
all_regions = set(iw.regions.values())

regions = [i for i in all_regions if i.size>mini]

print("Number of region segmented: %s"%len(all_regions))
print("Number of large enough regions : %s"%len(regions))

Number of region segmented: 82126
Number of large enough regions : 41333
In [19]:
#Remove peaks on masked region
sieved_region = [i for i in regions if not big_mask[(i.index//nimg.shape[-1], i.index%nimg.shape[-1])]]
print("Number of peaks not on masked areea : %s"%len(sieved_region))
Number of peaks not on masked areea : 30001
In [20]:
# Histogram of peak height:
s = numpy.array([i.maxi for i in sieved_region])

fig, ax = subplots()
ax.hist(s, 100)
Data type cannot be displayed: application/javascript
Out[20]:
(array([1.0000e+00, 0.0000e+00, 1.0000e+01, 3.9940e+03, 2.2274e+04,
        1.4800e+03, 1.1700e+02, 3.4000e+01, 8.0000e+00, 2.0000e+00,
        2.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00, 1.0000e+00,
        1.0000e+00, 0.0000e+00, 0.0000e+00, 1.0000e+00, 0.0000e+00,
        1.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00, 1.0000e+00,
        0.0000e+00, 1.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00,
        1.0000e+00, 2.0000e+00, 2.0000e+00, 1.0000e+00, 0.0000e+00,
        1.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00,
        0.0000e+00, 0.0000e+00, 0.0000e+00, 1.0000e+00, 0.0000e+00,
        1.0000e+00, 3.0000e+00, 1.0000e+00, 2.0000e+00, 0.0000e+00,
        0.0000e+00, 2.0000e+00, 1.0000e+00, 0.0000e+00, 0.0000e+00,
        0.0000e+00, 1.0000e+00, 0.0000e+00, 0.0000e+00, 2.0000e+00,
        0.0000e+00, 0.0000e+00, 2.0000e+00, 0.0000e+00, 0.0000e+00,
        2.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00,
        0.0000e+00, 1.0000e+00, 1.0000e+00, 3.0000e+00, 3.0000e+00,
        7.0000e+00, 1.0000e+01, 1.0000e+01, 2.1000e+01, 1.9000e+01,
        2.9000e+01, 3.5000e+01, 5.7000e+01, 7.3000e+01, 9.9000e+01,
        1.1800e+02, 1.5900e+02, 1.9400e+02, 1.9500e+02, 1.9700e+02,
        1.9500e+02, 1.6300e+02, 1.6200e+02, 1.0200e+02, 8.1000e+01,
        6.6000e+01, 3.9000e+01, 7.0000e+00, 1.0000e+00, 1.0000e+00]),
 array([0.04771975, 0.05009548, 0.05247121, 0.05484694, 0.05722266,
        0.05959839, 0.06197412, 0.06434985, 0.06672558, 0.06910131,
        0.07147704, 0.07385276, 0.07622849, 0.07860422, 0.08097995,
        0.08335568, 0.08573141, 0.08810713, 0.09048286, 0.09285859,
        0.09523432, 0.09761005, 0.09998578, 0.1023615 , 0.10473723,
        0.10711296, 0.10948869, 0.11186442, 0.11424015, 0.11661588,
        0.1189916 , 0.12136733, 0.12374306, 0.12611879, 0.12849452,
        0.13087025, 0.13324597, 0.1356217 , 0.13799743, 0.14037316,
        0.14274889, 0.14512462, 0.14750035, 0.14987607, 0.1522518 ,
        0.15462753, 0.15700326, 0.15937899, 0.16175472, 0.16413044,
        0.16650617, 0.1688819 , 0.17125763, 0.17363336, 0.17600909,
        0.17838482, 0.18076054, 0.18313627, 0.185512  , 0.18788773,
        0.19026346, 0.19263919, 0.19501491, 0.19739064, 0.19976637,
        0.2021421 , 0.20451783, 0.20689356, 0.20926929, 0.21164501,
        0.21402074, 0.21639647, 0.2187722 , 0.22114793, 0.22352366,
        0.22589938, 0.22827511, 0.23065084, 0.23302657, 0.2354023 ,
        0.23777803, 0.24015376, 0.24252948, 0.24490521, 0.24728094,
        0.24965667, 0.2520324 , 0.25440813, 0.25678385, 0.25915958,
        0.26153531, 0.26391104, 0.26628677, 0.2686625 , 0.27103822,
        0.27341395, 0.27578968, 0.27816541, 0.28054114, 0.28291687,
        0.2852926 ]),
 <a list of 100 Patch objects>)
In [21]:
#sieve-out for peak intensity
int_mini = 0.1
peaks = [(i.index//nimg.shape[-1], i.index%nimg.shape[-1]) for i in sieved_region if (i.maxi)>int_mini]
print("Number of remaining peaks with I>%s: %s"%(int_mini, len(peaks)))

peaks_raw = numpy.array(peaks)
Number of remaining peaks with I>0.1: 2075
In [22]:
# Finally the peak positions are interpolated using a second order taylor expansion
# in thevinicy of the maximum value of the signal:

#Create bilinear interpolator
bl = Bilinear(my_smooth)

#Overlay raw peak coordinate and refined peak positions

ref_peaks = [bl.local_maxi(p) for p in peaks]
fig, ax = subplots()
ax.imshow(img.clip(0,1000), interpolation="nearest")
peaks_ref = numpy.array(ref_peaks)
ax.plot(peaks_raw[:,1], peaks_raw[:, 0], ".r")
ax.plot(peaks_ref[:,1],peaks_ref[:, 0], ".b")
ax.set_title("Extracted peak position (red: raw, blue: refined)")
print("Refined peak coordinate:")
print(ref_peaks[:10])

Data type cannot be displayed: application/javascript
Refined peak coordinate:
[(1595.8290861696005, 1013.8245996534824), (1595.8770727142692, 1043.3853384256363), (1595.9537781216204, 1072.6417447328568), (1596.0508789345622, 1102.100757934153), (1595.9913188870996, 1131.4514838457108), (1596.1773583739996, 1160.8746052980423), (1596.1950097680092, 1190.3706273138523), (1596.2930068075657, 1219.7231954336166), (1596.3396399617195, 1249.2461778968573), (1596.385121703148, 1278.6429178714752)]

At this stage we have about 2000 peaks (with sub-pixel precision) which are visually distributed on all modules. Some modules have their peaks located along sub-module boundaries which are masked out, hence they have fewer ontrol point for the calculation. Let’s assign each peak to a module identifier. This allows to print out the number of peaks per module:

In [23]:

yxi = numpy.array([i+(mid[round(i[0]),round(i[1])],)
                   for i in ref_peaks], dtype=dt)
print("Number of keypoint per module:")
for i in range(1,mid.max()+1):
    print("Module id:",i, "cp:", (yxi[:]["i"] == i).sum())
Number of keypoint per module:
Module id: 1 cp: 48
Module id: 2 cp: 30
Module id: 3 cp: 48
Module id: 4 cp: 46
Module id: 5 cp: 42
Module id: 6 cp: 47
Module id: 7 cp: 47
Module id: 8 cp: 30
Module id: 9 cp: 48
Module id: 10 cp: 48
Module id: 11 cp: 41
Module id: 12 cp: 39
Module id: 13 cp: 48
Module id: 14 cp: 30
Module id: 15 cp: 47
Module id: 16 cp: 48
Module id: 17 cp: 42
Module id: 18 cp: 48
Module id: 19 cp: 47
Module id: 20 cp: 30
Module id: 21 cp: 48
Module id: 22 cp: 47
Module id: 23 cp: 42
Module id: 24 cp: 47
Module id: 25 cp: 48
Module id: 26 cp: 30
Module id: 27 cp: 48
Module id: 28 cp: 47
Module id: 29 cp: 41
Module id: 30 cp: 50
Module id: 31 cp: 46
Module id: 32 cp: 30
Module id: 33 cp: 48
Module id: 34 cp: 42
Module id: 35 cp: 47
Module id: 36 cp: 48
Module id: 37 cp: 48
Module id: 38 cp: 28
Module id: 39 cp: 48
Module id: 40 cp: 42
Module id: 41 cp: 44
Module id: 42 cp: 48
Module id: 43 cp: 47
Module id: 44 cp: 26
Module id: 45 cp: 46
Module id: 46 cp: 42
Module id: 47 cp: 45
Module id: 48 cp: 48

Grid assignment

The calibration is performed using a regular grid, the idea is to assign to each peak of coordinates (x,y) the integer value (X, Y) which correspond to the grid corrdinate system.

The first step is to measure the grid pitch which correspond to the distance (in pixels) from one peak to the next. This is easily obtained from a pair-wise distribution function.

In [24]:
# pairwise distance calculation using scipy.spatial.distance_matrix

dist = distance_matrix(peaks_ref, peaks_ref)

fig, ax = subplots()
ax.hist(dist.ravel(), 100, range=(0,100))
ax.set_title("Pair-wise distribution function")
Data type cannot be displayed: application/javascript
Out[24]:
Text(0.5, 1.0, 'Pair-wise distribution function')

The histogram of the pair-distribution function has a first peak at 0 and the second peak between 29 and 30. Let’s start the fit with this value

Two other parameters correspond to the offset, in pixel for the grid index (X,Y) = (0,0). The easiest is to measure the smallest x and y for the first module.

In [25]:
#from pair-wise distribution histogram
step = 29
#work with the first module and fit the peak positions
first = yxi[yxi[:]["i"] == 1]
y_min = first[:]["y"].min()
x_min = first[:]["x"].min()
print("offset for the first peak: ", x_min, y_min)
offset for the first peak:  16.269793540239334 7.186804354190826

The grid looks very well aligned with the axes which makes this step easier but nothing garanties it is perfect, so the rotation of the grid has to be measured as well.

The default rotation will be zero and will be fitted later on.

Once the indexes X,Y determined for eack peak, one can fit the parameter to properly align the grid with the first module. Those 4 parameters are step-size, x_min, y_min and angle

In [26]:
#Assign each peak to an index
indexed1 = numpy.zeros(len(first), dtype=dl)

for i,v in enumerate(first):
    Y = int(round((v["y"]-y_min)/step))
    X = int(round((v["x"]-x_min)/step))
    indexed1[i]["y"] = v["y"]
    indexed1[i]["x"] = v["x"]
    indexed1[i]["i"] = v["i"]
    indexed1[i]["Y"] = Y
    indexed1[i]["X"] = X
    print("peak id: %s %20s Y:%d (Δ=%.3f) X:%s (Δ=%.3f)"%
          (i,v, Y, (v["y"]-Y*step-y_min)/step, X, (v["x"]-X*step-x_min)/step))

peak id: 0 (124.69245848, 16.47663069, 1) Y:4 (Δ=0.052) X:0 (Δ=0.007)
peak id: 1 (124.74490786, 45.56257215, 1) Y:4 (Δ=0.054) X:1 (Δ=0.010)
peak id: 2 (124.81674041, 75.13362026, 1) Y:4 (Δ=0.056) X:2 (Δ=0.030)
peak id: 3 (124.75223885, 104.64793801, 1) Y:4 (Δ=0.054) X:3 (Δ=0.048)
peak id: 4 (124.84565204, 133.79748473, 1) Y:4 (Δ=0.057) X:4 (Δ=0.053)
peak id: 5 (124.85729225, 163.3060154, 1) Y:4 (Δ=0.058) X:5 (Δ=0.070)
peak id: 6 (124.89991033, 192.57832247, 1) Y:4 (Δ=0.059) X:6 (Δ=0.080)
peak id: 7 (124.96431521, 222.11254385, 1) Y:4 (Δ=0.061) X:7 (Δ=0.098)
peak id: 8 (7.18680435, 16.41273034, 1) Y:0 (Δ=0.000) X:0 (Δ=0.005)
peak id: 9 (7.23123039, 45.74501115, 1) Y:0 (Δ=0.002) X:1 (Δ=0.016)
peak id: 10 (7.29059145, 75.19019249, 1) Y:0 (Δ=0.004) X:2 (Δ=0.032)
peak id: 11 (7.19876912, 104.56766725, 1) Y:0 (Δ=0.000) X:3 (Δ=0.045)
peak id: 12 (7.28363913, 134.01557775, 1) Y:0 (Δ=0.003) X:4 (Δ=0.060)
peak id: 13 (7.25490889, 163.38282073, 1) Y:0 (Δ=0.002) X:5 (Δ=0.073)
peak id: 14 (7.36717248, 192.76562056, 1) Y:0 (Δ=0.006) X:6 (Δ=0.086)
peak id: 15 (7.37469381, 222.16684446, 1) Y:0 (Δ=0.006) X:7 (Δ=0.100)
peak id: 16 (65.94893547, 16.45467511, 1) Y:2 (Δ=0.026) X:0 (Δ=0.006)
peak id: 17 (65.94483702, 45.61336857, 1) Y:2 (Δ=0.026) X:1 (Δ=0.012)
peak id: 18 (66.03506869, 75.15414216, 1) Y:2 (Δ=0.029) X:2 (Δ=0.030)
peak id: 19 (65.89957739, 104.38019937, 1) Y:2 (Δ=0.025) X:3 (Δ=0.038)
peak id: 20 (66.12287057, 133.89461011, 1) Y:2 (Δ=0.032) X:4 (Δ=0.056)
peak id: 21 (66.1012452, 163.39174998, 1) Y:2 (Δ=0.032) X:5 (Δ=0.073)
peak id: 22 (66.10113759, 192.66102123, 1) Y:2 (Δ=0.032) X:6 (Δ=0.082)
peak id: 23 (66.11095764, 222.20659751, 1) Y:2 (Δ=0.032) X:7 (Δ=0.101)
peak id: 24 (36.49696946, 16.50531429, 1) Y:1 (Δ=0.011) X:0 (Δ=0.008)
peak id: 25 (36.54493174, 45.66864574, 1) Y:1 (Δ=0.012) X:1 (Δ=0.014)
peak id: 26 (36.57917181, 75.22501931, 1) Y:1 (Δ=0.014) X:2 (Δ=0.033)
peak id: 27 (36.48521471, 104.38776881, 1) Y:1 (Δ=0.010) X:3 (Δ=0.039)
peak id: 28 (36.5958997, 133.89308376, 1) Y:1 (Δ=0.014) X:4 (Δ=0.056)
peak id: 29 (36.56109262, 163.54238695, 1) Y:1 (Δ=0.013) X:5 (Δ=0.078)
peak id: 30 (36.62988734, 192.63348889, 1) Y:1 (Δ=0.015) X:6 (Δ=0.082)
peak id: 31 (36.66130301, 222.22433692, 1) Y:1 (Δ=0.016) X:7 (Δ=0.102)
peak id: 32 (183.49989897, 16.39247844, 1) Y:6 (Δ=0.080) X:0 (Δ=0.004)
peak id: 33 (183.50560439, 45.46551961, 1) Y:6 (Δ=0.080) X:1 (Δ=0.007)
peak id: 34 (183.60742423, 75.08415242, 1) Y:6 (Δ=0.083) X:2 (Δ=0.028)
peak id: 35 (183.51583004, 104.62201929, 1) Y:6 (Δ=0.080) X:3 (Δ=0.047)
peak id: 36 (183.66220078, 133.73585817, 1) Y:6 (Δ=0.085) X:4 (Δ=0.051)
peak id: 37 (183.65553248, 163.30831155, 1) Y:6 (Δ=0.085) X:5 (Δ=0.070)
peak id: 38 (183.60811904, 192.46339762, 1) Y:6 (Δ=0.083) X:6 (Δ=0.076)
peak id: 39 (183.75545964, 222.06266521, 1) Y:6 (Δ=0.089) X:7 (Δ=0.096)
peak id: 40 (154.25518942, 16.26979354, 1) Y:5 (Δ=0.071) X:0 (Δ=0.000)
peak id: 41 (154.21079713, 45.57556251, 1) Y:5 (Δ=0.070) X:1 (Δ=0.011)
peak id: 42 (154.28871083, 75.04928852, 1) Y:5 (Δ=0.072) X:2 (Δ=0.027)
peak id: 43 (154.21604921, 104.4366582, 1) Y:5 (Δ=0.070) X:3 (Δ=0.040)
peak id: 44 (154.37423909, 133.8660319, 1) Y:5 (Δ=0.075) X:4 (Δ=0.055)
peak id: 45 (154.35255641, 163.24378996, 1) Y:5 (Δ=0.075) X:5 (Δ=0.068)
peak id: 46 (154.32582822, 192.64246738, 1) Y:5 (Δ=0.074) X:6 (Δ=0.082)
peak id: 47 (154.46003968, 222.04596291, 1) Y:5 (Δ=0.078) X:7 (Δ=0.096)

The error in positionning each of the pixel is less than 0.1 pixel which is already excellent and will allow a straight forward fit.

The cost function for the first module is calculated as the sum of distances squared in pixel space. It uses 4 parameters which are step-size, x_min, y_min and angle

In [27]:
#Calculate the center of every single module for rotation around this center.
centers = {i: numpy.array([[numpy.where(mid == i)[1].mean()], [numpy.where(mid == i)[0].mean()]]) for i in range(1, 49)}
for k,v in centers.items():
    print(k,v.ravel())
1 [121.  97.]
2 [364.5  97. ]
3 [615.  97.]
4 [858.5  97. ]
5 [1109.   97.]
6 [1352.5   97. ]
7 [121. 309.]
8 [364.5 309. ]
9 [615. 309.]
10 [858.5 309. ]
11 [1109.  309.]
12 [1352.5  309. ]
13 [121. 521.]
14 [364.5 521. ]
15 [615. 521.]
16 [858.5 521. ]
17 [1109.  521.]
18 [1352.5  521. ]
19 [121. 733.]
20 [364.5 733. ]
21 [615. 733.]
22 [858.5 733. ]
23 [1109.  733.]
24 [1352.5  733. ]
25 [121. 945.]
26 [364.5 945. ]
27 [615. 945.]
28 [858.5 945. ]
29 [1109.  945.]
30 [1352.5  945. ]
31 [ 121. 1157.]
32 [ 364.5 1157. ]
33 [ 615. 1157.]
34 [ 858.5 1157. ]
35 [1109. 1157.]
36 [1352.5 1157. ]
37 [ 121. 1369.]
38 [ 364.5 1369. ]
39 [ 615. 1369.]
40 [ 858.5 1369. ]
41 [1109. 1369.]
42 [1352.5 1369. ]
43 [ 121. 1581.]
44 [ 364.5 1581. ]
45 [ 615. 1581.]
46 [ 858.5 1581. ]
47 [1109. 1581.]
48 [1352.5 1581. ]
In [28]:
# Define a rotation of a module around the center of the module ...

def rotate(angle, xy, module):
    "Perform the rotation of the xy points around the center of the given module"
    rot = [[cos(angle),-sin(angle)],
           [sin(angle), cos(angle)]]
    center = centers[module]
    return numpy.dot(rot, xy - center) + center
In [29]:
guess1 = [step, y_min, x_min, 0]

def cost1(param):
    """contains: step, y_min, x_min, angle for the first module
    returns the sum of distance squared in pixel space
    """
    step = param[0]
    y_min = param[1]
    x_min = param[2]
    angle = param[3]
    XY = numpy.vstack((indexed1["X"], indexed1["Y"]))
#     rot = [[cos(angle),-sin(angle)],
#            [sin(angle), cos(angle)]]
    xy_min = [[x_min], [y_min]]
    xy_guess = rotate(angle, step * XY + xy_min, module=1)
    delta = xy_guess - numpy.vstack((indexed1["x"], indexed1["y"]))
    return (delta*delta).sum()
In [30]:
print("Before optimization", guess1, "cost=", cost1(guess1))
res1 = minimize(cost1, guess1, method = "slsqp")
print(res1)
print("After optimization", res1.x, "cost=", cost1(res1.x))
print("Average displacement (pixels): ",sqrt(cost1(res1.x)/len(indexed1)))
Before optimization [29, 7.186804354190826, 16.269793540239334, 0] cost= 250.36710826038234
     fun: 0.6337541947714507
     jac: array([-2.32458115e-05, -2.83122063e-07, -2.89082527e-06,  8.84100795e-04])
 message: 'Optimization terminated successfully.'
    nfev: 62
     nit: 8
    njev: 8
  status: 0
 success: True
       x: array([2.93980816e+01, 7.24220064e+00, 1.63128833e+01, 8.76927129e-04])
After optimization [2.93980816e+01 7.24220064e+00 1.63128833e+01 8.76927129e-04] cost= 0.6337541947714507
Average displacement (pixels):  0.11490523221799732

At this step, the grid is perfectly aligned with the first module. This module is used as the reference one and all other are aligned along it, using this first fit:

In [31]:
#retrieve the result of the first module fit:
step, y_min, x_min, angle = res1.x
indexed = numpy.zeros(yxi.shape, dtype=dl)

# rot =  [[cos(angle),-sin(angle)],
#         [sin(angle), cos(angle)]]
# irot =  [[cos(angle), sin(angle)],
#          [-sin(angle), cos(angle)]]

print("cost1: ",cost1([step, y_min, x_min, angle]), "for:", step, y_min, x_min, angle)

xy_min = numpy.array([[x_min], [y_min]])
xy = numpy.vstack((yxi["x"], yxi["y"]))
indexed["y"] = yxi["y"]
indexed["x"] = yxi["x"]
indexed["i"] = yxi["i"]
XY_app = (rotate(-angle, xy, 1)-xy_min) / step
XY_int = numpy.round((XY_app)).astype("int")
indexed["X"] = XY_int[0]
indexed["Y"] = XY_int[1]
xy_guess = rotate(angle, step * XY_int + xy_min, 1)

thres = 1.2
delta = abs(xy_guess - xy)
print((delta>thres).sum(), "suspicious peaks:")
suspicious = indexed[numpy.where(abs(delta>thres))[1]]
print(suspicious)
cost1:  0.6337541947714507 for: 29.398081603464487 7.242200644494111 16.312883343792684 0.0008769271285785412
6 suspicious peaks:
[(1448.30209509, 748.8784325 , 40, 49, 25)
 (1448.66381988, 895.86545633, 40, 49, 30)
 (1448.84506877, 954.56920397, 40, 49, 32)
 (1595.5792897 , 954.49445057, 46, 54, 32)
 (1654.4202804 , 954.44135761, 46, 56, 32)
 (1624.68617365, 807.47743243, 46, 55, 27)]
In [32]:
fig,ax = subplots()
ax.imshow(img.clip(0,1000))
ax.plot(indexed["x"], indexed["y"],".g")
ax.plot(suspicious["x"], suspicious["y"],".r")
Data type cannot be displayed: application/javascript
Out[32]:
[<matplotlib.lines.Line2D at 0x7fed1907b970>]

Only 6 peaks have an initial displacement of more than 1.2 pixel, all located in modules 40 and 46. The visual inspection confirms their localization is valid.

There are 48 (half-)modules which have each of them 2 translations and one rotation. In addition to the step size, this represents 145 degrees of freedom for the fit. The first module is used to align the grid, all other modules are then aligned along this grid.

In [33]:
def submodule_cost(param, module=1):
    """contains: step, y_min_1, x_min_1, angle_1, y_min_2, x_min_2, angle_2, ...
    returns the sum of distance squared in pixel space
    """

    step = param[0]
    y_min1 = param[1]
    x_min1 = param[2]
    angle1 = param[3]

    mask = indexed["i"] == module
    substack = indexed[mask]

    XY = numpy.vstack((substack["X"], substack["Y"]))
#     rot1 = [[cos(angle1), -sin(angle1)],
#             [sin(angle1), cos(angle1)]]
    xy_min1 = numpy.array([[x_min1], [y_min1]])
    xy_guess1 = rotate(angle1, step * XY + xy_min1, module=1)
    #This is guessed spot position for module #1
    if module == 1:
        "Not much to do for module 1"
        delta = xy_guess1 - numpy.vstack((substack["x"], substack["y"]))
    else:
        "perform the correction for given module"
        y_min = param[(module-1)*3+1]
        x_min = param[(module-1)*3+2]
        angle = param[(module-1)*3+3]

#         rot = numpy.array([[cos(angle),-sin(angle)],
#                            [sin(angle), cos(angle)]])
        xy_min = numpy.array([[x_min], [y_min]])
        xy_guess = rotate(angle, xy_guess1+xy_min, module)
        delta = xy_guess - numpy.vstack((substack["x"], substack["y"]))

    return (delta*delta).sum()

guess145 = numpy.zeros(48*3+1)
guess145[:4] = res1.x
for i in range(1, 49):
    print("Cost for module #",i, submodule_cost(guess145, i))

Cost for module # 1 0.6337541947714507
Cost for module # 2 1.8140315882086673
Cost for module # 3 1.8508566011459433
Cost for module # 4 4.910326690902635
Cost for module # 5 1.7748752717866814
Cost for module # 6 7.4825104876812
Cost for module # 7 14.440095344574033
Cost for module # 8 5.21996743058032
Cost for module # 9 23.407235041866166
Cost for module # 10 5.689981348501305
Cost for module # 11 4.125576564142333
Cost for module # 12 8.410826405649853
Cost for module # 13 29.85982067776588
Cost for module # 14 12.628452803186745
Cost for module # 15 3.1377093423071276
Cost for module # 16 5.27731012712747
Cost for module # 17 8.49732773975373
Cost for module # 18 4.653149069135429
Cost for module # 19 6.007095860780655
Cost for module # 20 4.980267155066706
Cost for module # 21 5.136967385086116
Cost for module # 22 5.048721619614182
Cost for module # 23 1.4373118813151964
Cost for module # 24 6.046290046100743
Cost for module # 25 38.254467838332445
Cost for module # 26 19.509797122441007
Cost for module # 27 2.429709408897061
Cost for module # 28 5.488601083628455
Cost for module # 29 10.764108672684777
Cost for module # 30 14.042159706096896
Cost for module # 31 22.94012332153916
Cost for module # 32 18.49373324261777
Cost for module # 33 12.941765597171702
Cost for module # 34 15.371708004318194
Cost for module # 35 39.66702184529083
Cost for module # 36 37.282796287146354
Cost for module # 37 63.700701291866444
Cost for module # 38 23.549518475576456
Cost for module # 39 34.00428432966999
Cost for module # 40 43.80371113090832
Cost for module # 41 15.505216762171113
Cost for module # 42 8.04882054200732
Cost for module # 43 22.735363081615375
Cost for module # 44 10.796022788479295
Cost for module # 45 41.01968148231025
Cost for module # 46 46.258943865873476
Cost for module # 47 24.546579904950047
Cost for module # 48 34.76960408313171

On retrieves that the modules 40 and 46 have large errors. Module 37 as well.

The total cost funtion is hence the sum of all cost function for all modules:

In [34]:
def total_cost(param):
    """contains: step, y_min_1, x_min_1, angle_1, ...
    returns the sum of distance squared in pixel space
    """
    return sum(submodule_cost(param, module=i) for i in range(1,49))
total_cost(guess145)
Out[34]:
778.394900545775
In [35]:
%%time
print("Before optimization", guess145[:10], "cost=", total_cost(guess145))
res_all = minimize(total_cost, guess145, method = "slsqp")
print(res_all)
print("After optimization", res_all.x[:10], "cost=", total_cost(res_all.x))
Before optimization [2.93980816e+01 7.24220064e+00 1.63128833e+01 8.76927129e-04
 0.00000000e+00 0.00000000e+00 0.00000000e+00 0.00000000e+00
 0.00000000e+00 0.00000000e+00] cost= 778.394900545775
     fun: 27.18456488444523
     jac: array([-1.69528580e+00, -2.30972767e-02, -2.18615532e-02,  4.29267883e-01,
        4.63390350e-03,  7.77387619e-03,  2.07757950e-03, -2.22206116e-04,
       -4.09746170e-03, -4.58765030e-03,  8.50105286e-03, -3.49807739e-03,
        1.58166885e-03, -6.04009628e-03,  6.13832474e-03,  5.81264496e-04,
       -5.59329987e-04,  3.29256058e-03, -1.82199478e-03,  2.01272964e-03,
       -2.25973129e-03,  1.72662735e-03,  5.64002991e-03, -1.42302513e-02,
        2.12907791e-03, -2.83336639e-03,  7.68232346e-03,  3.60441208e-03,
        2.16245651e-03, -2.26259232e-03,  4.21571732e-03, -1.73807144e-03,
       -9.25064087e-04,  4.85134125e-03,  1.65052414e-02, -2.67362595e-03,
        8.95690918e-03,  1.40237808e-03, -2.19011307e-03,  3.64971161e-03,
       -6.96182251e-04,  4.17757034e-03,  4.35590744e-03, -4.05645370e-03,
       -2.70509720e-03, -8.60452652e-03,  4.45175171e-03, -3.31354141e-03,
        1.63745880e-03,  7.55310059e-04,  4.28915024e-03, -2.76565552e-03,
        5.60283661e-04, -4.96959686e-03,  4.63199615e-03, -6.83259964e-03,
       -3.91292572e-03, -3.13329697e-03,  8.79812241e-03,  5.94949722e-03,
        8.56351852e-03,  5.25045395e-03,  2.71987915e-03, -1.35469437e-03,
       -3.14664841e-03, -2.15721130e-03, -1.57451630e-03, -1.01771355e-02,
        2.13236809e-02, -8.17918777e-03, -1.89399719e-03, -1.20100975e-02,
       -7.21359253e-03, -2.29454041e-03, -3.47995758e-03,  2.61354446e-03,
        6.97994232e-03, -1.16729736e-03,  2.26497650e-03, -7.73668289e-03,
       -5.27286530e-03, -1.13992691e-02,  8.76617432e-03, -1.02357864e-02,
       -3.89623642e-03, -3.05175781e-03,  7.08103180e-03, -6.40487671e-03,
       -2.88295746e-03, -2.62260437e-05,  7.30991364e-04,  1.69467926e-03,
       -1.10545158e-02,  3.86905670e-03,  2.54058838e-03,  4.36496735e-03,
        1.60989761e-02, -2.31266022e-03, -4.92525101e-03,  2.74658203e-04,
       -1.32822990e-02,  1.42421722e-02,  1.33924484e-02, -4.65440750e-03,
        7.62748718e-03, -1.11608505e-02, -1.42002106e-03, -2.97546387e-04,
        5.89656830e-03, -1.70412064e-02, -2.28986740e-02,  4.25291061e-03,
        1.65419579e-02,  6.25991821e-03,  1.13039017e-02, -2.06947327e-04,
       -5.17129898e-03,  1.42908096e-03,  2.45571136e-03,  5.43212891e-03,
        1.70331001e-02, -1.47132874e-02, -2.09856033e-03, -9.98163223e-03,
       -1.34849548e-03,  1.35526657e-02,  6.79302216e-03, -3.90005112e-03,
       -6.71577454e-03,  1.24020576e-02, -4.85897064e-03,  2.33845711e-02,
        3.63302231e-03, -1.02996826e-04, -1.64990425e-02, -1.99508667e-03,
        4.46748734e-03,  5.60951233e-03,  1.91512108e-02, -5.91278076e-03,
       -1.16286278e-02, -3.27157974e-03, -6.02722168e-03, -7.75861740e-03,
        3.23224068e-03])
 message: 'Optimization terminated successfully.'
    nfev: 13462
     nit: 89
    njev: 89
  status: 0
 success: True
       x: array([ 2.94082830e+01,  7.21153864e+00,  1.62772293e+01,  8.29133756e-04,
       -1.49134822e-01, -1.98989780e-01,  3.22787818e-04,  1.25919992e-01,
       -2.38726673e-01,  9.70909922e-04,  2.89556028e-01, -2.25602900e-01,
        1.98576011e-03, -7.46090857e-02, -4.65022384e-01,  3.64513000e-04,
        3.15530465e-01, -6.41090374e-01,  2.12977843e-03, -5.99393163e-01,
        7.53606591e-02,  1.39091865e-03, -4.29454741e-01, -1.34262298e-01,
        1.13086966e-03, -6.99061363e-01, -4.13976094e-01,  9.42292710e-04,
       -2.09729817e-01, -4.54233391e-01,  2.15329270e-03,  2.37398684e-01,
       -2.66279738e-01,  5.18065989e-04,  3.44153205e-01, -1.72231949e-01,
        6.95029703e-04, -6.85125793e-01, -5.70947884e-01,  1.19308963e-03,
       -5.09775526e-01, -5.82568789e-01,  1.35911198e-03, -2.88791293e-01,
       -3.34382644e-01,  1.01371768e-03, -1.77849834e-02, -5.55338288e-01,
        1.31963160e-03, -2.95278480e-01, -7.63523064e-01,  4.81229249e-04,
       -2.79515374e-02, -6.83761510e-01,  1.58385910e-03, -4.81814445e-01,
       -2.29286463e-01,  7.55124131e-04, -2.17764489e-01, -4.94392240e-01,
        7.63855235e-04, -4.20354977e-01, -4.13314189e-01, -2.52887412e-04,
       -2.37437497e-01, -5.54789783e-01,  1.67238170e-03, -1.67640239e-01,
       -2.91835448e-01,  1.28641845e-03,  1.28060987e-01, -4.13686920e-01,
        1.70257833e-03, -7.41004109e-01, -7.96588778e-01,  9.92908160e-04,
       -3.97497662e-01, -8.79400338e-01,  1.80002721e-03, -8.53283743e-02,
       -2.33829471e-01, -4.77172627e-04,  5.75139985e-02, -2.65470780e-01,
        6.42371680e-04, -8.87043889e-02, -8.52484896e-01,  6.04836505e-04,
        1.76609589e-01, -6.43890828e-01,  2.37902297e-03, -5.56858961e-01,
       -7.12322422e-01, -6.98909929e-04, -6.74802085e-01, -8.12725741e-01,
        1.48447777e-03, -1.63480015e-01, -6.92393269e-01, -5.89772345e-04,
       -3.02803408e-02, -8.15326373e-01,  1.60480957e-03, -1.94268423e-01,
       -1.29973810e+00,  6.85898340e-04, -3.91252582e-02, -1.30838275e+00,
        6.07108528e-04, -1.02703769e+00, -1.04809975e+00,  1.36880494e-05,
       -9.30826763e-01, -9.01418700e-01,  9.22938499e-06, -5.76233210e-01,
       -1.05403930e+00,  1.56133801e-04, -2.50843882e-01, -1.28588025e+00,
        2.48009460e-03, -1.63832916e-01, -9.40310826e-01,  3.87868140e-04,
       -2.02473380e-01, -8.29605292e-01,  6.61447664e-04, -8.32764047e-01,
       -6.68905202e-01,  1.03494547e-03, -6.26782545e-01, -7.70933409e-01,
        5.99300599e-04, -5.39338254e-01, -1.17800489e+00, -3.08115549e-04,
       -3.71533757e-01, -1.35700499e+00,  1.33100853e-03, -7.80768029e-02,
       -1.02121301e+00,  1.20183923e-03,  2.54813189e-01, -9.24032021e-01,
        1.89070869e-03])
After optimization [ 2.94082830e+01  7.21153864e+00  1.62772293e+01  8.29133756e-04
 -1.49134822e-01 -1.98989780e-01  3.22787818e-04  1.25919992e-01
 -2.38726673e-01  9.70909922e-04] cost= 27.18456488444523
CPU times: user 35.2 s, sys: 6.73 ms, total: 35.2 s
Wall time: 35.2 s
In [36]:
for i in range(1,49):
    print("Module id: %d cost: %.3f Δx: %.3f, Δy: %.3f rot: %.3f°"%
          (i, submodule_cost(res_all.x, i), res_all.x[-2+i*3], res_all.x[-1+i*3], numpy.rad2deg(res_all.x[i*3])))
Module id: 1 cost: 0.684 Δx: 7.212, Δy: 16.277 rot: 0.048°
Module id: 2 cost: 0.473 Δx: -0.149, Δy: -0.199 rot: 0.018°
Module id: 3 cost: 0.650 Δx: 0.126, Δy: -0.239 rot: 0.056°
Module id: 4 cost: 0.613 Δx: 0.290, Δy: -0.226 rot: 0.114°
Module id: 5 cost: 0.566 Δx: -0.075, Δy: -0.465 rot: 0.021°
Module id: 6 cost: 0.666 Δx: 0.316, Δy: -0.641 rot: 0.122°
Module id: 7 cost: 0.696 Δx: -0.599, Δy: 0.075 rot: 0.080°
Module id: 8 cost: 0.341 Δx: -0.429, Δy: -0.134 rot: 0.065°
Module id: 9 cost: 0.566 Δx: -0.699, Δy: -0.414 rot: 0.054°
Module id: 10 cost: 0.661 Δx: -0.210, Δy: -0.454 rot: 0.123°
Module id: 11 cost: 0.616 Δx: 0.237, Δy: -0.266 rot: 0.030°
Module id: 12 cost: 0.495 Δx: 0.344, Δy: -0.172 rot: 0.040°
Module id: 13 cost: 0.763 Δx: -0.685, Δy: -0.571 rot: 0.068°
Module id: 14 cost: 0.353 Δx: -0.510, Δy: -0.583 rot: 0.078°
Module id: 15 cost: 0.645 Δx: -0.289, Δy: -0.334 rot: 0.058°
Module id: 16 cost: 0.492 Δx: -0.018, Δy: -0.555 rot: 0.076°
Module id: 17 cost: 0.440 Δx: -0.295, Δy: -0.764 rot: 0.028°
Module id: 18 cost: 0.779 Δx: -0.028, Δy: -0.684 rot: 0.091°
Module id: 19 cost: 0.748 Δx: -0.482, Δy: -0.229 rot: 0.043°
Module id: 20 cost: 0.423 Δx: -0.218, Δy: -0.494 rot: 0.044°
Module id: 21 cost: 0.520 Δx: -0.420, Δy: -0.413 rot: -0.014°
Module id: 22 cost: 0.676 Δx: -0.237, Δy: -0.555 rot: 0.096°
Module id: 23 cost: 0.503 Δx: -0.168, Δy: -0.292 rot: 0.074°
Module id: 24 cost: 0.822 Δx: 0.128, Δy: -0.414 rot: 0.098°
Module id: 25 cost: 0.575 Δx: -0.741, Δy: -0.797 rot: 0.057°
Module id: 26 cost: 0.320 Δx: -0.397, Δy: -0.879 rot: 0.103°
Module id: 27 cost: 0.642 Δx: -0.085, Δy: -0.234 rot: -0.027°
Module id: 28 cost: 0.599 Δx: 0.058, Δy: -0.265 rot: 0.037°
Module id: 29 cost: 0.498 Δx: -0.089, Δy: -0.852 rot: 0.035°
Module id: 30 cost: 0.830 Δx: 0.177, Δy: -0.644 rot: 0.136°
Module id: 31 cost: 0.687 Δx: -0.557, Δy: -0.712 rot: -0.040°
Module id: 32 cost: 0.422 Δx: -0.675, Δy: -0.813 rot: 0.085°
Module id: 33 cost: 0.616 Δx: -0.163, Δy: -0.692 rot: -0.034°
Module id: 34 cost: 0.393 Δx: -0.030, Δy: -0.815 rot: 0.092°
Module id: 35 cost: 0.447 Δx: -0.194, Δy: -1.300 rot: 0.039°
Module id: 36 cost: 0.587 Δx: -0.039, Δy: -1.308 rot: 0.035°
Module id: 37 cost: 0.603 Δx: -1.027, Δy: -1.048 rot: 0.001°
Module id: 38 cost: 0.295 Δx: -0.931, Δy: -0.901 rot: 0.001°
Module id: 39 cost: 0.558 Δx: -0.576, Δy: -1.054 rot: 0.009°
Module id: 40 cost: 0.498 Δx: -0.251, Δy: -1.286 rot: 0.142°
Module id: 41 cost: 0.468 Δx: -0.164, Δy: -0.940 rot: 0.022°
Module id: 42 cost: 0.569 Δx: -0.202, Δy: -0.830 rot: 0.038°
Module id: 43 cost: 0.633 Δx: -0.833, Δy: -0.669 rot: 0.059°
Module id: 44 cost: 0.413 Δx: -0.627, Δy: -0.771 rot: 0.034°
Module id: 45 cost: 0.671 Δx: -0.539, Δy: -1.178 rot: -0.018°
Module id: 46 cost: 0.542 Δx: -0.372, Δy: -1.357 rot: 0.076°
Module id: 47 cost: 0.537 Δx: -0.078, Δy: -1.021 rot: 0.069°
Module id: 48 cost: 0.593 Δx: 0.255, Δy: -0.924 rot: 0.108°

Analysis: Modules 40, 46 and 48 show large displacement but the fitting precedure allowed to reduce the residual cost to the same value as other modules.

Reconstruction of the pixel position

The pixel position can be obtained from the standard Pilatus detector. Each module is then displaced according to the fitted values, except the first one which is left where it is.

In [37]:
def correct(x, y, dx, dy, angle, module):
    "apply the correction dx, dy and angle to those pixels ..."
    trans = numpy.array([[dx],
                         [dy]])
    xy_guess = numpy.vstack((x.ravel(),
                             y.ravel()))
    xy_cor = rotate(-angle, xy_guess, module) - trans
    xy_cor.shape = ((2,)+x.shape)
    return xy_cor[0], xy_cor[1]
In [38]:
pixel_coord = pyFAI.detector_factory("Pilatus2MCdTe").get_pixel_corners()
pixel_coord_raw = pixel_coord.copy()
for i in range(2, 49):
    # Extract the pixel corners for one module
    module_idx = numpy.where(mid == i)
    one_module = pixel_coord_raw[module_idx]
    #retrieve the fitted values
    dy, dx, angle = res_all.x[-2+i*3:1+3*i]

    y = one_module[..., 1]/pilatus.pixel1
    x = one_module[..., 2]/pilatus.pixel2

    #apply the correction the other way around
    x_cor, y_cor = correct(x, y, dx, dy, angle, i)
    one_module[...,1] = y_cor * pilatus.pixel1 #y
    one_module[...,2] = x_cor * pilatus.pixel2 #x
    #Update the array
    pixel_coord_raw[module_idx] = one_module

Update the detector and save it in HDF5

In [39]:
pilatus.set_pixel_corners(pixel_coord_raw)
pilatus.mask = all_masks
pilatus.save("Pilatus_ID15_raw.h5")
In [40]:
displ = numpy.sqrt(((pixel_coord - pixel_coord_raw)**2).sum(axis=-1))
displ /= pilatus.pixel1 #convert in pixel units
fig, ax = subplots()
ax.hist(displ.ravel(), 100)
ax.set_title("Displacement of pixels versus the reference representation")
ax.set_xlabel("Error in pixel size (172µm)")
Data type cannot be displayed: application/javascript
Out[40]:
Text(0.5, 0, 'Error in pixel size (172µm)')
In [41]:
misaligned = numpy.vstack((pixel_coord_raw[..., 2].ravel(), #x
                           pixel_coord_raw[..., 1].ravel())) #y

reference = numpy.vstack((pixel_coord[..., 2].ravel(), #x
                          pixel_coord[..., 1].ravel())) #y
In [42]:
#Kabsch alignment of the whole detector ...

def kabsch(P, R):
    "Align P on R"
    centroid_P = P.mean(axis=0)
    centroid_R = R.mean(axis=0)
    centered_P = P - centroid_P
    centered_R = R - centroid_R
    C = numpy.dot(centered_P.T, centered_R)
    V, S, W = numpy.linalg.svd(C)
    d = (numpy.linalg.det(V) * numpy.linalg.det(W)) < 0.0
    if d:
        S[-1] = -S[-1]
        V[:, -1] = -V[:, -1]
    # Create Rotation matrix U
    U = numpy.dot(V, W)
    P = numpy.dot(centered_P, U)
    return P + centroid_R

%time aligned = kabsch(misaligned.T, reference.T).T
CPU times: user 628 ms, sys: 548 ms, total: 1.18 s
Wall time: 206 ms
In [43]:
displ = numpy.sqrt(((aligned-reference)**2).sum(axis=0))
displ /= pilatus.pixel1 #convert in pixel units
fig, ax = subplots()
ax.hist(displ.ravel(), 100)
ax.set_title("Displacement of pixels versus the reference representation")
ax.set_xlabel("Pixel size (172µm)")
Data type cannot be displayed: application/javascript
Out[43]:
Text(0.5, 0, 'Pixel size (172µm)')
In [44]:
pixel_coord_aligned = pixel_coord.copy()
pixel_coord_aligned[...,1] = aligned[1,:].reshape(pixel_coord.shape[:-1])
pixel_coord_aligned[...,2] = aligned[0,:].reshape(pixel_coord.shape[:-1])

pilatus.set_pixel_corners(pixel_coord_aligned)
pilatus.mask = all_masks
pilatus.save("Pilatus_ID15_Kabsch.h5")
In [45]:
fig, ax = subplots(1, 2, figsize=(8, 4))
ax[0].imshow((pixel_coord_aligned[...,2].mean(axis=-1) - pixel_coord[...,2].mean(axis=-1))/pilatus.pixel2)
ax[0].set_title("Displacement x (in pixel)")
ax[1].imshow((pixel_coord_aligned[...,1].mean(axis=-1) - pixel_coord[...,1].mean(axis=-1))/pilatus.pixel1)
ax[1].set_title("Displacement y (in pixel)")
Data type cannot be displayed: application/javascript
Out[45]:
Text(0.5, 1.0, 'Displacement y (in pixel)')

Validation of the result

To validate the improvement obtained, one can perform the experiment calibration and the 2D integration of a reference compound, the 2D integration with either the reference from Dectris or this freshly refined detector.

In [46]:
# The geometry has been obtained from pyFAI
geo = { "dist":  0.8001094657585498,
        "poni1": 0.14397714477803805,
        "poni2": 0.12758748978422835,
        "rot1":  0.0011165686147339689,
        "rot2":  0.0002214091645638961,
        "rot3":  0,
        "detector": "Pilatus2MCdTe"}
ai_unc = AzimuthalIntegrator(**geo)
geo["detector"] = "Pilatus_ID15_Kabsch.h5"
ai_cor = AzimuthalIntegrator(**geo)
fig, ax = subplots(1, 2, figsize=(8,4))
res_unc = ai_unc.integrate2d(rings, 100, 100, radial_range=(7.9, 8.2), unit="2th_deg", method="nosplit", mask=all_masks)
res_cor = ai_cor.integrate2d(rings, 100, 100, radial_range=(7.9, 8.2), unit="2th_deg", method="nosplit", mask=all_masks)
opts = {"origin":"lower",
        "extent": [res_unc.radial.min(), res_unc.radial.max(), -180, 180],
        "aspect":"auto",
        "cmap":"inferno"}
ax[0].imshow(res_unc[0], **opts)
ax[1].imshow(res_cor[0], **opts)
ax[0].set_xlabel(res_unc.unit.label)
ax[1].set_xlabel(res_cor.unit.label)
ax[0].set_ylabel(r"Azimuthal angle $\chi$ ($^{o}$)")
# ax[0].set_ylabel(r"Azimuthal angle $\chi$ ($^{o}$)")
ax[0].set_title("Uncorrected")
ax[1].set_title("Corrected")
Data type cannot be displayed: application/javascript
WARNING:pyFAI.DEPRECATION:Function _integrate2d_legacy is deprecated since pyFAI version 0.20.
  File "<ipython-input-46-7337d7138cba>", line 13, in <module>
    res_unc = ai_unc.integrate2d(rings, 100, 100, radial_range=(7.9, 8.2), unit="2th_deg", method="nosplit", mask=all_masks)
WARNING:pyFAI.DEPRECATION:Function _integrate2d_legacy is deprecated since pyFAI version 0.20.
  File "<ipython-input-46-7337d7138cba>", line 14, in <module>
    res_cor = ai_cor.integrate2d(rings, 100, 100, radial_range=(7.9, 8.2), unit="2th_deg", method="nosplit", mask=all_masks)
Out[46]:
Text(0.5, 1.0, 'Corrected')

Conclusion

This tutorial presents the way to calibrate a module based detector using the Pilatus2M CdTe from ESRF-ID15. The HDF5 file generated is directly useable by any parts of pyFAI, the reader is invited in calibrating the rings images with the default definition and with this optimized definition and check the residual error is almost divided by a factor two.

To come back on the precision of the localization of the pixel: not all the pixel are within the specifications provided by Dectris which claims the misaliment of the modules is within one pixel.

In [47]:
print("Total execution time: ", time.time()-start_time)
Total execution time:  71.63153123855591