;;;;  -*- Mode: Lisp; Syntax: Common-Lisp; Package: C -*-
;;;;
;;;;  Copyright (c) 2011, Juan Jose Garcia-Ripoll
;;;;
;;;;    This program is free software; you can redistribute it and/or
;;;;    modify it under the terms of the GNU Library General Public
;;;;    License as published by the Free Software Foundation; either
;;;;    version 2 of the License, or (at your option) any later version.
;;;;
;;;;    See file '../Copyright' for full details.

;;;; CMPTYPE-ASSERT  Type assertions automatically generated

(in-package "COMPILER")

(defun c1compiler-typecase (args)
  (let* ((var-name (pop args))
	 (var (c1vref var-name))
	 (first-case (car args)))
    ;; If the first type, which is supposedly the most specific
    ;; already includes the form, we keep it. This optimizes
    ;; most cases of CHECKED-VALUE.
    (if (subtypep (var-type var) (car first-case))
	(c1progn (cdr first-case))
	(let* ((types '())
	       (expressions (loop for (type . forms) in args
			       for c1form = (c1progn forms)
			       for c1form-type = (c1form-primary-type c1form)
			       do (push c1form-type types)
			       collect (list type c1form))))
	  (make-c1form* 'EXT:COMPILER-TYPECASE
			:type (reduce #'type-or types)
			:args var expressions)))))

(defun c2compiler-typecase (c1form var expressions)
  (declare (ignore c1form))
  (loop with var-type = (var-type var)
     for (type form) in expressions
     when (or (member type '(t otherwise))
	      (subtypep var-type type))
     return (c2expr form)))

(defconstant +simple-type-assertions+
  '((cons . "if (ecl_unlikely(ECL_ATOM(#0))) FEtype_error_cons(#0);")
    (array . "if (ecl_unlikely(!ECL_ARRAYP(#0))) FEtype_error_array(#0);")
    (list . "if (ecl_unlikely(!ECL_LISTP(#0))) FEtype_error_list(#0);")
    (sequence . "if (ecl_unlikely(!(ECL_LISTP(#0) || ECL_VECTORP(#0))))
           FEtype_error_sequence(#0);")
    (vector . "if (ecl_unlikely(!ECL_VECTORP(#0))) FEtype_error_vector(#0);")))

(defun simple-type-assertion (value type env)
  (let ((simple-form (cdr (assoc type +simple-type-assertions+))))
    (if simple-form
	`(ffi:c-inline (,value) (:object) :void ,simple-form
		       :one-liner nil)
	`(ffi:c-inline
	  ((typep ,value ',type) ',type ,value)
	  (:bool :object :object) :void
	  "if (ecl_unlikely(!(#0)))
         FEwrong_type_argument(#1,#2);" :one-liner nil))))

(defun expand-type-assertion (value type env compulsory)
  (cond ((or (not (symbolp value))
	     (special-variable-p value)
	     (symbol-macro-p value))
	 ;; If multiple references to the value cost time and space,
	 ;; or may cause side effects, we save it.
	 (with-clean-symbols (%asserted-value)
	   `(let* ((%asserted-value ,value))
	      (declare (:read-only %asserted-value))
	      ,(expand-type-assertion '%asserted-value type env compulsory))))
	(compulsory
	 ;; The check has to be produced, independent of the declared
	 ;; value of the variable (for instance, in LAMBDA arguments).
	 (simple-type-assertion value type env))
	(t
	 ;; We may rely on the compiler to choose the appropriate
	 ;; branch once type propagation has happened.
	 `(ext:compiler-typecase ,value
            (,type)
	    (t ,(simple-type-assertion value type env))))))

(defun c1checked-value (args)
  (let* ((type (pop args))
	 (value (pop args))
	 form form-type and-type)
    (cond ((or (trivial-type-p args) (not (policy-type-assertions)))
	   value)
	  ((and (consp type)
		(eq (first type) 'values))
	   (c1checked-value (list (values-type-primary-type type)
				  value)))
	  ((and (policy-evaluate-forms) (constantp value))
	   (unless (typep (cmp-eval value) type)
	     (cmpwarn "Failed type assertion for value ~A and type ~A"
		      value type))
	   value)
	  ;; Is the form type contained in the test?
	  ((progn
	     (setf form (c1expr value)
		   form-type (c1form-primary-type form)
		   and-type (type-and form-type type))
	     (eq and-type form-type))
	   form)
	  ;; Are the form type and the test disjoint types?
	  ((null and-type)
	   (cmpwarn "The expression ~S is not of the expected type ~S"
		    value type)
	   form)
	  ;; Otherwise, emit a full test
	  (t
	   (cmpnote "Checking type of ~S to be ~S" value type)
	   (let ((full-check
		  (with-clean-symbols (%checked-value)
		    `(let* ((%checked-value ,value))
		       (declare (:read-only %checked-value))
		       ,(expand-type-assertion '%checked-value type *cmp-env* nil)
		       (truly-the ,type %checked-value)))))
	     (make-c1form* 'CHECKED-VALUE
			   :type type
			   :args type form (c1expr full-check)))))))

(defun c2checked-value (c1form type value let-form)
  (c2expr (if (subtypep (c1form-primary-type value) type)
	      value
	      let-form)))

(defmacro optional-type-assertion (&whole whole value type &environment env)
  "If safety settings are high enough, generates a type check on an
expression, ensuring that it is satisfied."
  (when (and (policy-type-assertions env)
	     (not (trivial-type-p type)))
    (cmpnote "Checking type of ~A to be ~A" value type)
    `(checked-value ,type ,value)))

(defmacro type-assertion (&whole whole value type &environment env)
  "Generates a type check on an expression, ensuring that it is satisfied."
  (cmpnote "Checking type of ~A to be ~A" value type)
  (unless (trivial-type-p type)
    (expand-type-assertion value type env t)))
