process_rasters()

static int process_rasters ( RTLOADERCFG *  config, STRINGBUFFER *  buffer  )

static

Definition at line 2003 of file raster2pgsql.c.

References _, add_overview_constraints(), add_raster_constraints(), analyze_table(), append_sql_to_buffer(), build_overview(), raster_loader_config::constraints, convert_raster(), copy_rastinfo(), raster_loader_config::copy_statements, create_index(), create_table(), diff_rastinfo(), drop_table(), raster_loader_config::file_column, raster_loader_config::file_column_name, flush_stringbuffer(), raster_loader_config::idx, raster_loader_config::idx_tablespace, init_rastinfo(), init_stringbuffer(), insert_records(), stringbuffer_t::length, raster_loader_config::maintenance, raster_loader_config::max_extent, raster_loader_config::opt, raster_loader_config::out_srid, raster_loader_config::overview, raster_loader_config::overview_count, raster_loader_config::overview_table, raster_loader_config::raster_column, raster_loader_config::regular_blocking, raster_loader_config::rt_file, raster_loader_config::rt_file_count, raster_loader_config::rt_filename, rtdealloc_rastinfo(), rtdealloc_stringbuffer(), rterror(), raster_loader_config::schema, raster_loader_config::table, raster_loader_config::tablespace, raster_loader_config::transaction, and vacuum_table().

