diff options
author | Jacques Lucke <jacques@blender.org> | 2022-09-15 20:13:01 +0300 |
---|---|---|
committer | Jacques Lucke <jacques@blender.org> | 2022-09-15 20:13:13 +0300 |
commit | 8b26349d5777179808792ab123f98f5a90428754 (patch) | |
tree | 35c9536193344664790b95e82c22fb488e1d4e5c /source/blender/blenloader/intern/readblenentry.cc | |
parent | 9f76d0c8e6dd2570e0da1763fcadfb738add4be2 (diff) |
BLO: move blenloader to C++
Differential Revision: https://developer.blender.org/D15965
Diffstat (limited to 'source/blender/blenloader/intern/readblenentry.cc')
-rw-r--r-- | source/blender/blenloader/intern/readblenentry.cc | 458 |
1 files changed, 458 insertions, 0 deletions
diff --git a/source/blender/blenloader/intern/readblenentry.cc b/source/blender/blenloader/intern/readblenentry.cc new file mode 100644 index 00000000000..e3d6864b962 --- /dev/null +++ b/source/blender/blenloader/intern/readblenentry.cc @@ -0,0 +1,458 @@ +/* SPDX-License-Identifier: GPL-2.0-or-later + * Copyright 2001-2002 NaN Holding BV. All rights reserved. */ + +/** \file + * \ingroup blenloader + * `.blend` file reading entry point. + */ + +#include <stddef.h> + +#include <math.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include "MEM_guardedalloc.h" + +#include "BLI_ghash.h" +#include "BLI_linklist.h" +#include "BLI_listbase.h" +#include "BLI_string.h" +#include "BLI_utildefines.h" + +#include "DNA_genfile.h" +#include "DNA_sdna_types.h" + +#include "BKE_icons.h" +#include "BKE_idtype.h" +#include "BKE_main.h" + +#include "BLO_blend_defs.h" +#include "BLO_readfile.h" +#include "BLO_undofile.h" + +#include "readfile.h" + +#include "BLI_sys_types.h" /* Needed for `intptr_t`. */ + +#ifdef WIN32 +# include "BLI_winstuff.h" +#endif + +/* local prototypes --------------------- */ +void BLO_blendhandle_print_sizes(BlendHandle *bh, void *fp); + +/* Access routines used by filesel. */ + +BlendHandle *BLO_blendhandle_from_file(const char *filepath, BlendFileReadReport *reports) +{ + BlendHandle *bh; + + bh = (BlendHandle *)blo_filedata_from_file(filepath, reports); + + return bh; +} + +BlendHandle *BLO_blendhandle_from_memory(const void *mem, + int memsize, + BlendFileReadReport *reports) +{ + BlendHandle *bh; + + bh = (BlendHandle *)blo_filedata_from_memory(mem, memsize, reports); + + return bh; +} + +void BLO_blendhandle_print_sizes(BlendHandle *bh, void *fp) +{ + FileData *fd = (FileData *)bh; + BHead *bhead; + + fprintf(static_cast<FILE *>(fp), "[\n"); + for (bhead = blo_bhead_first(fd); bhead; bhead = blo_bhead_next(fd, bhead)) { + if (bhead->code == ENDB) { + break; + } + + const SDNA_Struct *struct_info = fd->filesdna->structs[bhead->SDNAnr]; + const char *name = fd->filesdna->types[struct_info->type]; + char buf[4]; + + buf[0] = (bhead->code >> 24) & 0xFF; + buf[1] = (bhead->code >> 16) & 0xFF; + buf[2] = (bhead->code >> 8) & 0xFF; + buf[3] = (bhead->code >> 0) & 0xFF; + + buf[0] = buf[0] ? buf[0] : ' '; + buf[1] = buf[1] ? buf[1] : ' '; + buf[2] = buf[2] ? buf[2] : ' '; + buf[3] = buf[3] ? buf[3] : ' '; + + fprintf(static_cast<FILE *>(fp), + "['%.4s', '%s', %d, %ld ],\n", + buf, + name, + bhead->nr, + (long int)(bhead->len + sizeof(BHead))); + } + fprintf(static_cast<FILE *>(fp), "]\n"); +} + +LinkNode *BLO_blendhandle_get_datablock_names(BlendHandle *bh, + int ofblocktype, + const bool use_assets_only, + int *r_tot_names) +{ + FileData *fd = (FileData *)bh; + LinkNode *names = NULL; + BHead *bhead; + int tot = 0; + + for (bhead = blo_bhead_first(fd); bhead; bhead = blo_bhead_next(fd, bhead)) { + if (bhead->code == ofblocktype) { + const char *idname = blo_bhead_id_name(fd, bhead); + if (use_assets_only && blo_bhead_id_asset_data_address(fd, bhead) == NULL) { + continue; + } + + BLI_linklist_prepend(&names, BLI_strdup(idname + 2)); + tot++; + } + else if (bhead->code == ENDB) { + break; + } + } + + *r_tot_names = tot; + return names; +} + +LinkNode *BLO_blendhandle_get_datablock_info(BlendHandle *bh, + int ofblocktype, + const bool use_assets_only, + int *r_tot_info_items) +{ + FileData *fd = (FileData *)bh; + LinkNode *infos = NULL; + BHead *bhead; + int tot = 0; + + for (bhead = blo_bhead_first(fd); bhead; bhead = blo_bhead_next(fd, bhead)) { + if (bhead->code == ENDB) { + break; + } + if (bhead->code == ofblocktype) { + const char *name = blo_bhead_id_name(fd, bhead) + 2; + AssetMetaData *asset_meta_data = blo_bhead_id_asset_data_address(fd, bhead); + + const bool is_asset = asset_meta_data != NULL; + const bool skip_datablock = use_assets_only && !is_asset; + if (skip_datablock) { + continue; + } + struct BLODataBlockInfo *info = static_cast<BLODataBlockInfo *>( + MEM_mallocN(sizeof(*info), __func__)); + + /* Lastly, read asset data from the following blocks. */ + if (asset_meta_data) { + bhead = blo_read_asset_data_block(fd, bhead, &asset_meta_data); + /* blo_read_asset_data_block() reads all DATA heads and already advances bhead to the + * next non-DATA one. Go back, so the loop doesn't skip the non-DATA head. */ + bhead = blo_bhead_prev(fd, bhead); + } + + STRNCPY(info->name, name); + info->asset_data = asset_meta_data; + + BLI_linklist_prepend(&infos, info); + tot++; + } + } + + *r_tot_info_items = tot; + return infos; +} + +/** + * Read the preview rects and store in `result`. + * + * `bhead` should point to the block that sourced the `preview_from_file` + * parameter. + * `bhead` parameter is consumed. The correct bhead pointing to the next bhead in the file after + * the preview rects is returned by this function. + * \param fd: The filedata to read the data from. + * \param bhead: should point to the block that sourced the `preview_from_file parameter`. + * bhead is consumed. the new bhead is returned by this function. + * \param result: the Preview Image where the preview rect will be stored. + * \param preview_from_file: The read PreviewImage where the bhead points to. The rects of this + * \return PreviewImage or NULL when no preview Images have been found. Caller owns the returned + */ +static BHead *blo_blendhandle_read_preview_rects(FileData *fd, + BHead *bhead, + PreviewImage *result, + const PreviewImage *preview_from_file) +{ + for (int preview_index = 0; preview_index < NUM_ICON_SIZES; preview_index++) { + if (preview_from_file->rect[preview_index] && preview_from_file->w[preview_index] && + preview_from_file->h[preview_index]) { + bhead = blo_bhead_next(fd, bhead); + BLI_assert((preview_from_file->w[preview_index] * preview_from_file->h[preview_index] * + sizeof(uint)) == bhead->len); + result->rect[preview_index] = static_cast<uint *>( + BLO_library_read_struct(fd, bhead, "PreviewImage Icon Rect")); + } + else { + /* This should not be needed, but can happen in 'broken' .blend files, + * better handle this gracefully than crashing. */ + BLI_assert(preview_from_file->rect[preview_index] == NULL && + preview_from_file->w[preview_index] == 0 && + preview_from_file->h[preview_index] == 0); + result->rect[preview_index] = NULL; + result->w[preview_index] = result->h[preview_index] = 0; + } + BKE_previewimg_finish(result, preview_index); + } + + return bhead; +} + +PreviewImage *BLO_blendhandle_get_preview_for_id(BlendHandle *bh, + int ofblocktype, + const char *name) +{ + FileData *fd = (FileData *)bh; + bool looking = false; + const int sdna_preview_image = DNA_struct_find_nr(fd->filesdna, "PreviewImage"); + + for (BHead *bhead = blo_bhead_first(fd); bhead; bhead = blo_bhead_next(fd, bhead)) { + if (bhead->code == DATA) { + if (looking && bhead->SDNAnr == sdna_preview_image) { + PreviewImage *preview_from_file = static_cast<PreviewImage *>( + BLO_library_read_struct(fd, bhead, "PreviewImage")); + + if (preview_from_file == NULL) { + break; + } + + PreviewImage *result = static_cast<PreviewImage *>(MEM_dupallocN(preview_from_file)); + bhead = blo_blendhandle_read_preview_rects(fd, bhead, result, preview_from_file); + MEM_freeN(preview_from_file); + return result; + } + } + else if (looking || bhead->code == ENDB) { + /* We were looking for a preview image, but didn't find any belonging to block. So it doesn't + * exist. */ + break; + } + else if (bhead->code == ofblocktype) { + const char *idname = blo_bhead_id_name(fd, bhead); + if (STREQ(&idname[2], name)) { + looking = true; + } + } + } + + return NULL; +} + +LinkNode *BLO_blendhandle_get_previews(BlendHandle *bh, int ofblocktype, int *r_tot_prev) +{ + FileData *fd = (FileData *)bh; + LinkNode *previews = NULL; + BHead *bhead; + int looking = 0; + PreviewImage *prv = NULL; + PreviewImage *new_prv = NULL; + int tot = 0; + + for (bhead = blo_bhead_first(fd); bhead; bhead = blo_bhead_next(fd, bhead)) { + if (bhead->code == ofblocktype) { + const char *idname = blo_bhead_id_name(fd, bhead); + switch (GS(idname)) { + case ID_MA: /* fall through */ + case ID_TE: /* fall through */ + case ID_IM: /* fall through */ + case ID_WO: /* fall through */ + case ID_LA: /* fall through */ + case ID_OB: /* fall through */ + case ID_GR: /* fall through */ + case ID_SCE: /* fall through */ + case ID_AC: /* fall through */ + case ID_NT: /* fall through */ + new_prv = static_cast<PreviewImage *>(MEM_callocN(sizeof(PreviewImage), "newpreview")); + BLI_linklist_prepend(&previews, new_prv); + tot++; + looking = 1; + break; + default: + break; + } + } + else if (bhead->code == DATA) { + if (looking) { + if (bhead->SDNAnr == DNA_struct_find_nr(fd->filesdna, "PreviewImage")) { + prv = static_cast<PreviewImage *>(BLO_library_read_struct(fd, bhead, "PreviewImage")); + + if (prv) { + memcpy(new_prv, prv, sizeof(PreviewImage)); + bhead = blo_blendhandle_read_preview_rects(fd, bhead, new_prv, prv); + MEM_freeN(prv); + } + } + } + } + else if (bhead->code == ENDB) { + break; + } + else { + looking = 0; + new_prv = NULL; + prv = NULL; + } + } + + *r_tot_prev = tot; + return previews; +} + +LinkNode *BLO_blendhandle_get_linkable_groups(BlendHandle *bh) +{ + FileData *fd = (FileData *)bh; + GSet *gathered = BLI_gset_ptr_new("linkable_groups gh"); + LinkNode *names = NULL; + BHead *bhead; + + for (bhead = blo_bhead_first(fd); bhead; bhead = blo_bhead_next(fd, bhead)) { + if (bhead->code == ENDB) { + break; + } + if (BKE_idtype_idcode_is_valid(bhead->code)) { + if (BKE_idtype_idcode_is_linkable(bhead->code)) { + const char *str = BKE_idtype_idcode_to_name(bhead->code); + + if (BLI_gset_add(gathered, (void *)str)) { + BLI_linklist_prepend(&names, BLI_strdup(str)); + } + } + } + } + + BLI_gset_free(gathered, NULL); + + return names; +} + +void BLO_blendhandle_close(BlendHandle *bh) +{ + FileData *fd = (FileData *)bh; + + blo_filedata_free(fd); +} + +/**********/ + +BlendFileData *BLO_read_from_file(const char *filepath, + eBLOReadSkip skip_flags, + BlendFileReadReport *reports) +{ + BlendFileData *bfd = NULL; + FileData *fd; + + fd = blo_filedata_from_file(filepath, reports); + if (fd) { + fd->skip_flags = skip_flags; + bfd = blo_read_file_internal(fd, filepath); + blo_filedata_free(fd); + } + + return bfd; +} + +BlendFileData *BLO_read_from_memory(const void *mem, + int memsize, + eBLOReadSkip skip_flags, + ReportList *reports) +{ + BlendFileData *bfd = NULL; + FileData *fd; + BlendFileReadReport bf_reports{}; + bf_reports.reports = reports; + + fd = blo_filedata_from_memory(mem, memsize, &bf_reports); + if (fd) { + fd->skip_flags = skip_flags; + bfd = blo_read_file_internal(fd, ""); + blo_filedata_free(fd); + } + + return bfd; +} + +BlendFileData *BLO_read_from_memfile(Main *oldmain, + const char *filepath, + MemFile *memfile, + const struct BlendFileReadParams *params, + ReportList *reports) +{ + BlendFileData *bfd = NULL; + FileData *fd; + ListBase old_mainlist; + BlendFileReadReport bf_reports{}; + bf_reports.reports = reports; + + fd = blo_filedata_from_memfile(memfile, params, &bf_reports); + if (fd) { + fd->skip_flags = eBLOReadSkip(params->skip_flags); + BLI_strncpy(fd->relabase, filepath, sizeof(fd->relabase)); + + /* separate libraries from old main */ + blo_split_main(&old_mainlist, oldmain); + /* add the library pointers in oldmap lookup */ + blo_add_library_pointer_map(&old_mainlist, fd); + + if ((params->skip_flags & BLO_READ_SKIP_UNDO_OLD_MAIN) == 0) { + /* Build idmap of old main (we only care about local data here, so we can do that after + * split_main() call. */ + blo_make_old_idmap_from_main(fd, static_cast<Main *>(old_mainlist.first)); + } + + /* removed packed data from this trick - it's internal data that needs saves */ + + /* Store all existing ID caches pointers into a mapping, to allow restoring them into newly + * read IDs whenever possible. */ + blo_cache_storage_init(fd, oldmain); + + bfd = blo_read_file_internal(fd, filepath); + + /* Ensure relinked caches are not freed together with their old IDs. */ + blo_cache_storage_old_bmain_clear(fd, oldmain); + + /* Still in-use libraries have already been moved from oldmain to new mainlist, + * but oldmain itself shall *never* be 'transferred' to new mainlist! */ + BLI_assert(old_mainlist.first == oldmain); + + /* That way, libs (aka mains) we did not reuse in new undone/redone state + * will be cleared together with oldmain... */ + blo_join_main(&old_mainlist); + + blo_filedata_free(fd); + } + + return bfd; +} + +void BLO_blendfiledata_free(BlendFileData *bfd) +{ + if (bfd->main) { + BKE_main_free(bfd->main); + } + + if (bfd->user) { + MEM_freeN(bfd->user); + } + + MEM_freeN(bfd); +} |