9768ffd0c4ac59f068406c95d23b9e717160ce27
[moodle.git] / dataformat / pdf / classes / writer.php
1 <?php
2 // This file is part of Moodle - http://moodle.org/
3 //
4 // Moodle is free software: you can redistribute it and/or modify
5 // it under the terms of the GNU General Public License as published by
6 // the Free Software Foundation, either version 3 of the License, or
7 // (at your option) any later version.
8 //
9 // Moodle is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 // GNU General Public License for more details.
13 //
14 // You should have received a copy of the GNU General Public License
15 // along with Moodle.  If not, see <http://www.gnu.org/licenses/>.
17 /**
18  * pdf data format writer
19  *
20  * @package    dataformat_pdf
21  * @copyright  2019 Shamim Rezaie <shamim@moodle.com>
22  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
23  */
25 namespace dataformat_pdf;
27 defined('MOODLE_INTERNAL') || die();
29 /**
30  * pdf data format writer
31  *
32  * @package    dataformat_pdf
33  * @copyright  2019 Shamim Rezaie <shamim@moodle.com>
34  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
35  */
36 class writer extends \core\dataformat\base {
38     public $mimetype = "application/pdf";
40     public $extension = ".pdf";
42     /**
43      * @var \pdf The pdf object that is used to generate the pdf file.
44      */
45     protected $pdf;
47     /**
48      * @var float Each column's width in the current sheet.
49      */
50     protected $colwidth;
52     /**
53      * @var string[] Title of columns in the current sheet.
54      */
55     protected $columns;
57     /**
58      * writer constructor.
59      */
60     public function __construct() {
61         global $CFG;
62         require_once($CFG->libdir . '/pdflib.php');
64         $this->pdf = new \pdf();
65         $this->pdf->setPrintHeader(false);
66         $this->pdf->SetFooterMargin(PDF_MARGIN_FOOTER);
68         // Set background color for headings.
69         $this->pdf->SetFillColor(238, 238, 238);
70     }
72     public function send_http_headers() {
73     }
75     /**
76      * Start output to file, note that the actual writing of the file is done in {@see close_output_to_file()}
77      */
78     public function start_output_to_file(): void {
79         $this->start_output();
80     }
82     public function start_output() {
83         $this->pdf->AddPage('L');
84     }
86     public function start_sheet($columns) {
87         $margins = $this->pdf->getMargins();
88         $pagewidth = $this->pdf->getPageWidth() - $margins['left'] - $margins['right'];
90         $this->colwidth = $pagewidth / count($columns);
91         $this->columns = $columns;
93         $this->print_heading();
94     }
96     /**
97      * Method to define whether the dataformat supports export of HTML
98      *
99      * @return bool
100      */
101     public function supports_html(): bool {
102         return true;
103     }
105     /**
106      * Write a single record
107      *
108      * @param array $record
109      * @param int $rownum
110      */
111     public function write_record($record, $rownum) {
112         $rowheight = 0;
114         $record = $this->format_record($record);
115         foreach ($record as $cell) {
116             $rowheight = max($rowheight, $this->pdf->getStringHeight($this->colwidth, $cell, false, true, '', 1));
117         }
119         $margins = $this->pdf->getMargins();
120         if ($this->pdf->GetY() + $rowheight + $margins['bottom'] > $this->pdf->getPageHeight()) {
121             $this->pdf->AddPage('L');
122             $this->print_heading();
123         }
125         // Get the last key for this record.
126         end($record);
127         $lastkey = key($record);
129         // Reset the record pointer.
130         reset($record);
132         // Loop through each element.
133         foreach ($record as $key => $cell) {
134             // Determine whether we're at the last element of the record.
135             $nextposition = ($lastkey === $key) ? 1 : 0;
136             // Write the element.
137             $this->pdf->writeHTMLCell($this->colwidth, $rowheight, '', '', $cell, 1, $nextposition, false, true, 'L');
138         }
139     }
141     public function close_output() {
142         $filename = $this->filename . $this->get_extension();
144         $this->pdf->Output($filename, 'D');
145     }
147     /**
148      * Write data to disk
149      *
150      * @return bool
151      */
152     public function close_output_to_file(): bool {
153         $this->pdf->Output($this->filepath, 'F');
155         return true;
156     }
158     /**
159      * Prints the heading row.
160      */
161     private function print_heading() {
162         $fontfamily = $this->pdf->getFontFamily();
163         $fontstyle = $this->pdf->getFontStyle();
164         $this->pdf->SetFont($fontfamily, 'B');
165         $rowheight = 0;
166         foreach ($this->columns as $columns) {
167             $rowheight = max($rowheight, $this->pdf->getStringHeight($this->colwidth, $columns, false, true, '', 1));
168         }
170         $total = count($this->columns);
171         $counter = 1;
172         foreach ($this->columns as $columns) {
173             $nextposition = ($counter == $total) ? 1 : 0;
174             $this->pdf->Multicell($this->colwidth, $rowheight, $columns, 1, 'C', true, $nextposition);
175             $counter++;
176         }
178         $this->pdf->SetFont($fontfamily, $fontstyle);
179     }