xref: /rk3399_rockchip-uboot/tools/dtoc/fdt_util.py (revision 355c67c35a8ce5aa9e9e2e2e8df99413c8215093)
1ec564b47SSimon Glass#!/usr/bin/python
2ec564b47SSimon Glass#
3ec564b47SSimon Glass# Copyright (C) 2016 Google, Inc
4ec564b47SSimon Glass# Written by Simon Glass <sjg@chromium.org>
5ec564b47SSimon Glass#
6ec564b47SSimon Glass# SPDX-License-Identifier:      GPL-2.0+
7ec564b47SSimon Glass#
8ec564b47SSimon Glass
9*355c67c3SSimon Glassimport os
10ec564b47SSimon Glassimport struct
11*355c67c3SSimon Glassimport tempfile
12*355c67c3SSimon Glass
13*355c67c3SSimon Glassimport command
14*355c67c3SSimon Glassimport tools
15ec564b47SSimon Glass
16ec564b47SSimon Glassdef fdt32_to_cpu(val):
17ec564b47SSimon Glass    """Convert a device tree cell to an integer
18ec564b47SSimon Glass
19ec564b47SSimon Glass    Args:
20ec564b47SSimon Glass        Value to convert (4-character string representing the cell value)
21ec564b47SSimon Glass
22ec564b47SSimon Glass    Return:
23ec564b47SSimon Glass        A native-endian integer value
24ec564b47SSimon Glass    """
25ec564b47SSimon Glass    return struct.unpack(">I", val)[0]
26*355c67c3SSimon Glass
27*355c67c3SSimon Glassdef EnsureCompiled(fname):
28*355c67c3SSimon Glass    """Compile an fdt .dts source file into a .dtb binary blob if needed.
29*355c67c3SSimon Glass
30*355c67c3SSimon Glass    Args:
31*355c67c3SSimon Glass        fname: Filename (if .dts it will be compiled). It not it will be
32*355c67c3SSimon Glass            left alone
33*355c67c3SSimon Glass
34*355c67c3SSimon Glass    Returns:
35*355c67c3SSimon Glass        Filename of resulting .dtb file
36*355c67c3SSimon Glass    """
37*355c67c3SSimon Glass    _, ext = os.path.splitext(fname)
38*355c67c3SSimon Glass    if ext != '.dts':
39*355c67c3SSimon Glass        return fname
40*355c67c3SSimon Glass
41*355c67c3SSimon Glass    dts_input = tools.GetOutputFilename('source.dts')
42*355c67c3SSimon Glass    dtb_output = tools.GetOutputFilename('source.dtb')
43*355c67c3SSimon Glass
44*355c67c3SSimon Glass    search_paths = [os.path.join(os.getcwd(), 'include')]
45*355c67c3SSimon Glass    root, _ = os.path.splitext(fname)
46*355c67c3SSimon Glass    args = ['-E', '-P', '-x', 'assembler-with-cpp', '-D__ASSEMBLY__']
47*355c67c3SSimon Glass    args += ['-Ulinux']
48*355c67c3SSimon Glass    for path in search_paths:
49*355c67c3SSimon Glass        args.extend(['-I', path])
50*355c67c3SSimon Glass    args += ['-o', dts_input, fname]
51*355c67c3SSimon Glass    command.Run('cc', *args)
52*355c67c3SSimon Glass
53*355c67c3SSimon Glass    # If we don't have a directory, put it in the tools tempdir
54*355c67c3SSimon Glass    search_list = []
55*355c67c3SSimon Glass    for path in search_paths:
56*355c67c3SSimon Glass        search_list.extend(['-i', path])
57*355c67c3SSimon Glass    args = ['-I', 'dts', '-o', dtb_output, '-O', 'dtb']
58*355c67c3SSimon Glass    args.extend(search_list)
59*355c67c3SSimon Glass    args.append(dts_input)
60*355c67c3SSimon Glass    command.Run('dtc', *args)
61*355c67c3SSimon Glass    return dtb_output
62