GNU Linux-libre 4.9.306-gnu1
[releases.git] / drivers / media / platform / mtk-vcodec / venc_vpu_if.c
1 /*
2  * Copyright (c) 2016 MediaTek Inc.
3  * Author: PoChun Lin <pochun.lin@mediatek.com>
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify
7  * it under the terms of the GNU General Public License version 2 as
8  * published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  */
15
16 #include "mtk_vpu.h"
17 #include "venc_ipi_msg.h"
18 #include "venc_vpu_if.h"
19
20 static void handle_enc_init_msg(struct venc_vpu_inst *vpu, void *data)
21 {
22         struct venc_vpu_ipi_msg_init *msg = data;
23
24         vpu->inst_addr = msg->vpu_inst_addr;
25         vpu->vsi = vpu_mapping_dm_addr(vpu->dev, msg->vpu_inst_addr);
26 }
27
28 static void handle_enc_encode_msg(struct venc_vpu_inst *vpu, void *data)
29 {
30         struct venc_vpu_ipi_msg_enc *msg = data;
31
32         vpu->state = msg->state;
33         vpu->bs_size = msg->bs_size;
34         vpu->is_key_frm = msg->is_key_frm;
35 }
36
37 static void vpu_enc_ipi_handler(void *data, unsigned int len, void *priv)
38 {
39         struct venc_vpu_ipi_msg_common *msg = data;
40         struct venc_vpu_inst *vpu =
41                 (struct venc_vpu_inst *)(unsigned long)msg->venc_inst;
42
43         mtk_vcodec_debug(vpu, "msg_id %x inst %p status %d",
44                          msg->msg_id, vpu, msg->status);
45
46         switch (msg->msg_id) {
47         case VPU_IPIMSG_ENC_INIT_DONE:
48                 handle_enc_init_msg(vpu, data);
49                 break;
50         case VPU_IPIMSG_ENC_SET_PARAM_DONE:
51                 break;
52         case VPU_IPIMSG_ENC_ENCODE_DONE:
53                 handle_enc_encode_msg(vpu, data);
54                 break;
55         case VPU_IPIMSG_ENC_DEINIT_DONE:
56                 break;
57         default:
58                 mtk_vcodec_err(vpu, "unknown msg id %x", msg->msg_id);
59                 break;
60         }
61
62         vpu->signaled = 1;
63         vpu->failure = (msg->status != VENC_IPI_MSG_STATUS_OK);
64
65         mtk_vcodec_debug_leave(vpu);
66 }
67
68 static int vpu_enc_send_msg(struct venc_vpu_inst *vpu, void *msg,
69                             int len)
70 {
71         int status;
72
73         mtk_vcodec_debug_enter(vpu);
74
75         if (!vpu->dev) {
76                 mtk_vcodec_err(vpu, "inst dev is NULL");
77                 return -EINVAL;
78         }
79
80         status = vpu_ipi_send(vpu->dev, vpu->id, msg, len);
81         if (status) {
82                 uint32_t msg_id = *(uint32_t *)msg;
83
84                 mtk_vcodec_err(vpu, "vpu_ipi_send msg_id %x len %d fail %d",
85                                msg_id, len, status);
86                 return -EINVAL;
87         }
88         if (vpu->failure)
89                 return -EINVAL;
90
91         mtk_vcodec_debug_leave(vpu);
92
93         return 0;
94 }
95
96 int vpu_enc_init(struct venc_vpu_inst *vpu)
97 {
98         int status;
99         struct venc_ap_ipi_msg_init out;
100
101         mtk_vcodec_debug_enter(vpu);
102
103         init_waitqueue_head(&vpu->wq_hd);
104         vpu->signaled = 0;
105         vpu->failure = 0;
106
107         status = vpu_ipi_register(vpu->dev, vpu->id, vpu_enc_ipi_handler,
108                                   NULL, NULL);
109         if (status) {
110                 mtk_vcodec_err(vpu, "vpu_ipi_register fail %d", status);
111                 return -EINVAL;
112         }
113
114         memset(&out, 0, sizeof(out));
115         out.msg_id = AP_IPIMSG_ENC_INIT;
116         out.venc_inst = (unsigned long)vpu;
117         if (vpu_enc_send_msg(vpu, &out, sizeof(out))) {
118                 mtk_vcodec_err(vpu, "AP_IPIMSG_ENC_INIT fail");
119                 return -EINVAL;
120         }
121
122         mtk_vcodec_debug_leave(vpu);
123
124         return 0;
125 }
126
127 int vpu_enc_set_param(struct venc_vpu_inst *vpu,
128                       enum venc_set_param_type id,
129                       struct venc_enc_param *enc_param)
130 {
131         struct venc_ap_ipi_msg_set_param out;
132
133         mtk_vcodec_debug(vpu, "id %d ->", id);
134
135         memset(&out, 0, sizeof(out));
136         out.msg_id = AP_IPIMSG_ENC_SET_PARAM;
137         out.vpu_inst_addr = vpu->inst_addr;
138         out.param_id = id;
139         switch (id) {
140         case VENC_SET_PARAM_ENC:
141                 out.data_item = 0;
142                 break;
143         case VENC_SET_PARAM_FORCE_INTRA:
144                 out.data_item = 0;
145                 break;
146         case VENC_SET_PARAM_ADJUST_BITRATE:
147                 out.data_item = 1;
148                 out.data[0] = enc_param->bitrate;
149                 break;
150         case VENC_SET_PARAM_ADJUST_FRAMERATE:
151                 out.data_item = 1;
152                 out.data[0] = enc_param->frm_rate;
153                 break;
154         case VENC_SET_PARAM_GOP_SIZE:
155                 out.data_item = 1;
156                 out.data[0] = enc_param->gop_size;
157                 break;
158         case VENC_SET_PARAM_INTRA_PERIOD:
159                 out.data_item = 1;
160                 out.data[0] = enc_param->intra_period;
161                 break;
162         case VENC_SET_PARAM_SKIP_FRAME:
163                 out.data_item = 0;
164                 break;
165         default:
166                 mtk_vcodec_err(vpu, "id %d not supported", id);
167                 return -EINVAL;
168         }
169         if (vpu_enc_send_msg(vpu, &out, sizeof(out))) {
170                 mtk_vcodec_err(vpu,
171                                "AP_IPIMSG_ENC_SET_PARAM %d fail", id);
172                 return -EINVAL;
173         }
174
175         mtk_vcodec_debug(vpu, "id %d <-", id);
176
177         return 0;
178 }
179
180 int vpu_enc_encode(struct venc_vpu_inst *vpu, unsigned int bs_mode,
181                    struct venc_frm_buf *frm_buf,
182                    struct mtk_vcodec_mem *bs_buf,
183                    unsigned int *bs_size)
184 {
185         struct venc_ap_ipi_msg_enc out;
186
187         mtk_vcodec_debug(vpu, "bs_mode %d ->", bs_mode);
188
189         memset(&out, 0, sizeof(out));
190         out.msg_id = AP_IPIMSG_ENC_ENCODE;
191         out.vpu_inst_addr = vpu->inst_addr;
192         out.bs_mode = bs_mode;
193         if (frm_buf) {
194                 if ((frm_buf->fb_addr[0].dma_addr % 16 == 0) &&
195                     (frm_buf->fb_addr[1].dma_addr % 16 == 0) &&
196                     (frm_buf->fb_addr[2].dma_addr % 16 == 0)) {
197                         out.input_addr[0] = frm_buf->fb_addr[0].dma_addr;
198                         out.input_addr[1] = frm_buf->fb_addr[1].dma_addr;
199                         out.input_addr[2] = frm_buf->fb_addr[2].dma_addr;
200                 } else {
201                         mtk_vcodec_err(vpu, "dma_addr not align to 16");
202                         return -EINVAL;
203                 }
204         }
205         if (bs_buf) {
206                 out.bs_addr = bs_buf->dma_addr;
207                 out.bs_size = bs_buf->size;
208         }
209         if (vpu_enc_send_msg(vpu, &out, sizeof(out))) {
210                 mtk_vcodec_err(vpu, "AP_IPIMSG_ENC_ENCODE %d fail",
211                                bs_mode);
212                 return -EINVAL;
213         }
214
215         mtk_vcodec_debug(vpu, "bs_mode %d state %d size %d key_frm %d <-",
216                          bs_mode, vpu->state, vpu->bs_size, vpu->is_key_frm);
217
218         return 0;
219 }
220
221 int vpu_enc_deinit(struct venc_vpu_inst *vpu)
222 {
223         struct venc_ap_ipi_msg_deinit out;
224
225         mtk_vcodec_debug_enter(vpu);
226
227         memset(&out, 0, sizeof(out));
228         out.msg_id = AP_IPIMSG_ENC_DEINIT;
229         out.vpu_inst_addr = vpu->inst_addr;
230         if (vpu_enc_send_msg(vpu, &out, sizeof(out))) {
231                 mtk_vcodec_err(vpu, "AP_IPIMSG_ENC_DEINIT fail");
232                 return -EINVAL;
233         }
234
235         mtk_vcodec_debug_leave(vpu);
236
237         return 0;
238 }