EVE/Reve/TriangleSet.h

   1 // $Header$
   2
   3 #ifndef REVE_TriangleSet_H
   4 #define REVE_TriangleSet_H
   5
   6 #include "RenderElement.h"
   7 #include <TNamed.h>
   8 #include <TAttBBox.h>
   9 #include <TAtt3D.h>
  10
  11 #include "ZTrans.h"
  12
  13 class TGeoMatrix;
  14
  15 namespace Reve {
  16
  17 class TriangleSet : public RenderElement,
  18                     public TNamed,
  19                     public TAttBBox,
  20                     public TAtt3D
  21 {
  22   friend class TriangleSetEditor;
  23   friend class TriangleSetGL;
  24
  25   TriangleSet(const TriangleSet&);            // Not implemented
  26   TriangleSet& operator=(const TriangleSet&); // Not implemented
  27
  28 protected:
  29
  30   // Vertex data
  31   Int_t    fNVerts;
  32   Float_t* fVerts;        //[3*fNVerts]
  33
  34   // Triangle data
  35   Int_t    fNTrings;
  36   Int_t*   fTrings;       //[3*fNTrings]
  37   Float_t* fTringNorms;   //[3*fNTrings]
  38   UChar_t* fTringCols;    //[3*fNTrings]
  39
  40   // --------------------------------------------------------------
  41
  42   Color_t  fColor;
  43   UShort_t fTransp;
  44   ZTrans   fHMTrans;
  45
  46 public:
  47
  48   TriangleSet(Int_t nv, Int_t nt, Bool_t norms=false, Bool_t cols=false);
  49   ~TriangleSet();
  50
  51   virtual Bool_t CanEditMainColor() { return kTRUE; }
  52
  53   Float_t* Vertex(Int_t i)         { return &(fVerts[3*i]);      }
  54   Int_t*   Triangle(Int_t i)       { return &(fTrings[3*i]);     }
  55   Float_t* TriangleNormal(Int_t i) { return &(fTringNorms[3*i]); }
  56   UChar_t* TriangleColor(Int_t i)  { return &(fTringCols[3*i]);  }
  57
  58   void SetVertex(Int_t i, Float_t x, Float_t y, Float_t z)
  59   { Float_t* v = Vertex(i); v[0] = x; v[1] = y; v[2] = z; }
  60   void SetTriangle(Int_t i, Int_t v0, Int_t v1, Int_t v2)
  61   { Int_t* t = Triangle(i); t[0] = v0; t[1] = v1; t[2] = v2; }
  62   void SetTriangleColor(Int_t i, UChar_t r, UChar_t g, UChar_t b, UChar_t a=255)
  63   { UChar_t* c = TriangleColor(i); c[0] = r; c[1] = g; c[2] = b; c[3] = a; }
  64
  65   void GenerateTriangleNormals();
  66   void GenerateRandomColors();
  67   void GenerateZNormalColors(Float_t fac=20, Int_t min=-20, Int_t max=20,
  68                              Bool_t interp=kFALSE, Bool_t wrap=kFALSE);
  69
  70   virtual void ComputeBBox();
  71   virtual void Paint(Option_t* = "");
  72
  73   Color_t GetColor() const { return fColor; }
  74   void SetColor(Color_t c) { fColor = c; }
  75
  76   UShort_t GetTransparency() const { return fTransp; }
  77   void SetTransparency(UShort_t tr) { fTransp = tr; } // *MENU*
  78
  79   ZTrans& RefHMTrans() { return fHMTrans; }
  80   void SetTransMatrix(Double_t* carr)        { fHMTrans.SetFrom(carr); }
  81   void SetTransMatrix(const TGeoMatrix& mat) { fHMTrans.SetFrom(mat);  }
  82
  83   static TriangleSet* ReadTrivialFile(const char* file);
  84
  85   ClassDef(TriangleSet, 0)
  86 }; // endclass TriangleSet
  87
  88 }
  89
  90 #endif