### Navigation

- ☰
- PyTecplot-0.6 »
- Module code »

# Source code for tecplot.data.load

    from builtins import int

    import logging

    from os import path
    from six import string_types
    from textwrap import dedent

    from ..tecutil import _tecutil, lock, ArgList, IndexSet, StringList, sv
    from ..constant import ReadDataOption, VarLoadMode
    from .. import layout
    from ..exception import *

    log = logging.getLogger(__name__)

    @lock()
    def _load_data_v3(arglist,
                      frame,
                      append,
                      reset_style,
                      initial_plot_first_zone_only,
                      initial_plot_type,
                      assign_strand_ids,
                      add_zones_to_existing_strands):
        frame = frame or layout.active_frame()

        read_data_option = None
        if append and frame.has_dataset and frame.dataset.num_zones:
            read_data_option = ReadDataOption.AppendData
        elif frame.has_dataset and frame.dataset.num_zones:
            read_data_option = ReadDataOption.ReplaceData

        with frame.activated():
            arglist.update(**{
                sv.READDATAOPTION: read_data_option,
                sv.RESETSTYLE: reset_style,
                sv.INITIALPLOTFIRSTZONEONLY: initial_plot_first_zone_only,
                sv.INITIALPLOTTYPE: initial_plot_type,
                sv.ASSIGNSTRANDIDS: assign_strand_ids,
                sv.ADDZONESTOEXISTINGSTRANDS: add_zones_to_existing_strands})

            if __debug__:
                msg = 'loading data:'
                for k, v in arglist.items():
                    if k == sv.FILENAMESORINSTRUCTIONS:
                        msg += '\n    {}:'.format(k)
                        for i in v:
                            msg += '\n        {}'.format(i)
                    else:
                        msg += '\n    {}: {}'.format(k, v)
                log.debug(msg)

            if not _tecutil.DataSetReadX(arglist):
                raise TecplotSystemError()

            return frame.dataset

    @lock()
    def load_cgns(filenames,

                  frame=None,

                  append=True,
                  reset_style=None,
                  initial_plot_first_zone_only=None,
                  initial_plot_type=None,

                  zones=None,
                  variables=None,

                  load_convergence_history=None,
                  combine_fe_sections=None,
                  average_to_nodes=None,
                  uniform_grid=None,
                  assign_strand_ids=None,
                  add_zones_to_existing_strands=None,

                  include_boundary_conditions=None):
        """Read CGNS data files.

        Parameters:
            filenames (`string <str>` or `list` of `strings <str>`): CGNS data files
                to be read. **Note:** Relative paths are considered to be from
                the running python script's current working directory which can
                be obtained via :func:`os.getcwd`.

            frame (`Frame`, optional): The `Frame` to attach the
                resulting `Dataset`. If `None`, the currently active `Frame` is
                used and the zones are appended by default.

            append (`boolean <bool>`, optional): Append the data to the existing
                `Dataset`. If `False`, the existing data attached to the `Frame` is
                deleted and replaced. (default: `True`)
            reset_style (`boolean <bool>`, optional): Reset the style for
                destination `Frame`, if `False`, the `Frame`'s current style is
                preserved. (default: `True`)
            initial_plot_first_zone_only (`boolean <bool>`, optional): Informs
                the |Tecplot Engine| that after the data is loaded it only needs
                to activate the first enabled `Zone <data_access>` for the initial plot. This
                option is particularly useful if you have many `Zones <data_access>`
                and want to get the data into the |Tecplot Engine| and the first
                `Zone <data_access>` drawn as fast as possible. The inactive `Zones <data_access>` can
                always be activated when needed. (default: `False`)
            initial_plot_type (`PlotType`, optional): Forces a specific type of
                plot upon loading of the data. Only used if *resetstyle* is `True`.
                To have |Tecplot 360 EX| determine the most appropriate plot type
                for the data, use `PlotType.Automatic`. Possible values are:
                `PlotType.Automatic` (default), `Cartesian3D`, `Cartesian2D`,
                `XYLine`, `PlotType.Sketch`, `PolarLine`.

            zones (`list` of `integers <int>`, optional): List of zone indexes to
                load starting from zero. `None` implies loading all zones.
                (default: `None`)
            variables (`list` of `integers <int>`, optional): List of variable
                indexes, beyond the first coordinate variables, to load starting
                from zero. `None` implies loading all variables. The grid will
                always be loaded and an index of zero indicates the first
                non-coordinate variable. (default: `None`)

            load_convergence_history (`boolean <bool>`, optional): Load the
                global convergence history rather than any grid or solution data.
                (default: `False`)
            combine_fe_sections (`boolean <bool>`, optional): Combine all
                finite-element sections with the zone cell-dimension into one
                zone. (default: `False`)
            average_to_nodes (`string <str>`, optional): Average
                cell-centered data to grid nodes using the specified method.
                (Options: `None`, "Arithmetic", "Laplacian")
            uniform_grid (`boolean <bool>`, optional): Indicates the grid
                structure is the same for all time steps. (default: `True`)
            assign_strand_ids (`boolean <bool>`, optional): Assign strand ID's to
                zones that have a strand ID of -1. (default: `True`)
            add_zones_to_existing_strands (`boolean <bool>`, optional): Add the
                `Zones <data_access>` to matching strands, if they exist. Otherwise, if
                the new data specifies strands, new ones will be created beginning
                after the last strand in the `Dataset`. (default: `False`)

            include_boundary_conditions (`boolean <bool>`, optional): Load the
                boundary conditions along with the data. (default: `False`)

        Returns:
            `Dataset`: The `Dataset` holding the loaded data.

        Raises:
            `TecplotSystemError`: Internal error when loading data.
            `TecplotTypeError`: Invalid input.
        """
        if __debug__:
            avg_to_nodes_opts = [None, 'Arithmetic', 'Laplacian']
            if average_to_nodes not in avg_to_nodes_opts:
                msg = 'average_to_nodes must be one of: '
                msg += ', '.join(str(x) for x in avg_to_nodes_opts)
                raise TecplotTypeError(msg)

        with StringList('STANDARDSYNTAX', '1.0') as instr:
            with ArgList() as arglist:
                arglist[sv.DATASETREADER] = 'CGNS LOADER'
                instr += ['LoaderVersion', 'V3']

                # We need to add the cgns library version to the arglist,
                # although this setting is not exposed in the API.
                instr += ['CgnsLibraryVersion', '3.1.4']

                if isinstance(filenames, string_types):
                    filenames = filenames.split(',')
                instr += ['FILELIST_CGNSFILES', str(len(filenames))]
                instr += [path.abspath(f) for f in filenames]

                if zones is not None:
                    indexes = [str(int(z)) for z in zones]
                    instr += ['ZoneList', ','.join(indexes)]
                if variables is not None:
                    indexes = [str(int(v)) for v in variables]
                    instr += ['VarList', ','.join(indexes)]

                if average_to_nodes is not None:
                    instr += ['AverageToNodes', average_to_nodes]
                if uniform_grid is not None:
                    instr += ['UniformGridStructure',
                              'Yes' if uniform_grid else 'No']

                if combine_fe_sections is not None:
                    instr += [
                        'SectionLoad', 'SeparateZones'
                        if combine_fe_sections else 'Combine']
                if include_boundary_conditions is not None:
                    instr += ['LoadBCs',
                              'Yes' if include_boundary_conditions else 'No']
                if load_convergence_history is not None:
                    instr += ['LoadConvergenceHistory',
                              'Yes' if load_convergence_history else 'No']
                if assign_strand_ids is not None:
                    instr += ['AssignStrandIDs',
                              'Yes' if assign_strand_ids else 'No']

                arglist[sv.FILENAMESORINSTRUCTIONS] = instr

                return _load_data_v3(arglist,
                                     frame=frame,
                                     append=append,
                                     reset_style=reset_style,
                                     initial_plot_first_zone_only=initial_plot_first_zone_only,
                                     initial_plot_type=initial_plot_type,
                                     assign_strand_ids=None,
                                     add_zones_to_existing_strands=add_zones_to_existing_strands)

    @lock()
    def load_fluent(case_filenames=None,
                    data_filenames=None,

                    frame=None,

                    append=True,
                    reset_style=None,
                    initial_plot_first_zone_only=None,
                    initial_plot_type=None,

                    zones=None,
                    variables=None,

                    all_poly_zones=None,
                    average_to_nodes=None,
                    time_interval=None,
                    assign_strand_ids=True,
                    add_zones_to_existing_strands=None,

                    include_particle_data=None):
        """Read Fluent data files.

        Parameters:
            case_filenames (`string <str>` or `list` of `strings <str>`, optional):
                Case (*.cas*) files to be read.
            data_filenames (`string <str>` or `list` of `strings <str>`, optional):
                Data (*.dat*, *.xml* etc.) files to be read.

            frame (`Frame`, optional): The `Frame` to attach the
                resulting `Dataset`. If `None`, the currently active `Frame` is
                used and the zones are appended by default.

            append (`boolean <bool>`, optional): Append the data to the existing
                `Dataset`. If `False`, the existing data attached to the `Frame` is
                deleted and replaced. (default: `True`)
            reset_style (`boolean <bool>`, optional): Reset the style for
                destination `Frame`, if `False`, the `Frame`'s current style is
                preserved. (default: `True`)
            initial_plot_first_zone_only (`boolean <bool>`, optional): Informs
                the |Tecplot Engine| that after the data is loaded it only needs
                to activate the first enabled `Zone <data_access>` for the initial plot. This
                option is particularly useful if you have many `Zones <data_access>`
                and want to get the data into the |Tecplot Engine| and the first
                `Zone <data_access>` drawn as fast as possible. The inactive `Zones <data_access>` can
                always be activated when needed. (default: `False`)
            initial_plot_type (`PlotType`, optional): Forces a specific type of
                plot upon loading of the data. Only used if *resetstyle* is `True`.
                To have |Tecplot 360 EX| determine the most appropriate plot type
                for the data, use `PlotType.Automatic`. Possible values are:
                `PlotType.Automatic` (default), `Cartesian3D`, `Cartesian2D`,
                `XYLine`, `PlotType.Sketch`, `PolarLine`.

            zones (`string <str>` or `list` of `integers <int>`, optional): List of
                zone indexes (zero-based) to load or string specifying the type of
                zones to load. Possible values are: "CellsAndBoundaries",
                "CellsOnly" and "BoundariesOnly". Specifying one of these options
                is mutually exclusive with the ``variables`` option. (default:
                "CellsAndBoundaries")
            variables (`list` of `strings <str>`, optional): List of variable
                names to load. `None` implies loading all variables. (default:
                `None`)

            all_poly_zones (`boolean <bool>`, optional): Converts all zones to
                Tecplot polytope (polyhedral or polygonal) zones. (default:
                `False`)
            average_to_nodes (`string <str>`, optional): Average
                cell-centered data to grid nodes using the specified method.
                (Options: `None`, "Arithmetic", "Laplacian")
            time_interval (`float`, optional): Use a constant time interval between
                each *.dat* file. If `None`, the flow-data parameter of each
                solution *.dat* file is used. (default: `None`)
            assign_strand_ids (`boolean <bool>`, optional): Assign strand ID's to
                zones that have a strand ID of -1. (default: `True`)

                .. note::
                    assign_strand_ids only applies if you have also provided a
                    time_interval, otherwise it will be ignored.
            add_zones_to_existing_strands (`boolean <bool>`, optional): Add the
                `Zones <data_access>` to matching strands, if they exist. Otherwise, if
                the new data specifies strands, new ones will be created beginning
                after the last strand in the `Dataset`. (default: `False`)

            include_particle_data (`boolean <bool>`, optional): Load particle
                data from the *.dat* files. If loading particle data from an XML
                file, the XML file should be included in the ``data_filenames`` list.
                (default: `False`)

        Returns:
            `Dataset`: The `Dataset` holding the loaded data.

        Raises:
            `TecplotSystemError`: Internal error when loading data.
            `TecplotTypeError`: In-valid input.

        Notes:
            The ``zones`` option takes either a `list` of zone indexes to be
            imported or one of "CellsAndBoundaries", "CellsOnly" or
            "BoundariesOnly" to indicate the type of zones the user wants to load,
            however these options are mutually exclusive with the ``variables``
            option::

                >>> import tecplot
                >>> dataset = tecplot.data.load_fluent(['one.cas', 'two.cas'],
                ...     data_filenames=['one.dat', 'two.dat'],
                ...     variables = ['Pressure','Velocity'],
                ...     zones = [0,1,3])
        """
        if isinstance(case_filenames, string_types):
            case_filenames = [case_filenames]
        if isinstance(data_filenames, string_types):
            data_filenames = [data_filenames]

        if __debug__:
            zone_types_opts = ['CellsAndBoundaries', 'CellsOnly', 'BoundariesOnly']
            avg_to_nodes_opts = [None, 'Arithmetic', 'Laplacian']
            if isinstance(zones, string_types):
                if zones not in zone_types_opts:
                    msg = 'Zones must be list of indexes or one of: '
                    msg += ', '.join(zone_types_opts)
                    raise TecplotTypeError(msg)
                if variables is not None:
                    msg = 'Loading a subset of variables is allowed only if'
                    msg += ' zones is None or a list of zones.'
                    raise TecplotTypeError(msg)
            if (case_filenames or data_filenames) is None:
                raise TecplotTypeError('You must specify a case or data file.')
            if (case_filenames is None) and (len(data_filenames) != 1):
                msg = 'Case file is required when loading multiple data files'
                raise TecplotTypeError(msg)
            if average_to_nodes not in avg_to_nodes_opts:
                msg = 'average_to_nodes must be one of: '
                msg += ', '.join(str(x) for x in avg_to_nodes_opts)
                raise TecplotTypeError(msg)

        with StringList('STANDARDSYNTAX', '1.0') as instr:
            with ArgList() as arglist:
                arglist[sv.DATASETREADER] = 'Fluent Data Loader'

                if append:
                    instr += ['Append', 'Yes']

                if case_filenames is not None:
                    if data_filenames is not None:
                        nfiles = len(case_filenames) + len(data_filenames)
                        instr += ['LoadOption', 'MultipleCaseAndData',
                                  'FILELIST_Files', str(nfiles)]
                        instr += list(case_filenames) + list(data_filenames)
                    else:
                        instr += ['LoadOption', 'MultipleCaseAndData',
                                  'FILELIST_Files', str(len(case_filenames))]
                        instr += list(case_filenames)
                else:
                    instr += ['LoadOption', 'ResidualsOnly',
                              'FILENAME_DataFile', data_filenames[0]]

                if time_interval is not None:
                    instr += ["UnsteadyOption", 'ApplyConstantTimeInterval']
                    instr += ['TimeInterval', str(float(time_interval))]

                if isinstance(zones, string_types):
                    instr += ['GridZones', zones]
                elif (zones or variables) is not None:
                    instr += ["GridZones", 'SelectedZones']
                    if zones is not None:
                        zone_indexes = ','.join(str(int(z)+1) for z in zones)
                        instr += ['ZoneList', zone_indexes]
                    if variables is not None:
                        var_names = '\n'.join(variables)
                        instr += ['VarNameList', var_names]

                if include_particle_data is not None:
                    instr += ['IncludeParticleData',
                              'Yes' if include_particle_data else 'No']
                if all_poly_zones is not None:
                    instr += ['AllPolyZones',
                              'Yes' if all_poly_zones else 'No']
                if average_to_nodes is not None:
                    instr += ['AverageToNodes', average_to_nodes]

                if assign_strand_ids is not None:
                    instr += ['AssignStrandIDs',
                              'Yes' if assign_strand_ids else 'No']
                if add_zones_to_existing_strands is not None:
                    instr += ['AddZonesToExistingStrands',
                              'Yes' if add_zones_to_existing_strands else 'No']

                arglist[sv.FILENAMESORINSTRUCTIONS] = instr

                return _load_data_v3(arglist,
                                     frame=frame,
                                     append=append,
                                     reset_style=reset_style,
                                     initial_plot_first_zone_only=initial_plot_first_zone_only,
                                     initial_plot_type=initial_plot_type,
                                     assign_strand_ids=assign_strand_ids,
                                     add_zones_to_existing_strands=add_zones_to_existing_strands)

    @lock()
    def load_plot3d(grid_filenames=None,
                    solution_filenames=None,
                    function_filenames=None,
                    name_filename=None,

                    frame=None,

                    append=True,
                    reset_style=None,
                    initial_plot_first_zone_only=None,
                    initial_plot_type=None,

                    # setting any of these three implies auto_detect=False
                    data_structure=None,  # (1D|2D|3DP|3DW|UNSTRUCTURED)
                    is_multi_grid=None,  # (True|False)
                    style=None,  # (PLOT2DCLASSIC|PLOT3DFUNCTION|OVERFLOW)

                    ascii_is_double=None,  # (True|False)
                    ascii_has_blanking=None,  # (True|False)
                    uniform_grid=None,  # (True|False)

                    assign_strand_ids=True,
                    add_zones_to_existing_strands=True,
                    append_function_variables=None,  # (True|False)
                    include_boundaries=True):
        """Read Plot3D data files.

        Parameters:
            grid_filenames (`string <str>` or `list` of `strings <str>`, optional):
                One or more grid file names to be read.
            solution_filenames (`string <str>` or `list` of `strings <str>`, optional):
                One or more solution data file names to be read.
            function_filenames (`string <str>` or `list` of `strings <str>`, optional):
                One or more function file names.
            name_filename (`string <str>`, optional): Path to the name file.

            frame (`Frame`, optional): The `Frame` to attach the
                resulting `Dataset`. If `None`, the currently active `Frame` is
                used and the zones are appended by default.

            append (`boolean <bool>`, optional): Append the data to the existing
                `Dataset`. If `False`, the existing data attached to the `Frame` is
                deleted and replaced. (default: `True`)
            reset_style (`boolean <bool>`, optional): Reset the style for
                destination `Frame`, if `False`, the `Frame`'s current style is
                preserved. (default: `True`)
            initial_plot_first_zone_only (`boolean <bool>`, optional): Informs
                the |Tecplot Engine| that after the data is loaded it only needs
                to activate the first enabled `Zone <data_access>` for the initial plot. This
                option is particularly useful if you have many `Zones <data_access>`
                and want to get the data into the |Tecplot Engine| and the first
                `Zone <data_access>` drawn as fast as possible. The inactive `Zones <data_access>` can
                always be activated when needed. (default: `False`)
            initial_plot_type (`PlotType`, optional): Forces a specific type of
                plot upon loading of the data. Only used if *resetstyle* is `True`.
                To have |Tecplot 360 EX| determine the most appropriate plot type
                for the data, use `PlotType.Automatic`. Possible values are:
                `PlotType.Automatic` (default), `Cartesian3D`, `Cartesian2D`,
                `XYLine`, `PlotType.Sketch`, `PolarLine`.

            data_structure (`string <str>`, optional): Specifies the data
                structure and overrides the automatic detection. Options are:
                ``1D``, ``2D``, ``3DP``, ``3DW``, ``UNSTRUCTURED``. Setting this
                requires ``is_multi_grid`` and ``style`` to be set as well.
            is_multi_grid (`boolean <bool>`, optional): Sets data as multi-grid
                and overrides the automatic data structure detection. Setting
                this requires ``data_structure`` and ``style`` to be set as well.
            style (`boolean <bool>`, optional): Specifies the data style and
                overrides the automatic data structure detection. Options are:
                ``PLOT3DCLASSIC``, ``PLOT3DFUNCTION``, ``OVERFLOW``. Setting this
                requires ``data_structure`` and ``is_multi_grid`` to be set as
                well.

            ascii_is_double (`boolean <bool>`, optional): Indicates that
                floating-point numbers found in the text data files should be
                store with 64-bit precision. (default: `False`)
            ascii_has_blanking (`boolean <bool>`, optional): Indicates that the
                text data files contain blanking. (default: `False`)
            uniform_grid (`boolean <bool>`, optional): Indicates the grid
                structure is the same for all time steps. (default: `True`)

            assign_strand_ids (`boolean <bool>`, optional): Assign strand ID's to
                zones that have a strand ID of -1. (default: `True`)
            add_zones_to_existing_strands (`boolean <bool>`, optional): Add the
                `Zones <data_access>` to matching strands, if they exist. Otherwise, if
                the new data specifies strands, new ones will be created beginning
                after the last strand in the `Dataset`. (default: `True`)
            append_function_variables (`boolean <bool>`, optional): Append
                variables in function files to those found in solution files.
                (default: `False`)
            include_boundaries (`boolean <bool>`, optional): Loads boundary zones
                found in the ".g.fvbnd" file located in the same directory as the
                grid file, if available. (default: `True`)

        Returns:
            `Dataset`: The `Dataset` holding the loaded data.

        Raises:
            `TecplotSystemError`: Internal error when loading data.
            `TecplotValueError`: In-valid input.

        .. note:: Data structure is automatically detected by default.

            The options ``data_structure``, ``is_multi_grid`` and ``style``
            must be supplied together or not at all. When all of these are
            `None`, the data structure is automatically detected.

        The variables from the function files can be appended to the dataset
        upon loading::

            >>> import tecplot as tp
            >>> dataset = tp.data.load_plot3d(
            ...     grid_filenames = 'data.g',
            ...     solution_filenames = ['t0.q', 't1.q'],
            ...     function_filenames = ['t0.f', 't1.f'],
            ...     append_function_variables = True)
        """
        if isinstance(grid_filenames, string_types):
            grid_filenames = [grid_filenames]
        if isinstance(solution_filenames, string_types):
            solution_filenames = [solution_filenames]
        if isinstance(function_filenames, string_types):
            function_filenames = [function_filenames]

        if __debug__:
            data_structure_opts = [None, '1D', '2D', '3DP', '3DW', 'UNSTRUCTURED']
            style_opts = [None, 'PLOT3DCLASSIC', 'PLOT3DFUNCTION', 'OVERFLOW']
            if data_structure not in data_structure_opts:
                msg = 'data_structure must be one of: '
                msg += ', '.join(str(x) for x in data_structure_opts)
                raise TecplotValueError(msg)
            if style not in style_opts:
                msg = 'style must be one of: '
                msg += ', '.join(str(x) for x in style_opts)
                raise TecplotValueError(msg)
            if grid_filenames and solution_filenames:
                if len(grid_filenames) != 1 and \
                   len(grid_filenames) != len(solution_filenames):
                    raise TecplotValueError(dedent('''\
                        You must specify a single grid file or
                        the same number of grid and solution files.'''))
            if solution_filenames and function_filenames:
                if len(solution_filenames) != len(function_filenames):
                    raise TecplotValueError(dedent('''\
                        You must specify the same number of
                        solution and function files.'''))
            autodetect_opts = [data_structure, is_multi_grid, style]
            if sum([x is None for x in autodetect_opts]) not in [0, 3]:
                raise TecplotValueError(dedent('''\
                    The options: data_structure, is_multi_grid and style
                    must be supplied all together or not at all.'''))

        frame = frame or layout.active_frame()

        with StringList('STANDARDSYNTAX', '1.0') as instr:
            with ArgList() as arglist:
                arglist[sv.DATASETREADER] = 'PLOT3D Loader'

                preexisting_dataset = frame.has_dataset and frame.dataset.num_zones
                if append is not None:
                    if append:
                        if preexisting_dataset:
                            instr += ['Append', 'Yes']
                    else:
                        instr += ['Append', 'No']

                if grid_filenames is not None:
                    instr += ['FILELIST_GRIDFILES', str(len(grid_filenames))]
                    instr += grid_filenames
                if solution_filenames is not None:
                    instr += ['FILELIST_SOLUTIONFILES',
                              str(len(solution_filenames))]
                    instr += solution_filenames
                if function_filenames is not None:
                    instr += ['FILELIST_FUNCTIONFILES',
                              str(len(function_filenames))]
                    instr += function_filenames
                if name_filename is not None:
                    instr += ['FILENAME_NAMEFILE', name_filename]

                if (data_structure or is_multi_grid or style) is None:
                    instr += ['AUTODETECT', 'Yes']
                else:
                    instr += ['AUTODETECT', 'No']
                    instr += ['DATASTRUCTURE', data_structure]
                    instr += ['ISMULTIGRID', 'Yes' if is_multi_grid else 'No']
                    instr += ['STYLE', style]

                if ascii_is_double is not None:
                    instr += ['ASCIIISDOUBLE', 'Yes' if ascii_is_double else 'No']
                if ascii_has_blanking is not None:
                    instr += ['ASCIIHASBLANK',
                              'Yes' if ascii_has_blanking else 'No']

                if uniform_grid is not None:
                    instr += ['UNIFORMGRIDSTRUCTURE',
                              'Yes' if uniform_grid else 'No']
                if assign_strand_ids is not None:
                    instr += ['ASSIGNSTRANDIDS',
                              'Yes' if assign_strand_ids else 'No']
                if add_zones_to_existing_strands is not None:
                    if preexisting_dataset:
                        instr += ['ADDTOEXISTINGSTRANDS',
                                  'Yes' if add_zones_to_existing_strands else 'No']

                if append_function_variables is not None:
                    if solution_filenames or preexisting_dataset:
                        instr += ['APPENDFUNCTIONVARIABLES',
                                  'Yes' if append_function_variables else 'No']
                if include_boundaries is not None:
                    instr += ['LOADBOUNDARY', 'Yes' if include_boundaries else 'No']

                arglist[sv.FILENAMESORINSTRUCTIONS] = instr

                try:
                    return _load_data_v3(arglist,
                        frame=frame,
                        append=append,
                        reset_style=reset_style,
                        initial_plot_first_zone_only=initial_plot_first_zone_only,
                        initial_plot_type=initial_plot_type,
                        assign_strand_ids=None,
                        add_zones_to_existing_strands=None)
                except TecplotLogicError as e:
                    if str(e).startswith('The boundary file does not'):
                        log.warning(e)
                        return layout.active_frame().dataset
                    raise

    @lock()
    def load_tecplot(filenames,

                     frame=None,

                     append=True,
                     reset_style=None,
                     initial_plot_first_zone_only=None,
                     initial_plot_type=None,

                     zones=None,
                     variables=None,

                     collapse=None,
                     skip=None,
                     assign_strand_ids=True,
                     add_zones_to_existing_strands=None,

                     include_text=None,
                     include_geom=None,
                     include_custom_labels=None,
                     include_data=None):
        """Read a tecplot data file.

        Parameters:
            filenames (`string <str>` or `list` of `strings <str>`): Files to be
                read.

            frame (`Frame`, optional): The `Frame` to attach the
                resulting `Dataset`. If `None`, the currently active `Frame` is
                used and the zones are appended by default.

            append (`boolean <bool>`, optional): Append the data to the existing
                `Dataset`. If `False`, the existing data attached to the `Frame` is
                deleted and replaced. (default: `True`)
            reset_style (`boolean <bool>`, optional): Reset the style for
                destination `Frame`, if `False`, the `Frame`'s current style is
                preserved. (default: `True`)
            initial_plot_first_zone_only (`boolean <bool>`, optional): Informs
                the |Tecplot Engine| that after the data is loaded it only needs
                to activate the first enabled `Zone <data_access>` for the initial plot. This
                option is particularly useful if you have many `Zones <data_access>`
                and want to get the data into the |Tecplot Engine| and the first
                `Zone <data_access>` drawn as fast as possible. The inactive `Zones <data_access>` can
                always be activated when needed. (default: `False`)
            initial_plot_type (`PlotType`, optional): Forces a specific type of
                plot upon loading of the data. Only used if *resetstyle* is `True`.
                To have |Tecplot 360 EX| determine the most appropriate plot type
                for the data, use `PlotType.Automatic`. Possible values are:
                `PlotType.Automatic` (default), `Cartesian3D`, `Cartesian2D`,
                `XYLine`, `PlotType.Sketch`, `PolarLine`.

            zones (`set` of `integers <int>`, optional): Set of `Zones <data_access>`
                to load. Use `None` to load all zones. (default: `None`)
            variables (`set` of `strings <str>` or `integers <int>`, optional):
                Set of `Variables <Variable>` to load. Use `None` to load all
                variables. (default: `None`)

            collapse (`boolean <bool>`, optional): Reindex `Zones <data_access>` and
                `Variables <Variable>` if any are disabled. (default: `False`)
            skip: (3-`tuple` of `integers <int>`, optional) The *ijk*-skip. A value
                of (1,1,1) loads every data point in the *(i,j,k)* directions. A
                value of (2,2,2) loads every other data point and so forth. This
                only applies to ordered data. (default: (1,1,1))
            assign_strand_ids (`boolean <bool>`, optional): Assign strand ID's to
                zones that have a strand ID of -1. (default: `True`)
            add_zones_to_existing_strands (`boolean <bool>`, optional): Add the
                `Zones <data_access>` to matching strands, if they exist. Otherwise, if
                the new data specifies strands, new ones will be created beginning
                after the last strand in the `Dataset`. (default: `False`)

            include_text (`boolean <bool>`, optional): Load any text, geometries, or
                custom labels (default: `True`)
            include_geom (`boolean <bool>`, optional): Load geometries. (default:
                `True`)
            include_custom_labels (`boolean <bool>`, optional):  (default: `True`)
            include_data (`boolean <bool>`, optional): Load data. Set this to
                `False` if you only want annotations such as text or geometries.
                (default: `True`)

        Returns:
            `Dataset`: The `Dataset` holding the loaded data.

        Raises:
            `TecplotSystemError`: Internal error when loading data.
            `TecplotTypeError`: In-valid input.
        """
        with StringList(filenames) as fnames:
            with ArgList() as arglist:
                allocd = []

                arglist.update(**{
                    sv.DATASETREADER: 'TECPLOT',
                    sv.FILENAMESORINSTRUCTIONS: fnames,
                    sv.INCLUDETEXT: include_text,
                    sv.INCLUDEGEOM: include_geom,
                    sv.INCLUDECUSTOMLABELS: include_custom_labels,
                    sv.INCLUDEDATA: include_data,
                    sv.COLLAPSEZONESANDVARS: collapse,})

                if zones is not None:
                    zoneset = IndexSet(zones)
                    allocd.append(zoneset)
                    arglist[sv.ZONELIST] = zoneset
                if variables is not None:
                    if isinstance(variables[0], int):
                        arglist[sv.VARLOADMODE] = VarLoadMode.ByPosition
                        varset = IndexSet(*variables)
                        allocd.append(varset)
                        arglist[sv.VARPOSITIONLIST] = varset
                    elif isinstance(variables[0], string_types):
                        arglist[sv.VARLOADMODE] = VarLoadMode.ByName
                        var_string_list = StringList(*variables)
                        allocd.append(var_string_list)
                        arglist[sv.VARNAMELIST] = var_string_list
                    else:
                        err = 'unknown type for variables: {}'
                        raise TecplotTypeError(err.format(type(variables)))
                if skip is not None:
                    i, j, k = skip
                    arglist[sv.ISKIP] = i
                    arglist[sv.JSKIP] = j
                    arglist[sv.KSKIP] = k

                try:
                    return _load_data_v3(arglist, frame=frame,
                        append=append, reset_style=reset_style,
                        initial_plot_first_zone_only=initial_plot_first_zone_only,
                        initial_plot_type=initial_plot_type,
                        assign_strand_ids=assign_strand_ids,
                        add_zones_to_existing_strands=add_zones_to_existing_strands)
                finally:
                    for a in allocd:
                        a.dealloc()

    @lock()
    def load_tecplot_szl(filename, **kwargs):
        """Read tecplot SZL data file.

        .. note:: Parameters are the same as the default tecplot data loader:
            `data.load.tecplot`.

        Returns:
            `Dataset`: The `Dataset` holding the loaded data.
        """
        frame, kwargs = layout.util.kwargs_pop_frame(**kwargs)
        with frame.activated():
            with StringList('STANDARDSYNTAX', '1.0', 'FILENAME_FILE',
                            filename) as filenames:
                with ArgList() as arglist:
                    arglist['FILENAMESORINSTRUCTIONS'] = filenames
                    arglist['DATASETREADER'] = 'Tecplot Subzone Data Loader'
                    arglist.update(**{k.upper(): v for k, v in kwargs.items()})
                    _tecutil.DataSetReadX(arglist)

            return frame.dataset

### Navigation

- index
- modules |