示例#1
0
        public static int DumpTask(string path, bool imports, bool missinghashes, bool texinfo, bool classinfo)
        {
            #region checks

            var isDirectory   = false;
            var inputFileInfo = new FileInfo(path);
            var inputDirInfo  = new DirectoryInfo(path);
            if (!inputFileInfo.Exists)
            {
                if (!inputDirInfo.Exists)
                {
                    return(0);
                }
                else
                {
                    isDirectory = true;
                }
            }

            var archives = new List <Archive>();

            if (isDirectory)
            {
                archives.AddRange(inputDirInfo
                                  .GetFiles("*.archive", SearchOption.AllDirectories)
                                  .Select(_ => new Archive(_.FullName)));
            }
            else
            {
                archives.Add(new Archive(inputFileInfo.FullName));
            }

            #endregion

            var mainController = ServiceLocator.Default.ResolveType <IMainController>();
            var logger         = ServiceLocator.Default.ResolveType <ILoggerService>();

            var missinghashtxt = isDirectory
                ? Path.Combine(inputDirInfo.FullName, "missinghashes.txt")
                : $"{inputFileInfo.FullName}.missinghashes.txt";
            using var mwriter = File.CreateText(missinghashtxt);

            var typedict = new ConcurrentDictionary <string, IEnumerable <string> >();

            // Parallel
            foreach (var ar in archives)
            {
                if (classinfo)
                {
                    using var mmf = MemoryMappedFile.CreateFromFile(ar.Filepath, FileMode.Open,
                                                                    ar.Filepath.GetHashMD5(), 0,
                                                                    MemoryMappedFileAccess.Read);


                    var fileinfo = ar.Files.Values;
                    var query    = fileinfo.GroupBy(
                        ext => Path.GetExtension(ext.NameStr),
                        file => file,
                        (ext, finfo) => new
                    {
                        Key  = ext,
                        File = fileinfo.Where(_ => Path.GetExtension(_.NameStr) == ext)
                    }).ToList();



                    int progress = 0;
                    var total    = query.Count;

                    Console.Write($"Exporting {total} bundle entries ");

                    // foreach extension
                    Parallel.ForEach(query, new ParallelOptions {
                        MaxDegreeOfParallelism = 16
                    }, result =>
                    {
                        if (!string.IsNullOrEmpty(result.Key))
                        {
                            Parallel.ForEach(result.File, new ParallelOptions {
                                MaxDegreeOfParallelism = 16
                            }, fi =>
                            {
                                var(f, b)    = ar.GetFileData(fi.NameHash64, mmf);
                                using var ms = new MemoryStream(f);
                                using var br = new BinaryReader(ms);

                                var cr2w = new CR2WFile();
                                try
                                {
                                    cr2w.ReadImportsAndBuffers(br);
                                }
                                catch (Exception e)
                                {
                                    return;
                                }

                                foreach (var chunk in cr2w.Chunks)
                                {
                                    var o = chunk.GetDumpObject(br);
                                    if (o != null)
                                    {
                                        Register(o);
                                    }
                                }
                            });
                        }

                        Interlocked.Increment(ref progress);
                        logger.LogProgress(progress / (float)total);

                        Console.WriteLine($"Dumped extension {result.Key}");
                    });
                }
                if (imports || texinfo)
                {
                    using var mmf = MemoryMappedFile.CreateFromFile(ar.Filepath, FileMode.Open,
                                                                    ar.Filepath.GetHashMD5(), 0,
                                                                    MemoryMappedFileAccess.Read);

                    int progress = 0;

                    var fileDictionary = new ConcurrentDictionary <ulong, Cr2wChunkInfo>();
                    var texDictionary  = new ConcurrentDictionary <ulong, Cr2wTextureInfo>();

                    // get info
                    var count = ar.FileCount;

                    Console.Write($"Exporting {count} bundle entries ");

                    Parallel.For(0, count, new ParallelOptions {
                        MaxDegreeOfParallelism = 8
                    }, i =>
                    {
                        var entry    = ar.Files.ToList()[i];
                        var hash     = entry.Key;
                        var filename = string.IsNullOrEmpty(entry.Value.NameStr) ? hash.ToString() : entry.Value.NameStr;

                        if (imports)
                        {
                            var(f, buffers) = ar.GetFileData(hash, mmf);

                            // check if cr2w file
                            if (f.Length < 4)
                            {
                                return;
                            }
                            var id = f.Take(4);
                            if (!id.SequenceEqual(MAGIC))
                            {
                                return;
                            }

                            var cr2w = new CR2WFile();

                            using var ms = new MemoryStream(f);
                            using var br = new BinaryReader(ms);
                            cr2w.ReadImportsAndBuffers(br);

                            var obj = new Cr2wChunkInfo
                            {
                                Filename = filename,
                                Imports  = cr2w.Imports
                            };



                            fileDictionary.AddOrUpdate(hash, obj, (arg1, o) => obj);
                        }

                        if (texinfo)
                        {
                            if (!string.IsNullOrEmpty(entry.Value.NameStr) && entry.Value.NameStr.Contains(".xbm"))
                            {
                                var(f, buffers) = ar.GetFileData(hash, mmf);

                                // check if cr2w file
                                if (f.Length < 4)
                                {
                                    return;
                                }
                                var id = f.Take(4);
                                if (!id.SequenceEqual(MAGIC))
                                {
                                    return;
                                }

                                var cr2w = new CR2WFile();

                                using var ms = new MemoryStream(f);
                                using var br = new BinaryReader(ms);
                                var result   = cr2w.Read(br);

                                if (result != EFileReadErrorCodes.NoError)
                                {
                                    return;
                                }
                                if (!(cr2w.Chunks.FirstOrDefault()?.data is CBitmapTexture xbm) ||
                                    !(cr2w.Chunks[1]?.data is rendRenderTextureBlobPC blob))
                                {
                                    return;
                                }

                                // create dds header
                                var texinfo = new Cr2wTextureInfo()
                                {
                                    Filename    = filename,
                                    width       = blob.Header.SizeInfo.Width.val,
                                    height      = blob.Header.SizeInfo.Height.val,
                                    mips        = blob.Header.TextureInfo.MipCount.val,
                                    slicecount  = blob.Header.TextureInfo.SliceCount.val,
                                    alignment   = blob.Header.TextureInfo.DataAlignment.val,
                                    compression = xbm.Setup.Compression,
                                    Group       = xbm.Setup.Group,
                                    rawFormat   = xbm.Setup.RawFormat,
                                };

                                texDictionary.AddOrUpdate(hash, texinfo, (arg1, o) => texinfo);
                            }
                        }

                        Interlocked.Increment(ref progress);
                        logger.LogProgress(progress / (float)count);
                    });
示例#2
0
        public static int Cr2wTask(string path, bool all, bool chunks)
        {
            // initial checks
            var inputFileInfo = new FileInfo(path);

            if (!inputFileInfo.Exists)
            {
                return(0);
            }

            var f = File.ReadAllBytes(inputFileInfo.FullName);

            using var ms = new MemoryStream(f);
            using var br = new BinaryReader(ms);

            var cr2w = new CR2WFile();

            if (all)
            {
                cr2w.ReadImportsAndBuffers(br);

                var obj = new Cr2wChunkInfo {
                    Filename = inputFileInfo.FullName
                };

                obj.Stringdict = cr2w.StringDictionary;
                obj.Imports    = cr2w.Imports;
                obj.Buffers    = cr2w.Buffers;
                obj.Chunks     = cr2w.Chunks;
                foreach (var chunk in cr2w.Chunks)
                {
                    obj.ChunkData.Add(chunk.GetDumpObject(br));
                }

                //write
                File.WriteAllText($"{inputFileInfo.FullName}.info.json",
                                  JsonConvert.SerializeObject(obj, Formatting.Indented, new JsonSerializerSettings()
                {
                    ReferenceLoopHandling      = ReferenceLoopHandling.Ignore,
                    PreserveReferencesHandling = PreserveReferencesHandling.None,
                    TypeNameHandling           = TypeNameHandling.Auto
                }));
                Console.WriteLine($"Finished. Dump file written to {inputFileInfo.FullName}.info.json");
            }

            if (chunks)
            {
                cr2w.Read(br);

                //write
                File.WriteAllText($"{inputFileInfo.FullName}.json",
                                  JsonConvert.SerializeObject(cr2w, Formatting.Indented, new JsonSerializerSettings()
                {
                    ReferenceLoopHandling      = ReferenceLoopHandling.Ignore,
                    PreserveReferencesHandling = PreserveReferencesHandling.None,
                    TypeNameHandling           = TypeNameHandling.None
                }));
                Console.WriteLine($"Finished. Dump file written to {inputFileInfo.FullName}.json");
            }



            return(1);
        }
示例#3
0
        public static int DumpTask(string path, bool imports, bool missinghashes, bool info)
        {
            #region checks

            var isDirectory   = false;
            var inputFileInfo = new FileInfo(path);
            var inputDirInfo  = new DirectoryInfo(path);
            if (!inputFileInfo.Exists)
            {
                if (!inputDirInfo.Exists)
                {
                    return(0);
                }
                else
                {
                    isDirectory = true;
                }
            }

            var archives = new List <Archive>();

            if (isDirectory)
            {
                archives.AddRange(inputDirInfo
                                  .GetFiles("*.archive", SearchOption.AllDirectories)
                                  .Select(_ => new Archive(_.FullName)));
            }
            else
            {
                archives.Add(new Archive(inputFileInfo.FullName));
            }

            #endregion


            var missinghashtxt = isDirectory
                ? Path.Combine(inputDirInfo.FullName, "missinghashes.txt")
                : $"{inputFileInfo.FullName}.missinghashes.txt";
            using var mwriter = File.CreateText(missinghashtxt);

            foreach (var ar in archives)
            {
                using var pb = new ProgressBar();
                if (imports || info)
                {
                    using var mmf = MemoryMappedFile.CreateFromFile(ar.Filepath, FileMode.Open,
                                                                    ar.Filepath.GetHashMD5(), 0,
                                                                    MemoryMappedFileAccess.Read);
                    using var p1 = pb.Progress.Fork();
                    int progress = 0;

                    var fileDictionary = new ConcurrentDictionary <ulong, Cr2wChunkInfo>();
                    var texDictionary  = new ConcurrentDictionary <ulong, Cr2wTextureInfo>();
                    //foreach (var (hash, value) in ar.Files)
                    //    fileDictionary[hash] = new Cr2wChunkInfo();

                    // get info
                    var count = ar.FileCount;
                    Parallel.For(0, count, new ParallelOptions {
                        MaxDegreeOfParallelism = 8
                    }, i =>
                    {
                        var entry    = ar.Files.ToList()[i];
                        var hash     = entry.Key;
                        var filename = string.IsNullOrEmpty(entry.Value.NameStr) ? hash.ToString() : entry.Value.NameStr;

                        if (imports)
                        {
                            var(f, buffers) = ar.GetFileData(hash, mmf);

                            // check if cr2w file
                            if (f.Length < 4)
                            {
                                return;
                            }
                            var id = f.Take(4);
                            if (!id.SequenceEqual(MAGIC))
                            {
                                return;
                            }

                            var cr2w = new CR2WFile();

                            using var ms = new MemoryStream(f);
                            using var br = new BinaryReader(ms);
                            cr2w.ReadImportsAndBuffers(br);

                            var obj = new Cr2wChunkInfo
                            {
                                Filename = filename,
                                Imports  = cr2w.Imports
                            };


                            fileDictionary.AddOrUpdate(hash, obj, (arg1, o) => obj);
                        }

                        if (info)
                        {
                            if (!string.IsNullOrEmpty(entry.Value.NameStr) && entry.Value.NameStr.Contains(".xbm"))
                            {
                                var(f, buffers) = ar.GetFileData(hash, mmf);

                                // check if cr2w file
                                if (f.Length < 4)
                                {
                                    return;
                                }
                                var id = f.Take(4);
                                if (!id.SequenceEqual(MAGIC))
                                {
                                    return;
                                }

                                var cr2w = new CR2WFile();

                                using var ms = new MemoryStream(f);
                                using var br = new BinaryReader(ms);
                                var result   = cr2w.Read(br);

                                if (result != EFileReadErrorCodes.NoError)
                                {
                                    return;
                                }
                                if (!(cr2w.Chunks.FirstOrDefault()?.data is CBitmapTexture xbm) ||
                                    !(cr2w.Chunks[1]?.data is rendRenderTextureBlobPC blob))
                                {
                                    return;
                                }

                                // create dds header
                                var texinfo = new Cr2wTextureInfo()
                                {
                                    Filename    = filename,
                                    width       = blob.Header.SizeInfo.Width.val,
                                    height      = blob.Header.SizeInfo.Height.val,
                                    mips        = blob.Header.TextureInfo.MipCount.val,
                                    slicecount  = blob.Header.TextureInfo.SliceCount.val,
                                    alignment   = blob.Header.TextureInfo.DataAlignment.val,
                                    compression = xbm.Setup.Compression,
                                    Group       = xbm.Setup.Group,
                                    rawFormat   = xbm.Setup.RawFormat,
                                };

                                texDictionary.AddOrUpdate(hash, texinfo, (arg1, o) => texinfo);
                            }
                        }

                        progress += 1;
                        var perc  = progress / (double)count;
                        p1.Report(perc, $"Loading bundle entries: {progress}/{count}");
                    });