233 lines
7.0 KiB
C#
233 lines
7.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.ComponentModel;
|
|
using System.Drawing;
|
|
using System.Drawing.Design;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Text.RegularExpressions;
|
|
using System.Web.UI.Design;
|
|
using System.Windows.Forms;
|
|
using System.Windows.Forms.Design;
|
|
using System.Xml;
|
|
using System.Xml.Serialization;
|
|
using Sony.Vegas;
|
|
|
|
public class EntryPoint {
|
|
public void FromVegas(Vegas vegas) {
|
|
Config config = Config.Load();
|
|
ImportDialog importDialog = new ImportDialog(config, delegate { Import(config, vegas); });
|
|
importDialog.ShowDialog();
|
|
config.Save();
|
|
}
|
|
|
|
private void Import(Config config, Vegas vegas) {
|
|
// Load XML file
|
|
if (!File.Exists(config.XmlFile)) {
|
|
throw new Exception("XML file does not exist.");
|
|
}
|
|
XmlDocument xmlDocument = new XmlDocument();
|
|
xmlDocument.Load(config.XmlFile);
|
|
|
|
// Determine image file names
|
|
XmlNodeList mouthCueElements = xmlDocument.SelectNodes("//mouthCue");
|
|
List<string> shapeNames = new List<string>();
|
|
foreach (XmlElement mouthCueElement in mouthCueElements) {
|
|
if (!shapeNames.Contains(mouthCueElement.InnerText)) {
|
|
shapeNames.Add(mouthCueElement.InnerText);
|
|
}
|
|
}
|
|
Dictionary<string, string> imageFileNames = GetImageFileNames(config.OneImageFile, shapeNames.ToArray());
|
|
|
|
// Create new project
|
|
bool promptSave = !config.DiscardChanges;
|
|
bool showDialog = false;
|
|
Project project = new Project(promptSave, showDialog);
|
|
|
|
// Set frame size
|
|
Bitmap testImage = new Bitmap(config.OneImageFile);
|
|
project.Video.Width = testImage.Width;
|
|
project.Video.Height = testImage.Height;
|
|
|
|
// Set frame rate
|
|
if (config.FrameRate < 0.1 || config.FrameRate > 100) {
|
|
throw new Exception("Invalid frame rate.");
|
|
}
|
|
project.Video.FrameRate = config.FrameRate;
|
|
|
|
// Set other video settings
|
|
project.Video.FieldOrder = VideoFieldOrder.ProgressiveScan;
|
|
project.Video.PixelAspectRatio = 1;
|
|
|
|
// Add video track with images
|
|
VideoTrack videoTrack = vegas.Project.AddVideoTrack();
|
|
foreach (XmlElement mouthCueElement in mouthCueElements) {
|
|
Timecode start = GetTimecode(mouthCueElement.Attributes["start"]);
|
|
Timecode length = GetTimecode(mouthCueElement.Attributes["duration"]);
|
|
VideoEvent videoEvent = videoTrack.AddVideoEvent(start, length);
|
|
Media imageMedia = new Media(imageFileNames[mouthCueElement.InnerText]);
|
|
videoEvent.AddTake(imageMedia.GetVideoStreamByIndex(0));
|
|
}
|
|
|
|
// Add audio track with original sound file
|
|
AudioTrack audioTrack = vegas.Project.AddAudioTrack();
|
|
Media audioMedia = new Media(xmlDocument.SelectSingleNode("//soundFile").InnerText);
|
|
AudioEvent audioEvent = audioTrack.AddAudioEvent(new Timecode(0), audioMedia.Length);
|
|
audioEvent.AddTake(audioMedia.GetAudioStreamByIndex(0));
|
|
}
|
|
|
|
private static Timecode GetTimecode(XmlAttribute valueAttribute) {
|
|
double seconds = Double.Parse(valueAttribute.Value, CultureInfo.InvariantCulture);
|
|
return Timecode.FromSeconds(seconds);
|
|
}
|
|
|
|
private Dictionary<string, string> GetImageFileNames(string oneImageFile, string[] shapeNames) {
|
|
if (oneImageFile == null) {
|
|
throw new Exception("Image file name not set.");
|
|
}
|
|
Regex nameRegex = new Regex(@"(?<=-)([^-]*)(?=\.[^.]+$)");
|
|
if (!nameRegex.IsMatch(oneImageFile)) {
|
|
throw new Exception("Image file name doesn't have expected format.");
|
|
}
|
|
|
|
Dictionary<string, string> result = new Dictionary<string, string>();
|
|
foreach (string shapeName in shapeNames) {
|
|
string imageFileName = nameRegex.Replace(oneImageFile, shapeName);
|
|
if (!File.Exists(imageFileName)) {
|
|
throw new Exception(string.Format("Image file '{0}' not found.", imageFileName));
|
|
}
|
|
result[shapeName] = imageFileName;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
}
|
|
|
|
public class Config {
|
|
|
|
private string xmlFile;
|
|
private string oneImageFile;
|
|
private double frameRate = 100;
|
|
private bool discardChanges = false;
|
|
|
|
[DisplayName("XML File")]
|
|
[Description("An XML file generated by Rhubarb Lip Sync.")]
|
|
[Editor(typeof(XmlFileEditor), typeof(UITypeEditor))]
|
|
public string XmlFile {
|
|
get { return xmlFile; }
|
|
set { xmlFile = value; }
|
|
}
|
|
|
|
[DisplayName("One image file")]
|
|
[Description("Any image file out of the set of image files representing the mouth chart.")]
|
|
[Editor(typeof(FileNameEditor), typeof(UITypeEditor))]
|
|
public string OneImageFile {
|
|
get { return oneImageFile; }
|
|
set { oneImageFile = value; }
|
|
}
|
|
|
|
[DisplayName("Frame rate")]
|
|
[Description("The frame rate for the new project.")]
|
|
public double FrameRate {
|
|
get { return frameRate; }
|
|
set { frameRate = value; }
|
|
}
|
|
|
|
[DisplayName("Discard Changes")]
|
|
[Description("Discard all changes to the current project without prompting to save.")]
|
|
public bool DiscardChanges {
|
|
get { return discardChanges; }
|
|
set { discardChanges = value; }
|
|
}
|
|
|
|
private static string ConfigFileName {
|
|
get {
|
|
string folder = Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData);
|
|
return Path.Combine(folder, "ImportRhubarbSettings.xml");
|
|
}
|
|
}
|
|
|
|
public static Config Load() {
|
|
try {
|
|
XmlSerializer serializer = new XmlSerializer(typeof(Config));
|
|
using (FileStream file = File.OpenRead(ConfigFileName)) {
|
|
return (Config) serializer.Deserialize(file);
|
|
}
|
|
} catch (Exception) {
|
|
return new Config();
|
|
}
|
|
}
|
|
|
|
public void Save() {
|
|
XmlSerializer serializer = new XmlSerializer(typeof(Config));
|
|
using (StreamWriter file = File.CreateText(ConfigFileName)) {
|
|
XmlWriterSettings settings = new XmlWriterSettings();
|
|
settings.Indent = true;
|
|
settings.IndentChars = "\t";
|
|
using (XmlWriter writer = XmlWriter.Create(file, settings)) {
|
|
serializer.Serialize(writer, this);
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
public delegate void ImportAction();
|
|
|
|
public class ImportDialog : Form {
|
|
|
|
private readonly Config config;
|
|
private readonly ImportAction import;
|
|
|
|
public ImportDialog(Config config, ImportAction import) {
|
|
this.config = config;
|
|
this.import = import;
|
|
SuspendLayout();
|
|
InitializeComponent();
|
|
ResumeLayout(false);
|
|
}
|
|
|
|
private void InitializeComponent() {
|
|
// Configure dialog
|
|
Text = "Import Rhubarb";
|
|
Size = new Size(600, 400);
|
|
Font = new Font(Font.FontFamily, 10);
|
|
|
|
// Add property grid
|
|
PropertyGrid propertyGrid1 = new PropertyGrid();
|
|
propertyGrid1.SelectedObject = config;
|
|
Controls.Add(propertyGrid1);
|
|
propertyGrid1.Dock = DockStyle.Fill;
|
|
|
|
// Add button panel
|
|
FlowLayoutPanel buttonPanel = new FlowLayoutPanel();
|
|
buttonPanel.FlowDirection = FlowDirection.RightToLeft;
|
|
buttonPanel.AutoSize = true;
|
|
buttonPanel.Dock = DockStyle.Bottom;
|
|
Controls.Add(buttonPanel);
|
|
|
|
// Add Cancel button
|
|
Button cancelButton1 = new Button();
|
|
cancelButton1.Text = "Cancel";
|
|
cancelButton1.DialogResult = DialogResult.Cancel;
|
|
buttonPanel.Controls.Add(cancelButton1);
|
|
CancelButton = cancelButton1;
|
|
|
|
// Add OK button
|
|
Button okButton1 = new Button();
|
|
okButton1.Text = "OK";
|
|
okButton1.Click += OkButtonClickedHandler;
|
|
buttonPanel.Controls.Add(okButton1);
|
|
AcceptButton = okButton1;
|
|
}
|
|
|
|
private void OkButtonClickedHandler(object sender, EventArgs e) {
|
|
try {
|
|
import();
|
|
DialogResult = DialogResult.OK;
|
|
} catch (Exception exception) {
|
|
MessageBox.Show(exception.Message, "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
|
|
}
|
|
}
|
|
|
|
} |