Switch-Toolbox/Switch_FileFormatsMain/GUI/BFRES/TexturePattern/BfresTexturePatternEditor.cs
2019-07-09 20:25:32 -04:00

668 lines
22 KiB
C#

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Drawing;
using System.Drawing.Imaging;
using System.Data;
using System.Linq;
using System.Text;
using System.Threading;
using System.Windows.Forms;
using Switch_Toolbox.Library.Animations;
using Switch_Toolbox.Library.Forms;
using Switch_Toolbox.Library;
using Bfres.Structs;
namespace FirstPlugin.Forms
{
public partial class BfresTexturePatternEditor : STForm
{
public PlayerState AnimationPlayerState = PlayerState.Stop;
public enum PlayerState
{
Playing,
Pause,
Stop,
}
public bool IsLooping
{
get { return loopChkBox.Checked; }
set { loopChkBox.Checked = value; }
}
public bool IsPlaying
{
get
{
return AnimationPlayerState == PlayerState.Playing;
}
}
ImageList imgList = new ImageList();
private List<MaterialAnimation> MaterialAnimations;
public BfresTexturePatternEditor(TreeNodeCollection materialAnimations)
{
InitializeComponent();
listViewCustom1.HeaderStyle = ColumnHeaderStyle.None;
listViewCustom1.BackColor = FormThemes.BaseTheme.TextEditorBackColor;
imgList = new ImageList()
{
ColorDepth = ColorDepth.Depth32Bit,
ImageSize = new Size(32, 32),
};
stPanel4.BackColor = FormThemes.BaseTheme.FormBackColor;
timer1.Interval = 100 / 60;
treeView1.BackColor = FormThemes.BaseTheme.FormBackColor;
treeView1.ForeColor = FormThemes.BaseTheme.FormForeColor;
toolstripShiftUp.Image.RotateFlip(RotateFlipType.RotateNoneFlipY);
foreach (var type in Enum.GetValues(typeof(Runtime.PictureBoxBG)).Cast<Runtime.PictureBoxBG>())
backgroundCB.Items.Add(type);
backgroundCB.SelectedItem = Runtime.pictureBoxStyle;
MaterialAnimations = new List<MaterialAnimation>();
foreach (TreeNode matAnim in materialAnimations)
{
MaterialAnimations.Add((MaterialAnimation)matAnim);
activeAnimCB.Items.Add(matAnim.Text);
}
}
MaterialAnimation.SamplerKeyGroup activeSampler;
MaterialAnimation _activeMaterialAnim;
MaterialAnimation ActiveMaterialAnim
{
get
{
return _activeMaterialAnim;
}
set
{
_activeMaterialAnim = value;
maxFrameCounterUD.Maximum = value.FrameCount;
maxFrameCounterUD.Value = value.FrameCount;
currentFrameCounterUD.Maximum = value.FrameCount;
animationTrackBar.TickDivide = 1;
animationTrackBar.Maximum = value.FrameCount;
animationTrackBar.Minimum = 0;
currentFrameCounterUD.Value = 0;
frameCountUD.Maximum = value.FrameCount;
frameCountUD.Value = value.FrameCount;
}
}
private void OnPropertyChanged()
{
}
public bool IsLoaded = false;
public void LoadAnim(MaterialAnimation materialAnim)
{
if (materialAnim.Materials.Count <= 0)
return;
IsLoaded = false;
ReloadAnimationView(materialAnim);
IsLoaded = true;
activeAnimCB.SelectItemByText(materialAnim.Text);
animationTrackBar.Value = 0;
}
private void ReloadAnimationView(MaterialAnimation materialAnim)
{
treeView1.Nodes.Clear();
int Index = 0;
foreach (var mat in materialAnim.Materials)
{
mat.Nodes.Clear();
var matWrapper = new TreeNode(mat.Text) { Tag = mat, };
treeView1.Nodes.Add(matWrapper);
foreach (var sampler in mat.Samplers)
matWrapper.Nodes.Add(new TreeNode(sampler.Text) { Tag = sampler, });
if (matWrapper.Nodes.Count > 0 && Index == 0)
treeView1.SelectedNode = matWrapper.Nodes[0];
Index++;
}
ReloadAnimationView();
}
private void ReloadAnimationView()
{
editToolStripMenuItem.Enabled = false;
listViewCustom1.SuspendLayout();
listViewCustom1.Items.Clear();
LoadAniamtion(ActiveMaterialAnim, activeSampler);
listViewCustom1.ResumeLayout();
if (listViewCustom1.Items.Count > 0)
{
listViewCustom1.Items[0].Selected = true;
listViewCustom1.Select();
editToolStripMenuItem.Enabled = true;
}
}
Dictionary<int, Bitmap> Images = new Dictionary<int, Bitmap>();
public List<int> KeyFrames = new List<int>(); //Used for getting the frame of the list item
public bool IsLoading = false;
private void LoadAniamtion(MaterialAnimation anim, MaterialAnimation.SamplerKeyGroup activeSampler)
{
if (activeSampler == null || IsLoading)
return;
imgList.Images.Clear();
Images.Clear();
KeyFrames.Clear();
listViewCustom1.SmallImageList = imgList;
listViewCustom1.Items.Clear();
listViewCustom1.View = View.SmallIcon;
Thread Thread = new Thread((ThreadStart)(() =>
{
IsLoading = true;
for (int Frame = 0; Frame <= anim.FrameCount; Frame++)
{
//Constants always show so break after first frame
if (activeSampler.Constant && Frame != 0)
break;
var keyFrame = activeSampler.GetKeyFrame(Frame);
if (keyFrame.IsKeyed || activeSampler.Constant)
{
AddKeyFrame((int)keyFrame.Value, Frame);
}
}
if (listViewCustom1.InvokeRequired)
{
listViewCustom1.Invoke((MethodInvoker)delegate
{
if (listViewCustom1.Items.Count > 0)
{
listViewCustom1.Items[0].Selected = true;
listViewCustom1.Select();
}
});
}
else
{
if (listViewCustom1.Items.Count > 0)
{
listViewCustom1.Items[0].Selected = true;
listViewCustom1.Select();
}
}
IsLoading = false;
}));
Thread.Start();
}
private void AddKeyFrame(int Index, int Frame)
{
if (activeSampler == null)
return;
string TextureKey = activeSampler.GetActiveTextureNameByIndex((int)Index);
var tex = activeSampler.GetActiveTexture((int)Index);
int ImageIndex = imgList.Images.Count;
if (tex != null)
{
Bitmap temp = tex.GetBitmap();
Images.Add(Frame, temp);
KeyFrames.Add(Frame);
if (listViewCustom1.InvokeRequired)
{
listViewCustom1.Invoke((MethodInvoker)delegate {
// Running on the UI thread
listViewCustom1.Items.Add($"{Frame} / {ActiveMaterialAnim.FrameCount} \n" + tex.Text, ImageIndex);
imgList.Images.Add(temp);
var dummy = imgList.Handle;
});
}
else
listViewCustom1.Items.Add($"{Frame} / {ActiveMaterialAnim.FrameCount} \n" + tex.Text, ImageIndex);
}
else
{
if (listViewCustom1.InvokeRequired)
{
listViewCustom1.Invoke((MethodInvoker)delegate {
listViewCustom1.Items.Add($"{Frame} / {ActiveMaterialAnim.FrameCount} \n" + TextureKey, ImageIndex);
});
}
else
listViewCustom1.Items.Add($"{Frame} / {ActiveMaterialAnim.FrameCount} \n" + TextureKey, ImageIndex);
}
}
private void SelectThumbnailItems()
{
}
private void btnPlay_Click(object sender, EventArgs e)
{
if (AnimationPlayerState == PlayerState.Playing)
Pause();
else
Play();
}
private void btnStop_Click(object sender, EventArgs e) {
Stop();
}
private void Play()
{
timer1.Start();
AnimationPlayerState = PlayerState.Playing;
UpdateAnimationUI();
}
private void Pause()
{
timer1.Stop();
AnimationPlayerState = PlayerState.Stop;
UpdateAnimationUI();
}
private void Stop()
{
timer1.Stop();
animationTrackBar.Value = 0;
AnimationPlayerState = PlayerState.Stop;
UpdateAnimationUI();
}
private void UpdateAnimationUI()
{
btnPlay.BackgroundImage = IsPlaying ? Switch_Toolbox.Library.Properties.Resources.PauseBtn
: Switch_Toolbox.Library.Properties.Resources.PlayArrowR;
}
private void timer1_Tick(object sender, EventArgs e)
{
if (AnimationPlayerState == PlayerState.Playing)
{
if (animationTrackBar.Value == animationTrackBar.Maximum)
{
if (IsLooping)
animationTrackBar.Value = 0;
else
Stop();
}
else
{
animationTrackBar.Value++;
}
}
}
private void maxFrameCounterUD_ValueChanged(object sender, EventArgs e)
{
if (ActiveMaterialAnim == null) return;
if (maxFrameCounterUD.Value < 1)
{
maxFrameCounterUD.Value = 1;
}
else
{
ActiveMaterialAnim.FrameCount = (int)maxFrameCounterUD.Value;
animationTrackBar.Value = 0;
animationTrackBar.Maximum = ActiveMaterialAnim.FrameCount;
animationTrackBar.Minimum = 0;
}
}
private void animationTrackBar_ValueChanged(object sender, EventArgs e)
{
currentFrameCounterUD.Value = animationTrackBar.Value;
SetAnimationsToFrame(animationTrackBar.Value);
}
private void SelectListItem(int Frame)
{
if (KeyFrames.Contains(Frame))
{
int index = KeyFrames.IndexOf(Frame);
listViewCustom1.TrySelectItem(index);
}
}
private void SetAnimationsToFrame(int Frame)
{
if (activeSampler != null)
{
var keyFrame = activeSampler.GetKeyFrame(Frame);
var tex = activeSampler.GetActiveTexture((int)keyFrame.Value);
if (tex != null)
{
if (Images.ContainsKey(Frame))
pictureBoxCustom1.Image = Images[Frame];
}
}
}
private void currentFrameCounterUD_ValueChanged(object sender, EventArgs e)
{
if (currentFrameCounterUD.Value > maxFrameCounterUD.Value)
currentFrameCounterUD.Value = maxFrameCounterUD.Value;
animationTrackBar.Value = (int)currentFrameCounterUD.Value;
}
private void listViewCustom1_SelectedIndexChanged(object sender, EventArgs e)
{
if (listViewCustom1.SelectedItems.Count > 0 && KeyFrames.Count > 0)
{
editToolStripMenuItem.Enabled = true;
int SelectedFrame = KeyFrames[listViewCustom1.SelectedIndices[0]];
animationTrackBar.Value = SelectedFrame;
textureFrameUD.Value = SelectedFrame;
}
else
editToolStripMenuItem.Enabled = false;
}
private void animationTrackBar_Scroll(object sender, ScrollEventArgs e)
{
}
private void btnForward1_Click(object sender, EventArgs e) {
if (animationTrackBar.Value < animationTrackBar.Maximum)
animationTrackBar.Value++;
}
private void btnBackward1_Click(object sender, EventArgs e) {
if (animationTrackBar.Value > 0)
animationTrackBar.Value--;
}
private void btnEditMaterial_Click(object sender, EventArgs e)
{
TexPatternMaterialEditor editor = new TexPatternMaterialEditor();
editor.LoadAnim(ActiveMaterialAnim);
if (editor.ShowDialog() == DialogResult.OK)
{
}
}
private void btnRemove_Click(object sender, EventArgs e)
{
}
private void btnAdd_Click(object sender, EventArgs e)
{
bool IsWiiU = activeSampler is FTXP.BfresSamplerAnim;
Texture_Selector editor = new Texture_Selector();
editor.LoadTexture("", IsWiiU);
if (editor.ShowDialog() == DialogResult.OK)
{
string NewTex = editor.GetSelectedTexture();
if (ActiveMaterialAnim.Textures == null)
ActiveMaterialAnim.Textures = new List<string>();
if (!ActiveMaterialAnim.Textures.Contains(NewTex))
ActiveMaterialAnim.Textures.Add(NewTex);
activeSampler.AddKeyFrame(NewTex);
}
}
private void listViewCustom1_DoubleClick(object sender, EventArgs e)
{
if (listViewCustom1.SelectedItems.Count > 0)
{
bool IsWiiU = activeSampler is FTXP.BfresSamplerAnim;
int SelectedFrame = KeyFrames[listViewCustom1.SelectedIndices[0]];
string currentTex = listViewCustom1.SelectedItems[0].Text;
Texture_Selector editor = new Texture_Selector();
editor.LoadTexture(currentTex, IsWiiU);
if (editor.ShowDialog() == DialogResult.OK)
{
string NewTex = editor.GetSelectedTexture();
if (!ActiveMaterialAnim.Textures.Contains(NewTex))
ActiveMaterialAnim.Textures.Add(NewTex);
int index = ActiveMaterialAnim.Textures.IndexOf(NewTex);
activeSampler.SetValue(SelectedFrame, index);
ActiveMaterialAnim.UpdateAnimationData();
}
}
}
private void exportToolStripMenuItem_Click(object sender, EventArgs e)
{
if (listViewCustom1.SelectedItems.Count > 0 && activeSampler != null)
{
int Frame = animationTrackBar.Value;
var keyFrame = activeSampler.GetKeyFrame(Frame);
var tex = activeSampler.GetActiveTexture((int)keyFrame.Value);
if (tex != null)
{
using (var sfd = new SaveFileDialog())
{
if (tex is FTEX)
{
sfd.Filter = FileFilters.FTEX;
if (sfd.ShowDialog() == DialogResult.OK)
{
if (tex is FTEX)
((FTEX)tex).Export(sfd.FileName);
}
}
if (tex is TextureData)
{
sfd.Filter = FileFilters.BNTX_TEX;
if (sfd.ShowDialog() == DialogResult.OK)
{
if (tex is TextureData)
((TextureData)tex).Export(sfd.FileName);
}
}
}
}
}
}
private void replaceToolStripMenuItem_Click(object sender, EventArgs e)
{
if (listViewCustom1.SelectedItems.Count > 0 && activeSampler != null)
{
int Frame = animationTrackBar.Value;
var keyFrame = activeSampler.GetKeyFrame(Frame);
var tex = activeSampler.GetActiveTexture((int)keyFrame.Value);
if (tex != null)
{
using (var ofd = new OpenFileDialog())
{
if (tex is FTEX)
{
ofd.Filter = FileFilters.FTEX;
if (ofd.ShowDialog() == DialogResult.OK)
{
if (tex is FTEX)
((FTEX)tex).Replace(ofd.FileName);
}
}
if (tex is TextureData)
{
ofd.Filter = FileFilters.BNTX_TEX;
if (ofd.ShowDialog() == DialogResult.OK)
{
if (tex is TextureData)
((TextureData)tex).Replace(ofd.FileName);
}
}
}
}
ReloadAnimationView();
}
}
private void textureFrameUD_ValueChanged(object sender, EventArgs e)
{
if (listViewCustom1.SelectedItems.Count > 0 && KeyFrames.Count > 0)
{
int SelectedFrame = KeyFrames[listViewCustom1.SelectedIndices[0]];
}
}
private void treeView1_AfterSelect(object sender, TreeViewEventArgs e)
{
if (ActiveMaterialAnim == null || !IsLoaded)
return;
var node = treeView1.SelectedNode;
if (node.Tag is MaterialAnimation.Material)
{
}
if (node.Tag is MaterialAnimation.SamplerKeyGroup)
{
activeSampler = (MaterialAnimation.SamplerKeyGroup)node.Tag;
ReloadAnimationView();
}
}
private void addKeyFrameToolstrip_Click(object sender, EventArgs e)
{
if (activeSampler != null)
{
bool IsWiiU = activeSampler is FTXP.BfresSamplerAnim;
Texture_Selector editor = new Texture_Selector();
editor.LoadTexture("", IsWiiU);
if (editor.ShowDialog() == DialogResult.OK)
{
string NewTex = editor.GetSelectedTexture();
if (ActiveMaterialAnim.Textures == null)
ActiveMaterialAnim.Textures = new List<string>();
if (!ActiveMaterialAnim.Textures.Contains(NewTex))
ActiveMaterialAnim.Textures.Add(NewTex);
if (activeSampler != null)
activeSampler.AddKeyFrame(NewTex);
}
}
}
private void removeKeyFrameToolstrip_Click(object sender, EventArgs e)
{
if (activeSampler != null)
{
int Frame = animationTrackBar.Value;
activeSampler.RemoveKeyFrame(Frame);
}
}
private void toolstripShiftUp_Click(object sender, EventArgs e)
{
if (activeSampler != null)
{
int Frame = animationTrackBar.Value;
activeSampler.ShiftKeyUp(Frame);
}
}
private void toolstripShiftDown_Click(object sender, EventArgs e)
{
if (activeSampler != null)
{
int Frame = animationTrackBar.Value;
activeSampler.ShiftKeyDown(Frame);
}
}
private void activeAnimCB_SelectedIndexChanged(object sender, EventArgs e)
{
if (activeAnimCB.SelectedIndex != -1 && IsLoaded)
{
ActiveMaterialAnim = MaterialAnimations[activeAnimCB.SelectedIndex];
ReloadAnimationView(ActiveMaterialAnim);
}
}
private void backgroundCB_SelectedIndexChanged(object sender, EventArgs e)
{
Runtime.pictureBoxStyle = (Runtime.PictureBoxBG)backgroundCB.SelectedItem;
UpdateBackgroundImage();
}
private void UpdateBackgroundImage()
{
switch (Runtime.pictureBoxStyle)
{
case Runtime.PictureBoxBG.Black:
pictureBoxCustom1.BackgroundImage = null;
pictureBoxCustom1.BackColor = Color.Black;
break;
case Runtime.PictureBoxBG.White:
pictureBoxCustom1.BackgroundImage = null;
pictureBoxCustom1.BackColor = Color.White;
break;
case Runtime.PictureBoxBG.Checkerboard:
pictureBoxCustom1.BackgroundImage = Switch_Toolbox.Library.Properties.Resources.CheckerBackground;
break;
}
}
}
}