File: imageFormat.py

package info (click to toggle)
esys-particle 2.3.5%2Bdfsg2-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 13,132 kB
  • sloc: cpp: 81,480; python: 5,872; makefile: 1,259; sh: 313; perl: 225
file content (92 lines) | stat: -rw-r--r-- 3,048 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
#############################################################
##                                                         ##
## Copyright (c) 2003-2017 by The University of Queensland ##
## Centre for Geoscience Computing                         ##
## http://earth.uq.edu.au/centre-geoscience-computing      ##
##                                                         ##
## Primary Business: Brisbane, Queensland, Australia       ##
## Licensed under the Open Software License version 3.0    ##
## http://www.apache.org/licenses/LICENSE-2.0              ##
##                                                         ##
#############################################################

"""
Defines the L{ImageFormat} class and functions for mapping a
file name extension to an associated C{ImageFormat} object.
"""
import os
import os.path

class ImageFormat(object):
    """
    Class representing an image format.
    """
    def __init__(self, name):
        """
        Constructor.
        @type name: str
        @param name: Name assigned to this image format.
        """
        self.name = name

    def getName(self):
        """
        Returns the name associated with this image format.
        @rtype: str
        """
        return self.name

    def __str__(self):
        return self.getName()


PNG = ImageFormat("PNG")
PNM = ImageFormat("PNM")

_nameFormatDict = dict()
_nameFormatDict[str.upper(str(PNG))] = PNG
_nameFormatDict[str.upper(str(PNM))] = PNM

def _getDelimitedFormatNameString():
    return ", ".join(map(str,list(_nameFormatDict.keys())))

def getFormatFromName(formatName, ext=None):
    """
    Returns the C{{ImageFormat}} object which corresponds
    to a specified image-format name (string).
    @type formatName: str
    @param formatName: The name of an image format, one of: {0:s}
    @type ext: str
    @param ext: File name extension for error message string.
    """.format(_getDelimitedFormatNameString())
    if str.upper(formatName) in _nameFormatDict:
        return _nameFormatDict[str.upper(formatName)]
    raise \
      ValueError(
        (
            "No image format found which matched extension '{0:s}';" +
            " valid image file formats are: {1:s}"
        ).format(ext, _getDelimitedFormatNameString())
      )

def getFormatFromExtension(fileName):
    """
    Returns the C{ImageFormat} object which corresponds
    to a specified file name. Uses the C{fileName} extension
    to try and deduce the corresponding C{ImageFormat} object.
    @type fileName: str
    @param fileName: A file name.
    @rtype: C{ImageFormat}
    @return: An C{ImageFormat} object corresponding to the
    specified file name (and corresponding file name extension).
    """
    (base, ext) = os.path.splitext(fileName)
    if (len(ext) > 0):
        formatName = str.lstrip(ext, ".")
    else:
        raise ValueError(
            "Could not determine image format from file "
            +
            "name " + fileName + ", no extension."
        )
    return getFormatFromName(formatName, ext)