JPrintDetector.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <iomanip>
#include <set>
#include <iterator>
 
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JDetector/JDetectorAddressMap.hh"
#include "JDetector/JDetectorSupportkit.hh"
#include "JDetector/JModule.hh"
 
#include "JMath/JConstants.hh"
#include "JTools/JRange.hh"
#include "JSystem/JDateAndTime.hh"
#include "JGeometry3D/JGeometry3DToolkit.hh"
#include "JGeometry3D/JCylinder3D.hh"
#include "JGeometry3D/JQuaternion3D.hh"
 
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
namespace {
  /**
   * Print options.
   */
  const char* const default_t    =  "default";      //!< as-is
 
  const char* const modules_t    =  "modules";      //!< modules
  const char* const pmts_t       =  "pmts";         //!< PMTs
  const char* const geometry_t   =  "geometry";     //!< geometry
 
  const char* const comment_t    =  "comment";      //!< comment
  const char* const header_t     =  "header";       //!< header
  const char* const identifier_t =  "identifier";   //!< identifier
  const char* const version_t    =  "version";      //!< version
  const char* const can_t        =  "can";          //!< can
  const char* const center_t     =  "center";       //!< center
  const char* const summary_t    =  "summary";      //!< summary
 
  const char* const HEADER_t     =  "HEADER";       //!< header     in zsh format
  const char* const IDENTIFIER_t =  "IDENTIFIER";   //!< identifier in zsh format
  const char* const VERSION_t    =  "VERSION";      //!< version    in zsh format
  const char* const CAN_t        =  "CAN";          //!< can        in zsh format
  const char* const CENTER_t     =  "CENTER";       //!< center     in zsh format
  const char* const SUMMARY_t    =  "SUMMARY";      //!< summary    in zsh format
 
