public void TestProxyBuild()
{
/// Make sure the header file that gets generated is basically "ok" for use. These are simple tests.
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
FileInfo fhpp = new FileInfo("ntuple_dude.h");
FileInfo cfile = new FileInfo("TestProxyBuild.C");
using (var writer = cfile.CreateText())
{
writer.WriteLine("#include \"ntuple_dude.h\"");
writer.Close();
}
/// Make sure that ROOT has been correctly initalized!
ROOTNET.NTApplication.CreateApplication();
/// Now run the compile
var compile = ROOTNET.NTSystem.gSystem.CompileMacro("TestProxyBuild.C");
Assert.AreEqual(1, compile, "compile error for built macro - make sure that cl is a good command by starting devenv with vs command line!!");
}
public void TestGenerateSimpleItems()
{
var t = TTreeParserCPPTests.CreateTrees.CreateSingleItemTree();
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "expected only the top level class to come back");
string[] possibleTypes = new string[] {
"int",
"uint",
"short",
"ushort",
"long",
"ulong",
"bool"
};
Assert.IsTrue(result[0].Items.All(i => possibleTypes.Contains(i.ItemType)), "Some responses not correct!");
Assert.AreEqual(possibleTypes.Length, result[0].Items.Select(i => i.ItemType).Distinct().Count(), "Incorrect number of branches found!");
CheckSerialization(result, "GenerateClassesTestVectorIntAndDoubleAndShort");
}
public void GenerateClassesTestCPPName()
{
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithIndexedSimpleVector(20);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "should only be top level class");
var item = result[0];
Assert.AreEqual(2, item.Items.Count(), "# of items found");
Assert.AreEqual(1, item.Items.Where(i => i.ItemType == "int").Count(), "# found variables");
var i1 = item.Items.Where(i => i.ItemType == "int").First();
Assert.AreEqual("n", i1.Name, "index name");
Assert.AreEqual(1, item.Items.Where(i => i.ItemType == "int[]").Count(), "# of int[] variables");
var i2 = item.Items.Where(i => i.ItemType == "int[]").First();
Assert.AreEqual("arr", i2.Name, "arr name");
var i2asA = i2 as ItemCStyleArray;
Assert.AreEqual(1, i2asA.Indicies.Count, "# of indicies");
Assert.AreEqual("n", i2asA.Indicies[0].indexBoundName, "index name");
Assert.AreEqual(0, i2asA.Indicies[0].indexPosition, "index name");
Assert.IsFalse(i2asA.Indicies[0].indexConst, "index const");
CheckSerialization(result, "GenerateClassesTestCPPName");
}
public void GenerateClassesTestCPPName2D()
{
// Test an array that is 2D C style
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithIndexedSimpleVector2D(20);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "should only be top level class");
var item = result[0];
Assert.AreEqual(1, item.Items.Count(), "# of items found");
var item1 = item.Items[0];
Assert.AreEqual("int[][]", item1.ItemType, "2d array type");
Assert.AreEqual("arr", item1.Name, "arr name");
var i2asA = item1 as ItemCStyleArray;
Assert.AreEqual(2, i2asA.Indicies.Count, "# of indicies");
Assert.AreEqual("5", i2asA.Indicies[0].indexBoundName, "index name");
Assert.AreEqual(0, i2asA.Indicies[0].indexPosition, "index name");
Assert.IsTrue(i2asA.Indicies[0].indexConst, "index const");
Assert.AreEqual("4", i2asA.Indicies[1].indexBoundName, "index name");
Assert.AreEqual(0, i2asA.Indicies[1].indexPosition, "index name");
Assert.IsTrue(i2asA.Indicies[1].indexConst, "index const");
CheckSerialization(result, "GenerateClassesTestCPPName");
}
public void TestGenerateWithStringVectorTypes()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithStringTypes();
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "Empty ntuple should have fired");
Assert.AreEqual(0, result[0].Items.Count, "Expected nothing to be pasred");
}
public void TestClassWithDoubleColonName()
{
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithDoubleColorName();
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
var cls = result[0];
var item = cls.Items[0];
Assert.AreEqual("dude__fork", item.Name, "Name not translated correctly");
}
public void GenerateClassesTestBadNameEvent()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntName("event");
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
var item = result[0];
var first = item.Items[0];
var st = first as ItemSimpleType;
Assert.AreEqual("event_", st.Name, "Specialized name not correct");
}
public void TestTreesWithPounds()
{
var f = new ROOTNET.NTFile("atest.root", "READ");
var t = f.Get("##Shapes") as ROOTNET.Interface.NTTree;
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "# of classes");
var obj = result[0];
Assert.IsFalse(obj.UserInfoPath.Contains("#"), "user path: " + obj.UserInfoPath);
}
public void GenerateClassesTestListOfTLZ()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithTLZVector();
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "Only the top level class was expected");
Assert.AreEqual(1, result[0].Items.Count(), "only one TLZ should have been in there");
var i = result[0].Items.First();
Assert.AreEqual("TLorentzVector[]", i.ItemType, "incorrect item type for vector of tlz");
}
public void TestProxyGenerationContents()
{
/// Make sure the header file that gets generated is basically "ok" for use. These are simple tests.
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
Assert.IsNotNull(t);
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
FileInfo fhpp = new FileInfo("ntuple_dude.h");
//Assert.IsFalse(CheckForLineContaining(fhpp, "junk_macro_parsettree.C"), "Checking for junk_macro in the file");
}
public void TestComplexObjectNiceObjectFile()
{
// Do we correctly parse an ATLAS MC file?
var f = ROOTNET.NTFile.Open("btagobjs.root", "READ");
var t = f.Get("btagging") as ROOTNET.Interface.NTTree;
var p = new ParseTTree();
var r = p.GenerateClasses(t).ToArray();
f.Close();
Assert.Inconclusive();
}
public void GenerateAndGC()
{
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithConstIndexedSimpleVector(20);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result_itr = p.GenerateClasses(t);
GC.Collect(GC.MaxGeneration);
var result = result_itr.ToArray();
CheckSerialization(result, "GenerateClassesTestCPPName");
}
public void TestSimpleParse()
{
var t = GetTree("short-nonsplit.root", "CollectionTree");
var p = new ParseTTree();
var result = p.GenerateClasses(t.Item1).ToArray();
Assert.AreEqual(1, result.Where(c => c.IsTopLevelClass).Count(), "# of top level classes");
Assert.AreEqual(6, result.Count(), "Total number of classes");
var classMap = result.ToDictionary(i => i.Name, i => i);
Assert.IsTrue(classMap.ContainsKey("EventInfo_p3"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("McEventCollection_p5"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("GenParticle_p5"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("GenVertex_p5"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("GenEvent_p5"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("CollectionTree"), "CollectionTree");
// Check collection tree has the right top level stuff
var ct = classMap["CollectionTree"];
Assert.AreEqual(2, ct.Items.Count, "# of items in collection tree");
var ctitems = ct.Items.ToDictionary(citem => citem.Name, citem => citem);
Assert.IsTrue(ctitems.ContainsKey("EventInfo_p3_McEventInfo"), "EventInfo_p3_McEventInfo");
Assert.IsTrue(ctitems.ContainsKey("McEventCollection_p5_GEN_EVENT"), "McEventCollection_p5_GEN_EVENT");
Assert.AreEqual("EventInfo_p3", ctitems["EventInfo_p3_McEventInfo"].ItemType, "McEventCollection_p5_GEN_EVENT type");
Assert.AreEqual("McEventCollection_p5", ctitems["McEventCollection_p5_GEN_EVENT"].ItemType, "McEventCollection_p5_GEN_EVENT type");
// Check the McEventCollection to make sure things work one level down.
var mcec = classMap["McEventCollection_p5"];
ctitems = mcec.Items.ToDictionary(citem => citem.Name, citem => citem);
Assert.AreEqual(3, ctitems.Count, "# of items in McEventCollection_p5 class");
Assert.IsTrue(ctitems.ContainsKey("m_genEvents"), "m_genEvents member missing from McEventCollection_p5");
Assert.IsTrue(ctitems.ContainsKey("m_genVertices"), "m_genVerticies member missing from McEventCollection_p5");
Assert.IsTrue(ctitems.ContainsKey("m_genParticles"), "m_genParticles member missing from McEventCollection_p5");
Assert.AreEqual("GenParticle_p5[]", ctitems["m_genParticles"].ItemType, "m_genParticles member type");
Assert.AreEqual("GenVertex_p5[]", ctitems["m_genVertices"].ItemType, "m_genVertices member type");
Assert.AreEqual("GenEvent_p5[]", ctitems["m_genEvents"].ItemType, "m_genEvents member type");
// spot check one of the low-level objects.
var gevts = classMap["GenEvent_p5"];
ctitems = gevts.Items.ToDictionary(citem => citem.Name, citem => citem);
Assert.AreEqual(20, ctitems.Count, "# of items in GenEvent_p5 class (minus # that aren't basic types)");
Assert.IsTrue(ctitems.ContainsKey("m_mpi"), "m_mpi member missing");
Assert.AreEqual("int", ctitems["m_mpi"].ItemType, "m_mpi type");
Assert.IsTrue(ctitems.ContainsKey("m_weights"), "m_weight smember missing");
Assert.AreEqual("double[]", ctitems["m_weights"].ItemType, "m_weights type");
}
public void TestBasicProxyAndUserInfoGeneration()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
FileInfo fhpp = new FileInfo("ntuple_dude.h");
Assert.IsTrue(fhpp.Exists, "check for hpp file existance");
Assert.IsTrue(File.Exists(result[0].UserInfoPath), "user info file missing");
}
public void GenerateClassesTestTLorentzVector()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithTLZOnly(5);
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "should only be top level class");
var item = result[0];
Assert.AreEqual(5, item.Items.Count(), "Expected 5 items in there");
Assert.IsTrue(item.Items.All(i => i.ItemType == "ROOTNET.Interface.NTLorentzVector"), "Not everything is a TLZ!");
CheckSerialization(result, "GenerateClassesTestTLorentzVector");
}
public void GenerateClassesEmptyTree()
{
var emptyT = new ROOTNET.NTTree("dude", "empty");
var p = new ParseTTree();
var result = p.GenerateClasses(emptyT).ToArray();
Assert.AreEqual(1, result.Length, "Expected only the top level class");
Assert.AreEqual("dude", result[0].Name, "class name incorrect");
Assert.AreEqual(0, result[0].Items.Count(), "empty tree...");
Assert.IsTrue(result[0].IsTopLevelClass, "top level class");
Assert.IsFalse(result[0].IsTClonesArrayClass, "tclones array");
CheckSerialization(result, "GenerateClassesEmptyTree");
}
public void GenerateClassesTestSingleBasicType()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "should only be top level class");
var item = result[0];
Assert.AreEqual(5, item.Items.Count(), "Expected 5 items in there");
Assert.IsTrue(item.Items.All(i => i.ItemType == "int"), "Not everything is an int!");
Assert.IsTrue(item.Items.All(i => !i.NotAPointer), "Not A Pointer is set");
CheckSerialization(result, "GenerateClassesTestSingleBasicType");
}
public void GenerateClassesEmptyTree()
{
var emptyT = new ROOTNET.NTTree("dude", "empty");
var p = new ParseTTree();
var result = p.GenerateClasses(emptyT).ToArray();
Assert.AreEqual(1, result.Length, "Expected only the top level class");
Assert.AreEqual("dude", result[0].Name, "class name incorrect");
Assert.AreEqual(0, result[0].Items.Count(), "empty tree...");
Assert.IsTrue(result[0].IsTopLevelClass, "top level class");
Assert.IsFalse(result[0].IsTClonesArrayClass, "tclones array");
CheckSerialization(result, "GenerateClassesEmptyTree");
}
public void GenerateClassesTestVectorVector()
{
var t = TTreeParserCPPTests.CreateTrees.CreateVectorVectorTree();
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "expected only the top level class to come back");
Assert.AreEqual(1, result[0].Items.Count(), "incorrect # of items found for this guy");
Assert.AreEqual("double[][]", result[0].Items.First().ItemType, "improper double vector type!");
CheckSerialization(result, "GenerateClassesTestVectorVector");
}
public void TestGenerateClasses2DCStyleVector()
{
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithCStyleVectorArray();
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "only one top level class here");
var citem = result[0];
Assert.AreEqual(1, citem.Items.Count, "# of items");
var item = citem.Items[0];
Assert.IsInstanceOfType(item, typeof(ItemCStyleArray), "item type");
}
public void GenerateClassesTestSingleBasicType()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "should only be top level class");
var item = result[0];
Assert.AreEqual(5, item.Items.Count(), "Expected 5 items in there");
Assert.IsTrue(item.Items.All(i => i.ItemType == "int"), "Not everything is an int!");
Assert.IsTrue(item.Items.All(i => !i.NotAPointer), "Not A Pointer is set");
CheckSerialization(result, "GenerateClassesTestSingleBasicType");
}
public void GenerateClassesTestListOfLeaves()
{
var t = TTreeParserCPPTests.CreateTrees.CreateListOfLeavesTree();
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "expected only the top level class to come back");
Assert.AreEqual(3, result[0].Items.Count(), "incorrect # of items found for this guy");
string[] leafTypes = new string[] { "int", "float", "double" };
Assert.IsTrue(result[0].Items.All(i => leafTypes.Contains(i.ItemType)), "bad item type in the list!");
Assert.AreEqual(leafTypes.Length, result[0].Items.Select(l => l.ItemType).Distinct().Count(), "Bad # of differen tytpes");
CheckSerialization(result, "GenerateClassesTestListOfLeaves");
}
public void ParseArrayWithPairs()
{
// Do we correctly parse an ATLAS MC file?
var f = ROOTNET.NTFile.Open("pairedarray.root", "READ");
var t = f.Get("MetaData") as ROOTNET.Interface.NTTree;
Assert.IsNotNull(t, "no tree found");
var p = new ParseTTree();
var r = p.GenerateClasses(t).ToArray();
f.Close();
Assert.IsFalse(r.Any(c => c.Name.Contains(",")), "a name contains a comma");
Assert.IsFalse(r.Any(c => c.Name.Contains("<")), "a name contains a <");
Assert.IsFalse(r.Any(c => c.Name.Contains(">")), "a name contains a >");
Assert.IsFalse(r.Any(c => c.Name.Contains(" ")), "a name contains a space ");
}
public void TestSimpleParse()
{
var t = GetTree("short-nonsplit.root", "CollectionTree");
var p = new ParseTTree();
var result = p.GenerateClasses(t.Item1).ToArray();
Assert.AreEqual(1, result.Where(c => c.IsTopLevelClass).Count(), "# of top level classes");
Assert.AreEqual(6, result.Count(), "Total number of classes");
var classMap = result.ToDictionary(i => i.Name, i => i);
Assert.IsTrue(classMap.ContainsKey("EventInfo_p3"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("McEventCollection_p5"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("GenParticle_p5"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("GenVertex_p5"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("GenEvent_p5"), "EventInfo_p3");
Assert.IsTrue(classMap.ContainsKey("CollectionTree"), "CollectionTree");
// Check collection tree has the right top level stuff
var ct = classMap["CollectionTree"];
Assert.AreEqual(2, ct.Items.Count, "# of items in collection tree");
var ctitems = ct.Items.ToDictionary(citem => citem.Name, citem => citem);
Assert.IsTrue(ctitems.ContainsKey("EventInfo_p3_McEventInfo"), "EventInfo_p3_McEventInfo");
Assert.IsTrue(ctitems.ContainsKey("McEventCollection_p5_GEN_EVENT"), "McEventCollection_p5_GEN_EVENT");
Assert.AreEqual("EventInfo_p3", ctitems["EventInfo_p3_McEventInfo"].ItemType, "McEventCollection_p5_GEN_EVENT type");
Assert.AreEqual("McEventCollection_p5", ctitems["McEventCollection_p5_GEN_EVENT"].ItemType, "McEventCollection_p5_GEN_EVENT type");
// Check the McEventCollection to make sure things work one level down.
var mcec = classMap["McEventCollection_p5"];
ctitems = mcec.Items.ToDictionary(citem => citem.Name, citem => citem);
Assert.AreEqual(3, ctitems.Count, "# of items in McEventCollection_p5 class");
Assert.IsTrue(ctitems.ContainsKey("m_genEvents"), "m_genEvents member missing from McEventCollection_p5");
Assert.IsTrue(ctitems.ContainsKey("m_genVertices"), "m_genVerticies member missing from McEventCollection_p5");
Assert.IsTrue(ctitems.ContainsKey("m_genParticles"), "m_genParticles member missing from McEventCollection_p5");
Assert.AreEqual("GenParticle_p5[]", ctitems["m_genParticles"].ItemType, "m_genParticles member type");
Assert.AreEqual("GenVertex_p5[]", ctitems["m_genVertices"].ItemType, "m_genVertices member type");
Assert.AreEqual("GenEvent_p5[]", ctitems["m_genEvents"].ItemType, "m_genEvents member type");
// spot check one of the low-level objects.
var gevts = classMap["GenEvent_p5"];
ctitems = gevts.Items.ToDictionary(citem => citem.Name, citem => citem);
Assert.AreEqual(20, ctitems.Count, "# of items in GenEvent_p5 class (minus # that aren't basic types)");
Assert.IsTrue(ctitems.ContainsKey("m_mpi"), "m_mpi member missing");
Assert.AreEqual("int", ctitems["m_mpi"].ItemType, "m_mpi type");
Assert.IsTrue(ctitems.ContainsKey("m_weights"), "m_weight smember missing");
Assert.AreEqual("double[]", ctitems["m_weights"].ItemType, "m_weights type");
}
public void TestBasicProxyGenerationSpecial()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
DirectoryInfo newdir = new DirectoryInfo("./TestBasicProxyGenerationSpecial");
if (!newdir.Exists)
{
newdir.Create();
}
p.ProxyGenerationLocation = newdir;
var result = p.GenerateClasses(t).ToArray();
FileInfo fhpp = new FileInfo(newdir.FullName + "\\ntuple_dude.h");
Assert.IsTrue(fhpp.Exists, "check for hpp file existance");
}
public void TestSingleBasicTypeIsUngrouped()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
///
/// See if the ntuple output file that groups things exists
///
using (var reader = File.OpenText(result[0].UserInfoPath))
{
var parser = new XmlSerializer(typeof(TTreeUserInfo));
var classInfo = parser.Deserialize(reader) as TTreeUserInfo;
Assert.IsNotNull(classInfo, "Unable to read back the user info");
Assert.AreEqual(1, classInfo.Groups.Length, "# of groups");
var grp = classInfo.Groups[0];
Assert.AreEqual("ungrouped", grp.Name, "group name");
Assert.AreEqual(5, grp.Variables.Length, "# of group variables");
Assert.AreEqual("item_", grp.Variables[0].NETName, "first variable name incorrect");
}
}
public void GenerateClassesWithCustomClassesAndSubClasses()
{
var f = new ROOTNET.NTFile("ComplexNtupleTestInput.root", "READ");
try {
Assert.IsTrue(f.IsOpen(), "Test file not found");
var t = f.Get("btag") as ROOTNET.Interface.NTTree;
Assert.IsNotNull(t.Name); // Make sure we got the tree correctly
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
// Make sure the global features are right
Assert.AreEqual(1, result.Where(c => c.IsTopLevelClass).Count(), "# of top level classes");
Assert.AreEqual(3, result.Length, "# of classes");
var classMap = result.ToDictionary(r => r.Name, r => r);
Assert.IsTrue(classMap.ContainsKey("btag"), "btag class present");
Assert.IsTrue(classMap.ContainsKey("MuonInBJet"), "MuonInBJet class present");
Assert.IsTrue(classMap.ContainsKey("BTagJet"), "btag class present");
} finally
{
f.Close();
}
}
public void GenerateClassesTestVectorIntAndDoubleAndShort()
{
var t = TTreeParserCPPTests.CreateTrees.CreateVectorTree();
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "expected only the top level class to come back");
string[] possibleTypes = new string[] {
"int[]",
"double[]",
// "short[]", // Not known to root dictionary by default.
"bool[]"
//"float[]" // Weird - it doesn't work either.
};
Assert.IsTrue(result[0].Items.All(i => possibleTypes.Contains(i.ItemType)), "Some responses not correct!");
Assert.AreEqual(possibleTypes.Length, result[0].Items.Select(i => i.ItemType).Distinct().Count(), "Incorrect number of branches found!");
CheckSerialization(result, "GenerateClassesTestVectorIntAndDoubleAndShort");
}
public void TestForSTLVectorPickup()
{
///
/// If a tree contains a vector<float> or similar, then that will generate
/// a #pragma link and link6 line. We need to make sure that our code correctly
/// removes that from the .h file and also adds them to the list of variables
/// that should be added later. This is due to a bug in the ROOT on windows that
/// means unloading the query after using one of these becomes unsafe (and unstable!).
///
var t = TTreeParserCPPTests.CreateTrees.CreateVectorVectorTree();
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.IsNotNull(result[0].ClassesToGenerate, "Expected some classes to generate");
Assert.AreEqual(3, result[0].ClassesToGenerate.Length, "expected 3 different vector class references");
Assert.IsTrue(result[0].ClassesToGenerate.All(c => c.includeFiles == "vector"), "vector needed to be the include file");
var names = new List <string>()
{
"vector<int>", "vector<short>", "vector<bool>"
};
Assert.IsTrue(result[0].ClassesToGenerate.All(c => names.Contains(c.classSpec)), "Some bad class spec names lifted from C++");
var fi = new FileInfo(result[0].NtupleProxyPath);
using (var reader = fi.OpenText())
{
while (!reader.EndOfStream)
{
var line = reader.ReadLine();
Assert.IsFalse(line.Contains("#pragma") && line.Contains("vector"), "line '" + line + "' seems to have a bad pragma");
}
}
}
public void GenerateClassesTestCPPName()
{
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithIndexedSimpleVector(20);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "should only be top level class");
var item = result[0];
Assert.AreEqual(2, item.Items.Count(), "# of items found");
Assert.AreEqual(1, item.Items.Where(i => i.ItemType == "int").Count(), "# found variables");
var i1 = item.Items.Where(i => i.ItemType == "int").First();
Assert.AreEqual("n", i1.Name, "index name");
Assert.AreEqual(1, item.Items.Where(i => i.ItemType == "int[]").Count(), "# of int[] variables");
var i2 = item.Items.Where(i => i.ItemType == "int[]").First();
Assert.AreEqual("arr", i2.Name, "arr name");
var i2asA = i2 as ItemCStyleArray;
Assert.AreEqual(1, i2asA.Indicies.Count, "# of indicies");
Assert.AreEqual("n", i2asA.Indicies[0].indexBoundName, "index name");
Assert.AreEqual(0, i2asA.Indicies[0].indexPosition, "index name");
Assert.IsFalse(i2asA.Indicies[0].indexConst, "index const");
CheckSerialization(result, "GenerateClassesTestCPPName");
}
public void TestBasicProxyGenerationSpecial()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
DirectoryInfo newdir = new DirectoryInfo("./TestBasicProxyGenerationSpecial");
if (!newdir.Exists)
newdir.Create();
p.ProxyGenerationLocation = newdir;
var result = p.GenerateClasses(t).ToArray();
FileInfo fhpp = new FileInfo(newdir.FullName + "\\ntuple_dude.h");
Assert.IsTrue(fhpp.Exists, "check for hpp file existance");
Assert.AreEqual(fhpp.FullName, result[0].NtupleProxyPath, "ntuple proxy path incorrect");
}
public void GenerateClassesTestTLorentzVector()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithTLZOnly(5);
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "should only be top level class");
var item = result[0];
Assert.AreEqual(5, item.Items.Count(), "Expected 5 items in there");
Assert.IsTrue(item.Items.All(i => i.ItemType == "ROOTNET.Interface.NTLorentzVector"), "Not everything is a TLZ!");
CheckSerialization(result, "GenerateClassesTestTLorentzVector");
}
public void TestProxyBuild()
{
/// Make sure the header file that gets generated is basically "ok" for use. These are simple tests.
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
FileInfo fhpp = new FileInfo("ntuple_dude.h");
FileInfo cfile = new FileInfo("TestProxyBuild.C");
using (var writer = cfile.CreateText())
{
writer.WriteLine("#include \"ntuple_dude.h\"");
writer.Close();
}
/// Make sure that ROOT has been correctly initalized!
ROOTNET.NTApplication.CreateApplication();
/// Now run the compile
var compile = ROOTNET.NTSystem.gSystem.CompileMacro("TestProxyBuild.C");
Assert.AreEqual(1, compile, "compile error for built macro - make sure that cl is a good command by starting devenv with vs command line!!");
}
public void TestProxyGenerationContents()
{
/// Make sure the header file that gets generated is basically "ok" for use. These are simple tests.
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
Assert.IsNotNull(t);
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
FileInfo fhpp = new FileInfo("ntuple_dude.h");
//Assert.IsFalse(CheckForLineContaining(fhpp, "junk_macro_parsettree.C"), "Checking for junk_macro in the file");
}
public void TestGenerateWithOddVectorTypes()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithOddTypes();
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "expected only the top level class to come back");
string[] possibleTypes = new string[] {
"uint[]",
"int[]"
};
Assert.IsTrue(result[0].Items.All(i => possibleTypes.Contains(i.ItemType)), "Some responses not correct!");
Assert.AreEqual(possibleTypes.Length, result[0].Items.Select(i => i.ItemType).Distinct().Count(), "Incorrect number of branches found!");
CheckSerialization(result, "GenerateClassesTestVectorIntAndDoubleAndShort");
}
public void TestClassWithDoubleColonName()
{
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithDoubleColorName();
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
var cls = result[0];
var item = cls.Items[0];
Assert.AreEqual("dude__fork", item.Name, "Name not translated correctly");
}
public void GenerateClassesTestListOfTLZ()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithTLZVector();
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "Only the top level class was expected");
Assert.AreEqual(1, result[0].Items.Count(), "only one TLZ should have been in there");
var i = result[0].Items.First();
Assert.AreEqual("TLorentzVector[]", i.ItemType, "incorrect item type for vector of tlz");
}
public void TestSingleBasicTypeIsUngrouped()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
///
/// See if the ntuple output file that groups things exists
///
using (var reader = File.OpenText(result[0].UserInfoPath))
{
var parser = new XmlSerializer(typeof(TTreeUserInfo));
var classInfo = parser.Deserialize(reader) as TTreeUserInfo;
Assert.IsNotNull(classInfo, "Unable to read back the user info");
Assert.AreEqual(1, classInfo.Groups.Length, "# of groups");
var grp = classInfo.Groups[0];
Assert.AreEqual("ungrouped", grp.Name, "group name");
Assert.AreEqual(5, grp.Variables.Length, "# of group variables");
Assert.AreEqual("item_", grp.Variables[0].NETName, "first variable name incorrect");
}
}
public void TestBasicProxyAndUserInfoGeneration()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntOnly(5);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
FileInfo fhpp = new FileInfo("ntuple_dude.h");
Assert.IsTrue(fhpp.Exists, "check for hpp file existance");
Assert.AreEqual(fhpp.FullName, result[0].NtupleProxyPath, "ntuple proxy path incorrect");
Assert.IsTrue(File.Exists(result[0].UserInfoPath), "user info file missing");
}
public void ParseArrayWithPairs()
{
// Do we correctly parse an ATLAS MC file?
var f = ROOTNET.NTFile.Open("pairedarray.root", "READ");
var t = f.Get("MetaData") as ROOTNET.Interface.NTTree;
Assert.IsNotNull(t, "no tree found");
var p = new ParseTTree();
var r = p.GenerateClasses(t).ToArray();
f.Close();
Assert.IsFalse(r.Any(c => c.Name.Contains(",")), "a name contains a comma");
Assert.IsFalse(r.Any(c => c.Name.Contains("<")), "a name contains a <");
Assert.IsFalse(r.Any(c => c.Name.Contains(">")), "a name contains a >");
Assert.IsFalse(r.Any(c => c.Name.Contains(" ")), "a name contains a space ");
}
public void GenerateClassesTestVectorVector()
{
var t = TTreeParserCPPTests.CreateTrees.CreateVectorVectorTree();
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "expected only the top level class to come back");
Assert.AreEqual(1, result[0].Items.Count(), "incorrect # of items found for this guy");
Assert.AreEqual("double[][]", result[0].Items.First().ItemType, "improper double vector type!");
CheckSerialization(result, "GenerateClassesTestVectorVector");
}
public void TestComplexObjectNiceObjectFile()
{
// Do we correctly parse an ATLAS MC file?
var f = ROOTNET.NTFile.Open("btagobjs.root", "READ");
var t = f.Get("btagging") as ROOTNET.Interface.NTTree;
var p = new ParseTTree();
var r = p.GenerateClasses(t).ToArray();
f.Close();
Assert.Inconclusive();
}
public void GenerateClassesTestVectorIntAndDoubleAndShort()
{
var t = TTreeParserCPPTests.CreateTrees.CreateVectorTree();
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "expected only the top level class to come back");
string[] possibleTypes = new string[] {
"int[]",
"double[]",
// "short[]", // Not known to root dictionary by default.
"bool[]"
//"float[]" // Weird - it doesn't work either.
};
Assert.IsTrue(result[0].Items.All(i => possibleTypes.Contains(i.ItemType)), "Some responses not correct!");
Assert.AreEqual(possibleTypes.Length, result[0].Items.Select(i => i.ItemType).Distinct().Count(), "Incorrect number of branches found!");
CheckSerialization(result, "GenerateClassesTestVectorIntAndDoubleAndShort");
}
public void TestForSTLVectorPickup()
{
///
/// If a tree contains a vector<float> or similar, then that will generate
/// a #pragma link and link6 line. We need to make sure that our code correctly
/// removes that from the .h file and also adds them to the list of variables
/// that should be added later. This is due to a bug in the ROOT on windows that
/// means unloading the query after using one of these becomes unsafe (and unstable!).
///
var t = TTreeParserCPPTests.CreateTrees.CreateVectorVectorTree();
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.IsNotNull(result[0].ClassesToGenerate, "Expected some classes to generate");
Assert.AreEqual(3, result[0].ClassesToGenerate.Length, "expected 3 different vector class references");
Assert.IsTrue(result[0].ClassesToGenerate.All(c => c.includeFiles == "vector"), "vector needed to be the include file");
var names = new List<string>()
{
"vector<int>", "vector<short>", "vector<bool>"
};
Assert.IsTrue(result[0].ClassesToGenerate.All(c => names.Contains(c.classSpec)), "Some bad class spec names lifted from C++");
var fi = new FileInfo(result[0].NtupleProxyPath);
using (var reader = fi.OpenText())
{
while (!reader.EndOfStream)
{
var line = reader.ReadLine();
Assert.IsFalse(line.Contains("#pragma") && line.Contains("vector"), "line '" + line + "' seems to have a bad pragma");
}
}
}
public void GenerateClassesTestBadNameEvent()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithIntName("event");
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
var item = result[0];
var first = item.Items[0];
var st = first as ItemSimpleType;
Assert.AreEqual("event_", st.Name, "Specialized name not correct");
}
public void TestComplexObjectATLASMCFile()
{
// Do we correctly parse an ATLAS MC file?
var f = ROOTNET.NTFile.Open("EVNT-short.root", "READ");
var t = f.Get("CollectionTree") as ROOTNET.Interface.NTTree;
var p = new ParseTTree();
var r = p.GenerateClasses(t).ToArray();
f.Close();
// Get the root class first.
var mainClass = r.FindClass("CollectionTree");
Assert.IsNotNull(mainClass, "CollectionTree class not found");
Assert.IsFalse(mainClass.IsTClonesArrayClass, "main class isn't a sub-tree class!");
Assert.IsTrue(mainClass.IsTopLevelClass, "main class shoudl be top level class");
// Check that the top level classes are present.
var eventInfo = mainClass.FindItem("EventInfo_p3_McEventInfo");
Assert.IsNotNull(eventInfo, "EventInfo_p3_McEventInfo");
Assert.IsTrue(eventInfo.NotAPointer, "event info not a pointer");
var eventInfoClass = r.FindClass(eventInfo.ItemType);
Assert.IsNotNull(eventInfoClass, string.Format("Event info class {0} wasn't found in the list", eventInfo.ItemType));
var mcCollection = mainClass.FindItem("McEventCollection_p4_GEN_EVENT");
Assert.IsNotNull(mcCollection, "McEventCollection");
Assert.IsTrue(mcCollection.NotAPointer, "mc event collection not a pointer");
var mcCollectionClass = r.FindClass(mcCollection.ItemType);
Assert.IsNotNull(mcCollectionClass, string.Format("Mc Collection class {0} wasn't foudn in the list", mcCollection.ItemType));
// Check that the sub classes are set up here.
Assert.IsFalse(eventInfoClass.IsTClonesArrayClass, "event info sub class setting");
Assert.IsFalse(eventInfoClass.IsTopLevelClass, "event info sub class setting");
Assert.IsFalse(mcCollectionClass.IsTClonesArrayClass, "mc collection class sub-class setting");
Assert.IsFalse(mcCollectionClass.IsTClonesArrayClass, "mc collection class sub-class setting");
// The McEventInfo guy should have one item in it.
Assert.AreEqual(1, eventInfoClass.Items.Count, "# items in the event info class");
Assert.AreEqual("m_AllTheData", eventInfoClass.Items[0].Name, "m_AllTheData name");
Assert.AreEqual("uint[]", eventInfoClass.Items[0].ItemType, "m_AllTheData type");
Assert.IsFalse(eventInfoClass.Items[0].NotAPointer, "array on event info class not a pointer");
// The McCollection has a bunch more stuff in it, namely 3 items.
Assert.AreEqual(3, mcCollectionClass.Items.Count, "# items in the McCollection class");
var rGenParticle = mcCollectionClass.FindItem("m_genParticles");
Assert.IsNotNull(rGenParticle, "gen particle");
Assert.IsTrue(rGenParticle.NotAPointer, "gen particle not a item");
Assert.AreEqual("GenParticle_p4", rGenParticle.ItemType, "gen particle type");
// And go down one more level just to check. :-)
var rGenParticle_p4Class = r.FindClass(rGenParticle.ItemType);
Assert.IsNotNull(rGenParticle_p4Class, "gen particle class spec");
Assert.IsFalse(rGenParticle_p4Class.IsTopLevelClass, "get particle class top level class");
Assert.IsTrue(rGenParticle_p4Class.IsTClonesArrayClass, "gen particle class tclones array");
var rpx = rGenParticle_p4Class.FindItem("m_px");
Assert.IsNotNull(rpx, "m_px in the gen particle class");
Assert.IsTrue(rpx.NotAPointer, "m_px in the gen particle class not a pointer");
Assert.AreEqual("float[]", rpx.ItemType, "m_px type");
Assert.IsInstanceOfType(rpx, typeof(ItemCStyleArray), "m_px should be a c-style array");
var rpxC = rpx as ItemCStyleArray;
Assert.AreEqual(1, rpxC.Indicies.Count, "# of indicies on m_px");
Assert.AreEqual(0, rpxC.Indicies[0].indexPosition, "m_px index boundary");
Assert.AreEqual(false, rpxC.Indicies[0].indexConst, "m_px index boundary");
Assert.AreEqual("implied", rpxC.Indicies[0].indexBoundName, "m_px index boundary");
// getVerticies has a vector in it - so we get a 2D guy. Check that.
var rGenVerticiesClass = r.FindClass("GenVertex_p4");
var rWeights = rGenVerticiesClass.FindItem("m_weights");
Assert.IsNotNull(rWeights, "the m_weights member of genVerticies");
Assert.AreEqual("float[][]", rWeights.ItemType, "m_weights type");
Assert.IsInstanceOfType(rWeights, typeof(ItemCStyleArray), "type of rWeigths guy");
var rWeightsC = rWeights as ItemCStyleArray;
Assert.AreEqual("implied", rWeightsC.Indicies[0].indexBoundName, "rWeights index boundary");
CheckSerialization(r, "TestComplexObjectATLASMCFile");
}
public void TestComplexObjectATLASMCFile()
{
// Do we correctly parse an ATLAS MC file?
var f = ROOTNET.NTFile.Open("EVNT-short.root", "READ");
var t = f.Get("CollectionTree") as ROOTNET.Interface.NTTree;
var p = new ParseTTree();
var r = p.GenerateClasses(t).ToArray();
f.Close();
// Get the root class first.
var mainClass = r.FindClass("CollectionTree");
Assert.IsNotNull(mainClass, "CollectionTree class not found");
Assert.IsFalse(mainClass.IsTClonesArrayClass, "main class isn't a sub-tree class!");
Assert.IsTrue(mainClass.IsTopLevelClass, "main class shoudl be top level class");
// Check that the top level classes are present.
var eventInfo = mainClass.FindItem("EventInfo_p3_McEventInfo");
Assert.IsNotNull(eventInfo, "EventInfo_p3_McEventInfo");
Assert.IsTrue(eventInfo.NotAPointer, "event info not a pointer");
var eventInfoClass = r.FindClass(eventInfo.ItemType);
Assert.IsNotNull(eventInfoClass, string.Format("Event info class {0} wasn't found in the list", eventInfo.ItemType));
var mcCollection = mainClass.FindItem("McEventCollection_p4_GEN_EVENT");
Assert.IsNotNull(mcCollection, "McEventCollection");
Assert.IsTrue(mcCollection.NotAPointer, "mc event collection not a pointer");
var mcCollectionClass = r.FindClass(mcCollection.ItemType);
Assert.IsNotNull(mcCollectionClass, string.Format("Mc Collection class {0} wasn't foudn in the list", mcCollection.ItemType));
// Check that the sub classes are set up here.
Assert.IsFalse(eventInfoClass.IsTClonesArrayClass, "event info sub class setting");
Assert.IsFalse(eventInfoClass.IsTopLevelClass, "event info sub class setting");
Assert.IsFalse(mcCollectionClass.IsTClonesArrayClass, "mc collection class sub-class setting");
Assert.IsFalse(mcCollectionClass.IsTClonesArrayClass, "mc collection class sub-class setting");
// The McEventInfo guy should have one item in it.
Assert.AreEqual(1, eventInfoClass.Items.Count, "# items in the event info class");
Assert.AreEqual("m_AllTheData", eventInfoClass.Items[0].Name, "m_AllTheData name");
Assert.AreEqual("uint[]", eventInfoClass.Items[0].ItemType, "m_AllTheData type");
Assert.IsFalse(eventInfoClass.Items[0].NotAPointer, "array on event info class not a pointer");
// The McCollection has a bunch more stuff in it, namely 3 items.
Assert.AreEqual(3, mcCollectionClass.Items.Count, "# items in the McCollection class");
var rGenParticle = mcCollectionClass.FindItem("m_genParticles");
Assert.IsNotNull(rGenParticle, "gen particle");
Assert.IsTrue(rGenParticle.NotAPointer, "gen particle not a item");
Assert.AreEqual("GenParticle_p4", rGenParticle.ItemType, "gen particle type");
// And go down one more level just to check. :-)
var rGenParticle_p4Class = r.FindClass(rGenParticle.ItemType);
Assert.IsNotNull(rGenParticle_p4Class, "gen particle class spec");
Assert.IsFalse(rGenParticle_p4Class.IsTopLevelClass, "get particle class top level class");
Assert.IsTrue(rGenParticle_p4Class.IsTClonesArrayClass, "gen particle class tclones array");
var rpx = rGenParticle_p4Class.FindItem("m_px");
Assert.IsNotNull(rpx, "m_px in the gen particle class");
Assert.IsTrue(rpx.NotAPointer, "m_px in the gen particle class not a pointer");
Assert.AreEqual("float[]", rpx.ItemType, "m_px type");
Assert.IsInstanceOfType(rpx, typeof(ItemCStyleArray), "m_px should be a c-style array");
var rpxC = rpx as ItemCStyleArray;
Assert.AreEqual(1, rpxC.Indicies.Count, "# of indicies on m_px");
Assert.AreEqual(0, rpxC.Indicies[0].indexPosition, "m_px index boundary");
Assert.AreEqual(false, rpxC.Indicies[0].indexConst, "m_px index boundary");
Assert.AreEqual("implied", rpxC.Indicies[0].indexBoundName, "m_px index boundary");
// getVerticies has a vector in it - so we get a 2D guy. Check that.
var rGenVerticiesClass = r.FindClass("GenVertex_p4");
var rWeights = rGenVerticiesClass.FindItem("m_weights");
Assert.IsNotNull(rWeights, "the m_weights member of genVerticies");
Assert.AreEqual("float[][]", rWeights.ItemType, "m_weights type");
Assert.IsInstanceOfType(rWeights, typeof(ItemCStyleArray), "type of rWeigths guy");
var rWeightsC = rWeights as ItemCStyleArray;
Assert.AreEqual("implied", rWeightsC.Indicies[0].indexBoundName, "rWeights index boundary");
CheckSerialization(r, "TestComplexObjectATLASMCFile");
}
public void TestTreesWithPounds()
{
var f = new ROOTNET.NTFile("atest.root", "READ");
var t = f.Get("##Shapes") as ROOTNET.Interface.NTTree;
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "# of classes");
var obj = result[0];
Assert.IsFalse(obj.NtupleProxyPath.Contains("#"), "proxy path: " + obj.NtupleProxyPath);
Assert.IsFalse(obj.UserInfoPath.Contains("#"), "user path: " + obj.UserInfoPath);
}
public void GenerateClassesTestListOfLeaves()
{
var t = TTreeParserCPPTests.CreateTrees.CreateListOfLeavesTree();
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "expected only the top level class to come back");
Assert.AreEqual(3, result[0].Items.Count(), "incorrect # of items found for this guy");
string[] leafTypes = new string[] { "int", "float", "double" };
Assert.IsTrue(result[0].Items.All(i => leafTypes.Contains(i.ItemType)), "bad item type in the list!");
Assert.AreEqual(leafTypes.Length, result[0].Items.Select(l => l.ItemType).Distinct().Count(), "Bad # of differen tytpes");
CheckSerialization(result, "GenerateClassesTestListOfLeaves");
}
public void GenerateClassesWithCustomClassesAndSubClasses()
{
var f = new ROOTNET.NTFile("ComplexNtupleTestInput.root", "READ");
try {
Assert.IsTrue(f.IsOpen(), "Test file not found");
var t = f.Get("btag") as ROOTNET.Interface.NTTree;
Assert.IsNotNull(t.Name); // Make sure we got the tree correctly
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
// Make sure the global features are right
Assert.AreEqual(1, result.Where(c => c.IsTopLevelClass).Count(), "# of top level classes");
Assert.AreEqual(3, result.Length, "# of classes");
var classMap = result.ToDictionary(r => r.Name, r => r);
Assert.IsTrue(classMap.ContainsKey("btag"), "btag class present");
Assert.IsTrue(classMap.ContainsKey("MuonInBJet"), "MuonInBJet class present");
Assert.IsTrue(classMap.ContainsKey("BTagJet"), "btag class present");
} finally
{
f.Close();
}
}
public void TestGenerateClasses2DCStyleVector()
{
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithCStyleVectorArray();
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "only one top level class here");
var citem = result[0];
Assert.AreEqual(1, citem.Items.Count, "# of items");
var item = citem.Items[0];
Assert.IsInstanceOfType(item, typeof(ItemCStyleArray), "item type");
}
public void GenerateAndGC()
{
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithConstIndexedSimpleVector(20);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result_itr = p.GenerateClasses(t);
GC.Collect(GC.MaxGeneration);
var result = result_itr.ToArray();
CheckSerialization(result, "GenerateClassesTestCPPName");
}
public void GenerateClassesTestCPPName2D()
{
// Test an array that is 2D C style
var t = TTreeParserCPPTests.CreateTrees.CreateTreeWithIndexedSimpleVector2D(20);
var p = new ParseTTree();
p.ProxyGenerationLocation = new DirectoryInfo(".");
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "should only be top level class");
var item = result[0];
Assert.AreEqual(1, item.Items.Count(), "# of items found");
var item1 = item.Items[0];
Assert.AreEqual("int[][]", item1.ItemType, "2d array type");
Assert.AreEqual("arr", item1.Name, "arr name");
var i2asA = item1 as ItemCStyleArray;
Assert.AreEqual(2, i2asA.Indicies.Count, "# of indicies");
Assert.AreEqual("5", i2asA.Indicies[0].indexBoundName, "index name");
Assert.AreEqual(0, i2asA.Indicies[0].indexPosition, "index name");
Assert.IsTrue(i2asA.Indicies[0].indexConst, "index const");
Assert.AreEqual("4", i2asA.Indicies[1].indexBoundName, "index name");
Assert.AreEqual(0, i2asA.Indicies[1].indexPosition, "index name");
Assert.IsTrue(i2asA.Indicies[1].indexConst, "index const");
CheckSerialization(result, "GenerateClassesTestCPPName");
}
public void TestGenerateWithStringVectorTypes()
{
var t = TTreeParserCPPTests.CreateTrees.CreateWithStringTypes();
var p = new ParseTTree();
var result = p.GenerateClasses(t).ToArray();
Assert.AreEqual(1, result.Length, "Empty ntuple should have fired");
Assert.AreEqual(0, result[0].Items.Count, "Expected nothing to be pasred");
}