Issue 654 - gzipping asset bundles

bouncer/src/repo/core/handler/fileservice/repo_file_manager.cpp

@@ -18,9 +18,15 @@
 #include "repo_file_manager.h"
 #include "../../../lib/repo_exception.h"
 #include "../../model/repo_model_global.h"
+#include "../../model/bson/repo_bson_builder.h"
 #include "../../model/bson/repo_bson_factory.h"
 #include "repo_file_handler_fs.h"
 #include "repo_file_handler_gridfs.h"
+#include <boost/iostreams/filtering_stream.hpp>
+#include <boost/iostreams/filter/gzip.hpp>
+#include <boost/iostreams/copy.hpp>
+#include <boost/interprocess/streams/bufferstream.hpp>
+#include <istream>
 
 using namespace repo::core::handler::fileservice;
 
@@ -46,20 +52,63 @@ bool FileManager::uploadFileAndCommit(
 	const std::string                            &collectionNamePrefix,
 	const std::string                            &fileName,
 	const std::vector<uint8_t>                   &bin,
-	const repo::core::model::RepoBSON            &metadata)
+	const repo::core::model::RepoBSON            &metadata,
+	const Encoding                               &encoding)
 {
 	bool success = true;
 	auto fileUUID = repo::lib::RepoUUID::createUUID();
-	auto linkName = defaultHandler->uploadFile(databaseName, collectionNamePrefix, fileUUID.toString(), bin);
+
+	// Create local references that can be reassigned, if we need to do any
+	// further processing...
+
+	const std::vector<uint8_t>* fileContents = &bin;
+	repo::core::model::RepoBSON fileMetadata = metadata;
+
+	switch (encoding)
+	{
+		case Encoding::Gzip:
+		{
+			// Use stringstream as a binary container to hold the compressed data
+			std::ostringstream compressedstream;
+
+			// Bufferstream operates directly over the user provided array
+			boost::interprocess::bufferstream uncompressed((char*)bin.data(), bin.size());
+
+			boost::iostreams::filtering_istream in;
+			in.push(boost::iostreams::gzip_compressor());
+			in.push(std::istream(uncompressed.rdbuf())); // For some reason bufferstream is ambigous between stream and streambuf, so wrap it unambiguously
+
+			boost::iostreams::copy(in, compressedstream);
+
+			auto compresseddata = compressedstream.str();
+			fileContents = new std::vector<uint8_t>(compresseddata.begin(), compresseddata.end());
+
+			repo::core::model::RepoBSONBuilder builder;
+			builder.append("encoding", "gzip");
+			fileMetadata = metadata.cloneAndAddFields(&builder.obj());
+		}
+		break;
+	}
+
+	auto linkName = defaultHandler->uploadFile(databaseName, collectionNamePrefix, fileUUID.toString(), *fileContents);
 	if (success = !linkName.empty()) {
 		success = upsertFileRef(
 			databaseName,
 			collectionNamePrefix,
 			cleanFileName(fileName),
 			linkName,
 			defaultHandler->getType(),
-			bin.size(),
-			metadata);
+			fileContents->size(),
+			fileMetadata);
+	}
+
+	// If we've created a new vector, clean it up. metadata doesn't need to be
+	// deleted because Mongo BSONs have built in smart-pointers allowing them to
+	// be passed by value.
+
+	if (fileContents != &bin)
+	{
+		delete fileContents;
 	}
 
 	return success;

bouncer/src/repo/core/handler/fileservice/repo_file_manager.h

@@ -54,6 +54,14 @@ namespace repo {
 						repo::core::handler::AbstractDatabaseHandler *dbHandler
 					);
 
+					/*
+					* Possible options for static compression of stored files
+					*/
+					enum Encoding {
+						None = 0,
+						Gzip = 1
+					};
+
 					/**
 					 * Upload file and commit ref entry to database.
 					 */
