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

/*
 *    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: 2281 $
 * @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;
    BerryBasketModel berryBasketModel;
    

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

    public QueryHistoryModel getQueryHistoryModel() {
    return queryHistoryModel;
    }

    public BerryBasketModel getBerryBasketModel() {
    return berryBasketModel;
    }
    
    /* 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
    
    }
    
    
}