Dépôt officiel du code source de l'ERP OpenConcerto

package org.jopencalendar.print;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.image.BufferedImage;
import java.awt.print.PageFormat;
import java.awt.print.Pageable;
import java.awt.print.Printable;
import java.awt.print.PrinterException;
import java.awt.print.PrinterJob;
import java.text.DateFormat;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;

import javax.print.attribute.HashPrintRequestAttributeSet;
import javax.print.attribute.PrintRequestAttributeSet;
import javax.print.attribute.standard.PrintQuality;
import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.SwingUtilities;
import javax.swing.UIManager;

import org.jopencalendar.LayoutUtils;
import org.jopencalendar.model.Flag;
import org.jopencalendar.model.JCalendarItem;

public class CalendarItemPrinter implements Printable, Pageable {

    private static final int RIGHT_MARGIN = 50;
    public static final Font FONT_NORMAL = new Font("Arial", Font.PLAIN, 11);
    public static final Font FONT_BOLD = FONT_NORMAL.deriveFont(Font.BOLD);

    private List<JCalendarItem> items;
    private List<CalendarItemPage> pages = null;
    private String title;
    private boolean showDuration = true;
    private PageFormat calendarPageFormat;

    public CalendarItemPrinter(String title, List<JCalendarItem> items, PageFormat pf) {
        this.title = title;
        this.items = items;
        this.calendarPageFormat = pf;
    }

    public String getTitle() {
        return title;
    }

    @Override
    public int print(Graphics g, PageFormat pf, int pageIndex) throws PrinterException {
        if (pages == null) {
            computeLayout(g, pf);
        }

        if (pageIndex >= pages.size()) {
            return NO_SUCH_PAGE;
        }

        final Graphics2D g2d = (Graphics2D) g;
        g2d.translate(pf.getImageableX(), pf.getImageableY());
        final CalendarItemPage page = this.pages.get(pageIndex);

        printPage(g, page);

        /* tell the caller that this page is part of the printed document */
        return PAGE_EXISTS;
    }

    public void printPage(Graphics g, final CalendarItemPage page) {
        // We use our own page format to avoid issue between formatting and printing
        PageFormat pf = this.calendarPageFormat;

        final Graphics2D g2d = (Graphics2D) g;
        // Title
        g.setFont(getTitleFont());
        g.setColor(getTitleColor());
        final int xTitle = ((int) pf.getImageableWidth() - (int) g2d.getFontMetrics().getStringBounds(getTitle(), g2d).getWidth()) / 2;
        g.drawString(getTitle(), xTitle, g2d.getFontMetrics().getHeight());
        // Page number
        g.setFont(getPageNumberFont());
        g.setColor(getPageNumberColor());
        int y = getTitleHeight();
        final DateFormat df = DateFormat.getDateInstance(DateFormat.FULL);
        final int size = page.getItems().size();
        final String strPages = (page.getPageIndex() + 1) + " / " + pages.size();
        int xStrPages = (int) pf.getImageableWidth() - (int) g2d.getFontMetrics().getStringBounds(strPages, g2d).getWidth();
        g.drawString(strPages, xStrPages, g2d.getFontMetrics().getHeight());

        final double hourColumnWidth = getHourColumnWidth();
        final double maxWidth = pf.getImageableWidth() - hourColumnWidth - getMarginRight();
        for (int i = 0; i < size; i++) {
            final JCalendarItem item = page.getItems().get(i);
            final double lineHeight = page.getHeights().get(i);
            int lY = y;
            drawBackground(0, lY, (int) pf.getImageableWidth(), (int) lineHeight);
            if (page.getShowDates().get(i)) {
                g2d.setColor(getHourColor(item));
                g.setFont(getHourFont(item));
                int dateHeight = g2d.getFontMetrics().getHeight();
                lY += dateHeight;
                final String format = getDate(df, item);
                g2d.drawString(format, 2, lY);
            }
            // Vertical bar
            g2d.setColor(getVerticalBarColor(item));
            g2d.drawLine((int) hourColumnWidth - 5, lY + 6, (int) hourColumnWidth - 5, (int) (y + lineHeight));

            final Calendar dtStart = item.getDtStart();
            final Calendar dtEnd = item.getDtEnd();
            // Hour
            g.setColor(getHourColor(item));
            g.setFont(getHourFont(item));
            final int hourY = lY + g2d.getFontMetrics().getHeight();
            g2d.drawString(formatTime(dtStart) + " - " + formatTime(dtEnd), 3, hourY);
            // Duration
            if (this.showDuration) {
                g.setColor(getDurationColor(item));
                g.setFont(getDurationFont(item));
                final int durationY = hourY + g2d.getFontMetrics().getHeight();
                g2d.drawString(formatDuration(dtStart, dtEnd), 3, durationY);
            }

            // Summary
            g.setFont(getLine1Font(item));

            final List<String> l1 = LayoutUtils.wrap(getLine1Text(item), g.getFontMetrics(), (int) maxWidth);
            if (item.hasFlag(Flag.getFlag("warning"))) {
                g2d.setColor(new Color(255, 249, 144));
                g2d.fillRect((int) hourColumnWidth - 1, lY + 3, (int) maxWidth, g2d.getFontMetrics().getHeight() * l1.size());
            }
            g.setColor(getLine1Color(item));
            for (String string : l1) {
                lY += g2d.getFontMetrics().getHeight();
                g2d.drawString(string, (int) hourColumnWidth, lY);
            }
            // Description
            g.setFont(getLine2Font(item));
            g.setColor(getLine2Color(item));
            final List<String> l2 = LayoutUtils.wrap(getLine2Text(item), g.getFontMetrics(), (int) maxWidth);
            for (String string : l2) {
                lY += g2d.getFontMetrics().getHeight();
                g2d.drawString(string, (int) hourColumnWidth, lY);
            }
            //
            y += lineHeight;
        }
    }

