public void EncodeInteractionFeature(string fileName)
{
dynamic h5py = Py.Import("h5py");
StreamReader file = new StreamReader(fileName);
PyList[] features = new PyList[6];
for (int i = 0; i < 6; i++)
{
features[i] = new PyList();
}
PyList targetList = new PyList();
string line = null;
string[] featureNames = null;
int count = 0;
while ((line = file.ReadLine()) != null)
{
GameRecord gameRecord = JsonConvert.DeserializeObject <GameRecord>(line);
List <StateKeyInfo> playSec = new List <StateKeyInfo>();
int lastEntity = 1000;
foreach (StateKeyInfo stKeyInfo in gameRecord.playSec)
{
int result = (stKeyInfo.attackPlayer.turn == gameRecord.result) ? 1 : 0;
PlayerKeyInfo p1Info = stKeyInfo.attackPlayer;
PlayerKeyInfo p2Info = stKeyInfo.defensePlayer;
bool isOwnTurn = p1Info.turn == 0 ? true : false;
Playfield tempPf = null;
if (isOwnTurn)
{
tempPf = new Playfield(lastEntity, isOwnTurn, p1Info, p2Info);
}
else
{
tempPf = new Playfield(lastEntity, isOwnTurn, p2Info, p1Info);
}
Player mPlayer = tempPf.getCurrentPlayer(true);
Player ePlayer = tempPf.getCurrentPlayer(false);
StateFeature interFeature = Featurization.interactionFeaturization(tempPf);
foreach (PlayerKeyInfo.ActionKeyInfo actionKeyInfo in p1Info.playedActionJsonList)
{
Action action = CreateActionFromInfo(tempPf, actionKeyInfo);
int target = 0;
if (action.actionType == actionEnum.playcard)
{
target = FeatureConst.Instance.cardIdxDict[action.card.card.name];
}
else if (action.actionType == actionEnum.useHeroPower)
{
target = FeatureConst.Instance.cardIdxDict[CardDB.cardName.fireblast];
}
tempPf.getNextEntity();
tempPf.doAction(action);
PyList[] featureList = interFeature.getPyData();
if (featureNames == null)
{
featureNames = interFeature.getFeatureNames();
}
for (int i = 0; i < interFeature.numFeatures; i++)
{
PythonUtils.AppendRecycle(features[i], featureList[i]);
}
PythonUtils.AppendRecycle(targetList, FeatureConst.Instance.pyIntMap[target]);
interFeature = Featurization.interactionFeaturization(tempPf);
}
lastEntity = tempPf.getNextEntity() + 1;
}
count++;
if (count % 1000 == 0)
{
Console.WriteLine(count);
}
}
string outFileName = fileName + "inter.hdf5";
dynamic outFile = h5py.File(outFileName, "w");
for (int i = 0; i < features.Length; i++)
{
outFile.create_dataset(featureNames[i], Py.kw("data", features[i]));
}
outFile.create_dataset("Target", Py.kw("data", targetList));
outFile.close();
}
