REveDataClasses.hxx

Go to the documentation of this file.

// @(#)root/eve7:$Id$
// Authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007, 2018
 
/*************************************************************************
 * Copyright (C) 1995-2019, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
 
#ifndef ROOT7_REveDataClasses
#define ROOT7_REveDataClasses
 
#include <ROOT/REveElement.hxx>
 
#include <functional>
#include <vector>
#include <iostream>
 
class TClass;
 
namespace ROOT {
namespace Experimental {
 
class REveDataItem;
 
//==============================================================================
 
class REveDataCollection : public REveElement
{
public:
   typedef std::vector<int> Ids_t;
 
private:
   std::function<void (REveDataCollection*)>               _handler_func;
   std::function<void (REveDataCollection*, const Ids_t&)> _handler_func_ids;
 
public:
   static Color_t fgDefaultColor;
 
   TClass *fItemClass{nullptr}; // so far only really need class name
 
   struct ItemInfo_t
   {
      void *fDataPtr{nullptr};
      REveDataItem *fItemPtr{nullptr};
 
      ItemInfo_t() = default;
      ItemInfo_t(void *dp, REveDataItem *di) : fDataPtr(dp), fItemPtr(di) {}
   };
 
   std::vector<ItemInfo_t> fItems;
 
   TString fFilterExpr;
   std::function<bool(void *)> fFilterFoo = [](void *) { return true; };
 
   REveDataCollection(const std::string& n = "REveDataCollection", const std::string& t = "");
   virtual ~REveDataCollection() {}
 
   Bool_t SingleRnrState() const override { return kTRUE; }
   Bool_t SetRnrState(Bool_t) override;
 
   TClass *GetItemClass() const { return fItemClass; }
   void SetItemClass(TClass *cls) { fItemClass = cls; }
 
   void ReserveItems(Int_t items_size) { fItems.reserve(items_size); }
   void AddItem(void *data_ptr, const std::string& n, const std::string& t);
   void ClearItems() { fItems.clear(); }
 
   void SetFilterExpr(const TString &filter);
   void ApplyFilter();
 
   Int_t GetNItems() const { return (Int_t)fItems.size(); }
   void *GetDataPtr(Int_t i) const { return fItems[i].fDataPtr; }
   REveDataItem *GetDataItem(Int_t i) const { return fItems[i].fItemPtr; }
 
   Int_t WriteCoreJson(nlohmann::json &cj, Int_t rnr_offset) override;
 
   void SetMainColor(Color_t) override;
   virtual void ItemChanged(REveDataItem *item);
 
   void SetHandlerFunc (std::function<void (REveDataCollection*)> handler_func)
   {
      _handler_func = handler_func;
   }
   void SetHandlerFuncIds (std::function<void (REveDataCollection*, const Ids_t&)> handler_func)
   {
      _handler_func_ids= handler_func;
   }
};
 
//==============================================================================
 
class REveDataItem : public REveElement,
                     public REveAuntAsList
{
protected:
   Bool_t fFiltered{false};
 
public:
   REveDataItem(const std::string& n = "REveDataItem", const std::string& t = "");
   virtual ~REveDataItem() {}
 
   Bool_t GetFiltered() const { return fFiltered; }
   void   SetFiltered(Bool_t f);
 
   virtual void SetItemColorRGB(UChar_t r, UChar_t g, UChar_t b);
   Bool_t SetRnrSelf(Bool_t) override;
 
   virtual void FillImpliedSelectedSet(Set_t& impSelSet) override;
 
   Int_t WriteCoreJson(nlohmann::json &cj, Int_t rnr_offset) override;
};
 
//==============================================================================
 
class REveDataTable : public REveElement
{
protected:
   const REveDataCollection *fCollection{nullptr};
 
public:
   REveDataTable(const std::string& n = "REveDataTable", const std::string& t = "");
   virtual ~REveDataTable() {}
 
   void SetCollection(const REveDataCollection *col) { fCollection = col; }
   const REveDataCollection *GetCollection() const { return fCollection; }
 
   void PrintTable();
   virtual Int_t WriteCoreJson(nlohmann::json &cj, Int_t rnr_offset);
 
   void AddNewColumn(const std::string& expr, const std::string& title, int prec = 2);
};
 
//==============================================================================
 
class REveDataColumn : public REveElement
{
public:
   enum FieldType_e { FT_Double = 0, FT_Bool, FT_String };
 
protected:
public:
   TString fExpression;
   FieldType_e fType; // can we auto detect this?
   Int_t fPrecision{2};
 
   std::string fTrue{"*"};
   std::string fFalse{" "};
 
   std::function<double(void *)> fDoubleFoo;
   std::function<bool(void *)> fBoolFoo;
   std::function<std::string(void *)> fStringFoo;
 
public:
   REveDataColumn(const std::string& n = "REveDataColumn", const std::string& t = "");
   virtual ~REveDataColumn() {}
 
   void SetExpressionAndType(const std::string &expr, FieldType_e type);
   void SetPrecision(Int_t prec);
 
   std::string EvalExpr(void *iptr);
};
 
 
} // namespace Experimental
} // namespace ROOT
 
#endif