public static List <SpectrumSourceGroup> SetInitialStructure(TreeNode node, ISession session, string sourceGroupFile)
{
if (node == null)
{
if (sourceGroupFile != null)
{
var dummy = new GroupingControlForm(session.SessionFactory);
dummy.applyAssemblyText(session, sourceGroupFile);
return(SetGroupingHierarchy(dummy._rootNode, dummy.session));
}
else
{
return(new List <SpectrumSourceGroup>());
}
}
_processedSources = new HashSet <string>();
tlvBranch convertedNode = GenericNodeToBranch(node, session, null);
var transaction = session.BeginTransaction();
try
{
return(SetGroupingHierarchy(convertedNode, session));
}
finally
{
transaction.Commit();
}
}
private void addGroupToolStripMenuItem_Click(object sender, EventArgs e)
{
var selNode = (tlvBranch)tlvGroupedFiles.GetModelObject(_clickedItem.Index);
_numberNewNodes++;
var ssg = new SpectrumSourceGroup();
ssg.Name = (string.Format("{0}/New Group({1})", selNode.Text, _numberNewNodes).Replace(@"//", @"/"));
var newNode = new tlvBranch
{
Text = Path.GetFileName(ssg.Name),
cms = cmRightClickGroupNode,
Parent = selNode,
Children = new List <tlvBranch>(),
Data = ssg
};
selNode.Children.Add(newNode);
OrganizeNode(selNode);
tlvGroupedFiles.RefreshObject(selNode);
if (selNode.Children.Count == 1)
{
tlvGroupedFiles.Expand(selNode);
}
tlvGroupedFiles.EditSubItem(tlvGroupedFiles.ModelToItem(newNode), 0);
}
private void RemoveGroupNode(tlvBranch selNode, bool refresh)
{
var abandonedSpectraSources = new List <tlvBranch>();
getListOfSprectrumSourcesRecursively(selNode, ref abandonedSpectraSources);
if (selNode.Parent.Text != null)
{
selNode.Parent.Children.Remove(selNode);
if (refresh)
{
tlvGroupedFiles.RefreshObject(selNode.Parent);
}
}
else
{
selNode.Children.Clear();
if (refresh)
{
tlvGroupedFiles.RefreshObject(selNode);
}
}
foreach (tlvBranch tn in abandonedSpectraSources)
{
var lvi = new ListViewItem {
Name = Path.GetFileName(tn.Text), Text = Path.GetFileName(tn.Text), Tag = tn
};
lvNonGroupedFiles.Items.Add(lvi);
}
}
private int OrganizeNode(tlvBranch target)
{
var sources = from tlvBranch item in target.Children
where item.Data is SpectrumSource
select item;
var groups = from tlvBranch item in target.Children
where item.Data is SpectrumSourceGroup
select item;
if (!sources.Any() && !groups.Any())
{
return(-1);
}
target.Children = new List <tlvBranch>();
foreach (var item in groups)
{
target.Children.Add(item);
}
foreach (var item in sources)
{
target.Children.Add(item);
}
return(groups.Count());
}
private void AddAssociatedSpectra(ref tlvBranch groupNode, Dictionary <int, List <SpectrumSource> > spectraDictionary)
{
try
{
int groupID = (int)(groupNode.Data as SpectrumSourceGroup).Id;
if (spectraDictionary.ContainsKey(groupID))
{
foreach (SpectrumSource ss in spectraDictionary[groupID])
{
var newNode = new tlvBranch
{
Text = Path.GetFileName(ss.Name),
Parent = groupNode,
Data = ss,
cms = cmRightClickFileNode
};
groupNode.Children.Add(newNode);
}
}
}
catch
{
MessageBox.Show("Could not add spectra to branch");
}
}
private void RemoveFileNode(tlvBranch selNode, bool refresh)
{
selNode.Parent.Children.Remove(selNode);
if (refresh)
{
tlvGroupedFiles.RefreshObject(selNode.Parent);
}
var lvi = new ListViewItem {
Name = Path.GetFileName(selNode.Text), Text = Path.GetFileName(selNode.Text), Tag = selNode
};
lvNonGroupedFiles.Items.Add(lvi);
}
/// <summary>
/// Check to prevent group nodes from being dragged into their children
/// group nodes.
/// </summary>
/// <returns></returns>
private bool checkIfDestGroupAChildNodeOfMe(tlvBranch destNode, tlvBranch dragNode)
{
try
{
return(destNode.Data is SpectrumSourceGroup &&
dragNode.Data is SpectrumSourceGroup &&
((SpectrumSourceGroup)destNode.Data).Name
.Contains(((SpectrumSourceGroup)dragNode.Data).Name));
}
catch (Exception exc)
{
throw new Exception("Error validating group move: checkIfDestGroupAChildNodeOfMe()\r\n", exc);
}
}
private static void getSprectrumSourceGroupsRecursively(tlvBranch treeNode, List <tlvBranch> nodeList, string rootname)
{
if (treeNode.Data is SpectrumSourceGroup)
{
(treeNode.Data as SpectrumSourceGroup).Name =
(rootname + "/" + treeNode.Text)
.Replace("//", "/");
nodeList.Add(treeNode);
foreach (tlvBranch subNode in treeNode.Children)
{
getSprectrumSourceGroupsRecursively(subNode, nodeList, (treeNode.Data as SpectrumSourceGroup).Name);
}
}
}
private static tlvBranch GenericNodeToBranch(TreeNode node, ISession session, tlvBranch parent)
{
var thisBranch = new tlvBranch
{
Text = node.Text,
Parent = parent ?? new tlvBranch()
};
if (node.Tag.ToString() == "Source")
{
var source = session.QueryOver <SpectrumSource>().Where(x => x.Name == node.Text).List().FirstOrDefault();
if (source == null || _processedSources.Contains(source.Name))
{
return(null);
}
_processedSources.Add(source.Name);
thisBranch.Data = source;
thisBranch.Children = null;
}
else
{
var trimmedName = node.Text == "/" ? "/" : Path.GetFileName(node.Text);
var ssg = session.QueryOver <SpectrumSourceGroup>().Where(x => x.Name == trimmedName).SingleOrDefault() ??
new SpectrumSourceGroup {
Name = trimmedName
};
thisBranch.Text = trimmedName;
thisBranch.Data = ssg;
thisBranch.Children = new List <tlvBranch>();
foreach (TreeNode childNode in node.Nodes)
{
var newChild = GenericNodeToBranch(childNode, session, thisBranch);
if (newChild != null)
{
thisBranch.Children.Add(newChild);
}
}
if (!thisBranch.Children.Any())
{
return(null);
}
}
return(thisBranch);
}
private static void getListOfSprectrumSourcesRecursively(tlvBranch treeNode, ref List <tlvBranch> spectraNodes)
{
try
{
if (treeNode.Data is SpectrumSource)
{
spectraNodes.Add(treeNode);
}
else
{
foreach (tlvBranch subNode in treeNode.Children)
{
getListOfSprectrumSourcesRecursively(subNode, ref spectraNodes);
}
}
}
catch (Exception exc)
{
throw new Exception("Error in retrieving spectra", exc);
}
}
private tlvBranch FillBranch(tlvBranch groupNode, IList <SpectrumSourceGroup> groups, Dictionary <int, List <SpectrumSource> > spectraDictionary)
{
List <SpectrumSourceGroup> potentialChildren;
string fullPath = (groupNode.Data as SpectrumSourceGroup).Name;
if (fullPath == "/")
{
potentialChildren = (from g in groups
where g.Name != fullPath &&
!g.Name.Remove(0, 1).Contains("/")
select g).ToList();
}
else
{
potentialChildren = (from g in groups
where g.Name.Contains(fullPath + "/") &&
!g.Name.Remove(0, fullPath.Length + 1).Contains("/")
select g).ToList();
}
foreach (SpectrumSourceGroup ssg in potentialChildren.OrderBy(o => o.Name))
{
var newNode = new tlvBranch
{
Text = Path.GetFileName(ssg.Name),
Children = new List <tlvBranch>(),
Data = ssg,
Parent = groupNode,
cms = cmRightClickGroupNode
};
groups.Remove(newNode.Data as SpectrumSourceGroup);
AddAssociatedSpectra(ref newNode, spectraDictionary);
groupNode.Children.Add(FillBranch(newNode, groups, spectraDictionary));
}
OrganizeNode(groupNode);
return(groupNode);
}
private void miResetFiles_Click(object sender, EventArgs e)
{
if (MessageBox.Show("Are you sure you want to remove all groups?", "Remove All", MessageBoxButtons.YesNo) != DialogResult.Yes)
{
return;
}
tlvGroupedFiles.ClearObjects();
_rootNode = new tlvBranch
{
Text = "/",
Parent = new tlvBranch {
Text = null
},
Children = new List <tlvBranch>(),
Data = new SpectrumSourceGroup {
Name = "/"
}
};
tlvGroupedFiles.AddObject(_rootNode);
foreach (var ss in session.Query <SpectrumSource>())
{
var lvTag = new tlvBranch
{
Text = ss.Name,
Parent = new tlvBranch {
Text = null
},
Children = new List <tlvBranch>(),
Data = ss
};
var lvi = new ListViewItem {
Text = ss.Name, Tag = lvTag
};
lvNonGroupedFiles.Items.Add(lvi);
}
}
private static tlvBranch GenericNodeToBranch(TreeNode node, ISession session, tlvBranch parent)
{
var thisBranch = new tlvBranch
{
Text = node.Text,
Parent = parent ?? new tlvBranch()
};
if (node.Tag.ToString() == "Source")
{
var source = session.QueryOver<SpectrumSource>().Where(x => x.Name == node.Text).List().FirstOrDefault();
if (source == null || _processedSources.Contains(source.Name))
return null;
_processedSources.Add(source.Name);
thisBranch.Data = source;
thisBranch.Children = null;
}
else
{
var trimmedName = node.Text == "/" ? "/" : Path.GetFileName(node.Text);
var ssg = session.QueryOver<SpectrumSourceGroup>().Where(x => x.Name == trimmedName).SingleOrDefault() ??
new SpectrumSourceGroup { Name = trimmedName };
thisBranch.Text = trimmedName;
thisBranch.Data = ssg;
thisBranch.Children = new List<tlvBranch>();
foreach (TreeNode childNode in node.Nodes)
{
var newChild = GenericNodeToBranch(childNode, session, thisBranch);
if (newChild != null)
thisBranch.Children.Add(newChild);
}
if (!thisBranch.Children.Any())
return null;
}
return thisBranch;
}
private int OrganizeNode(tlvBranch target)
{
var sources = from tlvBranch item in target.Children
where item.Data is SpectrumSource
select item;
var groups = from tlvBranch item in target.Children
where item.Data is SpectrumSourceGroup
select item;
if (!sources.Any() && !groups.Any())
return -1;
target.Children = new List<tlvBranch>();
foreach (var item in groups)
target.Children.Add(item);
foreach (var item in sources)
target.Children.Add(item);
return groups.Count();
}
private void ApplyDefaultGroups(object sender, EventArgs e)
{
if (sender == miDefaultGroups &&
MessageBox.Show("Are you sure you want to reset the groups to their initial values?",
"Reset Groups?", MessageBoxButtons.YesNo) != DialogResult.Yes)
return;
var obj = tlvGroupedFiles.Objects;
foreach(var item in obj)
tlvGroupedFiles.RemoveObject(item);
lvNonGroupedFiles.Items.Clear();
var sourcesByGroup = session.Query<SpectrumSource>().Where(o => o.Group != null)
.ToLookup(o => (int)o.Group.Id.Value)
.ToDictionary(o => o.Key, o => o.ToList());
var groups = session.Query<SpectrumSourceGroup>().ToList();
var groupNode = new tlvBranch
{
Text = "/",
Parent = new tlvBranch { Text = null },
Children = new List<tlvBranch>(),
Data = (from g in groups where g.Name == "/" select g).Single()
};
groups.Remove(groupNode.Data as SpectrumSourceGroup);
groupNode.Text = (groupNode.Data as SpectrumSourceGroup).Name;
groupNode.cms = cmRightClickGroupNode;
AddAssociatedSpectra(ref groupNode, sourcesByGroup);
groupNode = FillBranch(groupNode, groups, sourcesByGroup);
tlvGroupedFiles.AddObject(groupNode);
_rootNode = groupNode;
tlvGroupedFiles.ExpandAll();
// add ungrouped sources
foreach (var ss in session.Query<SpectrumSource>().Where(g => g.Group == null))
{
var lvTag = new tlvBranch
{
Text = ss.Name,
Parent = new tlvBranch { Text = null },
Children = new List<tlvBranch>(),
Data = ss
};
var lvi = new ListViewItem { Text = ss.Name, Tag = lvTag };
lvNonGroupedFiles.Items.Add(lvi);
}
}
private void RemoveFileNode(tlvBranch selNode, bool refresh)
{
selNode.Parent.Children.Remove(selNode);
if (refresh)
tlvGroupedFiles.RefreshObject(selNode.Parent);
var lvi = new ListViewItem { Name = Path.GetFileName(selNode.Text), Text = Path.GetFileName(selNode.Text), Tag = selNode };
lvNonGroupedFiles.Items.Add(lvi);
}
private void RemoveGroupNode(tlvBranch selNode, bool refresh)
{
var abandonedSpectraSources = new List<tlvBranch>();
getListOfSprectrumSourcesRecursively(selNode, ref abandonedSpectraSources);
if (selNode.Parent.Text != null)
{
selNode.Parent.Children.Remove(selNode);
if (refresh)
tlvGroupedFiles.RefreshObject(selNode.Parent);
}
else
{
selNode.Children.Clear();
if (refresh)
tlvGroupedFiles.RefreshObject(selNode);
}
foreach (tlvBranch tn in abandonedSpectraSources)
{
var lvi = new ListViewItem { Name = Path.GetFileName(tn.Text), Text = Path.GetFileName(tn.Text), Tag = tn };
lvNonGroupedFiles.Items.Add(lvi);
}
}
private void addGroupToolStripMenuItem_Click(object sender, EventArgs e)
{
var selNode = (tlvBranch)tlvGroupedFiles.GetModelObject(_clickedItem.Index);
_numberNewNodes++;
var ssg = new SpectrumSourceGroup();
ssg.Name = (string.Format("{0}/New Group({1})", selNode.Text, _numberNewNodes).Replace(@"//",@"/"));
var newNode = new tlvBranch
{
Text = Path.GetFileName(ssg.Name),
cms = cmRightClickGroupNode,
Parent = selNode,
Children = new List<tlvBranch>(),
Data = ssg
};
selNode.Children.Add(newNode);
OrganizeNode(selNode);
tlvGroupedFiles.RefreshObject(selNode);
if (selNode.Children.Count == 1)
tlvGroupedFiles.Expand(selNode);
tlvGroupedFiles.EditSubItem(tlvGroupedFiles.ModelToItem(newNode),0);
}
private static void getSprectrumSourceGroupsRecursively(tlvBranch treeNode, List<tlvBranch> nodeList, string rootname)
{
if (treeNode.Data is SpectrumSourceGroup)
{
(treeNode.Data as SpectrumSourceGroup).Name =
(rootname + "/"+ treeNode.Text)
.Replace("//", "/");
nodeList.Add(treeNode);
foreach (tlvBranch subNode in treeNode.Children)
getSprectrumSourceGroupsRecursively(subNode, nodeList, (treeNode.Data as SpectrumSourceGroup).Name);
}
}
private static void getListOfSprectrumSourcesRecursively(tlvBranch treeNode, ref List<tlvBranch> spectraNodes)
{
try
{
if (treeNode.Data is SpectrumSource)
spectraNodes.Add(treeNode);
else
foreach (tlvBranch subNode in treeNode.Children)
getListOfSprectrumSourcesRecursively(subNode, ref spectraNodes);
}
catch (Exception exc)
{
throw new Exception("Error in retrieving spectra", exc);
}
}
/// <summary>
/// Check to prevent group nodes from being dragged into their children
/// group nodes.
/// </summary>
/// <returns></returns>
private bool checkIfDestGroupAChildNodeOfMe(tlvBranch destNode, tlvBranch dragNode)
{
try
{
return destNode.Data is SpectrumSourceGroup
&& dragNode.Data is SpectrumSourceGroup
&& ((SpectrumSourceGroup)destNode.Data).Name
.Contains(((SpectrumSourceGroup)dragNode.Data).Name);
}
catch (Exception exc)
{
throw new Exception("Error validating group move: checkIfDestGroupAChildNodeOfMe()\r\n", exc);
}
}
private static List<SpectrumSourceGroup> SetGroupingHierarchy(tlvBranch currentRoot, ISession currentSession)
{
var spectraLocations = new List<tlvBranch>();
var groupsToSave = new List<tlvBranch>();
// find all groups still present
getSprectrumSourceGroupsRecursively(currentRoot, groupsToSave,string.Empty);
// remove old groups and links
currentSession.CreateQuery("DELETE SpectrumSourceGroupLink").ExecuteUpdate();
var unusedGroups = currentSession.Query<SpectrumSourceGroup>().ToList();
foreach (tlvBranch tn in groupsToSave)
unusedGroups.Remove((tn.Data as SpectrumSourceGroup));
foreach (SpectrumSourceGroup ssg in unusedGroups)
currentSession.Delete(ssg);
// save group layout
foreach (tlvBranch treeNode in groupsToSave)
currentSession.SaveOrUpdate(treeNode.Data as SpectrumSourceGroup);
// get new spectra locations
getListOfSprectrumSourcesRecursively(currentRoot, ref spectraLocations);
// update SpectrumSource.Group_ and insert new SpectrumSourceGroupLinks;
// using prepared SQL commands for speed
var cmd1 = currentSession.Connection.CreateCommand();
var cmd2 = currentSession.Connection.CreateCommand();
cmd1.CommandText = "UPDATE SpectrumSource SET Group_ = ? WHERE Id = ?";
cmd2.CommandText = "INSERT INTO SpectrumSourceGroupLink (Group_, Source) VALUES (?,?)";
var parameters1 = new List<IDbDataParameter>();
var parameters2 = new List<IDbDataParameter>();
for (int i = 0; i < 2; ++i)
{
parameters1.Add(cmd1.CreateParameter());
parameters2.Add(cmd2.CreateParameter());
cmd1.Parameters.Add(parameters1[i]);
cmd2.Parameters.Add(parameters2[i]);
}
cmd1.Prepare();
cmd2.Prepare();
foreach (tlvBranch tn in spectraLocations)
{
parameters1[0].Value = ((SpectrumSourceGroup)tn.Parent.Data).Id;
parameters1[1].Value = ((SpectrumSource)tn.Data).Id;
cmd1.ExecuteNonQuery();
var tempNode = tn;
while (tempNode.Parent.Text != null)
{
parameters2[0].Value = ((SpectrumSourceGroup)tempNode.Parent.Data).Id;
parameters2[1].Value = ((SpectrumSource)tn.Data).Id;
tempNode = tempNode.Parent;
cmd2.ExecuteNonQuery();
}
}
return groupsToSave.Select(treeNode => treeNode.Data as SpectrumSourceGroup).ToList();
}
private tlvBranch FillBranch(tlvBranch groupNode, IList<SpectrumSourceGroup> groups, Dictionary<int, List<SpectrumSource>> spectraDictionary)
{
List<SpectrumSourceGroup> potentialChildren;
string fullPath = (groupNode.Data as SpectrumSourceGroup).Name;
if (fullPath == "/")
potentialChildren = (from g in groups
where g.Name != fullPath &&
!g.Name.Remove(0,1).Contains("/")
select g).ToList();
else
potentialChildren = (from g in groups
where g.Name.Contains(fullPath + "/") &&
!g.Name.Remove(0,fullPath.Length+1).Contains("/")
select g).ToList();
foreach (SpectrumSourceGroup ssg in potentialChildren.OrderBy(o => o.Name))
{
var newNode = new tlvBranch
{
Text = Path.GetFileName(ssg.Name),
Children = new List<tlvBranch>(),
Data = ssg,
Parent = groupNode,
cms = cmRightClickGroupNode
};
groups.Remove(newNode.Data as SpectrumSourceGroup);
AddAssociatedSpectra(ref newNode, spectraDictionary);
groupNode.Children.Add(FillBranch(newNode, groups, spectraDictionary));
}
OrganizeNode(groupNode);
return groupNode;
}
private static List <SpectrumSourceGroup> SetGroupingHierarchy(tlvBranch currentRoot, ISession currentSession)
{
var spectraLocations = new List <tlvBranch>();
var groupsToSave = new List <tlvBranch>();
// find all groups still present
getSprectrumSourceGroupsRecursively(currentRoot, groupsToSave, string.Empty);
// remove old groups and links
currentSession.CreateQuery("DELETE SpectrumSourceGroupLink").ExecuteUpdate();
var unusedGroups = currentSession.Query <SpectrumSourceGroup>().ToList();
foreach (tlvBranch tn in groupsToSave)
{
unusedGroups.Remove((tn.Data as SpectrumSourceGroup));
}
foreach (SpectrumSourceGroup ssg in unusedGroups)
{
currentSession.Delete(ssg);
}
// save group layout
foreach (tlvBranch treeNode in groupsToSave)
{
currentSession.SaveOrUpdate(treeNode.Data as SpectrumSourceGroup);
}
// get new spectra locations
getListOfSprectrumSourcesRecursively(currentRoot, ref spectraLocations);
// update SpectrumSource.Group_ and insert new SpectrumSourceGroupLinks;
// using prepared SQL commands for speed
var cmd1 = currentSession.Connection.CreateCommand();
var cmd2 = currentSession.Connection.CreateCommand();
cmd1.CommandText = "UPDATE SpectrumSource SET Group_ = ? WHERE Id = ?";
cmd2.CommandText = "INSERT INTO SpectrumSourceGroupLink (Group_, Source) VALUES (?,?)";
var parameters1 = new List <IDbDataParameter>();
var parameters2 = new List <IDbDataParameter>();
for (int i = 0; i < 2; ++i)
{
parameters1.Add(cmd1.CreateParameter());
parameters2.Add(cmd2.CreateParameter());
cmd1.Parameters.Add(parameters1[i]);
cmd2.Parameters.Add(parameters2[i]);
}
cmd1.Prepare();
cmd2.Prepare();
foreach (tlvBranch tn in spectraLocations)
{
parameters1[0].Value = ((SpectrumSourceGroup)tn.Parent.Data).Id;
parameters1[1].Value = ((SpectrumSource)tn.Data).Id;
cmd1.ExecuteNonQuery();
var tempNode = tn;
while (tempNode.Parent.Text != null)
{
parameters2[0].Value = ((SpectrumSourceGroup)tempNode.Parent.Data).Id;
parameters2[1].Value = ((SpectrumSource)tn.Data).Id;
tempNode = tempNode.Parent;
cmd2.ExecuteNonQuery();
}
}
return(groupsToSave.Select(treeNode => treeNode.Data as SpectrumSourceGroup).ToList());
}
private List<SpectrumSourceGroup> applyAssemblyText(ISession session, string filepath)
{
var spectrumSources = session.Query<SpectrumSource>().ToList();
var sourcesByGroup = new Map<string, List<SpectrumSource>>();
var alreadyGroupedSources = new Set<string>();
var sourceGroups = new List<SpectrumSourceGroup>();
// open the assembly.txt file
using (var assembleTxtFile = File.OpenText(filepath))
{
string line;
while ((line = assembleTxtFile.ReadLine()) != null)
{
if (line.Length == 0)
continue;
try
{
Regex groupFilemaskPair = new Regex("((\"(.+)\")|(\\S+))\\s+((\"(.+)\")|(\\S+))");
Match lineMatch = groupFilemaskPair.Match(line);
string group = lineMatch.Groups[3].ToString() + lineMatch.Groups[4].ToString();
string filemask = lineMatch.Groups[7].ToString() + lineMatch.Groups[8].ToString();
// for wildcards, use old style behavior
if (filemask.IndexOfAny("*?".ToCharArray()) > -1)
{
if (!Path.IsPathRooted(filemask))
filemask = Path.Combine(Path.GetDirectoryName(filepath), filemask);
if (!sourcesByGroup.Contains(group))
sourcesByGroup[group] = new List<SpectrumSource>();
if (!Directory.Exists(Path.GetDirectoryName(filemask)))
continue;
var files = Directory.GetFiles(Path.GetDirectoryName(filemask), Path.GetFileName(filemask));
var sourceNames = files.Select(o => Path.GetFileNameWithoutExtension(o));
foreach (string sourceName in sourceNames)
{
var spectrumSource = spectrumSources.SingleOrDefault(o => o.Name == sourceName);
if (spectrumSource == null)
continue;
var insertResult = alreadyGroupedSources.Insert(sourceName);
if (insertResult.WasInserted)
sourcesByGroup[group].Add(spectrumSource);
}
}
else
{
// otherwise, match directly to source names
string sourceName = Path.GetFileNameWithoutExtension(filemask);
var spectrumSource = spectrumSources.SingleOrDefault(o => o.Name == sourceName);
if (spectrumSource == null)
continue;
var insertResult = alreadyGroupedSources.Insert(sourceName);
if (insertResult.WasInserted)
sourcesByGroup[group].Add(spectrumSource);
}
}
catch (Exception ex)
{
Program.HandleException(new Exception("Error reading assembly text from \"" + filepath + "\": " + ex.Message, ex));
}
}
}
// remove existing groups
RemoveGroupNode(_rootNode, false);
sourceGroups.Add(new SpectrumSourceGroup { Name = "/" });
// build new group hierarchy
foreach (var itr in sourcesByGroup)
{
if (itr.Value.IsNullOrEmpty())
continue;
var ssg = new SpectrumSourceGroup { Name = itr.Key };
if (!alreadyGroupedSources.Contains(ssg.Name))
sourceGroups.Add(ssg);
// decompose group path into segments, e.g. /foo/bar/ -> {foo, bar}
IEnumerable<string> segments = ssg.Name.Split("/".ToCharArray(), StringSplitOptions.RemoveEmptyEntries);
segments = segments.Take(segments.Count() - 1); // ignore the last segment
var parentNode = _rootNode;
foreach(string segment in segments)
{
var segmentNode = parentNode.Children.FirstOrDefault(o => o.Text == segment);
if (segmentNode == null)
{
var segmentGroup = new SpectrumSourceGroup { Name = (parentNode.Text + "/").Replace("//", "/") + segment };
if (!alreadyGroupedSources.Contains(segmentGroup.Name))
sourceGroups.Add(segmentGroup);
segmentNode = new tlvBranch
{
Text = segment,
cms = cmRightClickGroupNode,
Parent = parentNode,
Children = new List<tlvBranch>(),
Data = segmentGroup
};
parentNode.Children.Add(segmentNode);
}
parentNode = segmentNode;
}
// parentNode is now the immediate parent of the current group
var groupNode = new tlvBranch
{
Text = Path.GetFileName(ssg.Name),
cms = cmRightClickGroupNode,
Parent = parentNode,
Children = new List<tlvBranch>(),
Data = ssg
};
foreach (var source in itr.Value)
{
var sourceNode = new tlvBranch
{
Text = Path.GetFileName(source.Name),
Parent = groupNode,
Data = source,
cms = cmRightClickFileNode
};
groupNode.Children.Add(sourceNode);
lvNonGroupedFiles.Items.RemoveByKey(sourceNode.Text);
}
parentNode.Children.Add(groupNode);
}
tlvGroupedFiles.RefreshObject(_rootNode);
tlvGroupedFiles.ExpandAll();
return sourceGroups.ToList();
}
private void ApplyDefaultGroups(object sender, EventArgs e)
{
if (sender == miDefaultGroups &&
MessageBox.Show("Are you sure you want to reset the groups to their initial values?",
"Reset Groups?", MessageBoxButtons.YesNo) != DialogResult.Yes)
{
return;
}
var obj = tlvGroupedFiles.Objects;
foreach (var item in obj)
{
tlvGroupedFiles.RemoveObject(item);
}
lvNonGroupedFiles.Items.Clear();
var groups = session.Query <SpectrumSourceGroup>().ToList();
// if root group is only group, put sources with null group into root group
var sourcesByGroup = session.Query <SpectrumSource>().Where(o => o.Group != null || groups.Count == 1)
.ToLookup(o => (int)(o.Group?.Id ?? 1))
.ToDictionary(o => o.Key, o => o.ToList());
var groupNode = new tlvBranch
{
Text = "/",
Parent = new tlvBranch {
Text = null
},
Children = new List <tlvBranch>(),
Data = (from g in groups where g.Name == "/" select g).First()
};
groups.Remove(groupNode.Data as SpectrumSourceGroup);
groupNode.Text = (groupNode.Data as SpectrumSourceGroup).Name;
groupNode.cms = cmRightClickGroupNode;
AddAssociatedSpectra(ref groupNode, sourcesByGroup);
groupNode = FillBranch(groupNode, groups, sourcesByGroup);
tlvGroupedFiles.AddObject(groupNode);
_rootNode = groupNode;
tlvGroupedFiles.ExpandAll();
// add ungrouped sources
foreach (var ss in session.Query <SpectrumSource>().Where(g => g.Group == null))
{
var lvTag = new tlvBranch
{
Text = ss.Name,
Parent = new tlvBranch {
Text = null
},
Children = new List <tlvBranch>(),
Data = ss
};
var lvi = new ListViewItem {
Text = ss.Name, Tag = lvTag
};
lvNonGroupedFiles.Items.Add(lvi);
}
}
private void AddAssociatedSpectra(ref tlvBranch groupNode, Dictionary<int, List<SpectrumSource>> spectraDictionary)
{
try
{
int groupID = (int)(groupNode.Data as SpectrumSourceGroup).Id;
if (spectraDictionary.ContainsKey(groupID))
{
foreach (SpectrumSource ss in spectraDictionary[groupID])
{
var newNode = new tlvBranch
{
Text = Path.GetFileName(ss.Name),
Parent = groupNode,
Data = ss,
cms = cmRightClickFileNode
};
groupNode.Children.Add(newNode);
}
}
}
catch
{
MessageBox.Show("Could not add spectra to branch");
}
}
private List <SpectrumSourceGroup> applyAssemblyText(ISession session, string filepath)
{
var spectrumSources = session.Query <SpectrumSource>().ToList();
var sourcesByGroup = new Map <string, List <SpectrumSource> >();
var alreadyGroupedSources = new Set <string>();
var sourceGroups = new List <SpectrumSourceGroup>();
// open the assembly.txt file
using (var assembleTxtFile = File.OpenText(filepath))
{
string line;
while ((line = assembleTxtFile.ReadLine()) != null)
{
if (line.Length == 0)
{
continue;
}
try
{
Regex groupFilemaskPair = new Regex("((\"(.+)\")|(\\S+))\\s+((\"(.+)\")|(\\S+))");
Match lineMatch = groupFilemaskPair.Match(line);
string group = lineMatch.Groups[3].ToString() + lineMatch.Groups[4].ToString();
string filemask = lineMatch.Groups[7].ToString() + lineMatch.Groups[8].ToString();
group = group.TrimEnd(' ', '/');
// for wildcards, use old style behavior
if (filemask.IndexOfAny("*?".ToCharArray()) > -1)
{
if (!Path.IsPathRooted(filemask))
{
filemask = Path.Combine(Path.GetDirectoryName(filepath), filemask);
}
if (!sourcesByGroup.Contains(group))
{
sourcesByGroup[group] = new List <SpectrumSource>();
}
if (!Directory.Exists(Path.GetDirectoryName(filemask)))
{
continue;
}
var files = Directory.GetFiles(Path.GetDirectoryName(filemask), Path.GetFileName(filemask));
var sourceNames = files.Select(o => Path.GetFileNameWithoutExtension(o));
foreach (string sourceName in sourceNames)
{
var spectrumSource = spectrumSources.SingleOrDefault(o => o.Name == sourceName);
if (spectrumSource == null)
{
continue;
}
var insertResult = alreadyGroupedSources.Insert(sourceName);
if (insertResult.WasInserted)
{
sourcesByGroup[group].Add(spectrumSource);
}
}
}
else
{
// otherwise, match directly to source names
string sourceName = Path.GetFileNameWithoutExtension(filemask);
var spectrumSource = spectrumSources.SingleOrDefault(o => o.Name == sourceName);
if (spectrumSource == null)
{
continue;
}
var insertResult = alreadyGroupedSources.Insert(sourceName);
if (insertResult.WasInserted)
{
sourcesByGroup[group].Add(spectrumSource);
}
}
}
catch (Exception ex)
{
Program.HandleException(new Exception("Error reading assembly text from \"" + filepath + "\": " + ex.Message, ex));
}
}
}
// remove existing groups
RemoveGroupNode(_rootNode, false);
sourceGroups.Add(new SpectrumSourceGroup {
Name = "/"
});
// build new group hierarchy
foreach (var itr in sourcesByGroup)
{
if (itr.Value.IsNullOrEmpty())
{
continue;
}
var ssg = new SpectrumSourceGroup {
Name = itr.Key
};
if (!alreadyGroupedSources.Contains(ssg.Name))
{
sourceGroups.Add(ssg);
}
// decompose group path into segments, e.g. /foo/bar/ -> {foo, bar}
IEnumerable <string> segments = ssg.Name.Split("/".ToCharArray(), StringSplitOptions.RemoveEmptyEntries);
segments = segments.Take(segments.Count() - 1); // ignore the last segment
var parentNode = _rootNode;
foreach (string segment in segments)
{
var segmentNode = parentNode.Children.FirstOrDefault(o => o.Text == segment);
if (segmentNode == null)
{
var segmentGroup = new SpectrumSourceGroup {
Name = (parentNode.Text + "/").Replace("//", "/") + segment
};
if (!alreadyGroupedSources.Contains(segmentGroup.Name))
{
sourceGroups.Add(segmentGroup);
}
segmentNode = new tlvBranch
{
Text = segment,
cms = cmRightClickGroupNode,
Parent = parentNode,
Children = new List <tlvBranch>(),
Data = segmentGroup
};
parentNode.Children.Add(segmentNode);
}
parentNode = segmentNode;
}
// parentNode is now the immediate parent of the current group
var groupNode = new tlvBranch
{
Text = Path.GetFileName(ssg.Name),
cms = cmRightClickGroupNode,
Parent = parentNode,
Children = new List <tlvBranch>(),
Data = ssg
};
foreach (var source in itr.Value)
{
var sourceNode = new tlvBranch
{
Text = Path.GetFileName(source.Name),
Parent = groupNode,
Data = source,
cms = cmRightClickFileNode
};
groupNode.Children.Add(sourceNode);
lvNonGroupedFiles.Items.RemoveByKey(sourceNode.Text);
}
parentNode.Children.Add(groupNode);
}
tlvGroupedFiles.RefreshObject(_rootNode);
tlvGroupedFiles.ExpandAll();
return(sourceGroups.ToList());
}