@@ -62,7 +70,8 @@ namespace repo {
 						const std::string                            &collectionNamePrefix,
 						const std::string                            &fileName,
 						const std::vector<uint8_t>                   &bin,
-						const repo::core::model::RepoBSON            &metadata = repo::core::model::RepoBSON()
+						const repo::core::model::RepoBSON            &metadata = repo::core::model::RepoBSON(),
+						const Encoding                               &encoding = Encoding::None
 					);
 
 					/**

bouncer/src/repo/lib/datastructure/repo_structs.h

@@ -25,11 +25,14 @@
 #include "repo_vector.h"
 #include <boost/crc.hpp>
 
-typedef struct {
-	std::unordered_map<std::string, std::vector<uint8_t>> geoFiles; //files where geometery are stored
-	std::unordered_map<std::string, std::vector<uint8_t>> jsonFiles; //JSON mapping files
+using repo_web_geo_files_t = std::unordered_map<std::string, std::vector<uint8_t>>;
+using repo_web_json_files_t = std::unordered_map<std::string, std::vector<uint8_t>>;
+
+using repo_web_buffers_t = struct {
+	repo_web_geo_files_t geoFiles; //files where geometery are stored
+	repo_web_json_files_t jsonFiles; //JSON mapping files
 	repo::core::model::RepoUnityAssets unityAssets; //Unity assets list
-}repo_web_buffers_t;
+};
 
 //This is used to map info for multipart optimization
 typedef struct {

bouncer/src/repo/manipulator/modelconvertor/export/repo_model_export_asset.cpp

@@ -85,7 +85,7 @@ AssetModelExport::~AssetModelExport()
 repo_web_buffers_t AssetModelExport::getAllFilesExportedAsBuffer() const
 {
 	return {
-		std::unordered_map<std::string, std::vector<uint8_t>>(),
+		repo_web_geo_files_t(),
 		getJSONFilesAsBuffer(),
 		getUnityAssets()
 	};

bouncer/src/repo/manipulator/modelconvertor/export/repo_model_export_gltf.cpp

@@ -564,9 +564,9 @@ repo_web_buffers_t GLTFModelExport::getAllFilesExportedAsBuffer() const
 	return{ getGLTFFilesAsBuffer(), getJSONFilesAsBuffer() };
 }
 
-std::unordered_map<std::string, std::vector<uint8_t>> GLTFModelExport::getGLTFFilesAsBuffer() const
+repo_web_geo_files_t GLTFModelExport::getGLTFFilesAsBuffer() const
 {
-	std::unordered_map<std::string, std::vector<uint8_t>> files;
+	repo_web_geo_files_t files;
 	//GLTF files
 	for (const auto &pair : trees)
 	{

bouncer/src/repo/manipulator/modelconvertor/export/repo_model_export_gltf.h

@@ -236,7 +236,7 @@ namespace repo{
 				* Return the GLTF file as raw bytes buffer
 				* returns an empty vector if the export has failed
 				*/
-				std::unordered_map<std::string, std::vector<uint8_t>> getGLTFFilesAsBuffer() const;
+				repo_web_geo_files_t getGLTFFilesAsBuffer() const;
 
 				/**
 				* Reindex the given faces base on the given information

bouncer/src/repo/manipulator/modelconvertor/export/repo_model_export_src.cpp

@@ -136,9 +136,9 @@ SRCModelExport::~SRCModelExport()
 {
 }
 
-std::unordered_map<std::string, std::vector<uint8_t>> SRCModelExport::getSRCFilesAsBuffer() const
+repo_web_geo_files_t SRCModelExport::getSRCFilesAsBuffer() const
 {
-	std::unordered_map < std::string, std::vector<uint8_t> > fileBuffers;
+	repo_web_geo_files_t fileBuffers;
 
 	for (const auto &treePair : trees)
 	{

bouncer/src/repo/manipulator/modelconvertor/export/repo_model_export_src.h

@@ -94,7 +94,7 @@ namespace repo{
 				* Return the SRC file as raw bytes buffer
 				* returns an empty vector if the export has failed
 				*/
-				std::unordered_map<std::string, std::vector<uint8_t>> getSRCFilesAsBuffer() const;
+				repo_web_geo_files_t getSRCFilesAsBuffer() const;
 			};
 		} //namespace modelconvertor
 	} //namespace manipulator

bouncer/src/repo/manipulator/modeloptimizer/repo_optimizer_multipart.cpp

@@ -39,8 +39,14 @@ using Scalar = float;
 using Bvh = bvh::Bvh<Scalar>;
 using BvhVector3 = bvh::Vector3<Scalar>;
 
-static const size_t REPO_MP_MAX_VERTEX_COUNT = 65536;
+// The vertex count is used as a rough approximation of the total geometry size.
+// This figure is empirically set to end up with an average bundle size of 24 Mb.
+static const size_t REPO_MP_MAX_VERTEX_COUNT = 1200000;
+
+// This limit is used to prevent metadata files becoming unwieldly, and the
+// supermesh UV resolution falling below the quantisation noise floor.
 static const size_t REPO_MP_MAX_MESHES_IN_SUPERMESH = 5000;
+
 static const size_t REPO_BVH_MAX_LEAF_SIZE = 16;
 static const size_t REPO_MODEL_LOW_CLUSTERING_RATIO = 0.2f;
 

bouncer/src/repo/manipulator/modelutility/repo_scene_manager.cpp

@@ -43,7 +43,7 @@ bool SceneManager::commitWebBuffers(
 	//Upload the files
 	for (const auto &bufferPair : resultBuffers.geoFiles)
 	{
-		if (success &= fileManager->uploadFileAndCommit(databaseName, projectName + "." + geoStashExt, bufferPair.first, bufferPair.second))
+		if (success &= fileManager->uploadFileAndCommit(databaseName, projectName + "." + geoStashExt, bufferPair.first, bufferPair.second, {}, repo::core::handler::fileservice::FileManager::Encoding::Gzip)) // Web geometry files are gzipped by default
 		{
 			repoInfo << "File (" << bufferPair.first << ") added successfully to file storage.";
 		}