software/JAcoustics/JCreep.cc

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
 
#include "JSupport/JMultipleFileScanner.hh"
 
#include "JTools/JQuantile.hh"
 
#include "JAcoustics/JEvt.hh"
#include "JAcoustics/JSupport.hh"
 
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
 
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \file
 *
 * Auxialiry program to determine average creep of strings.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  JMultipleFileScanner<JEvt> inputFile;
  JLimit_t&    numberOfEvents = inputFile.getLimit();
  string       detectorFile;
  bool         overwriteDetector;
  int          debug;
 
  try {
 
    JParser<> zap("Auxialiry program to determine average creep of strings.");
 
    zap['f'] = make_field(inputFile,         "input file (output of JKatoomba[.sh])");
    zap['n'] = make_field(numberOfEvents)   = JLimit::max();
    zap['a'] = make_field(detectorFile,      "detector file.")                           = "";
    zap['A'] = make_field(overwriteDetector, "overwrite detector file provided through '-a' with average stretching.");
    zap['d'] = make_field(debug)            = 2;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
  JDetector detector;
 
  if (detectorFile != "") {
    try {
      load(detectorFile, detector);
    }
    catch(const JException& error) {
      FATAL(error);
    }
  }
 
  map<int, JQuantile> Q;
 
  while (inputFile.hasNext()) {
 
    STATUS("event: " << setw(10) << inputFile.getCounter() << '\r'); DEBUG(endl);
 
    const JEvt* evt = inputFile.next();
 
    for (JEvt::const_iterator i = evt->begin(); i != evt->end(); ++i) {
      Q[i->id].put(i->vs);
    }
  }
  STATUS(endl);
 
  if (debug >= status_t) {
    for (map<int, JQuantile>::const_iterator i = Q.begin(); i != Q.end(); ++i) {
      cout << setw(4)    << i->first                << ' '
           << setw(6)    << i->second.getCount()    << ' '
           << FIXED(9,6) << i->second.getMean (0.0) << ' '
           << FIXED(9,6) << i->second.getSTDev(0.0) << endl;
    }
  }
 
  if (overwriteDetector) {
 
    map<int, map<int, JModule*> > buffer;
 
    for (JDetector::iterator module = detector.begin(); module != detector.end(); ++module) {
      buffer[module->getString()][module->getFloor()] = &(*module);
    }
    
    for (map<int, JQuantile>::const_iterator i = Q.begin(); i != Q.end(); ++i) {
 
      if (buffer.count(i->first) != 0 && buffer[i->first].count(0) != 0) {
        
        const double z0 = buffer[i->first][0]->getZ();
 
        for (auto& module : buffer[i->first]) {
 
          if (module.second->getFloor() != 0) {
 
            const double z1 = module.second->getZ();
 
            module.second->add(JPosition3D(0.0, 0.0, i->second.getMean(0.0) * (z1 - z0)));
          }
        }
        
      } else {
 
        FATAL("Missing location " << FILL(4,'0') << i->first << '.' << FILL(2,'0') << 0 << FILL() << endl);
      }
    }
 
    NOTICE("Store stretching data on file " << detectorFile << endl);
 
    store(detectorFile, detector);
  }
}