vtkPostgreSQLDatabasePrivate.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   Visualization Toolkit
  Module:    vtkPostgreSQLDatabasePrivate.h
 
  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
 
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.
 
=========================================================================*/
/*-------------------------------------------------------------------------
  Copyright 2008 Sandia Corporation.
  Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
  the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------*/
 
#ifndef vtkPostgreSQLDatabasePrivate_h
#define vtkPostgreSQLDatabasePrivate_h
 
#include "vtkStdString.h"
#include "vtkTimeStamp.h"
#include "vtkType.h"
 
#include <libpq-fe.h>
#include <map>
 
class vtkPostgreSQLDatabasePrivate
{
public:
  vtkPostgreSQLDatabasePrivate() { this->Connection = nullptr; }
 
  virtual ~vtkPostgreSQLDatabasePrivate()
  {
    if (this->Connection)
    {
      PQfinish(this->Connection);
    }
  }
 
  // Given a Postgres column type OID, return a VTK array type (see vtkType.h).
  int GetVTKTypeFromOID(Oid pgtype)
  {
    std::map<Oid, int>::const_iterator it = this->DataTypeMap.find(pgtype);
    if (it == this->DataTypeMap.end())
    {
      return VTK_STRING;
    }
    return it->second;
  }
 
  // This is the actual database connection.  It will be nullptr if no
  // connection is open.
  PGconn* Connection;
 
  // Map Postgres column types to VTK types.
  std::map<Oid, int> DataTypeMap;
};
 
#endif // vtkPostgreSQLDatabasePrivate_h
// VTK-HeaderTest-Exclude: vtkPostgreSQLDatabasePrivate.h