/data/refman/pasoursint/CMSSW_4_2_9_HLT1_bphpatch4/src/Fireworks/Geometry/plugins/MFAnalyzer.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

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

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"

#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "MagneticField/Engine/interface/MagneticField.h"

class MFAnalyzer : public edm::EDAnalyzer
{
public:
  explicit MFAnalyzer( const edm::ParameterSet& );
  ~MFAnalyzer( void );

  bool         valid (void) { return m_valid; }

private:
  virtual void  analyze( const edm::Event&, const edm::EventSetup& );
  void          evaluate( const double point[3], double field[3] ) const;
  unsigned      m_mapDensityX;
  unsigned      m_mapDensityY;
  unsigned      m_mapDensityZ;
  double        m_minX;
  double        m_maxX;
  double        m_minY;
  double        m_maxY;
  double        m_minZ;
  double        m_maxZ;
  double        m_xBaseDir;
  double        m_yBaseDir;
  double        m_zBaseDir;
  bool          m_valid;
  edm::ESHandle<MagneticField> m_mf;
};

MFAnalyzer::MFAnalyzer( const edm::ParameterSet& iPset )
  : m_valid( false )
{
  m_mapDensityX = iPset.getUntrackedParameter<unsigned>( "mapDensityX", 10 );
  m_mapDensityY = iPset.getUntrackedParameter<unsigned>( "mapDensityY", 10 );
  m_mapDensityZ = iPset.getUntrackedParameter<unsigned>( "mapDensityY", 10 );
  m_minX = iPset.getUntrackedParameter<double>( "minX", -18.0 );
  m_maxX = iPset.getUntrackedParameter<double>( "maxX", 18.0 );
  m_minY = iPset.getUntrackedParameter<double>( "minY", -18.0 );
  m_maxY = iPset.getUntrackedParameter<double>( "maxY", 18.0 );
  m_minZ = iPset.getUntrackedParameter<double>( "minZ", -18.0 );
  m_maxZ = iPset.getUntrackedParameter<double>( "maxZ", 18.0 );

  m_xBaseDir = ( m_maxX - m_minX ) / m_mapDensityX;
  m_yBaseDir = ( m_maxY - m_minY ) / m_mapDensityY;
  m_zBaseDir = ( m_maxZ - m_minZ ) / m_mapDensityZ;
}

MFAnalyzer::~MFAnalyzer( void )
{}

void
MFAnalyzer::analyze( const edm::Event& iEvent, const edm::EventSetup& iSetup )
{
  iSetup.get<IdealMagneticFieldRecord>().get( m_mf );
  m_mf.isValid() ? m_valid = true : m_valid = false;
   
  for( unsigned i = 0; i <= m_mapDensityX; ++i )
  {
    for( unsigned j = 0; j <= m_mapDensityY; ++j )
    {                       
      for( unsigned k = 0; k <= m_mapDensityZ; ++k )
      {                       
        // Prepare field position and get value.
        double x =  m_minX + m_xBaseDir * i;
        double y = m_minY + m_yBaseDir * j;
        double z = m_minZ + m_zBaseDir * k;
        double  pt[3] = { x, y, z };
        double  val[3];
        evaluate( pt, val );
        std::cout << "(" << x << ", " << y << ", " << z << ") " << val[0] << ":" << val[1] << ":" << val[2] << "; ";
      }
      std::cout << std::endl;
    }
    std::cout << std::endl;
  }
}

void
MFAnalyzer::evaluate (const double point [3], double field [3]) const
{
  GlobalVector b = m_mf->inTesla( GlobalPoint( point[0], point[1], point[2] ));
  
  field [0] = b.x();
  field [1] = b.y();
  field [2] = b.z();
}

DEFINE_FWK_MODULE( MFAnalyzer );