source: trunk/java-client/org/nzdl/gsdl/SimpleGraphicalClient/CSModel.java@ 2278

/*
 *    SimpleGraphicalClient.java
 *    Copyright (C) 2001 New Zealand Digital Library Project
 *
 *    This program is free software; you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation; either version 2 of the License, or
 *    (at your option) any later version.
 *
 *    This program 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 General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with this program; if not, write to the Free Software
 *    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
package org.nzdl.gsdl.SimpleGraphicalClient;

import java.util.*;
import java.util.Set;
import java.util.TreeSet;
import java.util.Iterator;
import java.util.Vector;

// local libraries

import org.nzdl.gsdl.service.NzdlService;
import org.nzdl.gsdl.service.NzdlCollectionInfo;

/**
 * Class 
 *
 * A class to 
 * 
 * @author Dave Nichols ([email protected])
 * @author stuart yeates ([email protected])
 * @version $Revision: 2278 $
 * @see org.nzdl.gsdl.service.SimpleGraphicalClient.SimpleGraphicalClient
 */

class CSModel extends Object
{

  Vector collectionList;
  TreeMap collectionInfoMap;
    HashMap collectionNameMap;
  String currentCollection;
  ResultModel results;
  NzdlService nzdl;
  QueryHistoryModel queryHistoryModel;
    

  public CSModel(NzdlService newNzdl)
  {
    nzdl = newNzdl;
    results = new ResultModel();
    collectionList = new Vector();
    collectionInfoMap = new TreeMap();
    collectionNameMap = new HashMap();
    queryHistoryModel = new QueryHistoryModel();
    currentCollection = "";
    initialiseCollection();
  }
  
  
  public NzdlService getNzdlService()
  {
    return nzdl;
  }

    public QueryHistoryModel getQueryHistoryModel() {
    return queryHistoryModel;
    }
    

  /* Result methods */

  public ResultModel getResultsModel()
  {
    return results;
  }
    
  public void addResult(Result result)
  {
    results.add(result);
  }
    
  public void clearResults()
  {
    //System.out.println("Clearing resultModel");
    results.clear();
  }

  /* collectionList methods */

    public Vector getCollectionList() {
    return new Vector(collectionInfoMap.keySet());
    }

    public Collection getCollectionAsCollection() {
    return (Collection)collectionInfoMap;
    }

    public int getCollectionListSize() {
    return collectionInfoMap.size();
    }

    public NzdlCollectionInfo getCollectionInfo(String collectionName) {
    return (NzdlCollectionInfo) collectionInfoMap.get(collectionName);
    }


    public String getLongCollectionName(String collectionName) {
    return (String) collectionNameMap.get(collectionName);
    }


    public String getCollectionDescription(String collectionName) {
    Set descSet = nzdl.getMetaData(collectionName, "collection", "collectionextra");
    if (descSet.size() == 1) 
        return (String) descSet.toArray()[0];
    else
        return "collectionName has more than 1 collectionextra entry";
    }


    public void setCurrentCollection(String collection )
        {
        currentCollection = collection;
        }

    public String getCurrentCollection()
        {
        return currentCollection;
        }

    public String getFirstCollection()
        {
        if (getCollectionListSize() > 0)
            return (String)collectionInfoMap.firstKey();
        else
            return "no collections";
        }
    /**
     *
     */
    public void initialiseCollection() 
        {
        Set collectionSet = nzdl.getCollectionSet(); // set of strings of collection titles
        TreeSet orderedCollectionSet = new TreeSet(collectionSet);
        // TreeSet automatically puts strings in alphabetical order
        System.out.println("Number of collections on the server: " + 
                   orderedCollectionSet.size());
        Iterator collectionSetIterator = orderedCollectionSet.iterator();
        while (collectionSetIterator.hasNext() ) {
          // object -> String
          String collectionName = (String)collectionSetIterator.next(); 
          NzdlCollectionInfo info = nzdl.getCollectionInfo(collectionName);

          if (info != null && 
              info.getBuildDate() != 0 &&
              nzdl.pingCollection(collectionName)) {
            collectionInfoMap.put(collectionName, info);
            String longCollectionName = "";
            Set n = nzdl.getMetaData(collectionName, "collection", "collectionname");
            if (n != null) {
              if (n.size() == 1) 
            longCollectionName = (String) n.toArray()[0];
              else 
            longCollectionName = collectionName + " has more than 1 collectionname";
              collectionNameMap.put(collectionName,longCollectionName );
              System.err.println("Adding " + collectionName + " to collection list...");
            }
          } // end if
          else {
            System.err.println("Collection " + collectionName + " getBuildDate() returned 0");
          } // end else
        } // end while  
        setCurrentCollection(getFirstCollection()); //assume at least one collection
    
        }
    
    
    }