    private int getMarginRight() {
        return RIGHT_MARGIN;
    }

    public String getDate(final DateFormat df, final JCalendarItem item) {
        return LayoutUtils.firstUp(df.format(item.getDtStart().getTime()));
    }

    public String formatDuration(Calendar dtStart, Calendar dtEnd) {
        long t2 = dtEnd.getTimeInMillis();
        long t1 = dtStart.getTimeInMillis();
        long seconds = (t2 - t1) / 1000;

        final long minutes = seconds / 60;

        String l = String.valueOf(minutes % 60);
        if (l.length() < 2) {
            l = "0" + l;
        }
        return "(" + minutes / 60 + "h" + l + ")";
    }

    public void drawBackground(int x, int y, int w, int h) {
        // for subclass to add some background
    }

    public Color getPageNumberColor() {
        return Color.BLACK;
    }

    public Font getPageNumberFont() {
        return FONT_NORMAL;
    }

    // Title
    public Color getTitleColor() {
        return Color.BLACK;
    }

    public Font getTitleFont() {
        return FONT_NORMAL;
    }

    // Hour
    public Color getHourColor(JCalendarItem item) {
        return Color.BLACK;
    }

    public Font getHourFont(JCalendarItem item) {
        return FONT_BOLD;
    }

    // Duration
    public Color getDurationColor(JCalendarItem item) {
        return Color.BLACK;
    }

    public Font getDurationFont(JCalendarItem item) {
        return FONT_NORMAL;
    }

    // Line 1
    public Color getLine1Color(final JCalendarItem item) {
        return Color.BLACK;
    }

    public String getLine1Text(final JCalendarItem item) {
        return item.getSummary();
    }

    public Font getLine1Font(final JCalendarItem item) {
        return FONT_NORMAL;
    }

    // Line 2
    public Color getLine2Color(final JCalendarItem item) {
        return Color.BLACK;
    }

    public String getLine2Text(final JCalendarItem item) {
        return item.getDescription();
    }

    public Font getLine2Font(final JCalendarItem item) {
        return FONT_NORMAL;
    }

    public Color getVerticalBarColor(final JCalendarItem item) {
        return new Color(0, 68, 128);
    }

    public void setShowDuration(boolean showDuration) {
        this.showDuration = showDuration;
    }

    public String formatTime(final Calendar dtStart) {
        String h = String.valueOf(dtStart.get(Calendar.HOUR_OF_DAY));
        String m = String.valueOf(dtStart.get(Calendar.MINUTE));
        if (h.length() < 2) {
            h = " " + h;
        }
        if (m.length() < 2) {
            m = "0" + m;
        }
        return h + ":" + m;
    }

    private void computeLayout(Graphics g, PageFormat pf) {
        int pageIndex = 0;
        pages = new ArrayList<CalendarItemPage>();
        CalendarItemPage page = new CalendarItemPage();
        page.setPageIndex(pageIndex);
        this.pages.add(page);
        pageIndex++;
        if (items.isEmpty()) {
            return;
        }
        double remainingHeight = pf.getImageableHeight() - getTitleHeight();
        boolean showDate = true;
        Calendar c = Calendar.getInstance();
        c.setTimeInMillis(this.items.get(0).getDtStart().getTimeInMillis());
        for (int i = 0; i < this.items.size(); i++) {
            JCalendarItem item = this.items.get(i);
            if (i > 0) {
                showDate = (item.getDtStart().get(Calendar.YEAR) != c.get(Calendar.YEAR) || item.getDtStart().get(Calendar.DAY_OF_YEAR) != c.get(Calendar.DAY_OF_YEAR));
            }
            c = item.getDtStart();
            double h = getPrintHeight(g, pf, item);
            double secureMargin = 20D;
            if (remainingHeight < h + secureMargin) {
                page = new CalendarItemPage();
                page.setPageIndex(pageIndex);
                showDate = true;
                remainingHeight = pf.getImageableHeight() - getTitleHeight();
                this.pages.add(page);
                pageIndex++;

            }
            if (showDate) {
                g.setFont(getHourFont(item));
                h += g.getFontMetrics().getHeight();
            }
            remainingHeight -= h;

            page.add(item);
            page.addHeight(Double.valueOf(h));
            page.addShowDate(showDate);
        }

    }

