@Override
public void finalStatistics(EvolutionState state, int result) {
super.finalStatistics(state, result);
if (compress) {
try {
taos.close();
} catch (IOException ex) {
Logger.getLogger(SampleSolutionsStat.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
@Override
public void setup(EvolutionState state, Parameter base) {
super.setup(state, base);
outFile = state.parameters.getFile(base.push(P_FILE), null);
outFile = new File(outFile.getParent(), jobPrefix + outFile.getName());
compress = state.parameters.getBoolean(base.push(P_COMPRESS), null, true);
if (compress) {
try {
taos =
new TarArchiveOutputStream(
new GZIPOutputStream(new BufferedOutputStream(new FileOutputStream(outFile))));
} catch (IOException ex) {
Logger.getLogger(SampleSolutionsStat.class.getName()).log(Level.SEVERE, null, ex);
}
}
if (!compress && !outFile.exists()) {
outFile.mkdirs();
}
sampleSize = state.parameters.getInt(base.push(P_SAMPLE_SIZE), null);
}
@Override
public void postEvaluationStatistics(EvolutionState state) {
super.postEvaluationStatistics(state);
if (taos == null && compress) { // can happen in case of resuming from checkpoint
taos = SolutionPersistence.reopen(outFile);
}
int[] subs = new int[sampleSize];
int[] inds = new int[sampleSize];
for (int i = 0; i < sampleSize; ) {
int sub = state.random[0].nextInt(state.population.subpops.length);
int ind = state.random[0].nextInt(state.population.subpops[sub].individuals.length);
for (int j = 0; j < i; j++) { // check if this individual was already picked
if (subs[j] == sub && inds[j] == ind) {
continue;
}
}
subs[i] = sub;
inds[i] = ind;
i++;
}
for (int i = 0; i < sampleSize; i++) {
Individual ind = state.population.subpops[subs[i]].individuals[inds[i]];
PersistentSolution c =
SolutionPersistence.createPersistentController(state, ind, subs[i], inds[i]);
try {
if (compress) {
SolutionPersistence.writeSolutionToTar(c, taos);
} else {
SolutionPersistence.writeSolutionInFolder(c, outFile);
}
} catch (IOException e) {
e.printStackTrace();
}
}
}