: DirectorySelected(false)

, ContentScanned(false)

, UseFlatView(false)

{

ModalResult = SHOW;

if (!ShowModal("Choose a package to open", 500, 200))

return CANCEL;

UpdateSelectedPackages();

// controls are not released automatically when dialog closed to be able to

// poll them for user selection; release the memory now

ReleaseControls();

return ModalResult;

{

SelectedPackages.Empty();

SelectedPackages.Add(name);

SelectDirFromFilename(name);

{

param.Add(file);

return true;

{

(*this)

[

NewControl(UIGroup, GROUP_HORIZONTAL_LAYOUT|GROUP_NO_BORDER)

[

NewControl(UICheckbox, "Flat view", &UseFlatView)

.SetCallback(BIND_MEM_CB(&UIPackageDialog::OnFlatViewChanged, this))

+ NewControl(UISpacer)

+ NewControl(UILabel, "Filter:")

.SetY(2)

.SetAutoSize()

+ NewControl(UITextEdit, &PackageFilter)

.SetWidth(120)

.SetCallback(BIND_MEM_CB(&UIPackageDialog::OnFilterTextChanged, this))

]

+ NewControl(UIPageControl)

.Expose(FlatViewPager)

.SetHeight(500)

[

// page 0: TreeView + ListBox

NewControl(UIGroup, GROUP_HORIZONTAL_LAYOUT|GROUP_NO_BORDER)

.SetHeight(500)

[

NewControl(UITreeView)

.SetRootLabel("Game")

.SetWidth(EncodeWidth(0.3f))

.SetHeight(-1)

.SetCallback(BIND_MEM_CB(&UIPackageDialog::OnTreeItemSelected, this))

.UseFolderIcons()

.SetItemHeight(20)

.Expose(PackageTree)

+ NewControl(UISpacer)

+ NewControl(UIMulticolumnListbox, COLUMN_Count)

.SetHeight(-1)

.SetSelChangedCallback(BIND_MEM_CB(&UIPackageDialog::OnPackageSelected, this))

.SetDblClickCallback(BIND_MEM_CB(&UIPackageDialog::OnPackageDblClick, this))

.Expose(PackageListbox)

.AllowMultiselect()

//?? right-align text in numeric columns

.AddColumn("Package name")

.AddColumn("Skel", 35, TA_Right)

.AddColumn("Stat", 35, TA_Right)

.AddColumn("Anim", 35, TA_Right)

.AddColumn("Tex", 35, TA_Right)

.AddColumn("Size, Kb", 70, TA_Right)

]

// page 1: single ListBox

+ NewControl(UIMulticolumnListbox, COLUMN_Count)

.SetHeight(-1)

.SetSelChangedCallback(BIND_MEM_CB(&UIPackageDialog::OnPackageSelected, this))

.SetDblClickCallback(BIND_MEM_CB(&UIPackageDialog::OnPackageDblClick, this))

.Expose(FlatPackageList)

.AllowMultiselect()

//?? right-align text in numeric columns

.AddColumn("Package name")

.AddColumn("Skel", 35, TA_Right)

.AddColumn("Stat", 35, TA_Right)

.AddColumn("Anim", 35, TA_Right)

.AddColumn("Tex", 35, TA_Right)

.AddColumn("Size, Kb", 70, TA_Right)

]

];

if (!Packages.Num())

{

// not scanned yet

appEnumGameFiles(PackageListEnum, Packages);

}

// add paths of all found packages

if (SelectedPackages.Num()) DirectorySelected = true;

char prevPath[MAX_PACKAGE_PATH], path[MAX_PACKAGE_PATH];

prevPath[0] = 0;

for (int i = 0; i < Packages.Num(); i++)

{

appStrncpyz(path, Packages[i]->RelativeName, ARRAY_COUNT(path));

char* s = strrchr(path, '/');

if (s)

{

*s = 0;

// simple optimization - avoid calling PackageTree->AddItem() too frequently (assume package list is sorted)

if (!strcmp(prevPath, path)) continue;

strcpy(prevPath, path);

// add a directory to TreeView

PackageTree->AddItem(path);

}

if (!DirectorySelected)

{

// find the first directory with packages, but don't select /Game/Engine subdirectories by default

bool isUE4EnginePath = strnicmp(path, "Engine/", 7) == 0;

if (!isUE4EnginePath && (stricmp(path, SelectedDir) < 0 || SelectedDir.IsEmpty()))

{

// set selection to the first directory

SelectedDir = s ? path : "";

}

}

}

// "Tools" menu

UIMenu* toolsMenu = new UIMenu;

(*toolsMenu)

[

NewMenuItem("Scan content")

.Enable(!ContentScanned)

.Expose(ScanContentMenu)

.SetCallback(BIND_MEM_CB(&UIPackageDialog::ScanContent, this))

+ NewMenuItem("Scan versions")

.SetCallback(BIND_FREE_CB(&ShowPackageScanDialog))

+ NewMenuSeparator()

+ NewMenuItem("Save selected packages")

.Enable(SelectedPackages.Num() > 0)

.SetCallback(BIND_MEM_CB(&UIPackageDialog::SavePackages, this))

.Expose(SavePackagesMenu)

+ NewMenuSeparator()

+ NewMenuItem("About UModel")

.SetCallback(BIND_FREE_CB(&UIAboutDialog::Show))

];

// dialog buttons

NewControl(UIGroup, GROUP_HORIZONTAL_LAYOUT|GROUP_NO_BORDER)

.SetParent(this)

[

NewControl(UILabel, "Hint: you may open this dialog at any time by pressing \"O\"")

+ NewControl(UIMenuButton, "Tools")

.SetWidth(80)

.SetMenu(toolsMenu)

+ NewControl(UISpacer)

+ NewControl(UIButton, "Open")

.SetWidth(80)

.Enable(false)

.Expose(OkButton)

.SetOK()

+ NewControl(UISpacer)

+ NewControl(UIButton, "Export")

.SetWidth(80)

.Enable(false)

.Expose(ExportButton)

.SetCallback(BIND_MEM_CB(&UIPackageDialog::OnExportClicked, this))

+ NewControl(UISpacer)

+ NewControl(UIButton, "Cancel")

.SetWidth(80)

.SetCancel()

];

UpdateFlatMode();

{

SelectedPackages.Reset();

if (!UseFlatView)

{

// get package name from directory + name

for (int selIndex = 0; selIndex < PackageListbox->GetSelectionCount(); selIndex++)

{

const char* pkgInDir = PackageListbox->GetItem(PackageListbox->GetSelectionIndex(selIndex));

const char* dir = *SelectedDir;

FString* newPackageName = new (SelectedPackages) FString;

if (dir[0])

{

char buffer[MAX_PACKAGE_PATH];

appSprintf(ARRAY_ARG(buffer), "%s/%s", dir, pkgInDir);

*newPackageName = buffer;

}

else

{

*newPackageName = pkgInDir;

}

}

}

else

{

// use flat list, with relative filename (including path)

for (int selIndex = 0; selIndex < FlatPackageList->GetSelectionCount(); selIndex++)

{

FString* newPackageName = new (SelectedPackages) FString;

const char* text = FlatPackageList->GetItem(FlatPackageList->GetSelectionIndex(selIndex));

*newPackageName = text;

if (selIndex == 0)

SelectDirFromFilename(text);

}

}

{

// extract a directory name from 1st package name

char buffer[512];

appStrncpyz(buffer, filename, ARRAY_COUNT(buffer));

char* s = strrchr(buffer, '/');

if (s)

{

*s = 0;

SelectedDir = buffer;

}

else

{

SelectedDir = "";

}

{

PackageListbox->RemoveAllItems();

SelectedDir = text;

DirectorySelected = true;

const char* filter = *PackageFilter;

if (!filter[0]) filter = NULL;

for (int i = 0; i < Packages.Num(); i++)

{

const CGameFileInfo* package = Packages[i];

char buffer[MAX_PACKAGE_PATH];

appStrncpyz(buffer, package->RelativeName, ARRAY_COUNT(buffer));

char* s = strrchr(buffer, '/');

if (s) *s++ = 0;

if ((!s && !text[0]) || // root directory

(s && !strcmp(buffer, text))) // other directory

{

const char* packageName = s ? s : buffer;

if (!filter || appStristr(packageName, filter))

{

// this package is in selected directory

AddPackageToList(PackageListbox, package, true);

}

}

}

{

FlatPackageList->RemoveAllItems();

const char* filter = *PackageFilter;

if (!filter[0]) filter = NULL;

for (int i = 0; i < Packages.Num(); i++)

{

const CGameFileInfo* package = Packages[i];

if (!filter || appStristr(package->RelativeName, filter))

AddPackageToList(FlatPackageList, package, false);

}

{

const char* s = package->RelativeName;

if (stripPath)

{

const char* s2 = strrchr(s, '/');

if (s2) s = s2 + 1;

}

int index = listbox->AddItem(s);

char buf[32];

// object counts

if (package->PackageScanned)

{

#define ADD_COLUMN(ColumnEnum, Value) \

if (Value) \

{ \

appSprintf(ARRAY_ARG(buf), "%d", Value); \

listbox->AddSubItem(index, ColumnEnum, buf); \

}

ADD_COLUMN(COLUMN_NumSkel, package->NumSkeletalMeshes);

ADD_COLUMN(COLUMN_NumStat, package->NumStaticMeshes);

ADD_COLUMN(COLUMN_NumAnim, package->NumAnimations);

ADD_COLUMN(COLUMN_NumTex, package->NumTextures);

#undef ADD_COLUMN

}

// size

appSprintf(ARRAY_ARG(buf), "%d", package->SizeInKb);

listbox->AddSubItem(index, COLUMN_Size, buf);

{

UseFlatView = !UseFlatView;

UpdateSelectedPackages();

UseFlatView = !UseFlatView;

UpdateFlatMode();

{

#if 0

appPrintf("Selected packages:\n");

for (int i = 0; i < SelectedPackages.Num(); i++) appPrintf(" %s\n", *SelectedPackages[i]);

#endif

if (UseFlatView)

{

// switching to flat list

PackageListbox->RemoveAllItems();

FillFlatPackageList();

// select item which was active in tree+list

FlatPackageList->UnselectAllItems();

for (int i = 0; i < SelectedPackages.Num(); i++)

FlatPackageList->SelectItem(SelectedPackages[i], true);

// update buttons enable state

OnPackageSelected(FlatPackageList);

}

else

{

// switching to tree+list

FlatPackageList->RemoveAllItems();

PackageTree->SelectItem(*SelectedDir);

OnTreeItemSelected(PackageTree, *SelectedDir);

// select directory and package

PackageListbox->UnselectAllItems();

for (int i = 0; i < SelectedPackages.Num(); i++)

{

const char *s = SelectedPackages[i];

const char* s2 = strrchr(s, '/'); // strip path

if (s2) s = s2+1;

//!! todo: compare string between [s,s2] with SelectedDir, add only when strings are equal

PackageListbox->SelectItem(s, true);

}

// update buttons enable state

OnPackageSelected(PackageListbox);

}

FlatViewPager->SetActivePage(UseFlatView ? 1 : 0);

{

// re-filter lists

UpdateSelectedPackages();

UpdateFlatMode();

{

for (int idx = 0; idx < package->Summary.ExportCount; idx++)

{

const char* ObjectClass = package->GetObjectName(package->GetExport(idx).ClassIndex);

if (!stricmp(ObjectClass, "SkeletalMesh"))

file->NumSkeletalMeshes++;

else if (!stricmp(ObjectClass, "StaticMesh"))

file->NumStaticMeshes++;

else if (!stricmp(ObjectClass, "Animation") || !stricmp(ObjectClass, "AnimSequence")) // whole AnimSet for UE2 and number of sequences for UE3+

file->NumAnimations++;

else if (!strnicmp(ObjectClass, "Texture", 7))

file->NumTextures++;

}

{

UIProgressDialog progress;

progress.Show("Scanning packages");

progress.SetDescription("Scanning package");

bool cancelled = false;

for (int i = 0; i < Packages.Num(); i++)

{

CGameFileInfo* file = const_cast<CGameFileInfo*>(Packages[i]); // we'll modify this structure here

if (file->PackageScanned) continue;

if (!progress.Progress(file->RelativeName, i, GNumPackageFiles))

{

cancelled = true;

break;

}

UnPackage* package = UnPackage::LoadPackage(file->RelativeName, /*silent=*/ true); // should always return non-NULL

file->PackageScanned = true;

if (!package) continue; // should not happen

ScanPackageExports(package, file);

}

progress.CloseDialog();

if (cancelled) return;

ContentScanned = true;

// finished - no needs to perform scan again, disable button

ScanContentMenu->Enable(false);

// update package list with new data

UpdateSelectedPackages();

UpdateFlatMode();

{

byte buffer[16384];

while (Count > 0)

{

int Size = min(Count, sizeof(buffer));

Src->Serialize(buffer, Size);

if (fwrite(buffer, Size, 1, Dst) != 1) appError("Write failed");

Count -= Size;

}

{

guard(UIPackageDialog::SavePackages);

//!! Possible options:

//!! - save used packages (find better name - "imports", "links", "used packages", "referenced packages" ...)

//!! - decompress packages

//!! - preserve package paths

//!! - destination directory

UpdateSelectedPackages();

UIProgressDialog progress;

progress.Show("Saving packages");

progress.SetDescription("Saving package");

for (int i = 0; i < SelectedPackages.Num(); i++)

{

const CGameFileInfo* file = appFindGameFile(*SelectedPackages[i]);

assert(file);

if (!progress.Progress(file->RelativeName, i, GNumPackageFiles))

break;

FArchive *Ar = appCreateFileReader(file);

if (!Ar) continue;

// prepare destination file

char OutFile[1024];

appSprintf(ARRAY_ARG(OutFile), "UmodelSaved/%s", file->ShortFilename); //!! make an option, add menu item to open "saved" directory

appMakeDirectoryForFile(OutFile);

FILE *out = fopen(OutFile, "wb");

// copy data

CopyStream(Ar, out, Ar->GetFileSize());

// cleanup

delete Ar;

fclose(out);

}

progress.CloseDialog();

unguard;

{

bool enableButtons = (sender->GetSelectionCount() > 0);

OkButton->Enable(enableButtons);

ExportButton->Enable(enableButtons);

SavePackagesMenu->Enable(enableButtons);

{

if (value != -1)

CloseDialog();

{

ModalResult = EXPORT;

CloseDialog();