IMatrixTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKALIGNINTERFACES_IMATRIXTOOL_IH
#define TRKALIGNINTERFACES_IMATRIXTOOL_IH

 
#include "GaudiKernel/IAlgTool.h"
 
#include <fstream>
 
namespace Trk {
 
  class AlignModule;
 
  class AlVec;
  class AlSymMatBase;
 
  static const InterfaceID
    IID_TRKALIGNINTERFACES_IMatrixTool("IMatrixTool",1,0);
  
  class IMatrixTool : virtual public IAlgTool {
    
  public:

    IMatrixTool();

    virtual ~IMatrixTool() {}
    
    static const InterfaceID& interfaceID();

    virtual StatusCode allocateMatrix(int nDoF) = 0; 
    
    virtual void prepareBinaryFiles(int solveOption) = 0;

    virtual void addFirstDerivatives(AlVec* vector) = 0;

    virtual void addFirstDerivatives(std::list<int,double>& derivatives) = 0;
 
    virtual void addFirstDerivative(int irow, double val) = 0;

    virtual void addSecondDerivatives(AlSymMatBase* matrix) = 0;
    
    virtual void addSecondDerivatives(std::list<std::pair<int,int>,double >& derivatives) = 0;
    virtual void addSecondDerivative(int irow, int icol, double val) = 0;

    virtual bool accumulateFromFiles() = 0;

    virtual int solve() = 0;
    
    void addModule(int alignModuleIndex, int nAlignParam);
    int  entryNumber(int alignModuleIndex);

    //virtual void setAlignModule(int irow, const AlignModule* alignmodule,
    //          int ipar) = 0;

    void setNHits(int n) { m_nHits = n; }
    int nHits() const { return m_nHits; }

    void setNTracks(int n) { m_nTracks = n; }
    int nTracks() const { return m_nTracks; }

    void setNMeasurements(int n) { m_nMeasurements = n; }
    int nMeasurements() const { return m_nMeasurements; }

    virtual void setLogStream(std::ostream * os) { m_logStream = os; }
 
  protected:
    std::ostream * m_logStream; 
 
    int m_nHits;
    int m_nTracks;
    int m_nMeasurements;
 
  private:
    std::map<int,int> m_alignModuleMap;
    int m_nentries;
 
  }; // end of class
  
  inline const InterfaceID& IMatrixTool::interfaceID() {
    return IID_TRKALIGNINTERFACES_IMatrixTool;
  }
 
  inline IMatrixTool::IMatrixTool() : m_logStream(0), m_nHits(0), m_nTracks(0), m_nMeasurements(0), m_nentries(0) {}
 
  inline void IMatrixTool::addModule(int alignModuleIndex, int nAlignParam) { m_alignModuleMap[alignModuleIndex]=m_nentries; m_nentries += nAlignParam; }
 
  inline int IMatrixTool::entryNumber(int alignModuleIndex) { if ( m_alignModuleMap.find(alignModuleIndex) == m_alignModuleMap.end()) return -1; else return m_alignModuleMap[alignModuleIndex]; }
 
} // end of namespace
  
#endif // TRKALIGNINTERFACES_IMATRIXTOOL_IH