CondTools/Ecal/src/EcalAlignmentXMLTranslator.cc - master - githubmirror/cmssw

CondTools/Ecal/src/EcalAlignmentXMLTranslator.cc
128 строк4 KB
Christopher Jones
Move Xerces lock to Utilities/Xerces
15 июн 2019, 14:35
15 июн 2019, 14:35d1670cc
#include "CondTools/Ecal/interface/EcalAlignmentXMLTranslator.h"
#include "CondTools/Ecal/interface/DOMHelperFunctions.h"
#include "CondFormats/Alignment/interface/Alignments.h"
#include "Utilities/Xerces/interface/Xerces.h"
#include "Utilities/Xerces/interface/XercesStrUtils.h"
#include <xercesc/dom/DOMNode.hpp>
#include <xercesc/dom/DOM.hpp>
#include <xercesc/parsers/XercesDOMParser.hpp>
#include <xercesc/util/XMLString.hpp>
#include <xercesc/sax/SAXException.hpp>
#include <xercesc/framework/LocalFileFormatTarget.hpp>
#include <iostream>
#include <sstream>
#include <fstream>

using namespace XERCES_CPP_NAMESPACE;
using namespace xuti;
using namespace std;

int EcalAlignmentXMLTranslator::writeXML(const string& filename,
                                         const EcalCondHeader& header,
                                         const Alignments& record) {
  cms::concurrency::xercesInitialize();

  fstream fs(filename.c_str(), ios::out);
  fs << dumpXML(header, record);

  cms::concurrency::xercesTerminate();

  return 0;
}

string EcalAlignmentXMLTranslator::dumpXML(const EcalCondHeader& header, const Alignments& record) {
  unique_ptr<DOMImplementation> impl(DOMImplementationRegistry::getDOMImplementation(cms::xerces::uStr("LS").ptr()));

  DOMLSSerializer* writer = impl->createLSSerializer();
  if (writer->getDomConfig()->canSetParameter(XMLUni::fgDOMWRTFormatPrettyPrint, true))
    writer->getDomConfig()->setParameter(XMLUni::fgDOMWRTFormatPrettyPrint, true);

  DOMDocumentType* doctype = impl->createDocumentType(cms::xerces::uStr("XML").ptr(), nullptr, nullptr);
  DOMDocument* doc = impl->createDocument(nullptr, cms::xerces::uStr(AlignmentConstant_tag.c_str()).ptr(), doctype);
  DOMElement* root = doc->getDocumentElement();

  xuti::writeHeader(root, header);

  for (vector<AlignTransform>::const_iterator it = record.m_align.begin(); it != record.m_align.end(); it++) {
    int Id = (*it).rawId();
    int sub = (Id >> 24) & 0xF;
    stringstream subdet;
    if (sub == 2) {
      subdet << "EB";
      int SM;
      int phi = Id & 0x1FF;
      //    int eta = (Id>>9)&0x3F;
      int side = (Id >> 16) & 1;
      if (side == 0) {
        subdet << "-";
        SM = (phi + 19) / 20;
      } else {
        subdet << "+";
        SM = phi / 20;
      }
      if (SM < 10)
        subdet << "0" << SM;
      else
        subdet << SM;
    } else if (sub == 4) {
      subdet << "EE";
      //      int y = Id&0x7F;
      int x = (Id >> 7) & 0x7F;
      //      int Dee = x/70 + 1;
      int side = (Id >> 14) & 1;
      if (side == 0) {
        subdet << "-";
      } else {
        subdet << "+";
      }
      //      subdet << "0" << Dee;
      if (x == 20)
        subdet << "F";
      else if (x == 70)
        subdet << "N";
      else
        cout << " strange value for x " << x << endl;  // should never occur
    } else if (sub == 6) {
      subdet << "ES";
      //      int strip = Id&0x3F;
      int x = (Id >> 6) & 0x3F;
      //      int y = (Id>>12)&0x3F;
      int plane = (Id >> 18) & 1;
      int side = (Id >> 19) & 1;
      if (side == 0)
        subdet << "-";
      else
        subdet << "+";
      if (plane)
        subdet << "F";
      else
        subdet << "R";
      if (x / 30)
        subdet << "F";
      else
        subdet << "N";
    } else
      cout << " problem sub = " << sub << endl;
    cout << (*it).rawId() << " " << (*it).rotation().getPhi() << " " << (*it).rotation().getTheta() << " "
         << (*it).rotation().getPsi() << " " << (*it).translation().x() << " " << (*it).translation().y() << " "
         << (*it).translation().z() << endl;
    uint32_t rawid = (*it).rawId();
    DOMElement* cellnode = root->getOwnerDocument()->createElement(cms::xerces::uStr(Cell_tag.c_str()).ptr());
    root->appendChild(cellnode);
    cellnode->setAttribute(cms::xerces::uStr(subdet_tag.c_str()).ptr(), cms::xerces::uStr(subdet.str().c_str()).ptr());
    xuti::WriteNodeWithValue(cellnode, id_tag, rawid);
    xuti::WriteNodeWithValue(cellnode, x_tag, (*it).translation().x());
    xuti::WriteNodeWithValue(cellnode, y_tag, (*it).translation().y());
    xuti::WriteNodeWithValue(cellnode, z_tag, (*it).translation().z());
    xuti::WriteNodeWithValue(cellnode, Phi_tag, (*it).rotation().getPhi());
    xuti::WriteNodeWithValue(cellnode, Theta_tag, (*it).rotation().getTheta());
    xuti::WriteNodeWithValue(cellnode, Psi_tag, (*it).rotation().getPsi());
  }

  std::string dump = cms::xerces::toString(writer->writeToString(root));
  doc->release();
  doctype->release();
  writer->release();

  return dump;
}