// Copyright 2006 - 2008: Rory Plaire (codekaizen@gmail.com)

//

// This file is part of SharpMap.

// SharpMap is free software; you can redistribute it and/or modify

// it under the terms of the GNU Lesser General Public License as published by

// the Free Software Foundation; either version 2 of the License, or

// (at your option) any later version.

//

// SharpMap is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

// GNU Lesser General Public License for more details.

// You should have received a copy of the GNU Lesser General Public License

// along with SharpMap; if not, write to the Free Software

// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

using System;

using System.Collections.Generic;

using System.Data;

using System.Data.Common;

using GeoAPI.Geometries;

using SharpMap.Data;

namespace SharpMap.Data

{

internal sealed class LoadFeaturesAdapter : DataAdapter

{

private readonly IGeometryFactory _factory;

public LoadFeaturesAdapter(IGeometryFactory factory)

{

_factory = factory;

}

public Int32 Fill(FeatureDataTable table, IFeatureDataReader dataReader)

{

return Fill(new FeatureDataTable[] { table }, dataReader);

}

public Int32 Fill(ICollection<FeatureDataTable> dataTables, IFeatureDataReader dataReader)

{

FeatureDataTable[] tables = new FeatureDataTable[dataTables.Count];

dataTables.CopyTo(tables, 0);

return Fill(tables, dataReader, 0, 0);

}

protected override Int32 Fill(DataTable[] dataTables, IDataReader dataReader, Int32 startRecord, Int32 maxRecords)

{

if (dataTables.Length == 0)

{

return 0;

}

Int32 tableIndex = 0;

IFeatureDataReader featureReader = dataReader as IFeatureDataReader;

if (featureReader == null)

{

throw new ArgumentException("Parameter 'dataReader' " +

"must be a IFeatureDataReader instance.");

}

do

{

FeatureDataTable table = dataTables[tableIndex] as FeatureDataTable;

if (table == null)

{

throw new ArgumentException("Components of 'dataTables' must be " +

"FeatureDataTable instances.");

}

if (table.Columns.Count == 0)

{

FillSchema(table, SchemaType.Mapped, featureReader);

}

table.Merge((IEnumerable<IFeatureDataRecord>)featureReader, null, _factory, SchemaMergeAction.AddWithKey | SchemaMergeAction.CaseInsensitive);

tableIndex++;

} while (dataReader.NextResult());

return dataTables[0].Rows.Count;

}

}

}