ZTrSetStream.H

/* ZTrSetStream.H */
/* $Id: ZTrSetStream.H,v 1.7 2003/11/17 14:18:39 vlad Exp $ */
#ifndef __ZTrSetStream_H
#define __ZTrSetStream_H

#include <ldb/trset.h>
#include <ldb/trselect.h>
#include <zm/ZSeismicStream.H>
#include <zm/ZTracePar.H>


class ZTrSetStream : public ZSeismicStream
{
public:

  ZTrSetStream (TrSet* ts);

  ZTrSetStream (const ZTrSetStream& ztss);

  virtual ~ZTrSetStream ();

  const TrIndexTable*   index_table () const;

  TrSet*        trset () const;

  virtual const ZTracePar&  operator() () const;

  virtual void      set_selection_rule (TsExpr* expr);


  virtual void  set_custom_coord (const char* coord_name,
                  Compar order = ASCENT_ORDER);

  virtual void  set_secondary_order (const char* secondary_name,
                     Compar order = ASCENT_ORDER);

  virtual void  set_tertiary_order (const char* tertiary_name,
                    Compar order = ASCENT_ORDER);

  virtual Logic     get_coord_range (float c[2]);

  virtual float     average_coord_step ();


  void  set_custom_coord_vector (double* coordslist);

  /*
   * Iterator methods.
   */

  virtual Logic     go_first_coord ();

  virtual Logic     go_next_coord ();

  virtual float     coord ();

  virtual int       ensemble_length () const;

  virtual float*    trace_body (int ens_index = 0);

  virtual ZTracePar trace_par (int ens_index = 0) const;

  virtual long      trace_id (int ens_index = 0) const;

private:

  TrSet     *ts;    
  TrIndexTable  it; 
  char  *selrule;   
  char  *cname[3];  
  Compar corder[3]; 
  TraceAttrId ctaid[3]; 
  int   ens_start;  
  int   ens_length; 
  ZTracePar trpar;  
  double    avg_step;   
  double    *coords;    
  void          gather_ensemble ();

  void          apply_selection ();

};


#endif /* ZTrSetStream.H */