BeamSpotWrite2DB.cc

Go to the documentation of this file.

// C++ standard
#include <string>
// CMS
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "RecoVertex/BeamSpotProducer/interface/BeamSpotWrite2DB.h"
#include "RecoVertex/BeamSpotProducer/interface/BSFitter.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "CondFormats/BeamSpotObjects/interface/BeamSpotObjects.h"

BeamSpotWrite2DB::BeamSpotWrite2DB(const edm::ParameterSet& iConfig) {
  fasciiFileName = iConfig.getUntrackedParameter<std::string>("OutputFileName");

  fasciiFile.open(fasciiFileName.c_str());
}

BeamSpotWrite2DB::~BeamSpotWrite2DB() {}

void BeamSpotWrite2DB::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {}

void BeamSpotWrite2DB::beginJob() {}

void BeamSpotWrite2DB::endJob() {
  std::cout << " Read beam spot data from text file: " << fasciiFileName << std::endl;
  std::cout << " please see plugins/BeamSpotWrite2DB.cc for format of text file." << std::endl;
  /*
    std::cout << " Content of the file is expected to have this format with the first column as a keyword:" << std::endl;
    std::cout << " x\n y\n z\n sigmaZ\n dxdz\n dydz\n beamWidthX\n beamWidthY" << std::endl;
    for (int i =0; i<7; i++) {
        for (int j=0; j<7; j++ ) {
            
            std::cout << " cov["<<i<<"]["<<j<<"] cov["<<i<<"]["<<j<<"] cov["<<i<<"]["<<j<<"] cov["<<i<<"]["<<j<<"] cov["<<i<<"]["<<j<<"] cov["<<j<<"]["<<j<<"] cov["<<i<<"]["<<j<<"]" << std::endl;
        }
    }
    */

  // extract from file
  double x, y, z, sigmaZ, dxdz, dydz, beamWidthX, beamWidthY, emittanceX, emittanceY, betastar;
  std::string tag;
  double cov[7][7];
  int type;

  fasciiFile >> tag >> type;
  fasciiFile >> tag >> x;
  fasciiFile >> tag >> y;
  fasciiFile >> tag >> z;
  fasciiFile >> tag >> sigmaZ;
  fasciiFile >> tag >> dxdz;
  fasciiFile >> tag >> dydz;
  fasciiFile >> tag >> beamWidthX;
  fasciiFile >> tag >> beamWidthY;
  fasciiFile >> tag >> cov[0][0] >> cov[0][1] >> cov[0][2] >> cov[0][3] >> cov[0][4] >> cov[0][5] >> cov[0][6];
  fasciiFile >> tag >> cov[1][0] >> cov[1][1] >> cov[1][2] >> cov[1][3] >> cov[1][4] >> cov[1][5] >> cov[1][6];
  fasciiFile >> tag >> cov[2][0] >> cov[2][1] >> cov[2][2] >> cov[2][3] >> cov[2][4] >> cov[2][5] >> cov[2][6];
  fasciiFile >> tag >> cov[3][0] >> cov[3][1] >> cov[3][2] >> cov[3][3] >> cov[3][4] >> cov[3][5] >> cov[3][6];
  fasciiFile >> tag >> cov[4][0] >> cov[4][1] >> cov[4][2] >> cov[4][3] >> cov[4][4] >> cov[4][5] >> cov[4][6];
  fasciiFile >> tag >> cov[5][0] >> cov[5][1] >> cov[5][2] >> cov[5][3] >> cov[5][4] >> cov[5][5] >> cov[5][6];
  fasciiFile >> tag >> cov[6][0] >> cov[6][1] >> cov[6][2] >> cov[6][3] >> cov[6][4] >> cov[6][5] >> cov[6][6];
  fasciiFile >> tag >> emittanceX;
  fasciiFile >> tag >> emittanceY;
  fasciiFile >> tag >> betastar;

  BeamSpotObjects* abeam = new BeamSpotObjects();

  abeam->SetType(type);
  abeam->SetPosition(x, y, z);
  abeam->SetSigmaZ(sigmaZ);
  abeam->Setdxdz(dxdz);
  abeam->Setdydz(dydz);
  abeam->SetBeamWidthX(beamWidthX);
  abeam->SetBeamWidthY(beamWidthY);
  abeam->SetEmittanceX(emittanceX);
  abeam->SetEmittanceY(emittanceY);
  abeam->SetBetaStar(betastar);

  for (int i = 0; i < 7; ++i) {
    for (int j = 0; j < 7; ++j) {
      abeam->SetCovariance(i, j, cov[i][j]);
    }
  }

  std::cout << " write results to DB..." << std::endl;

  edm::Service<cond::service::PoolDBOutputService> poolDbService;
  if (poolDbService.isAvailable()) {
    std::cout << "poolDBService available" << std::endl;
    if (poolDbService->isNewTagRequest("BeamSpotObjectsRcd")) {
      std::cout << "new tag requested" << std::endl;
      poolDbService->createNewIOV<BeamSpotObjects>(
          abeam, poolDbService->beginOfTime(), poolDbService->endOfTime(), "BeamSpotObjectsRcd");
    } else {
      std::cout << "no new tag requested" << std::endl;
      poolDbService->appendSinceTime<BeamSpotObjects>(abeam, poolDbService->currentTime(), "BeamSpotObjectsRcd");
    }
  }

  std::cout << "[BeamSpotWrite2DB] endJob done \n" << std::endl;
}

//define this as a plug-in
DEFINE_FWK_MODULE(BeamSpotWrite2DB);