    public int getHourColumnWidth() {
        return 90;
    }

    public int getTitleHeight() {
        return 40;
    }

    private double getPrintHeight(Graphics g, PageFormat pf, JCalendarItem item) {
        double heigth = 0;
        int maxWidth = (int) pf.getImageableWidth() - getHourColumnWidth() - getMarginRight();
        // Line 1
        g.setFont(this.getLine1Font(item));
        int l1 = LayoutUtils.wrap(getLine1Text(item), g.getFontMetrics(), maxWidth).size();
        heigth += l1 * g.getFontMetrics().getHeight();
        // Line 2
        g.setFont(this.getLine2Font(item));
        int l2 = LayoutUtils.wrap(getLine2Text(item), g.getFontMetrics(), maxWidth).size();
        heigth += l2 * g.getFontMetrics().getHeight();
        return heigth;
    }

    public List<CalendarItemPage> getPages() {
        return pages;
    }

    public static void main(String[] args) {
        SwingUtilities.invokeLater(new Runnable() {

            @Override
            public void run() {
                try {
                    UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
                } catch (Exception ex) {
                    ex.printStackTrace();
                }
                final JFrame f = new JFrame("Printing Pagination Example");
                f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
                final JButton printButton = new JButton("Print test pages");
                printButton.addActionListener(new ActionListener() {

                    @Override
                    public void actionPerformed(ActionEvent e) {
                        PrinterJob job = PrinterJob.getPrinterJob();
                        List<JCalendarItem> cItems = new ArrayList<JCalendarItem>();
                        Calendar c = Calendar.getInstance();
                        Flag.register(new Flag("warning", null, "Warning", "A default warning"));
                        for (int i = 0; i < 50; i++) {
                            JCalendarItem item = new JCalendarItem();
                            item.setSummary("Item " + i);
                            StringBuilder d = new StringBuilder();
                            d.append("Description");
                            for (int j = 0; j < i; j++) {
                                if (i % 2 == 0)
                                    d.append(" Hello");
                                else
                                    d.append(" Wo");
                            }
                            d.append("END");
                            if (i % 6 == 0) {
                                item.addFlag(Flag.getFlag("warning"));
                            }
                            if (i != 5) {
                                item.setDescription(d.toString());
                            } else {
                                item.setDescription(
                                        "Escalier Ménage Complet - 2 escaliers : Interrupteurs, poignées portes, mains courantes côté route, traces de doigts sur vitres et ascenseur, aspiration moquettes et carrelage, lavage des halls.");
                            }
                            item.setDtStart(c);
                            c.add(Calendar.HOUR_OF_DAY, 1);
                            item.setDtEnd(c);
                            c.add(Calendar.HOUR_OF_DAY, 1);
                            cItems.add(item);
                        }
                        final PrintRequestAttributeSet printAttributes = new HashPrintRequestAttributeSet();
                        printAttributes.add(PrintQuality.HIGH);
                        job.setPrintable(new CalendarItemPrinter("OpenConcerto", cItems, job.getPageFormat(printAttributes)));
                        boolean ok = job.printDialog();
                        if (ok) {
                            try {
                                job.print();
                            } catch (PrinterException ex) {
                                /* The job did not successfully complete */
                                ex.printStackTrace();
                            }
                        }

                    }
                });
                f.add("Center", printButton);
                f.pack();
                f.setLocationRelativeTo(null);
                f.setVisible(true);
            }
        });
    }

    public List<JCalendarItem> getItems() {
        return this.items;
    }

    @Override
    public int getNumberOfPages() {
        if (this.pages == null) {
            BufferedImage offImage = new BufferedImage((int) calendarPageFormat.getHeight(), (int) calendarPageFormat.getWidth(), BufferedImage.TYPE_INT_ARGB);
            Graphics2D g2 = offImage.createGraphics();
            computeLayout(g2, calendarPageFormat);
        }
        return this.pages.size();
    }

    @Override
    public PageFormat getPageFormat(int pageIndex) {
        return this.calendarPageFormat;
    }

    @Override
    public Printable getPrintable(int pageIndex) {
        final CalendarItemPage page = this.pages.get(pageIndex);
        return new Printable() {

            @Override
            public int print(Graphics graphics, PageFormat pageFormat, int i) throws PrinterException {
                final Graphics2D g2d = (Graphics2D) graphics;
                g2d.translate(calendarPageFormat.getImageableX(), calendarPageFormat.getImageableY());
                printPage(graphics, page);
                return PAGE_EXISTS;
            }
        };
    }
}