/*
   ____    _ __           ____               __    ____
  / __/___(_) /  ___ ____/ __ \__ _____ ___ / /_  /  _/__  ____
 _\ \/ __/ / _ \/ -_) __/ /_/ / // / -_|_-</ __/ _/ // _ \/ __/
/___/\__/_/_.__/\__/_/  \___\_\_,_/\__/___/\__/ /___/_//_/\__(_)

Copyright 2012 SciberQuest Inc.
*/
// .NAME vtkSQBOVWriter -- Connects the VTK pipeline to BOVWriter class.
// .SECTION Description
//
// Implements the VTK style pipeline and manipulates and instance of
// BOVWriter so that "brick of values" datasets, including time series,
// can be writen in parallel.
//
// It's assumed that the executive is responsible for the domain
// decomposition so make sure to initialize it.
//
// .SECTION See Also
// BOVWriter

#ifndef __vtkSQBOVWriter_h
#define __vtkSQBOVWriter_h

#include "vtkDataSetAlgorithm.h"

// define this for cerr status.
// #define vtkSQBOVWriterDEBUG

//BTX
class BOVWriter;
class vtkPVXMLElement;
class vtkInformationStringKey;
class vtkInformationDoubleKey;
class vtkInformationDoubleVectorKey;
class vtkInformationIntegerKey;
class vtkInformationIntegerVectorKey;
//ETX

class VTK_EXPORT vtkSQBOVWriter : public vtkDataSetAlgorithm
{
public:
  static vtkSQBOVWriter *New();
  vtkTypeMacro(vtkSQBOVWriter,vtkDataSetAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent);

  // Description:
  // Initialize the writer from an xml document.
  int Initialize(vtkPVXMLElement *elem);

  // Description:
  // Get/Set the file to write. Setting the file name opens
  // the file. Perhaps it's bad style but this is where open
  // fits best in VTK/PV pipeline execution.
  void SetFileName(const char *file);
  vtkGetStringMacro(FileName);

  // Descritpion:
  // Get status indicating if a file is opened successfully.
  bool IsOpen();

  // Description:
  // Array selection.
  void SetPointArrayStatus(const char *name, int status);
  int GetPointArrayStatus(const char *name);
  int GetNumberOfPointArrays();
  const char* GetPointArrayName(int idx);
  void ClearPointArrayStatus();

  // Description:
  // Time domain discovery interface.
  int GetNumberOfTimeSteps();
  double GetTimeStep(int i);
  void GetTimeSteps(double *times);

  //BTX
  enum
    {
    HINT_DEFAULT=0,
    HINT_AUTOMATIC=0,
    HINT_DISABLED=1,
    HINT_ENABLED=2
    };
  //ETX
  // Description:
  // Set/Get MPI file hints.
  vtkSetMacro(UseCollectiveIO,int);
  vtkGetMacro(UseCollectiveIO,int);

  vtkSetMacro(NumberOfIONodes,int);
  vtkGetMacro(NumberOfIONodes,int);

  vtkSetMacro(CollectBufferSize,int);
  vtkGetMacro(CollectBufferSize,int);

  vtkSetMacro(UseDirectIO,int);
  vtkGetMacro(UseDirectIO,int);

  vtkSetMacro(UseDeferredOpen,int);
  vtkGetMacro(UseDeferredOpen,int);

  vtkSetMacro(UseDataSieving,int);
  vtkGetMacro(UseDataSieving,int);

  vtkSetMacro(SieveBufferSize,int);
  vtkGetMacro(SieveBufferSize,int);

  vtkSetMacro(StripeSize,int);
  vtkGetMacro(StripeSize,int);

  vtkSetMacro(StripeCount,int);
  vtkGetMacro(StripeCount,int);

  // Description:
  // Writes a bov metadata file.
  int WriteMetaData();

  // Description:
  // Force a write.
  void Write();

  // Description:
  // When set Write will drive the pipeline once for each
  // available timestep.
  vtkSetMacro(WriteAllTimeSteps,int);
  vtkGetMacro(WriteAllTimeSteps,int);

  // Description:
  // Set the log level.
  // 0 -- no logging
  // 1 -- basic logging
  // .
  // n -- advanced logging
  vtkSetMacro(LogLevel,int);
  vtkGetMacro(LogLevel,int);

protected:
  /// Pipeline internals.
  int RequestUpdateExtent(vtkInformation *, vtkInformationVector **, vtkInformationVector *);
  int RequestDataObject(vtkInformation*,vtkInformationVector**,vtkInformationVector*);
  int RequestData(vtkInformation*,vtkInformationVector**,vtkInformationVector*);
  int RequestInformation(vtkInformation*,vtkInformationVector**,vtkInformationVector*);

  vtkSQBOVWriter();
  ~vtkSQBOVWriter();

private:
  vtkSQBOVWriter(const vtkSQBOVWriter &); // Not implemented
  void operator=(const vtkSQBOVWriter &); // Not implemented
  //
  void Clear();
  //
  void SetMPIFileHints();

private:
  BOVWriter *Writer;       // Implementation
  int DimMode;             // 2d or 3d cases
  char *FileName;          // Name of data file to load.
  bool FileNameChanged;    // Flag indicating that the dataset needs to be opened
  int IncrementalMetaData; // append to metadata with each update
  int WriteAllTimeSteps;   // drive the pipeline for all timesteps.
  int TimeStepId;          // step id only used for writing all steps.
  int WorldRank;           // rank of this process
  int WorldSize;           // number of processes
  int UseCollectiveIO;     // Turn on/off collective IO
  int UseDirectIO;         //
  int NumberOfIONodes;     // Number of aggregator for CIO
  int CollectBufferSize;   // Gather buffer size (if small IO is staged).
  int UseDeferredOpen;     // Turn on/off deffered open (only agg.'s open)
  int UseDataSieving;      // Turn on/off data sieving
  int SieveBufferSize;     // Sieve size.
  int StripeSize;          // Stripe size in bytes
  int StripeCount;         // Stripe count in OST's
  int LogLevel;            // control timing/logging
};

#endif
