Merged EVE-dev-after-merge to EVE-dev into HEAD. Requires ROOT-5.17.04.

[u/mrichter/AliRoot.git] / EVE / Reve / ZTrans.h

// $Header$

// Copyright (C) 1999-2005, Matevz Tadel. All rights reserved.
// This file is part of GLED, released under GNU General Public License version 2.
// For the licensing terms see $GLEDSYS/LICENSE or http://www.gnu.org/.
//
// Taken from Gled, main changes:
// 1. Put into namespace Reve.
// 2. Remove references to ZNode.
// 3. Added bool-members fApplyTrans, fEditTrans
// 4. Added set-from-TGeoMatrix
// 5. Added method SetupBuffer3D()

#ifndef REVE_ZTrans_H
#define REVE_ZTrans_H

#include <TVector3.h>

class TGeoMatrix;
class TGeoHMatrix;
class TBuffer3D;

namespace Reve {

/**************************************************************************/
// ZTrans -- 3D transformation in generalised coordinates
/**************************************************************************/

class ZTrans : public TObject
{
  friend class ZTransSubEditor;
  friend class ZTransEditor;

protected:
  Double32_t            M[16];

  mutable Float_t	mA1;   //!
  mutable Float_t	mA2;   //!
  mutable Float_t	mA3;   //!
  mutable Bool_t	bAsOK; //!

  // Reve
  Bool_t                fUseTrans;
  Bool_t                fEditTrans;

  Double_t norm3_column(Int_t col);
  Double_t orto3_column(Int_t col, Int_t ref);

public:
  ZTrans();
  ZTrans(const ZTrans& t);
  ZTrans(const Double_t arr[16]);
  ZTrans(const Float_t  arr[16]);
  virtual ~ZTrans() {}

  // General operations

  void     UnitTrans();
  void     UnitRot();
  void     SetTrans(const ZTrans& t, Bool_t copyAngles=kTRUE);
  void     SetFromArray(const Double_t arr[16]);
  void     SetFromArray(const Float_t  arr[16]);
  ZTrans&  operator=(const ZTrans& t) { SetTrans(t); return *this; }
  void     SetupRotation(Int_t i, Int_t j, Double_t f);

  void     OrtoNorm3();
  Double_t Invert();

  void MultLeft(const ZTrans& t);
  void MultRight(const ZTrans& t);
  void operator*=(const ZTrans& t) { MultRight(t); }

  void TransposeRotationPart();

  ZTrans operator*(const ZTrans& t);

  // Move & Rotate

  void MoveLF(Int_t ai, Double_t amount);
  void Move3LF(Double_t x, Double_t y, Double_t z);
  void RotateLF(Int_t i1, Int_t i2, Double_t amount);

  void MovePF(Int_t ai, Double_t amount);
  void Move3PF(Double_t x, Double_t y, Double_t z);
  void RotatePF(Int_t i1, Int_t i2, Double_t amount);

  void Move(const ZTrans& a, Int_t ai, Double_t amount);
  void Move3(const ZTrans& a, Double_t x, Double_t y, Double_t z);
  void Rotate(const ZTrans& a, Int_t i1, Int_t i2, Double_t amount);

  // Element access

  Double_t* Array() { return M; }      const Double_t* Array() const { return M; }
  Double_t* ArrX()  { return M; }      const Double_t* ArrX()  const { return M; }
  Double_t* ArrY()  { return M +  4; } const Double_t* ArrY()  const { return M +  4; }
  Double_t* ArrZ()  { return M +  8; } const Double_t* ArrZ()  const { return M +  8; }
  Double_t* ArrT()  { return M + 12; } const Double_t* ArrT()  const { return M + 12; }

  Double_t  operator[](Int_t i) const { return M[i]; }
  Double_t& operator[](Int_t i)       { return M[i]; }

  Double_t  CM(Int_t i, Int_t j) const { return M[4*j + i]; }
  Double_t& CM(Int_t i, Int_t j)       { return M[4*j + i]; }

  Double_t  operator()(Int_t i, Int_t j) const { return M[4*j + i - 5]; }
  Double_t& operator()(Int_t i, Int_t j)       { return M[4*j + i - 5]; }

  // Base-vector interface

  void SetBaseVec(Int_t b, Double_t x, Double_t y, Double_t z);
  void SetBaseVec(Int_t b, const TVector3& v);

  TVector3 GetBaseVec(Int_t b) const;
  void     GetBaseVec(Int_t b, TVector3& v) const;

  // Position interface

  void SetPos(Double_t x, Double_t y, Double_t z);
  void SetPos(Double_t* x);
  void SetPos(Float_t * x);
  void SetPos(const ZTrans& t);

  void GetPos(Double_t& x, Double_t& y, Double_t& z) const;
  void GetPos(Double_t* x) const;
  void GetPos(Float_t * x) const;
  void GetPos(TVector3& v) const;  
  TVector3 GetPos() const;

  // Cardan angle interface

  void SetRotByAngles(Float_t a1, Float_t a2, Float_t a3);
  void SetRotByAnyAngles(Float_t a1, Float_t a2, Float_t a3, const Text_t* pat);
  void GetRotAngles(Float_t* x) const;

  // Scaling

  void     Scale(Double_t sx, Double_t sy, Double_t sz);
  void     GetScale(Double_t& sx, Double_t& sy, Double_t& sz) const;
  void     Unscale(Double_t& sx, Double_t& sy, Double_t& sz);
  Double_t Unscale();


  // Operations on vectors

  void     MultiplyIP(TVector3& v, Double_t w=1) const;
  void     MultiplyIP(Double_t* v, Double_t w=1) const;
  void     MultiplyIP(Float_t*  v, Double_t w=1) const;
  TVector3 Multiply(const TVector3& v, Double_t w=1) const;
  void     Multiply(const Double_t *vin, Double_t* vout, Double_t w=1) const;
  void     RotateIP(TVector3& v) const;
  TVector3 Rotate(const TVector3& v) const;


  virtual void Print(Option_t* option = "") const;


  // Reve stuff

  void SetFrom(Double_t* carr);
  void SetFrom(const TGeoMatrix& mat);
  void SetGeoHMatrix(TGeoHMatrix& mat);
  void SetBuffer3D(TBuffer3D& buff);

  Bool_t GetUseTrans()  const { return fUseTrans; }
  void SetUseTrans(Bool_t v)  { fUseTrans = v;    }
  Bool_t GetEditTrans() const { return fEditTrans; }
  void SetEditTrans(Bool_t v) { fEditTrans = v;    }

  Bool_t IsScale(Double_t low=0.9, Double_t high=1.1) const;

  ClassDef(ZTrans, 1) // Column-major 4x4 matrix for homogeneous coordinates.
};

ostream& operator<<(ostream& s, const ZTrans& t);

}

#endif