File: converters.py

package info (click to toggle)
python-fabio 0.11.0%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 9,092 kB
  • sloc: python: 19,244; ansic: 1,085; makefile: 219; sh: 215
file content (92 lines) | stat: -rw-r--r-- 3,126 bytes parent folder | download | duplicates (4)
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
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
# coding: utf-8
#
#    Project: X-ray image reader
#             https://github.com/silx-kit/fabio
#
#
#    Copyright (C) European Synchrotron Radiation Facility, Grenoble, France
#
#    Principal author:       Jérôme Kieffer (Jerome.Kieffer@ESRF.eu)
#
#  Permission is hereby granted, free of charge, to any person
#  obtaining a copy of this software and associated documentation files
#  (the "Software"), to deal in the Software without restriction,
#  including without limitation the rights to use, copy, modify, merge,
#  publish, distribute, sublicense, and/or sell copies of the Software,
#  and to permit persons to whom the Software is furnished to do so,
#  subject to the following conditions:
#
#  The above copyright notice and this permission notice shall be
#  included in all copies or substantial portions of the Software.
#
#  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
#  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
#  OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
#  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
#  HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
#  WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
#  FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
#  OTHER DEALINGS IN THE SOFTWARE.

"""Converter module.

This is for the moment empty (populated only with almost pass through anonymous functions)
but aims to be populated with more sofisticated translators...

"""

__author__ = "Jérôme Kieffer"
__contact__ = "jerome.kieffer@esrf.eu"
__license__ = "MIT"
__copyright__ = "European Synchrotron Radiation Facility, Grenoble, France"

import logging
logger = logging.getLogger(__name__)


def convert_data_integer(data):
    """
    convert data to integer
    """
    if data is not None:
        return data.astype(int)
    else:
        return data


CONVERSION_HEADER = {
    ("edfimage", "edfimage"): lambda header: header,
}

CONVERSION_DATA = {
    ("edfimage", "edfimage"): lambda data: data,
    ("edfimage", "cbfimage"): convert_data_integer,
    ("edfimage", "mar345image"): convert_data_integer,
    ("edfimage", "fit2dmaskimage"): convert_data_integer,
    ("edfimage", "kcdimage"): convert_data_integer,
    ("edfimage", "OXDimage"): convert_data_integer,
    ("edfimage", "pnmimage"): convert_data_integer,
}


def convert_data(inp, outp, data):
    """
    Return data converted to the output format ... over-simplistic
    implementation for the moment...

    :param str inp: input format (like "cbfimage")
    :param str outp: output format (like "cbfimage")
    :param numpy.ndarray data: the actual dataset to be transformed
    """
    return CONVERSION_DATA.get((inp, outp), lambda data: data)(data)


def convert_header(inp, outp, header):
    """
    Return header converted to the output format

    :param str inp: input format (like "cbfimage")
    :param str outp: output format (like "cbfimage")
    :param dict header: the actual set of headers to be transformed
    """
    return CONVERSION_HEADER.get((inp, outp), lambda header: header)(header)