  /**
   * Get zsh array name for given string.
   *
   * \param  id             string number
   * \return                zsh array name
   */
  inline std::string getString(const int id)
  {
    return MAKE_STRING("STRING_" << FILL(4,'0') << id);
  }
}
 
 
/**
 * \file
 *
 * Auxiliary program to print detector file in human friendly format.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  string         detectorFile;
  string         option;
  double         precision;
  int            debug;
 
  try {
 
    JParser<> zap("Auxiliary program to print detector file in human friendly format.");
 
    zap['a'] = make_field(detectorFile);
    zap['O'] = make_field(option)            = 
 
      default_t,
 
      pmts_t,
      modules_t,
      geometry_t,
 
      comment_t,
      header_t,
      version_t,
      identifier_t,
      can_t,
      center_t,
      summary_t,
 
      HEADER_t,
      VERSION_t,
      IDENTIFIER_t,
      CAN_t,
      CENTER_t,
      SUMMARY_t;
 
    zap['p'] = make_field(precision,    "precision for match with reference module") = 1.0e-5;
    zap['d'] = make_field(debug)             = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
  
  JDetector detector;
 
  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }
 
  if        (option == default_t) {
 
    cout << detector << endl;
 
  } else if (option == modules_t) {
      
    const JDetectorBuilder& demo = getDetectorBuilder(detector.getID());
 
    for (JDetector::const_iterator module = detector.begin(); module != detector.end(); ++module) {
 
      cout << "Module";
      cout << ' ' << noshowpos << setw(8) << right << module->getID();
      cout << ' ' << noshowpos << setw(3) << right << module->getString();
      cout << ' ' << noshowpos << setw(2) << right << module->getFloor();
      cout << ' ' << FIXED(7,2) << module->getX();
      cout << ' ' << FIXED(7,2) << module->getY();
      cout << ' ' << FIXED(7,2) << module->getZ();
      cout << ' ' << FIXED(8,2) << module->getT0();
      
      if (module->getFloor() != 0) {
        {
          JModule buffer = demo.getModule(module->getID(), module->getLocation());
 
          const JQuaternion3D Q = getRotation(buffer, *module);
          const JQuaternion3D::decomposition q1(Q, JVector3Z_t);
 
          const double phi = (JVector3Z_t.getDot(q1.twist) >= 0.0 ? +1.0 : -1.0) * q1.twist.getAngle();
 
          cout << ' ' << FIXED(7,2) << phi;
        }
        {
          const JQuaternion3D Q = module->getQuaternion();
          const JQuaternion3D::decomposition q1(Q, JVector3Z_t);
 
          const double phi = (JVector3Z_t.getDot(q1.twist) >= 0.0 ? +1.0 : -1.0) * q1.twist.getAngle();
       
          cout << ' ' << FIXED(7,2) << phi;         
        }
        {
          JModule buffer = demo.getModule(module->getID(), module->getLocation());
 
          const JRotation3D R = getRotation(buffer, *module);
 
          buffer.rotate(R);
 
          cout << ' ' << (JModule::compare(buffer, *module, precision) ? "==" : "!=") 
               << ' ' << (demo.get(module->getID()) == demo.getDefaultModuleAddressMap() ? "default" : "custom"); 
        }
      }
 
      cout << endl;
    }
    
  } else if (option == pmts_t) {
          
    for (JDetector::const_iterator module = detector.begin(); module != detector.end(); ++module) {
      for (JModule::const_iterator pmt = module->begin(); pmt != module->end(); ++pmt) {
        cout << "PMT";
        cout << ' ' << noshowpos << setw(8) << right << pmt->getID();
        cout << ' ' << FIXED(7,2) << pmt->getX();
        cout << ' ' << FIXED(7,2) << pmt->getY();
        cout << ' ' << FIXED(7,2) << pmt->getZ();
        cout << ' ' << FIXED(6,3) << pmt->getDX();
        cout << ' ' << FIXED(6,3) << pmt->getDY();
        cout << ' ' << FIXED(6,3) << pmt->getDZ();
        cout << ' ' << FIXED(8,2) << pmt->getT0();
        cout << ' ' << noshowpos << setw(8) << right << pmt->getStatus();
        cout << endl;
      }
    }
 
  } else if (option == geometry_t) {
 
    cout << "Maximal distance [m] = " << FIXED(6,1) << getMaximalDistance(detector) << endl;
    cout << "Maximal time [ns]    = " << FIXED(6,1) << getMaximalTime    (detector) << endl;
  
  } else if (option == comment_t) {
 
    cout << detector.comment << endl;
  
  } else if (option == header_t ||
             option == HEADER_t) {
 
    if (option == header_t) {
 
      cout << detector.getProperties() << endl;
      cout << "name = " << (isARCADetector(detector) ? "ARCA" : 
                            isORCADetector(detector) ? "ORCA" : 
                            "unknown") << endl;
      cout << "validity = " 
           << JDateAndTime(detector.getLowerLimit(), true).toString() << ' '
           << JDateAndTime(detector.getUpperLimit(), true).toString() << endl;
 
    } else {
 
      cout << "set_variable UTM_EAST         " << FIXED(12,2) << detector.getUTMEast()  << ";" << endl;
      cout << "set_variable UTM_NORTH        " << FIXED(12,2) << detector.getUTMNorth() << ";" << endl;
      cout << "set_variable UTM_Z            " << FIXED(12,2) << detector.getUTMZ()     << ";" << endl;
      cout << "set_variable UTM_ZONE         " << FIXED(12,2) << detector.getUTMZone()  << ";" << endl;
      cout << "set_variable UTM_WGS          " << FIXED(12,2) << detector.getWGS()      << ";" << endl;
    }
 
  } else if (option == version_t || 
             option == VERSION_t) {
 
    if (option == version_t) {
 
      cout << detector.getVersion() << endl;
 
    } else {
 
      cout << "set_variable DETECTOR_VERSION " << detector.getVersion() << ";" << endl;
    }
 
  } else if (option == identifier_t ||
             option == IDENTIFIER_t) {
 
    if (option == identifier_t) {
 
      cout << "Detector " << detector.getID() << endl;
 
    } else {
 
      cout << "set_variable DETECTOR_ID      " << detector.getID() << ";" << endl;
    }
 
  } else if (option == can_t ||
             option == CAN_t) {
 
    const JCylinder3D cylinder(detector.begin(), detector.end());
 
    const double      V = (cylinder.getZmax() - cylinder.getZmin()) * PI * cylinder.getRadius() * cylinder.getRadius(); 
    const double      D = getMaximalDistance(detector);
 
    if (option == can_t) {
      
      cout << "X        = " << FIXED(7,1)       << cylinder.getX()      << endl;
      cout << "Y        = " << FIXED(7,1)       << cylinder.getY()      << endl;
      cout << "Zmin     = " << FIXED(7,1)       << cylinder.getZmin()   << endl;
      cout << "Zmax     = " << FIXED(7,1)       << cylinder.getZmax()   << endl;
      cout << "Radius   = " << FIXED(7,1)       << cylinder.getRadius() << endl;
      cout << "Depth    = " << FIXED(7,1)       << detector.getUTMZ()   << endl;
      cout << "Volume   = " << SCIENTIFIC(12,3) << V                    << endl;
      cout << "Distance = " << FIXED(9,3)       << D                    << endl;
 
    } else {
 
      cout << "set_variable CAN_X_M        " << FIXED(7,1)       << cylinder.getX()           << ";" << endl;
      cout << "set_variable CAN_Y_M        " << FIXED(7,1)       << cylinder.getY()           << ";" << endl;
      cout << "set_variable CAN_ZMIN_M     " << FIXED(7,1)       << cylinder.getZmin()        << ";" << endl;
      cout << "set_variable CAN_ZMAX_M     " << FIXED(7,1)       << cylinder.getZmax()        << ";" << endl;
      cout << "set_variable CAN_RADIUS_M   " << FIXED(7,1)       << cylinder.getRadius()      << ";" << endl;
      cout << "set_variable CAN_DEPTH_M    " << FIXED(7,1)       << detector.getUTMZ()        << ";" << endl;
      cout << "set_variable CAN_VOLUME_M3  " << SCIENTIFIC(12,3) << V                         << ";" << endl;
      cout << "set_variable CAN_DISTANCE_M " << FIXED(9,3)       << D                         << ";" << endl;
    }
  
  } else if (option == center_t || 
             option == CENTER_t) {
      
    const JCenter3D center(detector.begin(), detector.end());
      
    if (option == center_t) {
 
      cout << "center = ";
      cout << showpos << FIXED(8,3) << center.getX() << ' ';
      cout << showpos << FIXED(8,3) << center.getY() << ' ';
      cout << showpos << FIXED(8,3) << center.getZ() << endl;
 
    } else {
 
      cout << "set_variable CENTER_X_M " << FIXED(7,1) << center.getX() << ";" << endl;
      cout << "set_variable CENTER_Y_M " << FIXED(7,1) << center.getY() << ";" << endl;
      cout << "set_variable CENTER_Z_M " << FIXED(7,1) << center.getZ() << ";" << endl;
    }
    
  } else if (option == summary_t ||
             option == SUMMARY_t) {
      
    const int numberOfStrings    = getNumberOfStrings(detector);
    const int numberOfFloors     = getNumberOfFloors (detector);
    const int numberOfModules    = getNumberOfModules(detector);
    const int numberOfPMTs       = getNumberOfPMTs   (detector);
 
    typedef JTOOLS::JRange<int>   JRange_t;
 
    const JRange_t string_t(make_array(detector.begin(), detector.end(), &JModule::getString));
    const JRange_t floor_t (make_array(detector.begin(), detector.end(), &JModule::getFloor));
 
    set<int> strings;
    set<int> modules;
 
    for (JDetector::const_iterator module = detector.begin(); module != detector.end(); ++module) {
      strings.insert(module->getString());
      modules.insert(module->getID());
    }
 
    if (option == summary_t) {
        
      cout << "Number of strings = " << setw(4) << numberOfStrings << endl;
      cout << "Number of floors  = " << setw(4) << numberOfFloors  << endl;
      cout << "Number of modules = " << setw(4) << numberOfModules << endl;
      cout << "Number of PMTs    = " << setw(4) << numberOfPMTs    << endl;
      cout << "First string      = " << setw(4) << string_t.first  << endl;
      cout << "Last  string      = " << setw(4) << string_t.second << endl;
      cout << "First floor       = " << setw(4) << floor_t .first  << endl;
      cout << "Last  floor       = " << setw(4) << floor_t .second << endl;
 
    } else {
 
      cout << "let \"NUMBER_OF_STRINGS = " << setw(5) << numberOfStrings << "\";" << endl;
      cout << "let \"NUMBER_OF_FLOORS  = " << setw(5) << numberOfFloors  << "\";" << endl;
      cout << "let \"NUMBER_OF_MODULES = " << setw(5) << numberOfModules << "\";" << endl;
      cout << "let \"NUMBER_OF_PMTS    = " << setw(5) << numberOfPMTs    << "\";" << endl;
      cout << "let \"FIRST_STRING      = " << setw(5) << string_t.first  << "\";" << endl;
      cout << "let \"LAST_STRING       = " << setw(5) << string_t.second << "\";" << endl;
      cout << "let \"FIRST_FLOOR       = " << setw(5) << floor_t .first  << "\";" << endl;
      cout << "let \"LAST_FLOOR        = " << setw(5) << floor_t .second << "\";" << endl;
      cout << "STRINGS=(";
      copy(strings.begin(), strings.end(), ostream_iterator<int>(cout, " "));
      cout << ");" << endl;
      cout << "MODULES=(";
      copy(modules.begin(), modules.end(), ostream_iterator<int>(cout, " "));
      cout << ");" << endl;
 
      // detector map
 
      for (const auto i : strings) {
        cout << "typeset -A " << getString(i) << ";"                                         << endl;
      }
      for (const auto& i : detector) {
        cout << getString(i.getString()) << "[" << i.getFloor() << "]=" << i.getID() << ";"  << endl;
      }
    }
  }
}