portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-09 21:50:45 +01:00
Code Releases Activity

Hadrons: less code duplication in eigenpack IO

Browse Source
This commit is contained in:
Antonin Portelli 2018-10-01 20:15:21 +01:00
parent 21bba95909
commit 8fbb27ce13
1 changed files with 53 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
Hadrons/EigenPack.hpp
View File

@ -39,6 +39,13 @@ BEGIN_HADRONS_NAMESPACE
#define HADRONS_DEFAULT_LANCZOS_NBASIS 60 #define HADRONS_DEFAULT_LANCZOS_NBASIS 60
#endif #endif
#define HADRONS_DUMP_EP_METADATA \
LOG(Message) << "Eigenpack metadata:" << std::endl;\
LOG(Message) << "* operator" << std::endl;\
LOG(Message) << record.operatorXml << std::endl;\
LOG(Message) << "* solver" << std::endl;\
LOG(Message) << record.solverXml << std::endl;
template <typename F> template <typename F>
class EigenPack class EigenPack
{ {
@ -81,11 +88,16 @@ public:
for(int k = 0; k < evec.size(); ++k) for(int k = 0; k < evec.size(); ++k)
{ {
basicReadSingle(evec[k], eval[k], evecFilename(fileStem, k, traj), k); basicReadSingle(evec[k], eval[k], evecFilename(fileStem, k, traj), k);
if (k == 0)
{
HADRONS_DUMP_EP_METADATA;
}
} }
} }
else else
{ {
basicRead(evec, eval, evecFilename(fileStem, -1, traj), evec.size()); basicRead(evec, eval, evecFilename(fileStem, -1, traj), evec.size());
HADRONS_DUMP_EP_METADATA;
} }
} }
@ -118,26 +130,38 @@ protected:
} }
} }
template <typename T> void readHeader(ScidacReader &binReader)
void basicRead(std::vector<T> &evec, std::vector<RealD> &eval,
const std::string filename, const unsigned int size)
{ {
ScidacReader binReader;
std::string recordXml; std::string recordXml;
binReader.open(filename);
binReader.readLimeObject(recordXml, SCIDAC_FILE_XML); binReader.readLimeObject(recordXml, SCIDAC_FILE_XML);
XmlReader xmlReader(recordXml, true, "eigenPackPar"); XmlReader xmlReader(recordXml, true, "eigenPackPar");
xmlReader.push(); xmlReader.push();
xmlReader.readCurrentSubtree(record.operatorXml); xmlReader.readCurrentSubtree(record.operatorXml);
xmlReader.nextElement(); xmlReader.nextElement();
xmlReader.readCurrentSubtree(record.solverXml); xmlReader.readCurrentSubtree(record.solverXml);
}
template <typename T>
void readElement(T &evec, VecRecord &vecRecord, ScidacReader &binReader)
{
binReader.readScidacFieldRecord(evec, vecRecord);
}
template <typename T>
void basicRead(std::vector<T> &evec, std::vector<RealD> &eval,
const std::string filename, const unsigned int size)
{
ScidacReader binReader;
binReader.open(filename);
readHeader(binReader);
for(int k = 0; k < size; ++k) for(int k = 0; k < size; ++k)
{ {
VecRecord vecRecord; VecRecord vecRecord;
LOG(Message) << "Reading eigenvector " << k << std::endl; LOG(Message) << "Reading eigenvector " << k << std::endl;
binReader.readScidacFieldRecord(evec[k], vecRecord); readElement(evec[k], vecRecord, binReader);
if (vecRecord.index != k) if (vecRecord.index != k)
{ {
HADRONS_ERROR(Io, "Eigenvector " + std::to_string(k) + " has a" HADRONS_ERROR(Io, "Eigenvector " + std::to_string(k) + " has a"
@ -155,17 +179,11 @@ protected:
{ {
ScidacReader binReader; ScidacReader binReader;
VecRecord vecRecord; VecRecord vecRecord;
std::string recordXml;
binReader.open(filename); binReader.open(filename);
binReader.readLimeObject(recordXml, SCIDAC_FILE_XML); readHeader(binReader);
XmlReader xmlReader(recordXml, true, "eigenPackPar");
xmlReader.push();
xmlReader.readCurrentSubtree(record.operatorXml);
xmlReader.nextElement();
xmlReader.readCurrentSubtree(record.solverXml);
LOG(Message) << "Reading eigenvector " << index << std::endl; LOG(Message) << "Reading eigenvector " << index << std::endl;
binReader.readScidacFieldRecord(evec, vecRecord); readElement(evec, vecRecord, binReader);
if (vecRecord.index != index) if (vecRecord.index != index)
{ {
HADRONS_ERROR(Io, "Eigenvector " + std::to_string(index) + " has a" HADRONS_ERROR(Io, "Eigenvector " + std::to_string(index) + " has a"
@ -176,18 +194,30 @@ protected:
binReader.close(); binReader.close();
} }
void writeHeader(ScidacWriter &binWriter)
{
XmlWriter xmlWriter("", "eigenPackPar");
xmlWriter.pushXmlString(record.operatorXml);
xmlWriter.pushXmlString(record.solverXml);
binWriter.writeLimeObject(1, 1, xmlWriter, "parameters", SCIDAC_FILE_XML);
}
template <typename T>
void writeElement(ScidacWriter &binWriter, T &evec, VecRecord &vecRecord)
{
binWriter.writeScidacFieldRecord(evec, vecRecord, DEFAULT_ASCII_PREC);
}
template <typename T> template <typename T>
void basicWrite(const std::string filename, std::vector<T> &evec, void basicWrite(const std::string filename, std::vector<T> &evec,
const std::vector<RealD> &eval, const unsigned int size) const std::vector<RealD> &eval, const unsigned int size)
{ {
ScidacWriter binWriter(evec[0]._grid->IsBoss()); ScidacWriter binWriter(evec[0]._grid->IsBoss());
XmlWriter xmlWriter("", "eigenPackPar");
makeFileDir(filename, evec[0]._grid); makeFileDir(filename, evec[0]._grid);
xmlWriter.pushXmlString(record.operatorXml);
xmlWriter.pushXmlString(record.solverXml);
binWriter.open(filename); binWriter.open(filename);
binWriter.writeLimeObject(1, 1, xmlWriter, "parameters", SCIDAC_FILE_XML); writeHeader(binWriter);
for(int k = 0; k < size; ++k) for(int k = 0; k < size; ++k)
{ {
VecRecord vecRecord; VecRecord vecRecord;
@ -195,7 +225,7 @@ protected:
vecRecord.index = k; vecRecord.index = k;
vecRecord.eval = eval[k]; vecRecord.eval = eval[k];
LOG(Message) << "Writing eigenvector " << k << std::endl; LOG(Message) << "Writing eigenvector " << k << std::endl;
binWriter.writeScidacFieldRecord(evec[k], vecRecord, DEFAULT_ASCII_PREC); writeElement(binWriter, evec[k], vecRecord);
} }
binWriter.close(); binWriter.close();
} }
@ -205,18 +235,15 @@ protected:
const RealD eval, const unsigned int index) const RealD eval, const unsigned int index)
{ {
ScidacWriter binWriter(evec._grid->IsBoss()); ScidacWriter binWriter(evec._grid->IsBoss());
XmlWriter xmlWriter("", "eigenPackPar");
VecRecord vecRecord; VecRecord vecRecord;
makeFileDir(filename, evec._grid); makeFileDir(filename, evec._grid);
xmlWriter.pushXmlString(record.operatorXml);
xmlWriter.pushXmlString(record.solverXml);
binWriter.open(filename); binWriter.open(filename);
binWriter.writeLimeObject(1, 1, xmlWriter, "parameters", SCIDAC_FILE_XML); writeHeader(binWriter);
vecRecord.index = index; vecRecord.index = index;
vecRecord.eval = eval; vecRecord.eval = eval;
LOG(Message) << "Writing eigenvector " << index << std::endl; LOG(Message) << "Writing eigenvector " << index << std::endl;
binWriter.writeScidacFieldRecord(evec, vecRecord, DEFAULT_ASCII_PREC); writeElement(binWriter, evec, vecRecord);
binWriter.close(); binWriter.close();
} }
}; };
@ -330,6 +357,8 @@ using CoarseFermionEigenPack = CoarseEigenPack<
typename FImpl::SiteComplex, typename FImpl::SiteComplex,
nBasis>::CoarseField>; nBasis>::CoarseField>;
#undef HADRONS_DUMP_EP_METADATA
END_HADRONS_NAMESPACE END_HADRONS_NAMESPACE
#endif // Hadrons_EigenPack_hpp_ #endif // Hadrons_EigenPack_hpp_