package org.paneris.bibliomania.loadtest;

 import java.io.FileOutputStream;

 import java.io.ObjectOutputStream;

 import java.io.Serializable;

 import java.util.Enumeration;

 import java.util.Vector;

 import org.melati.poem.AccessToken;

 import org.melati.poem.Persistent;

 import org.melati.poem.PoemTask;

 import org.melati.poem.util.EnumUtils;

 import org.melati.util.UnexpectedExceptionException;

 import org.paneris.bibliomania.BibliomaniaDatabase;

 import org.paneris.bibliomania.Book;

 import org.paneris.bibliomania.Chapter;

 import org.paneris.bibliomania.Unit;

 public class ExtractTree implements Serializable {

   private static final long serialVersionUID = 847223626240878960L;

   public DiscreteDistribution books = new DiscreteDistribution();

   public int searchsectiongroup;

   public void readDB() {

     final BibliomaniaDatabase bib = new BibliomaniaDatabase(false);

     bib.connect("bibliomania", "org.melati.poem.dbms.Postgresql",

                 "jdbc:postgresql:bibliomania", "postgres", "*",8);

     bib.inSession(

         AccessToken.root,

         new PoemTask() {

           @SuppressWarnings("unchecked")

           public void run() {

             try {

               searchsectiongroup =

                   bib.getSearchSectionGroup().troid().intValue();

               dobooks: for (Enumeration<Unit> e = bib.getBookTable().selection();

                             e.hasMoreElements();) {

                 Book book = (Book)e.nextElement();

                 Vector<Chapter> chapters = EnumUtils.vectorOf(book.getChapters());

                 if (chapters.size() > 0) {

                   ChapterDesc[] chapterDescs = new ChapterDesc[chapters.size()];

                   for (int i = 0; i < chapterDescs.length; ++i) {

                     Chapter chapter = (Chapter)chapters.elementAt(i);

                     int p = chapter.totalPages();

                     if (p == 0) continue dobooks;

                     chapterDescs[i] = new ChapterDesc(chapter.troid().intValue(),

                                                   chapter.totalPages());

                   }

                   books.add(

                       new BookDesc(

                           book.getSection().getGroupTroid().intValue(),

                           book.getSectionTroid().intValue(),

                           book.getAuthorTroid().intValue(),

                           book.troid().intValue(),

                           chapterDescs));

                 }

               }

             }

             catch (Exception e) {

               throw new UnexpectedExceptionException(e);

             }

           }

         });

   }

   public static void main(String[] args) throws Exception {

     ExtractTree t = new ExtractTree();

     t.readDB();

     ObjectOutputStream o =

         new ObjectOutputStream(new FileOutputStream("books.ser"));

     o.writeObject(t);

     o.close();

   }

 }