vtkIOPostgreSQL_AutoInit.cxx 1.61 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
/*=========================================================================

  Program:   Visualization Toolkit
  Module:    vtkIOPostgreSQL_AutoInit.cxx

  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.

  =========================================================================*/

#include "vtkPostgreSQLDatabase.h"

#include <vtksys/SystemTools.hxx>

#include <string>

// Registration of PostgreSQL dynamically with the vtkSQLDatabase factory method.
vtkSQLDatabase * PostgreSQLCreateFunction(const char* URL)
{
  std::string urlstr(URL ? URL : "");
  std::string protocol, unused;
  vtkPostgreSQLDatabase *db = 0;

  if (vtksys::SystemTools::ParseURLProtocol(urlstr, protocol, unused) &&
      protocol == "psql")
    {
    db = vtkPostgreSQLDatabase::New();
    db->ParseURL(URL);
    }

  return db;
}

static unsigned int vtkIOPostgreSQLCount;

struct VTKIOPOSTGRESQL_EXPORT vtkIOPostgreSQL_AutoInit
{
  vtkIOPostgreSQL_AutoInit();
  ~vtkIOPostgreSQL_AutoInit();
};

vtkIOPostgreSQL_AutoInit::vtkIOPostgreSQL_AutoInit()
{
  if (++vtkIOPostgreSQLCount == 1)
    {
    vtkSQLDatabase::RegisterCreateFromURLCallback(PostgreSQLCreateFunction);
    }
}

vtkIOPostgreSQL_AutoInit::~vtkIOPostgreSQL_AutoInit()
{
  if (--vtkIOPostgreSQLCount == 0)
    {
    vtkSQLDatabase::UnRegisterCreateFromURLCallback(PostgreSQLCreateFunction);
    }
}