void FlattenDataTable(int serverClassIndex)
{
SendTable table = DataTables[ServerClasses[serverClassIndex].DataTableID];
CurrentExcludes.Clear();
CurrentBaseclasses = new List <ServerClass> (); //NOT .clear because we use *this* reference
//LITERALLY 3 lines later. @main--, this is warning for you.
GatherExcludesAndBaseclasses(table, true);
ServerClasses [serverClassIndex].BaseClasses = CurrentBaseclasses;
GatherProps(table, serverClassIndex, "");
var flattenedProps = ServerClasses[serverClassIndex].FlattenedProps;
List <int> priorities = new List <int>();
priorities.Add(64);
priorities.AddRange(flattenedProps.Select(a => a.Prop.Priority).Distinct());
priorities.Sort();
int start = 0;
for (int priorityIndex = 0; priorityIndex < priorities.Count; priorityIndex++)
{
int priority = priorities[priorityIndex];
while (true)
{
int currentProp = start;
while (currentProp < flattenedProps.Count)
{
SendTableProperty prop = flattenedProps[currentProp].Prop;
if (prop.Priority == priority || (priority == 64 && prop.Flags.HasFlagFast(SendPropertyFlags.ChangesOften)))
{
if (start != currentProp)
{
FlattenedPropEntry temp = flattenedProps[start];
flattenedProps[start] = flattenedProps[currentProp];
flattenedProps[currentProp] = temp;
}
start++;
break;
}
currentProp++;
}
if (currentProp == flattenedProps.Count)
{
break;
}
}
}
}
void GatherProps(SendTable table, int serverClassIndex, string prefix)
{
List <FlattenedPropEntry> tmpFlattenedProps = new List <FlattenedPropEntry>();
GatherProps_IterateProps(table, serverClassIndex, tmpFlattenedProps, prefix);
List <FlattenedPropEntry> flattenedProps = ServerClasses[serverClassIndex].FlattenedProps;
flattenedProps.AddRange(tmpFlattenedProps);
}
void GatherProps(SendTable table, int serverClassIndex, string prefix)
{
List <FlattenedPropEntry> tmpFlattenedProps = new List <FlattenedPropEntry>();
GatherProps_IterateProps(table, serverClassIndex, tmpFlattenedProps, prefix);
List <FlattenedPropEntry> flattenedProps = ServerClasses[serverClassIndex].FlattenedProps;
tmpFlattenedProps.RemoveAll(prop => flattenedProps.FirstOrDefault(ep => ep.PropertyName == prop.PropertyName) != null);
flattenedProps.AddRange(tmpFlattenedProps);
}
public void ParsePacket(IBitStream bitstream)
{
while (true)
{
var type = (SVC_Messages)bitstream.ReadProtobufVarInt();
if (type != SVC_Messages.svc_SendTable)
{
throw new Exception("Expected SendTable, got " + type);
}
var size = bitstream.ReadProtobufVarInt();
bitstream.BeginChunk(size * 8);
var sendTable = new SendTable(bitstream);
bitstream.EndChunk();
if (sendTable.IsEnd)
{
break;
}
DataTables.Add(sendTable);
}
int serverClassCount = checked ((int)bitstream.ReadInt(16));
for (int i = 0; i < serverClassCount; i++)
{
ServerClass entry = new ServerClass();
entry.ClassID = checked ((int)bitstream.ReadInt(16));
if (entry.ClassID > serverClassCount)
{
throw new Exception("Invalid class index");
}
entry.Name = bitstream.ReadDataTableString();
entry.DTName = bitstream.ReadDataTableString();
entry.DataTableID = DataTables.FindIndex(a => a.Name == entry.DTName);
if (ServerClasses.FirstOrDefault(sc => sc.Name == entry.Name && sc.DTName == entry.DTName) == null)
{
ServerClasses.Add(entry);
}
}
for (int i = 0; i < serverClassCount; i++)
{
FlattenDataTable(i);
}
}
void GatherProps_IterateProps(SendTable table, int ServerClassIndex, List <FlattenedPropEntry> flattenedProps, string prefix)
{
for (int i = 0; i < table.Properties.Count; i++)
{
SendTableProperty property = table.Properties[i];
if (property.Flags.HasFlagFast(SendPropertyFlags.InsideArray) || property.Flags.HasFlagFast(SendPropertyFlags.Exclude) || IsPropExcluded(table, property))
{
continue;
}
if (property.Type == SendPropertyType.DataTable)
{
SendTable subTable = GetTableByName(property.DataTableName);
if (property.Flags.HasFlagFast(SendPropertyFlags.Collapsible))
{
//we don't prefix Collapsible stuff, since it is just derived mostly
GatherProps_IterateProps(subTable, ServerClassIndex, flattenedProps, prefix);
}
else
{
//We do however prefix everything else
string nfix = prefix + ((property.Name.Length > 0) ? property.Name + "." : "");
GatherProps(subTable, ServerClassIndex, nfix);
}
}
else
{
if (property.Type == SendPropertyType.Array)
{
flattenedProps.Add(new FlattenedPropEntry(prefix + property.Name, property, table.Properties[i - 1]));
}
else
{
flattenedProps.Add(new FlattenedPropEntry(prefix + property.Name, property, null));
}
}
}
}
void GatherExcludesAndBaseclasses(SendTable sendTable, bool collectBaseClasses)
{
CurrentExcludes.AddRange(
sendTable.Properties
.Where(a => a.Flags.HasFlagFast(SendPropertyFlags.Exclude))
.Select(a => new ExcludeEntry(a.Name, a.DataTableName, sendTable.Name))
);
foreach (var prop in sendTable.Properties.Where(a => a.Type == SendPropertyType.DataTable))
{
if (collectBaseClasses && prop.Name == "baseclass")
{
GatherExcludesAndBaseclasses(GetTableByName(prop.DataTableName), true);
CurrentBaseclasses.Add(FindByDTName(prop.DataTableName));
}
else
{
GatherExcludesAndBaseclasses(GetTableByName(prop.DataTableName), false);
}
}
}
bool IsPropExcluded(SendTable table, SendTableProperty prop)
{
return(CurrentExcludes.Exists(a => table.Name == a.DTName && prop.Name == a.VarName));
}