Referenced by main().

                                                           {
        int i = 0;

        assert(config != NULL);
        assert(config->table != NULL);
        assert(config->raster_column != NULL);

        if (config->transaction) {
                if (!append_sql_to_buffer(buffer, strdup("BEGIN;"))) {
                        rterror(_("process_rasters: Could not add BEGIN statement to string buffer"));
                        return 0;
                }
        }

        /* drop table */
        if (config->opt == 'd') {
                if (!drop_table(config->schema, config->table, buffer)) {
                        rterror(_("process_rasters: Could not add DROP TABLE statement to string buffer"));
                        return 0;
                }

                if (config->overview_count) {
                        for (i = 0; i < config->overview_count; i++) {
                                if (!drop_table(config->schema, config->overview_table[i], buffer)) {
                                        rterror(_("process_rasters: Could not add an overview's DROP TABLE statement to string buffer"));
                                        return 0;
                                }
                        }
                }
        }

        /* create table */
        if (config->opt != 'a') {
                if (!create_table(
                        config->schema, config->table, config->raster_column,
                        config->file_column, config->file_column_name,
                        config->tablespace, config->idx_tablespace,
                        buffer
                )) {
                        rterror(_("process_rasters: Could not add CREATE TABLE statement to string buffer"));
                        return 0;
                }

                if (config->overview_count) {
                        for (i = 0; i < config->overview_count; i++) {
                                if (!create_table(
                                        config->schema, config->overview_table[i], config->raster_column,
                                        config->file_column, config->file_column_name,
                                        config->tablespace, config->idx_tablespace,
                                        buffer
                                )) {
                                        rterror(_("process_rasters: Could not add an overview's CREATE TABLE statement to string buffer"));
                                        return 0;
                                }
                        }
                }
        }

        /* no need to run if opt is 'p' */
        if (config->opt != 'p') {
                RASTERINFO refinfo;
                init_rastinfo(&refinfo);

                /* process each raster */
                for (i = 0; i < config->rt_file_count; i++) {
                        RASTERINFO rastinfo;
                        STRINGBUFFER tileset;

                        fprintf(stderr, _("Processing %d/%d: %s\n"), i + 1, config->rt_file_count, config->rt_file[i]);

                        init_rastinfo(&rastinfo);
                        init_stringbuffer(&tileset);

                        /* convert raster */
                        if (!convert_raster(i, config, &rastinfo, &tileset, buffer)) {
                                rterror(_("process_rasters: Could not process raster: %s"), config->rt_file[i]);
                                rtdealloc_rastinfo(&rastinfo);
                                rtdealloc_stringbuffer(&tileset, 0);
                                return 0;
                        }

                        /* process raster tiles into COPY or INSERT statements */
                        if (tileset.length && !insert_records(
                                config->schema, config->table, config->raster_column,
                                (config->file_column ? config->rt_filename[i] : NULL),
        config->file_column_name,
                                config->copy_statements, config->out_srid,
                                &tileset, buffer
                        )) {
                                rterror(_("process_rasters: Could not convert raster tiles into INSERT or COPY statements"));
                                rtdealloc_rastinfo(&rastinfo);
                                rtdealloc_stringbuffer(&tileset, 0);
                                return 0;
                        }

                        rtdealloc_stringbuffer(&tileset, 0);

                        /* flush buffer after every raster */
                        flush_stringbuffer(buffer);

                        /* overviews */
                        if (config->overview_count) {
                                int j = 0;

                                for (j = 0; j < config->overview_count; j++) {

                                        if (!build_overview(i, config, &rastinfo, j, &tileset, buffer)) {
                                                rterror(_("process_rasters: Could not create overview of factor %d for raster %s"), config->overview[j], config->rt_file[i]);
                                                rtdealloc_rastinfo(&rastinfo);
                                                rtdealloc_stringbuffer(&tileset, 0);
                                                return 0;
                                        }

                                        if (tileset.length && !insert_records(
                                                config->schema, config->overview_table[j], config->raster_column,
                                                (config->file_column ? config->rt_filename[i] : NULL), config->file_column_name,
                                                config->copy_statements, config->out_srid,
                                                &tileset, buffer
                                        )) {
                                                rterror(_("process_rasters: Could not convert overview tiles into INSERT or COPY statements"));
                                                rtdealloc_rastinfo(&rastinfo);
                                                rtdealloc_stringbuffer(&tileset, 0);
                                                return 0;
                                        }

                                        rtdealloc_stringbuffer(&tileset, 0);

                                        /* flush buffer after every raster */
                                        flush_stringbuffer(buffer);
                                }
                        }

                        if (config->rt_file_count > 1) {
                                if (i < 1)
                                        copy_rastinfo(&refinfo, &rastinfo);
                                else {
                                        diff_rastinfo(&rastinfo, &refinfo);
                                }
                        }

                        rtdealloc_rastinfo(&rastinfo);
                }

                rtdealloc_rastinfo(&refinfo);
        }

        /* index */
        if (config->idx) {
                /* create index */
                if (!create_index(
                        config->schema, config->table, config->raster_column,
                        config->idx_tablespace,
                        buffer
                )) {
                        rterror(_("process_rasters: Could not add CREATE INDEX statement to string buffer"));
                        return 0;
                }

                /* analyze */
                if (config->opt != 'p') {
                        if (!analyze_table(
                                config->schema, config->table,
                                buffer
                        )) {
                                rterror(_("process_rasters: Could not add ANALYZE statement to string buffer"));
                                return 0;
                        }
                }

                if (config->overview_count) {
                        for (i = 0; i < config->overview_count; i++) {
                                /* create index */
                                if (!create_index(
                                        config->schema, config->overview_table[i], config->raster_column,
                                        config->idx_tablespace,
                                        buffer
                                )) {
                                        rterror(_("process_rasters: Could not add an overview's CREATE INDEX statement to string buffer"));
                                        return 0;
                                }

                                /* analyze */
                                if (config->opt != 'p') {
                                        if (!analyze_table(
                                                config->schema, config->overview_table[i],
                                                buffer
                                        )) {
                                                rterror(_("process_rasters: Could not add an overview's ANALYZE statement to string buffer"));
                                                return 0;
                                        }
                                }
                        }
                }
        }

        /* add constraints */
        if (config->constraints) {
                if (!add_raster_constraints(
                        config->schema, config->table, config->raster_column,
                        config->regular_blocking, config->max_extent,
                        buffer
                )) {
                        rterror(_("process:rasters: Could not add AddRasterConstraints statement to string buffer"));
                        return 0;
                }

                if (config->overview_count) {
                        for (i = 0; i < config->overview_count; i++) {
                                if (!add_raster_constraints(
                                        config->schema, config->overview_table[i], config->raster_column,
                                        config->regular_blocking, config->max_extent,
                                        buffer
                                )) {
                                        rterror(_("process_rasters: Could not add an overview's AddRasterConstraints statement to string buffer"));
                                        return 0;
                                }
                        }
                }
        }

        /* overview constraint is automatically added */
        if (config->overview_count) {
                for (i = 0; i < config->overview_count; i++) {
                        if (!add_overview_constraints(
                                config->schema, config->overview_table[i], config->raster_column,
                                config->schema, config->table, config->raster_column,
                                config->overview[i],
                                buffer
                        )) {
                                rterror(_("process_rasters: Could not add an overview's AddOverviewConstraints statement to string buffer"));
                                return 0;
                        }
                }
        }

        if (config->transaction) {
                if (!append_sql_to_buffer(buffer, strdup("END;"))) {
                        rterror(_("process_rasters: Could not add END statement to string buffer"));
                        return 0;
                }
        }

        /* maintenance */
        if (config->opt != 'p' && config->maintenance) {
                if (!vacuum_table(
                        config->schema, config->table,
                        buffer
                )) {
                        rterror(_("process_rasters: Could not add VACUUM statement to string buffer"));
                        return 0;
                }

                if (config->overview_count) {
                        for (i = 0; i < config->overview_count; i++) {
                                if (!vacuum_table(
                                        config->schema, config->overview_table[i],
                                        buffer
                                )) {
                                        rterror(_("process_rasters: Could not add an overview's VACUUM statement to string buffer"));
                                        return 0;
                                }
                        }
                }

        }

        return 1;
}

Here is the call graph for this function:

Here is the caller graph for this function: