JFixedCan.cc

Go to the documentation of this file.

 
#include <string>
#include <iostream>
#include <iomanip>
 
#include "evt/Head.hh"
#include "evt/Evt.hh"
 
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JMonteCarloFileSupportkit.hh"
#include "JSupport/JSupport.hh"
#include "JSystem/JDate.hh" //for getDate
#include "JROOT/JROOTClassSelector.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \file
 * Auxiliary program to shift the tracks to a given vector (horizontal shifts only!).
 *
 * \author vkulikovskiy
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  JMultipleFileScanner<Evt> inputFile;
  JFileRecorder       <JAAnetTypes_t> outputFile;
  JLimit_t&           numberOfEvents = inputFile.getLimit();
  int                 debug;
  double xshift;
  double yshift;
 
  try {
 
    JParser<> zap("Auxiliary program to convert data formats.");
 
    zap['f'] = make_field(inputFile);
    zap['o'] = make_field(outputFile);
    zap['n'] = make_field(numberOfEvents)      = JLimit::max();
    zap['x'] = make_field(xshift) = 0.0;
    zap['y'] = make_field(yshift) = 0.0;
    zap['d'] = make_field(debug)               = 1;
 
    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }
 
 
  outputFile.open();
 
  Vec  center(xshift,yshift,0);
 
  if (debug >= 3)
        NOTICE("Offset applied to true tracks is: " << center << endl);
 
  Head header;
 
  try {
 
    header = getHeader(inputFile);
 
    JHead buffer(header);
 
    JAANET::simul shiftinfo;
    shiftinfo.program = "JShift";
    shiftinfo.version = getGITVersion();
    shiftinfo.date    = getDate();
    shiftinfo.time    = getTime();
    buffer.simul.push_back(shiftinfo);
    buffer.push(&JHead::simul);  //simul tag is registered with this. Actually there should be already simul info registered from generator, so we don't really need it
 
    bool can_found      = buffer.pull(&JHead::can)      != buffer.end();
    bool fixedcan_found = buffer.pull(&JHead::fixedcan) != buffer.end();
 
    if (debug >= 3) {
      if (can_found) {
        NOTICE("can found" << endl);
      }
      if (fixedcan_found) {
        NOTICE("fixedcan found" << endl);
      }
    }
 
    if (can_found and !(fixedcan_found)) {
      buffer.fixedcan.xcenter = xshift;
      buffer.fixedcan.ycenter = yshift;
      buffer.fixedcan.zmin =  buffer.can.zmin;
      buffer.fixedcan.zmax =  buffer.can.zmax;
      buffer.fixedcan.radius =  buffer.can.r;
      buffer.erase(&JHead::can);
      buffer.push(&JHead::fixedcan);
    } else if (fixedcan_found and !(can_found)) {
      buffer.fixedcan.xcenter = buffer.fixedcan.xcenter + xshift;
      buffer.fixedcan.ycenter = buffer.fixedcan.ycenter + yshift;
    } else if (!(can_found) and !(fixedcan_found)) {
      FATAL("Neither can nor fixedcan is found" << endl);
    } else {
      FATAL("Both can and fixedcan are found" << endl);
    }
 
    copy(buffer, header);
  }
  catch(const JException& error) {
    FATAL(error);
  }
  outputFile.put(header);
 
 
  while (inputFile.hasNext()) {
 
    Evt* evt = inputFile.next();                              // input
 
    for (vector<Trk>::iterator track = evt->mc_trks.begin(); track != evt->mc_trks.end(); ++track) {
      track->pos += center;
    }
 
    outputFile.put(*evt);
  }
 
  outputFile.close();
}